原有功能回归测试通过。

This commit is contained in:
zhengchao
2018-09-24 19:48:18 +08:00
parent e2f4a583ad
commit 4bb67bf7fa
7 changed files with 82 additions and 65 deletions

View File

@@ -776,6 +776,7 @@ void maat_read_full_config(_Maat_feather_t* _feather)
int Maat_initiate_feather(Maat_feather_t feather) int Maat_initiate_feather(Maat_feather_t feather)
{ {
_Maat_feather_t* _feather=(_Maat_feather_t*)feather; _Maat_feather_t* _feather=(_Maat_feather_t*)feather;
system_cmd_mkdir(_feather->foreign_cont_dir);
if(_feather->DEFERRED_LOAD_ON==0) if(_feather->DEFERRED_LOAD_ON==0)
{ {
maat_read_full_config(_feather); maat_read_full_config(_feather);

View File

@@ -1463,6 +1463,7 @@ int get_foreign_keys(redisContext *ctx, struct serial_rule_t* rule_list, int rul
const char* foreign_key_prefix="__FILE_"; const char* foreign_key_prefix="__FILE_";
const char* p_foregin=NULL; const char* p_foregin=NULL;
struct _Maat_table_info_t* p_table=NULL; struct _Maat_table_info_t* p_table=NULL;
struct stat file_info;
for(i=0; i<rule_num; i++) for(i=0; i<rule_num; i++)
{ {
if(rule_list[i].table_line==NULL) if(rule_list[i].table_line==NULL)
@@ -1511,21 +1512,29 @@ int get_foreign_keys(redisContext *ctx, struct serial_rule_t* rule_list, int rul
memcpy(rule_list[i].f_keys[j].key, p_foregin, foregin_key_size); memcpy(rule_list[i].f_keys[j].key, p_foregin, foregin_key_size);
rule_list[i].f_keys[j].filename=get_foreign_cont_filename(rule_list[i].table_name, rule_list[i].rule_id, p_foregin, dir); rule_list[i].f_keys[j].filename=get_foreign_cont_filename(rule_list[i].table_name, rule_list[i].rule_id, p_foregin, dir);
ret=stat(rule_list[i].f_keys[j].filename, &file_info);
if(ret==0)
{
rule_list[i].f_keys[j].is_existed=1;
}
} }
rule_with_foreign_key++; rule_with_foreign_key++;
} }
return rule_with_foreign_key; return rule_with_foreign_key;
} }
struct foreign_conts_track
void get_foreign_conts(redisContext *ctx, struct serial_rule_t* rule_list, int rule_num, void *logger) {
int rule_idx;
int foreign_idx;
};
void _get_foreign_conts(redisContext *ctx, struct serial_rule_t* rule_list, int rule_num, void *logger)
{ {
int i=0, j=0, ret=0; int i=0, j=0, ret=0;
int key_num=0; int key_num=0;
struct foreign_conts_track* track=ALLOC(struct foreign_conts_track, rule_num*MAX_FOREIGN_CLMN_NUM);
char redis_cmd[256]; char redis_cmd[256];
redisReply* reply=NULL; redisReply* reply=NULL;
struct serial_rule_t*p=NULL; struct serial_rule_t*p=NULL;
char** filenames[MAX_FOREIGN_CLMN_NUM];
FILE* fp=NULL; FILE* fp=NULL;
for(i=0;i<rule_num;i++) for(i=0;i<rule_num;i++)
{ {
@@ -1536,39 +1545,39 @@ void get_foreign_conts(redisContext *ctx, struct serial_rule_t* rule_list, int r
} }
for(j=0; j<p->n_foreign; j++) for(j=0; j<p->n_foreign; j++)
{ {
if(p->f_keys[i].is_existed==1)
{
continue;
}
snprintf(redis_cmd,sizeof(redis_cmd),"GET %s", p->f_keys[j].key); snprintf(redis_cmd,sizeof(redis_cmd),"GET %s", p->f_keys[j].key);
ret=redisAppendCommand(ctx, redis_cmd); ret=redisAppendCommand(ctx, redis_cmd);
track[key_num].rule_idx=i;
track[key_num].foreign_idx=j;
key_num++; key_num++;
assert(ret==REDIS_OK); assert(ret==REDIS_OK);
} }
} }
for(i=0;i<rule_num;i++) for(i=0;i<key_num;i++)
{
p=rule_list+i;
if(p->op==MAAT_OP_DEL||p->n_foreign==0)
{
continue;
}
for(j=0; j<p->n_foreign; j++)
{ {
ret=_wrap_redisGetReply(ctx,&reply); ret=_wrap_redisGetReply(ctx,&reply);
if(reply->type!=REDIS_REPLY_STRING) if(reply->type!=REDIS_REPLY_STRING)
{ {
MESA_handle_runtime_log(logger, RLOG_LV_FATAL, maat_redis_monitor MESA_handle_runtime_log(logger, RLOG_LV_FATAL, maat_redis_monitor
,"Get %s,%d foreign key %s content failed." ,"Get %s,%d foreign key %s content failed."
,rule_list[i].table_name ,rule_list[track[i].rule_idx].table_name
,rule_list[i].rule_id ,rule_list[track[i].rule_idx].rule_id
,p->f_keys[j].key); ,rule_list[track[i].rule_idx].f_keys[track[i].foreign_idx].key);
continue; continue;
} }
else else
{ {
fp=fopen(p->f_keys[j].filename, "w"); p=rule_list+track[i].rule_idx;
fp=fopen(p->f_keys[track[i].foreign_idx].filename, "w");
if(fp==NULL) if(fp==NULL)
{ {
MESA_handle_runtime_log(logger, RLOG_LV_FATAL, maat_redis_monitor MESA_handle_runtime_log(logger, RLOG_LV_FATAL, maat_redis_monitor
, "Write foreign content failed: fopen %s error." , "Write foreign content failed: fopen %s error."
, filenames[j]); , p->f_keys[track[i].foreign_idx]);
} }
else else
{ {
@@ -1579,10 +1588,21 @@ void get_foreign_conts(redisContext *ctx, struct serial_rule_t* rule_list, int r
} }
freeReplyObject(reply); freeReplyObject(reply);
} }
free(track);
return;
}
void get_foreign_conts(redisContext *ctx, struct serial_rule_t* rule_list, int rule_num, void *logger)
{
int max_redis_batch=4*1024,batch_cnt=0;
int success_cnt=0;
while(success_cnt<rule_num)
{
batch_cnt=MIN(rule_num-success_cnt,max_redis_batch);
_get_foreign_conts(ctx,rule_list+success_cnt,batch_cnt,logger);
success_cnt+=batch_cnt;
} }
return; return;
} }
void redis_monitor_traverse(long long version,redisContext *c, void redis_monitor_traverse(long long version,redisContext *c,
void (*start)(long long,int ,void*),//vesion,CM_UPDATE_TYPE_*,u_para void (*start)(long long,int ,void*),//vesion,CM_UPDATE_TYPE_*,u_para
int (*update)(const char* ,const char*,void* ),//table name ,line ,u_para int (*update)(const char* ,const char*,void* ),//table name ,line ,u_para

View File

@@ -604,11 +604,11 @@ int _read_integer_arrary(char* string, int *array, int size)
int read_plugin_table_info(const char* line, struct _Maat_table_info_t* p) int read_plugin_table_info(const char* line, struct _Maat_table_info_t* p)
{ {
int i=0,ret=0; int i=0,ret=0;
size_t j=0;
cJSON* json=NULL, *tmp=NULL;
char* copy_line=NULL, *plug_info=NULL; char* copy_line=NULL, *plug_info=NULL;
char *token=NULL,*sub_token=NULL,*saveptr; char *token=NULL,*sub_token=NULL,*saveptr;
const char* plug_info_prefix[]={"valid","tag","foreign"};
const int VALID_PREFIX_IDX=0, TAG_PREFIX_IDX=1, FOREIGN_PREFIX_IDX=2;
p->valid_flag_column=-1; p->valid_flag_column=-1;
copy_line=_maat_strdup(line); copy_line=_maat_strdup(line);
for (token = copy_line, i=0; ; token= NULL, i++) for (token = copy_line, i=0; ; token= NULL, i++)
@@ -620,9 +620,8 @@ int read_plugin_table_info(const char* line, struct _Maat_table_info_t* p)
{ {
break; break;
} }
} }
if(i<3) if(i<2)
{ {
goto error_out; goto error_out;
} }
@@ -638,35 +637,28 @@ int read_plugin_table_info(const char* line, struct _Maat_table_info_t* p)
free(copy_line); free(copy_line);
return 0; return 0;
} }
json=cJSON_Parse(plug_info);
if(!json)
{
goto error_out;
}
tmp=cJSON_GetObjectItem(json, "valid");
if(tmp!=NULL)
{
p->valid_flag_column=tmp->valueint;
}
tmp=cJSON_GetObjectItem(json, "tag");
if(tmp!=NULL)
{
p->rule_tag_column=tmp->valueint;
}
tmp=cJSON_GetObjectItem(json, "foreign");
if(tmp!=NULL)
{
_read_integer_arrary(tmp->string, p->foreign_columns, MAX_FOREIGN_CLMN_NUM);
}
cJSON_Delete(json);
for (token = plug_info; ; token= NULL)
{
sub_token= strtok_r(token,";", &saveptr);
if (sub_token == NULL)
break;
for(j=0; j< sizeof(plug_info_prefix)/sizeof(const char*); j++)
{
if(0==strncasecmp(sub_token, plug_info_prefix[j], strlen(plug_info_prefix[j])))
{
break;
}
}
switch(j)
{
case VALID_PREFIX_IDX:
sscanf(sub_token+strlen(plug_info_prefix[j])+1,"%d",&(p->valid_flag_column));
break;
case TAG_PREFIX_IDX:
sscanf(sub_token+strlen(plug_info_prefix[j])+1,"%d",&(p->rule_tag_column));
break;
case FOREIGN_PREFIX_IDX:
p->n_foreign=_read_integer_arrary(sub_token+strlen(plug_info_prefix[j])+1, p->foreign_columns, MAX_FOREIGN_CLMN_NUM);
break;
default:
break;
}
}
free(copy_line); free(copy_line);
return 0; return 0;
error_out: error_out:

View File

@@ -462,6 +462,7 @@ struct _maat_garbage_t
struct foreign_key struct foreign_key
{ {
int column; int column;
int is_existed;
char* key; char* key;
char* filename; char* filename;
}; };

View File

@@ -2,6 +2,9 @@
#include <arpa/inet.h> #include <arpa/inet.h>
#include <unistd.h> #include <unistd.h>
#include <sys/syscall.h> #include <sys/syscall.h>
#include <sys/types.h>//fstat
#include <sys/types.h>//fstat
#include <sys/stat.h>//fstat
#define ALLOC(type, number) ((type *)calloc(sizeof(type), number)) #define ALLOC(type, number) ((type *)calloc(sizeof(type), number))
#define FREE(p) {free(*p);*p=NULL;} #define FREE(p) {free(*p);*p=NULL;}

View File

@@ -13,4 +13,4 @@ file(COPY testdata DESTINATION ./)
file(COPY testdata_uni2ascii DESTINATION ./) file(COPY testdata_uni2ascii DESTINATION ./)
include(GoogleTest) include(GoogleTest)
gtest_discover_tests(test_maatframe) #gtest_discover_tests(test_maatframe)

View File

@@ -27,5 +27,5 @@
8 HTTP_REGION expr_plus GBK GBK no 0 8 HTTP_REGION expr_plus GBK GBK no 0
9 SIM_URL similar -- 9 SIM_URL similar --
10 IMAGE_FP expr UTF8 UTF8 yes 128 quickoff 10 IMAGE_FP expr UTF8 UTF8 yes 128 quickoff
11 TEST_EFFECTIVE_RANGE_TABLE plugin valid_pos=4;tag_pos=5; -- 11 TEST_EFFECTIVE_RANGE_TABLE plugin {"valid":4,"tag":5} --
12 TEST_EXETERN_LINK plugin valid_pos=3;foreign_pos=6,7; 12 TEST_EXETERN_LINK plugin {"valid":4,"foreign":"6,7","tag":3} --