char *
conf_set_bool(struct conf *cf, struct command *cmd, void *conf)
{
uint8_t *p;
int *bp;
struct string *value, true_str, false_str;
p = conf;
bp = (int *)(p + cmd->offset);
if (*bp != CONF_UNSET_NUM) {
return "is a duplicate";
}
value = array_top(&cf->arg);
string_set_text(&true_str, "true");
string_set_text(&false_str, "false");
if (string_compare(value, &true_str) == 0) {
*bp = 1;
} else if (string_compare(value, &false_str) == 0) {
*bp = 0;
} else {
return "is not \"true\" or \"false\"";
}
return CONF_OK;
}
struct stats *
stats_create(uint16_t stats_port, char *stats_ip, int stats_interval,
char *source, struct array *server_pool)
{
rstatus_t status;
struct stats *st;
char *server_name;
st = nc_alloc(sizeof(*st));
if (st == NULL) {
return NULL;
}
st->port = stats_port;
st->interval = stats_interval;
string_set_raw(&st->addr, stats_ip);
st->start_ts = (int64_t)time(NULL);
st->buf.len = 0;
st->buf.data = NULL;
st->buf.size = 0;
array_null(&st->current);
array_null(&st->shadow);
array_null(&st->sum);
st->tid = (pthread_t) -1;
st->sd = -1;
string_set_text(&st->service_str, "service");
//string_set_text(&st->service, "bilitw");
server_name = nc_alloc(100);
memset(server_name, 0, 100);
sprintf(server_name, "bilitw worker %d", nc_worker_index);
string_set_raw(&st->service, server_name);
//&st->service->len = strlen(server_name) + 1;
//&st->service->data = (uint8_t *)server_name;
string_set_text(&st->source_str, "source");
string_set_raw(&st->source, source);
string_set_text(&st->version_str, "version");
string_set_text(&st->version, NC_VERSION_STRING);
string_set_text(&st->uptime_str, "uptime");
string_set_text(&st->timestamp_str, "timestamp");
string_set_text(&st->ntotal_conn_str, "total_connections");
string_set_text(&st->ncurr_conn_str, "curr_connections");
st->updated = 0;
st->aggregate = 0;
/* map server pool to current (a), shadow (b) and sum (c) */
status = stats_pool_map(&st->current, server_pool);
if (status != NC_OK) {
goto error;
}
status = stats_pool_map(&st->shadow, server_pool);
if (status != NC_OK) {
goto error;
}
status = stats_pool_map(&st->sum, server_pool);
if (status != NC_OK) {
goto error;
}
status = stats_create_buf(st);
if (status != NC_OK) {
goto error;
}
status = stats_start_aggregator(st);
if (status != NC_OK) {
goto error;
}
return st;
error:
stats_destroy(st);
return NULL;
}
struct stats *
stats_create(uint16_t stats_port, char *stats_ip, int stats_interval,
char *source, struct array *server_pool)
{
rstatus_t status;
struct stats *st;
st = nc_alloc(sizeof(*st));
if (st == NULL) {
return NULL;
}
st->port = stats_port;
st->interval = stats_interval;
string_set_raw(&st->addr, stats_ip);
st->start_ts = (int64_t)time(NULL);
st->buf.len = 0;
st->buf.data = NULL;
st->buf.size = 0;
array_null(&st->current);
array_null(&st->shadow);
array_null(&st->sum);
st->tid = (pthread_t) -1;
st->sd = -1;
string_set_text(&st->service_str, "service");
string_set_text(&st->service, "nutcracker");
string_set_text(&st->source_str, "source");
string_set_raw(&st->source, source);
string_set_text(&st->version_str, "version");
string_set_text(&st->version, NC_VERSION_STRING);
string_set_text(&st->uptime_str, "uptime");
string_set_text(&st->timestamp_str, "timestamp");
string_set_text(&st->ntotal_conn_str, "total_connections");
string_set_text(&st->ncurr_conn_str, "curr_connections");
st->updated = 0;
st->aggregate = 0;
/* map server pool to current (a), shadow (b) and sum (c) */
status = stats_pool_map(&st->current, server_pool);
if (status != NC_OK) {
goto error;
}
status = stats_pool_map(&st->shadow, server_pool);
if (status != NC_OK) {
goto error;
}
status = stats_pool_map(&st->sum, server_pool);
if (status != NC_OK) {
goto error;
}
status = stats_create_buf(st); //为stats格式信息分配buf空间
if (status != NC_OK) {
goto error;
}
//stats状态信息专门起一个线程来处理
status = stats_start_aggregator(st);
if (status != NC_OK) {
goto error;
}
return st;
error:
stats_destroy(st);
return NULL;
}
static rstatus_t
sentinel_proc_pub(struct context *ctx, struct msg *msg)
{
rstatus_t status;
struct string pool_name, server_name, server_ip,
tmp_string, pub_titile, pub_event;
struct mbuf *line_buf;
int server_port;
string_init(&tmp_string);
string_init(&pool_name);
string_init(&server_name);
string_init(&server_ip);
string_set_text(&pub_titile, "pmessage");
string_set_text(&pub_event, "+switch-master");
line_buf = mbuf_get();
if (line_buf == NULL) {
goto error;
}
/* get line in line num 3 for pub titile */
msg_read_line(msg, line_buf, 3);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel pmessage when skip line not used.");
goto error;
}
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK || string_compare(&pub_titile, &tmp_string)) {
log_error("pub title error(lineinfo %.*s)", tmp_string.len, tmp_string.data);
goto error;
}
/* get line in line num 7 for pub event */
msg_read_line(msg, line_buf, 4);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel pmessage when skip line not used.");
goto error;
}
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK || string_compare(&pub_event, &tmp_string)) {
log_error("pub channel error(lineinfo %.*s)", tmp_string.len, tmp_string.data);
goto error;
}
/* get line in line num 9 for pub info */
msg_read_line(msg, line_buf, 2);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel pmessage when skip line not used.");
goto error;
}
/* parse switch master info */
/* get pool name */
status = mbuf_read_string(line_buf, SENTINEL_SERVERNAME_SPLIT, &pool_name);
if (status != NC_OK) {
log_error("get pool name string failed.");
goto error;
}
/* get server name */
status = mbuf_read_string(line_buf, ' ', &server_name);
if (status != NC_OK) {
log_error("get server name string failed.");
goto error;
}
/* skip old ip and port string */
status = mbuf_read_string(line_buf, ' ', NULL);
if (status != NC_OK) {
log_error("skip old ip string failed.");
goto error;
}
status = mbuf_read_string(line_buf, ' ', NULL);
if (status != NC_OK) {
log_error("skip old port string failed.");
goto error;
}
/* get new server ip string */
status = mbuf_read_string(line_buf, ' ', &server_ip);
if (status != NC_OK) {
log_error("get new server ip string failed.");
goto error;
}
/* get new server port */
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK) {
log_error("get new server port string failed.");
goto error;
}
server_port = nc_atoi(tmp_string.data, tmp_string.len);
if (server_port < 0) {
log_error("tanslate server port string to int failed.");
goto error;
}
status = server_switch(ctx, &pool_name, &server_name, &server_ip, server_port);
if (status == NC_OK) {
conf_rewrite(ctx);
}
status = NC_OK;
done:
if (line_buf != NULL) {
mbuf_put(line_buf);
}
string_deinit(&tmp_string);
string_deinit(&server_ip);
string_deinit(&server_name);
string_deinit(&pool_name);
return status;
error:
status = NC_ERROR;
goto done;
}
static rstatus_t
sentinel_proc_acksub(struct context *ctx, struct msg *msg)
{
rstatus_t status;
struct string sub_titile, sub_channel, sub_ok, tmp_string;
struct mbuf *line_buf;
string_init(&tmp_string);
string_set_text(&sub_titile, "psubscribe");
string_set_text(&sub_channel, "+switch-master");
string_set_text(&sub_ok, ":1");
line_buf = mbuf_get();
if (line_buf == NULL) {
goto error;
}
/* get line in line num 3 for sub titile */
msg_read_line(msg, line_buf, 3);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel ack sub when skip line not used.");
goto error;
}
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK || string_compare(&sub_titile, &tmp_string)) {
goto error;
}
/* get line in line num 5 for sub channel */
msg_read_line(msg, line_buf, 2);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel ack sub when skip line not used.");
goto error;
}
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK || string_compare(&sub_channel, &tmp_string)) {
goto error;
}
/* get sub status */
msg_read_line(msg, line_buf, 1);
if (line_buf == 0) {
log_error("read line failed from sentinel ack sub when skip line not used.");
goto error;
}
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK || string_compare(&sub_ok, &tmp_string)) {
goto error;
}
log_debug(LOG_INFO, "success sub channel %.*s from sentinel", sub_channel.len, sub_channel.data);
status = NC_OK;
done:
if (line_buf != NULL) {
mbuf_put(line_buf);
}
string_deinit(&tmp_string);
return status;
error:
status = NC_ERROR;
goto done;
}
static rstatus_t
sentinel_proc_sentinel_info(struct context *ctx, struct msg *msg)
{
rstatus_t status;
int i, master_num, switch_num;
struct string pool_name, server_name, server_ip,
tmp_string, sentinel_masters_prefix, master_ok;
struct mbuf *line_buf;
int server_port;
string_init(&tmp_string);
string_init(&pool_name);
string_init(&server_name);
string_init(&server_ip);
string_set_text(&sentinel_masters_prefix, "sentinel_masters");
string_set_text(&master_ok, "status=ok");
line_buf = mbuf_get();
if (line_buf == NULL) {
goto error;
}
/* get sentinel master num at line 3 */
msg_read_line(msg, line_buf, 3);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel ack info when skip line not used.");
goto error;
}
status = mbuf_read_string(line_buf, ':', &tmp_string);
if (status != NC_OK || string_compare(&sentinel_masters_prefix, &tmp_string)) {
goto error;
}
status = mbuf_read_string(line_buf, CR, &tmp_string);
if (status != NC_OK) {
goto error;
}
master_num = nc_atoi(tmp_string.data, tmp_string.len);
if (master_num < 0) {
log_error("parse master number from sentinel ack info failed.");
goto error;
}
/* skip 3 line in ack info which is not used. */
msg_read_line(msg, line_buf, 3);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel ack info when skip line not used.");
goto error;
}
/* parse master info from sentinel ack info */
switch_num = 0;
for (i = 0; i < master_num; i++) {
msg_read_line(msg, line_buf, 1);
if (mbuf_length(line_buf) == 0) {
log_error("read line failed from sentinel ack info when parse master item.");
goto error;
}
log_debug(LOG_INFO, "master item line : %.*s", mbuf_length(line_buf), line_buf->pos);
/* skip master item prefix */
status = mbuf_read_string(line_buf, ':', NULL);
if (status != NC_OK) {
log_error("skip master item prefix failed");
goto error;
}
/* skip master item server name prefix */
status = mbuf_read_string(line_buf, '=', NULL);
if (status != NC_OK) {
log_error("skip master item server name prefix failed.");
goto error;
}
/* get server pool name */
status = mbuf_read_string(line_buf, SENTINEL_SERVERNAME_SPLIT, &pool_name);
if (status != NC_OK) {
log_error("get server pool name failed.");
goto error;
}
/* get server name */
status = mbuf_read_string(line_buf, ',', &server_name);
if (status != NC_OK) {
log_error("get server name failed.");
goto error;
}
/* get master status */
status = mbuf_read_string(line_buf, ',', &tmp_string);
if (status != NC_OK) {
log_error("get master status failed.");
goto error;
}
if (string_compare(&master_ok, &tmp_string)) {
log_error("master item status is not ok, use it anyway");
}
/* skip ip string prefix name */
status = mbuf_read_string(line_buf, '=', NULL);
if (status != NC_OK) {
log_error("skip master item address prefix failed.");
goto error;
}
/* get server ip string */
status = mbuf_read_string(line_buf, ':', &server_ip);
if (status != NC_OK) {
log_error("get server ip string failed.");
goto error;
}
/* get server port */
status = mbuf_read_string(line_buf, ',', &tmp_string);
if (status != NC_OK) {
log_error("get server port string failed.");
goto error;
}
server_port = nc_atoi(tmp_string.data, tmp_string.len);
if (server_port < 0) {
log_error("tanslate server port string to int failed.");
goto error;
}
status = server_switch(ctx, &pool_name, &server_name, &server_ip, server_port);
/* if server is switched, add switch number */
if (status == NC_OK) {
switch_num++;
}
}
if (switch_num > 0) {
conf_rewrite(ctx);
}
status = NC_OK;
done:
if (line_buf != NULL) {
mbuf_put(line_buf);
}
string_deinit(&tmp_string);
string_deinit(&pool_name);
string_deinit(&server_name);
string_deinit(&server_ip);
return status;
error:
status = NC_ERROR;
goto done;
}
struct stats *
stats_create(uint16_t stats_port, char *stats_ip, int stats_interval,
char *source, struct array *server_pool, struct context *ctx)
{
rstatus_t status;
struct stats *st;
st = dn_alloc(sizeof(*st));
if (st == NULL) {
return NULL;
}
st->port = stats_port;
st->interval = stats_interval;
string_set_raw(&st->addr, stats_ip);
st->start_ts = (int64_t)time(NULL);
st->buf.len = 0;
st->buf.data = NULL;
st->buf.size = 0;
array_null(&st->current);
array_null(&st->shadow);
array_null(&st->sum);
st->tid = (pthread_t) -1;
st->sd = -1;
string_set_text(&st->service_str, "service");
string_set_text(&st->service, "dynomite");
string_set_text(&st->source_str, "source");
string_set_raw(&st->source, source);
string_set_text(&st->version_str, "version");
string_set_text(&st->version, DN_VERSION_STRING);
string_set_text(&st->uptime_str, "uptime");
string_set_text(&st->timestamp_str, "timestamp");
//for latency histo
string_set_text(&st->latency_999th_str, "latency_999th");
string_set_text(&st->latency_99th_str, "latency_99th");
string_set_text(&st->latency_95th_str, "latency_95th");
string_set_text(&st->latency_mean_str, "latency_mean");
string_set_text(&st->latency_max_str, "latency_max");
//for payload size histo
string_set_text(&st->payload_size_999th_str, "payload_size_999th");
string_set_text(&st->payload_size_99th_str, "payload_size_99th");
string_set_text(&st->payload_size_95th_str, "payload_size_95th");
string_set_text(&st->payload_size_mean_str, "payload_size_mean");
string_set_text(&st->payload_size_max_str, "payload_size_max");
string_set_text(&st->alloc_msgs_str, "alloc_msgs");
//only display the first pool
struct server_pool *sp = (struct server_pool*) array_get(server_pool, 0);
string_set_text(&st->rack_str, "rack");
string_copy(&st->rack, sp->rack.data, sp->rack.len);
string_set_text(&st->dc_str, "dc");
string_copy(&st->dc, sp->dc.data, sp->dc.len);
st->updated = 0;
st->aggregate = 0;
histo_init(&st->latency_histo);
histo_init(&st->payload_size_histo);
st->alloc_msgs = 0;
/* map server pool to current (a), shadow (b) and sum (c) */
status = stats_pool_map(&st->current, server_pool);
if (status != DN_OK) {
goto error;
}
status = stats_pool_map(&st->shadow, server_pool);
if (status != DN_OK) {
goto error;
}
status = stats_pool_map(&st->sum, server_pool);
if (status != DN_OK) {
goto error;
}
status = stats_create_buf(st);
if (status != DN_OK) {
goto error;
}
status = stats_start_aggregator(st);
if (status != DN_OK) {
goto error;
}
st->ctx = ctx;
return st;
error:
stats_destroy(st);
return NULL;
}
