127 lines
4.4 KiB
C
127 lines
4.4 KiB
C
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <pthread.h>
|
|
#include <unistd.h>
|
|
#include "http_decoder_private.h"
|
|
|
|
static const struct hd_stat_config_tuple g_httpd_stat_tuple[HTTP_STAT_MAX] =
|
|
{
|
|
{HTTP_C2S_BYTES, "http_c2s_bytes"},
|
|
{HTTP_S2C_BYTES, "http_s2c_bytes"},
|
|
{HTTP_C2S_TCP_SEG, "http_c2s_tcp_seg"},
|
|
{HTTP_S2C_TCP_SEG, "http_s2c_tcp_seg"},
|
|
{HTTP_C2S_HEADERS, "http_c2s_headers"},
|
|
{HTTP_S2C_HEADERS, "http_s2c_headers"},
|
|
{HTTP_C2S_ZIP_BYTES, "http_c2s_zip_bytes"},
|
|
{HTTP_S2C_ZIP_BYTES, "http_s2c_zip_bytes"},
|
|
{HTTP_C2S_UNZIP_BYTES, "http_c2s_unzip_bytes"},
|
|
{HTTP_S2C_UNZIP_BYTES, "http_s2c_unzip_bytes"},
|
|
{HTTP_URL_BYTES, "http_url_bytes"},
|
|
{HTTP_SESSION_NEW, "http_session_new"},
|
|
{HTTP_SESSION_FREE, "http_session_free"},
|
|
{HTTP_TRANSACTION_NEW, "http_transaction_new"},
|
|
{HTTP_TRANSACTION_FREE, "http_transaction_free"},
|
|
{HTTP_C2S_ASYMMETRY_SESSION, "http_c2s_asymmetry_sess"},
|
|
{HTTP_S2C_ASYMMETRY_SESSION, "http_s2c_asymmetry_sess"},
|
|
{HTTP_C2S_ASYMMETRY_TRANSACTION, "http_c2s_asymmetry_trans"},
|
|
{HTTP_S2C_ASYMMETRY_TRANSACTION, "http_s2c_asymmetry_trans"},
|
|
{HTTP_STAT_PARSE_ERR, "http_parse_error"},
|
|
};
|
|
|
|
void http_decoder_stat_free(struct http_decoder_stat *hd_stat)
|
|
{
|
|
if (hd_stat->timer_pid != 0)
|
|
{
|
|
pthread_cancel(hd_stat->timer_pid);
|
|
}
|
|
if (hd_stat->stats != NULL)
|
|
{
|
|
free(hd_stat->stats);
|
|
}
|
|
if (hd_stat->fse != NULL)
|
|
{
|
|
fieldstat_easy_free(hd_stat->fse);
|
|
}
|
|
}
|
|
|
|
static void *httpd_stat_timer_thread(void *arg)
|
|
{
|
|
pthread_setname_np(pthread_self(), "http_decoder_timer_thread");
|
|
struct http_decoder_stat *hd_stat = (struct http_decoder_stat *)arg;
|
|
struct timespec res;
|
|
while (1)
|
|
{
|
|
clock_gettime(CLOCK_MONOTONIC, &res);
|
|
hd_stat->current_time_ms = (res.tv_sec * 1000) + (res.tv_nsec / 1000000);
|
|
usleep(800);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
int http_decoder_stat_init(struct http_decoder_stat *hd_stat, int thread_max, int stat_interval_pkts, int stat_interval_time)
|
|
{
|
|
assert(sizeof(g_httpd_stat_tuple) / sizeof(struct hd_stat_config_tuple) == HTTP_STAT_MAX);
|
|
if (sizeof(g_httpd_stat_tuple) / sizeof(struct hd_stat_config_tuple) != HTTP_STAT_MAX)
|
|
{
|
|
fprintf(stderr, "enum http_decoder_stat_type number not match with g_httpd_stat_tuple!");
|
|
return -1;
|
|
}
|
|
hd_stat->fse = fieldstat_easy_new(thread_max, "http_decoder_statistics", NULL, 0);
|
|
if (NULL == hd_stat->fse)
|
|
{
|
|
fprintf(stderr, "fieldstat_easy_new failed.");
|
|
return -1;
|
|
}
|
|
|
|
for (int i = 0; i < HTTP_STAT_MAX; i++)
|
|
{
|
|
hd_stat->field_stat_id[i] = fieldstat_easy_register_counter(hd_stat->fse, g_httpd_stat_tuple[i].name);
|
|
if (hd_stat->field_stat_id[i] < 0)
|
|
{
|
|
fprintf(stderr, "fieldstat_easy_register_counter %s failed.", g_httpd_stat_tuple[i].name);
|
|
fieldstat_easy_free(hd_stat->fse);
|
|
hd_stat->fse = NULL;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int ret = fieldstat_easy_enable_auto_output(hd_stat->fse, FILEDSTAT_OUTPUT_FILE, stat_interval_time);
|
|
if (ret < 0)
|
|
{
|
|
fprintf(stderr, "fieldstat_easy_enable_auto_output failed.");
|
|
fieldstat_easy_free(hd_stat->fse);
|
|
hd_stat->fse = NULL;
|
|
return -1;
|
|
}
|
|
hd_stat->stats = (struct hd_statistics *)calloc(thread_max, sizeof(struct hd_statistics));
|
|
hd_stat->stat_interval_pkts = stat_interval_pkts;
|
|
hd_stat->stat_interval_time = stat_interval_time;
|
|
pthread_create(&hd_stat->timer_pid, NULL, httpd_stat_timer_thread, hd_stat);
|
|
pthread_detach(hd_stat->timer_pid);
|
|
return 0;
|
|
}
|
|
|
|
void http_decoder_stat_update(struct http_decoder_stat *hd_stat, int thread_id, enum http_decoder_stat_type type, long long value)
|
|
{
|
|
assert(hd_stat);
|
|
assert(thread_id >= 0);
|
|
assert(type < HTTP_STAT_MAX);
|
|
|
|
if (unlikely(hd_stat->stats == NULL))
|
|
{
|
|
return;
|
|
}
|
|
|
|
struct hd_statistics *cur_hds = &hd_stat->stats[thread_id];
|
|
|
|
cur_hds->counter[type] += value;
|
|
cur_hds->batch[type]++;
|
|
|
|
if (cur_hds->batch[type] >= hd_stat->stat_interval_pkts || cur_hds->time_ms[type] + 1000 < hd_stat->current_time_ms)
|
|
{
|
|
fieldstat_easy_counter_incrby(hd_stat->fse, thread_id, hd_stat->field_stat_id[type], NULL, 0, cur_hds->counter[type]);
|
|
cur_hds->counter[type] = 0;
|
|
cur_hds->batch[type] = 0;
|
|
cur_hds->time_ms[type] = hd_stat->current_time_ms;
|
|
}
|
|
} |