396 lines
16 KiB
C++
396 lines
16 KiB
C++
#include <sys/stat.h>
|
||
#include <sys/types.h>
|
||
#include <unistd.h>
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <assert.h>
|
||
#include <errno.h>
|
||
#include <pthread.h>
|
||
#include <string.h>
|
||
|
||
#include <event2/thread.h>
|
||
|
||
#include <MESA/MESA_prof_load.h>
|
||
|
||
#include "doris_client.h"
|
||
#include "doris_server_main.h"
|
||
#include "doris_server_http.h"
|
||
|
||
struct doris_global_info g_doris_server_info;
|
||
static unsigned long doris_vesion_20210803=20210803L;
|
||
|
||
int doris_mkdir_according_path(const char * path)
|
||
{
|
||
char buffer[256];
|
||
const char *ps=path, *pc;
|
||
|
||
if(*ps == '/')
|
||
ps += 1;
|
||
|
||
while((pc = strchr(ps, '/')) != NULL)
|
||
{
|
||
while(*(pc+1) == '/')
|
||
pc++;
|
||
|
||
memcpy(buffer, path, pc - path);
|
||
buffer[pc-path] = '\0';
|
||
|
||
if(access(buffer, F_OK))
|
||
{
|
||
if(mkdir(buffer, 0777))
|
||
{
|
||
return -1;
|
||
}
|
||
}
|
||
ps = pc + 1;
|
||
}
|
||
if(access(path, F_OK))
|
||
{
|
||
if(mkdir(path, 0777))
|
||
{
|
||
return -1;
|
||
}
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
static int doris_chech_name_valid(const char *name)
|
||
{
|
||
size_t i, namelen=strlen(name);
|
||
|
||
for(i=0; i<namelen; i++)
|
||
{
|
||
if(!((name[i]>='a' && name[i]<='z')||(name[i]>='A' && name[i]<='Z') ||
|
||
(name[i]>='0' && name[i]<='9') || name[i]=='_' || name[i]==':'))
|
||
{
|
||
return false;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
int32_t doris_read_profile_configs(const char *config_file)
|
||
{
|
||
char tmp_buf[4096], tmp_dir[512];
|
||
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "run_log_dir", g_doris_server_info.root_log_dir, sizeof(g_doris_server_info.root_log_dir), "./log");
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "run_log_lv", &g_doris_server_info.log_level, 10);
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "statistic_log_interval", &g_doris_server_info.statistic_period, 300);
|
||
/*runtimelog*/
|
||
snprintf(tmp_dir, 256, "%s/runtime_log", g_doris_server_info.root_log_dir);
|
||
if(doris_mkdir_according_path(tmp_dir))
|
||
{
|
||
printf("mkdir %s for runtimelog failed: %s\n", tmp_dir, strerror(errno));
|
||
return -1;
|
||
}
|
||
snprintf(tmp_buf, 256, "%s/doris_runtime.log", tmp_dir);
|
||
g_doris_server_info.log_runtime = MESA_create_runtime_log_handle(tmp_buf, g_doris_server_info.log_level);
|
||
if(NULL==g_doris_server_info.log_runtime)
|
||
{
|
||
printf("MESA_create_runtime_log_handle %s failed: %s\n", tmp_buf, strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
/*System*/
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "worker_thread_num", &g_doris_server_info.iothreads, 1);
|
||
if(g_doris_server_info.iothreads > 255)
|
||
{
|
||
printf("%s: [DORIS_SERVER]worker_thread_num=%d should not be bigger than 255!", config_file, g_doris_server_info.iothreads);
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "%s: [DORIS_SERVER]worker_thread_num=%d should not be bigger than 255!", config_file, g_doris_server_info.iothreads);
|
||
assert(0);return -1;
|
||
}
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "cache_file_frag_size", &g_doris_server_info.cache_frag_size, 67108864);
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "doris_server_role_on", &g_doris_server_info.server_role_sw, 1);
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "index_file_format_maat", &g_doris_server_info.idx_file_maat, 0);
|
||
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "scan_index_file_interval", &g_doris_server_info.scan_idx_interval, 10);
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "https_connection_on", &g_doris_server_info.ssl_conn_on, 0);
|
||
if(g_doris_server_info.ssl_conn_on)
|
||
{
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "ssl_trusted_ca_path", g_doris_server_info.ssl_CA_path, 256, "./conf/ssl_CA_path/");
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "ssl_certificate_file", g_doris_server_info.ssl_cert_file, 256, "./conf/server_cert.pem");
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "ssl_private_key_file", g_doris_server_info.ssl_key_file, 256, "./conf/server_key.pem");
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "ssl_private_key_passwd", g_doris_server_info.ssl_key_passwd, 64, "12345678");
|
||
}
|
||
|
||
/*FiledStat*/
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "fsstat_log_appname", g_doris_server_info.fsstat_appname, 16, "DORIS_SERVER_S");
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "fsstat_log_filepath", g_doris_server_info.fsstat_filepath, 256, "./log/doris_server.fs");
|
||
MESA_load_profile_uint_def(config_file, "DORIS_SERVER", "fsstat_log_interval", &g_doris_server_info.fsstat_period, 10);
|
||
MESA_load_profile_int_def(config_file, "DORIS_SERVER", "fsstat_log_print_mode", &g_doris_server_info.fsstat_print_mode, 1);
|
||
MESA_load_profile_string_def(config_file, "DORIS_SERVER", "fsstat_log_dst_ip", g_doris_server_info.fsstat_dst_ip, 64, "127.0.0.1");
|
||
MESA_load_profile_int_def(config_file, "DORIS_SERVER", "fsstat_log_dst_port", &g_doris_server_info.fsstat_dst_port, 8125);
|
||
|
||
//LIBEVENT
|
||
MESA_load_profile_int_def(config_file, "DORIS_SERVER", "server_listen_port", &g_doris_server_info.server_port, 9898);
|
||
MESA_load_profile_int_def(config_file, "DORIS_SERVER", "manage_listen_port", &g_doris_server_info.manager_port, 2233);
|
||
MESA_load_profile_int_def(config_file, "DORIS_SERVER", "socket_recv_bufsize", &g_doris_server_info.sock_recv_bufsize, 524288);
|
||
return 0;
|
||
}
|
||
|
||
static int doris_server_register_field_stat(struct doris_global_info *param)
|
||
{
|
||
const char *field_names[DRS_FSSTAT_FIELD_NUM]={"RecvErrVer", "FileStarts", "FileComplete", "ClientInvReq",
|
||
"ClientMetaReq", "SendNoNewMeta", "ClientFileReq", "SendBytes", "SendFile404"};
|
||
const char *status_names[DRS_FSSTAT_STATUS_NUM]={"MemoryUsed"};
|
||
const char *column_names[DRS_FSSTAT_CLUMN_NUM]={"RecvFullVer", "RecvIncVer", "RecvFiles", "SendResMeta", "SendFiles",
|
||
"CurFullVer", "CurIncVer", "TotalCfgNum"};
|
||
int value;
|
||
|
||
param->fsstat_handle = FS_create_handle();
|
||
FS_set_para(param->fsstat_handle, OUTPUT_DEVICE, param->fsstat_filepath, strlen(param->fsstat_filepath)+1);
|
||
if(param->fsstat_print_mode == 1)
|
||
{
|
||
FS_set_para(param->fsstat_handle, PRINT_MODE, ¶m->fsstat_print_mode, sizeof(param->fsstat_print_mode));
|
||
}
|
||
else
|
||
{
|
||
FS_set_para(param->fsstat_handle, PRINT_MODE, ¶m->fsstat_print_mode, sizeof(param->fsstat_print_mode));
|
||
value = 1;
|
||
FS_set_para(param->fsstat_handle, FLUSH_BY_DATE, &value, sizeof(value));
|
||
}
|
||
value = param->fsstat_period;
|
||
FS_set_para(param->fsstat_handle, STAT_CYCLE, &value, sizeof(value));
|
||
value = 0;
|
||
FS_set_para(param->fsstat_handle, CREATE_THREAD, &value, sizeof(value));
|
||
FS_set_para(param->fsstat_handle, APP_NAME, param->fsstat_appname, strlen(param->fsstat_appname)+1);
|
||
FS_set_para(param->fsstat_handle, STATS_SERVER_IP, param->fsstat_dst_ip, strlen(param->fsstat_dst_ip)+1);
|
||
FS_set_para(param->fsstat_handle, STATS_SERVER_PORT, ¶m->fsstat_dst_port, sizeof(param->fsstat_dst_port));
|
||
|
||
for(int i=0; i<DRS_FSSTAT_FIELD_NUM; i++)
|
||
{
|
||
param->fsstat_field[i] = FS_register(param->fsstat_handle, FS_STYLE_FIELD, FS_CALC_CURRENT, field_names[i]);
|
||
}
|
||
for(int i=0; i<DRS_FSSTAT_STATUS_NUM; i++)
|
||
{
|
||
param->fsstat_status[i] = FS_register(param->fsstat_handle, FS_STYLE_STATUS, FS_CALC_CURRENT, status_names[i]);
|
||
}
|
||
for(int i=0; i<DRS_FSSTAT_CLUMN_NUM; i++)
|
||
{
|
||
param->fsstat_column[i] = FS_register(param->fsstat_handle, FS_STYLE_COLUMN, FS_CALC_CURRENT, column_names[i]);
|
||
}
|
||
FS_start(param->fsstat_handle);
|
||
return 0;
|
||
}
|
||
|
||
static void instance_fsstat_output_timer_cb(int fd, short kind, void *userp)
|
||
{
|
||
struct timeval tv;
|
||
|
||
FS_passive_output(g_doris_server_info.fsstat_handle);
|
||
|
||
tv.tv_sec = g_doris_server_info.fsstat_period;
|
||
tv.tv_usec = 0;
|
||
event_add((struct event*)userp, &tv);
|
||
}
|
||
|
||
static int32_t doris_init_config_for_business(struct doris_global_info *info, struct event_base *manage_evbase, const char *config_file)
|
||
{
|
||
char tmpbuffer[4096], tmp_dir[256], tmp_dir2[256], *bizname, *save=NULL;
|
||
struct doris_business *business;
|
||
map<string, struct doris_parameter *>::iterator iter;
|
||
|
||
if(0>=MESA_load_profile_string_nodef(config_file, "DORIS_SERVER", "business_system_list", tmpbuffer, sizeof(tmpbuffer)))
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "%s: [DORIS_SERVER]business_system_list not found!", config_file);
|
||
assert(0);return -1;
|
||
}
|
||
for(bizname=strtok_r(tmpbuffer, ";", &save); bizname!=NULL; bizname=strtok_r(NULL, ";", &save))
|
||
{
|
||
if(!doris_chech_name_valid(bizname))
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "%s: [DORIS_SERVER]business_system_list bizname %s invalid, name must match: [a-zA-Z_:][a-zA-Z0-9_:]", bizname, config_file);
|
||
assert(0);return -1;
|
||
}
|
||
business = &info->business[info->business_num++];
|
||
snprintf(business->bizname, sizeof(business->bizname), "%s", bizname);
|
||
pthread_rwlock_init(&business->rwlock, NULL);
|
||
|
||
business->cfgver_head = config_version_handle_new();
|
||
if(info->name2business->find(string(business->bizname)) != info->name2business->end())
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "%s: [%s]business_system_list duplicate system name: %s!", bizname, config_file, business->bizname);
|
||
assert(0);return -1;
|
||
}
|
||
info->name2business->insert(make_pair(string(business->bizname), business));
|
||
|
||
MESA_load_profile_uint_def(config_file, business->bizname, "grafana_monitor_status_id", &business->mm_status_codeid, 3);
|
||
MESA_load_profile_uint_def(config_file, business->bizname, "doris_write_file_on", &business->write_file_sw, 1);
|
||
if(0>MESA_load_profile_string_nodef(config_file, business->bizname, "store_config_path", business->store_path_root, sizeof(business->store_path_root)))
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "%s: [%s]store_config_path not found!", bizname, config_file);
|
||
assert(0);return -1;
|
||
}
|
||
snprintf(tmp_dir, 512, "%s/full/index", business->store_path_root);
|
||
snprintf(tmp_dir2,512, "%s/inc/index", business->store_path_root);
|
||
if(doris_mkdir_according_path(tmp_dir) || doris_mkdir_according_path(tmp_dir2))
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "mkdir %s failed: %s\n", tmp_dir, strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
MESA_load_profile_uint_def(config_file, business->bizname, "receive_config_way", &business->recv_way, RECV_WAY_DRS_CLIENT);
|
||
if(business->recv_way == RECV_WAY_DRS_CLIENT)
|
||
{
|
||
if(0>=MESA_load_profile_string_nodef(config_file, business->bizname, "doris_client_confile", tmp_dir, sizeof(tmp_dir)))
|
||
{
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "%s: [DORIS_SERVER]doris_client_confile not found!", config_file);
|
||
assert(0);return -1;
|
||
}
|
||
if((iter=info->confile2param->find(string(tmp_dir))) != info->confile2param->end())
|
||
{
|
||
business->param = iter->second;
|
||
}
|
||
else
|
||
{
|
||
business->param = doris_parameter_new(tmp_dir, manage_evbase, info->log_runtime);
|
||
if(business->param == NULL)
|
||
{
|
||
assert(0);return -2;
|
||
}
|
||
info->confile2param->insert(make_pair(string(tmp_dir), business->param));
|
||
}
|
||
}
|
||
else
|
||
{
|
||
if(0>MESA_load_profile_string_nodef(config_file, business->bizname, "receive_config_path_full", business->recv_path_full, sizeof(business->recv_path_full)))
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "%s: [%s]receive_config_path not found!", bizname, config_file);
|
||
assert(0);return -1;
|
||
}
|
||
if(0>MESA_load_profile_string_nodef(config_file, business->bizname, "receive_config_path_inc", business->recv_path_inc, sizeof(business->recv_path_inc)))
|
||
{
|
||
MESA_RUNTIME_LOGV3(info->log_runtime, RLOG_LV_FATAL, "%s: [%s]receive_config_path not found!", bizname, config_file);
|
||
assert(0);return -1;
|
||
}
|
||
}
|
||
business->fs_lineid = FS_register(info->fsstat_handle, FS_STYLE_LINE, FS_CALC_CURRENT, business->bizname);;
|
||
|
||
snprintf(tmp_dir, 512, "latest_cfg_version_%s", business->bizname);
|
||
business->mm_latest_ver = MESA_Monitor_register(info->monitor, tmp_dir, MONITOR_METRICS_GAUGE, "Latest doris config version.");
|
||
snprintf(tmp_dir, 512, "total_config_num_%s", business->bizname);
|
||
business->mm_total_cfgnum = MESA_Monitor_register(info->monitor, tmp_dir, MONITOR_METRICS_GAUGE, "Total config num from latest full version till now.");
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
static void manager_statistic_threads_requests_cb(struct evhttp_request *req, void *arg)
|
||
{
|
||
evhttp_send_error(req, HTTP_BADREQUEST, "Not Supported.");
|
||
}
|
||
|
||
static void manager_statistic_status_requests_cb(struct evhttp_request *req, void *arg)
|
||
{
|
||
evhttp_send_error(req, HTTP_BADREQUEST, "Not Supported.");
|
||
}
|
||
|
||
static void manager_statistic_field_requests_cb(struct evhttp_request *req, void *arg)
|
||
{
|
||
evhttp_send_error(req, HTTP_BADREQUEST, "Not Supported.");
|
||
}
|
||
|
||
void manager_generic_requests_cb(struct evhttp_request *req, void *arg)
|
||
{
|
||
evhttp_send_error(req, HTTP_BADREQUEST, "Not Supported.");
|
||
}
|
||
|
||
int main(int argc, char **argv)
|
||
{
|
||
struct event_base *manage_evbase;
|
||
pthread_t thread_desc;
|
||
pthread_attr_t attr;
|
||
struct timeval tv;
|
||
struct event timer_statistic_fsstat;
|
||
struct evhttp *manager_http;
|
||
|
||
if(doris_read_profile_configs(NIRVANA_CONFIG_FILE) || doris_server_register_field_stat(&g_doris_server_info))
|
||
{
|
||
return -1;
|
||
}
|
||
evthread_use_pthreads();
|
||
g_doris_server_info.name2business = new map<string, struct doris_business*>;
|
||
g_doris_server_info.confile2param = new map<string, struct doris_parameter *>;
|
||
manage_evbase = event_base_new();
|
||
|
||
/*Doris manager server*/
|
||
g_doris_server_info.manager = doris_create_listen_socket(g_doris_server_info.manager_port);
|
||
if(g_doris_server_info.manager < 0)
|
||
{
|
||
return -6;
|
||
}
|
||
if(NULL == (manager_http = evhttp_new(manage_evbase)))
|
||
{
|
||
printf("evhttp_new error!\n");
|
||
assert(0); return -2;
|
||
}
|
||
evhttp_set_cb(manager_http, "/doris/statistic/field", manager_statistic_field_requests_cb, NULL);
|
||
evhttp_set_cb(manager_http, "/doris/statistic/status", manager_statistic_status_requests_cb, NULL);
|
||
evhttp_set_cb(manager_http, "/doris/statistic/threads", manager_statistic_threads_requests_cb, NULL);
|
||
evhttp_set_gencb(manager_http, manager_generic_requests_cb, NULL);
|
||
g_doris_server_info.monitor = MESA_Monitor_instance_evhttp_new(manager_http, doris_vesion_20210803);
|
||
if(evhttp_accept_socket(manager_http, g_doris_server_info.manager))
|
||
{
|
||
printf("evhttp_accept_socket %d error!\n", g_doris_server_info.manager);
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "evhttp_accept_socket %d error!\n", g_doris_server_info.manager);
|
||
assert(0); return -7;
|
||
}
|
||
|
||
//Ϊÿ<CEAA><C3BF>ҵ<EFBFBD><D2B5>ϵͳ<CFB5><CDB3>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD>ȡ<EFBFBD><C8A1><EFBFBD>õĽṹ
|
||
if(doris_init_config_for_business(&g_doris_server_info, manage_evbase, NIRVANA_CONFIG_FILE))
|
||
{
|
||
return -8;
|
||
}
|
||
|
||
pthread_attr_init(&attr);
|
||
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
|
||
for(u_int32_t i=0; i<g_doris_server_info.business_num; i++)
|
||
{
|
||
if(g_doris_server_info.business[i].recv_way == RECV_WAY_DRS_CLIENT)
|
||
{
|
||
if(pthread_create(&thread_desc, &attr, thread_doris_client_recv_cfg, &g_doris_server_info.business[i]))
|
||
{
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "pthread_create(): %s", strerror(errno));
|
||
assert(0);return -4;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
if(pthread_create(&thread_desc, &attr, thread_index_file_recv_cfg, &g_doris_server_info.business[i]))
|
||
{
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "pthread_create(): %s", strerror(errno));
|
||
assert(0);return -4;
|
||
}
|
||
}
|
||
}
|
||
|
||
/*Doris http server*/
|
||
if(g_doris_server_info.server_role_sw)
|
||
{
|
||
g_doris_server_info.listener = doris_create_listen_socket(g_doris_server_info.server_port);
|
||
if(g_doris_server_info.listener < 0)
|
||
{
|
||
return -5;
|
||
}
|
||
for(u_int32_t i=0; i<g_doris_server_info.iothreads; i++)
|
||
{
|
||
if(pthread_create(&thread_desc, &attr, thread_doris_http_server, NULL))
|
||
{
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "pthread_create(): %s", strerror(errno));
|
||
assert(0);return -4;
|
||
}
|
||
}
|
||
}
|
||
|
||
tv.tv_sec = g_doris_server_info.fsstat_period;
|
||
tv.tv_usec = 0;
|
||
evtimer_assign(&timer_statistic_fsstat, manage_evbase, instance_fsstat_output_timer_cb, &timer_statistic_fsstat);
|
||
evtimer_add(&timer_statistic_fsstat, &tv);
|
||
|
||
event_base_dispatch(manage_evbase);
|
||
printf("Libevent dispath error, should not run here.\n");
|
||
MESA_RUNTIME_LOGV3(g_doris_server_info.log_runtime, RLOG_LV_FATAL, "Libevent dispath error, should not run here.");
|
||
return 0;
|
||
}
|
||
|
||
|