static ngx_int_t __on_write_master1_complete(
uint8_t method,
ngx_bool_t has_tb,
ngx_http_request_t *r,
ngx_http_hustdb_ha_write_ctx_t * ctx)
{
__update_error(method, r->headers_out.status, ctx);
ctx->base.peer = ctx->base.peer->next;
ngx_bool_t alive = ngx_http_peer_is_alive(ctx->base.peer->peer);
ngx_http_hustdb_ha_main_conf_t * mcf = hustdb_ha_get_module_main_conf(r);
if (mcf->debug_sync)
{
alive = false;
}
if (alive) // master2
{
ctx->state = STATE_WRITE_MASTER2;
return ngx_http_run_subrequest(r, &ctx->base.base, ctx->base.peer->peer);
}
// master2 dead
++ctx->error_count;
ctx->error_peer = ctx->base.peer;
if (ctx->error_count > 1) // both master1 & master2 fail
{
return hustdb_ha_send_response(NGX_HTTP_NOT_FOUND, NULL, NULL, r);
}
return __write_sync_data(method, has_tb, r, ctx);
}
static ngx_bool_t __set_write_context(const char * key, ngx_bool_t has_tb, ngx_http_request_t *r, write_ctx_t * ctx)
{
if (key)
{
ctx->key = key;
}
else
{
ctx->key = ngx_http_get_param_val(&r->args, "key", r->pool);
if (!ctx->key)
{
return false;
}
}
ctx->proto = HUSTDB_PROTO_BINARY;
char * proto = ngx_http_get_param_val(&r->args, "proto", r->pool);
if (proto)
{
int tmp = atoi(proto);
if (HUSTDB_PROTO_JSON == tmp)
{
ctx->proto = HUSTDB_PROTO_JSON;
}
else if (HUSTDB_PROTO_BINARY == tmp)
{
ctx->proto = HUSTDB_PROTO_BINARY;
}
else
{
return false;
}
}
ctx->tb = ngx_http_get_param_val(&r->args, "tb", r->pool);
if (has_tb && !ctx->tb)
{
return false;
}
ctx->ttl = 0;
char * val = ngx_http_get_param_val(&r->args, "ttl", r->pool);
if (val)
{
ctx->ttl = atoi(val);
}
ngx_http_hustdb_ha_main_conf_t * mcf = hustdb_ha_get_module_main_conf(r);
if (!mcf)
{
return false;
}
ctx->peer = hustdb_ha_get_writelist(ctx->key);
if (!ctx->peer)
{
return false;
}
ctx->error_count = 0;
ctx->error_peer = NULL;
ctx->state = STATE_WRITE_MASTER1;
ngx_bool_t alive = ngx_http_peer_is_alive(ctx->peer->peer);
if (!alive) // master1
{
++ctx->error_count;
ctx->error_peer = ctx->peer;
ctx->peer = ctx->peer->next;
ctx->state = STATE_WRITE_MASTER2;
alive = ngx_http_peer_is_alive(ctx->peer->peer);
if (!alive) // master2
{
return false;
}
}
return true;
}
static ngx_bool_t __parse_args(
ngx_bool_t has_tb,
ngx_bool_t hash_by_tb,
ngx_http_request_t *r,
write_ctx_t * ctx)
{
ctx->tb = ngx_http_get_param_val(&r->args, "tb", r->pool);
if (has_tb && !ctx->tb)
{
return false;
}
ctx->ttl = 0;
char * val = ngx_http_get_param_val(&r->args, "ttl", r->pool);
if (val)
{
ctx->ttl = atoi(val);
}
ctx->score = 0;
val = ngx_http_get_param_val(&r->args, "score", r->pool);
if (val)
{
char * endptr;
ctx->score = strtoull(val, &endptr, 10);
}
ctx->opt = 0;
val = ngx_http_get_param_val(&r->args, "opt", r->pool);
if (val)
{
ctx->opt = atoi(val);
}
ngx_http_hustdb_ha_main_conf_t * mcf = hustdb_ha_get_module_main_conf(r);
if (!mcf)
{
return false;
}
ctx->peer = hustdb_ha_get_writelist(hash_by_tb ? ctx->tb : ctx->key);
if (!ctx->peer)
{
return false;
}
ctx->error_count = 0;
ctx->error_peer = NULL;
ctx->state = STATE_WRITE_MASTER1;
ngx_bool_t alive = ngx_http_peer_is_alive(ctx->peer->peer);
if (!alive) // master1
{
++ctx->error_count;
ctx->error_peer = ctx->peer;
ctx->peer = ctx->peer->next;
ctx->state = STATE_WRITE_MASTER2;
alive = ngx_http_peer_is_alive(ctx->peer->peer);
if (!alive) // master2
{
return false;
}
}
return true;
}