xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

enh: add csv-related parameters

This commit is contained in:
Yaming Pei 2025-02-20 18:02:04 +08:00
parent 6f0bdb882c
commit 4fe7ed9a75
2 changed files with 33 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/taos-tools/inc/bench.h
View File

@ -780,7 +780,11 @@ typedef struct SArguments_S {
bool mistMode;
bool escape_character;
bool pre_load_tb_meta;
char csvPath[MAX_FILE_NAME_LEN];
char* csv_output_dir;
char* csv_file_prefix;
char* csv_ts_format;
char* csv_ts_interval;
bool bind_vgroup;
} SArguments;

46
tools/taos-tools/src/benchJsonOpt.c
View File

@ -1586,25 +1586,37 @@ static int getMetaFromCommonJsonFile(tools_cJSON *json) {
}
}
g_arguments->csvPath[0] = 0;
tools_cJSON *csv = tools_cJSON_GetObjectItem(json, "csvPath");
if (csv && (csv->type == tools_cJSON_String)
&& (csv->valuestring != NULL)) {
tstrncpy(g_arguments->csvPath, csv->valuestring, MAX_FILE_NAME_LEN);
}
size_t len = strlen(g_arguments->csvPath);
if(len == 0) {
// set default with current path
strcpy(g_arguments->csvPath, "./output/");
mkdir(g_arguments->csvPath, 0775);
// csv output dir
tools_cJSON* csv_od = tools_cJSON_GetObjectItem(json, "csv_output_dir");
if (csv_od && csv_od->type == tools_cJSON_String && csv_od->valuestring != NULL) {
g_arguments->csv_output_dir = csv_od->valuestring;
} else {
// append end
if (g_arguments->csvPath[len-1] != '/' ) {
strcat(g_arguments->csvPath, "/");
g_arguments->csv_output_dir = "./output/";
}
mkdir(g_arguments->csvPath, 0775);
(void)mkdir(g_arguments->csv_output_dir, 0775);
// csv file prefix
tools_cJSON* csv_fp = tools_cJSON_GetObjectItem(json, "csv_file_prefix");
if (csv_fp && csv_fp->type == tools_cJSON_String && csv_fp->valuestring != NULL) {
g_arguments->csv_file_prefix = csv_fp->valuestring;
} else {
g_arguments->csv_file_prefix = "data";
}
// csv timestamp format
tools_cJSON* csv_tf = tools_cJSON_GetObjectItem(json, "csv_ts_format");
if (csv_tf && csv_tf->type == tools_cJSON_String && csv_tf->valuestring != NULL) {
g_arguments->csv_ts_format = csv_tf->valuestring;
} else {
g_arguments->csv_ts_format = "YYYYMMDDHHmmSS";
}
// csv timestamp format
tools_cJSON* csv_ti = tools_cJSON_GetObjectItem(json, "csv_ts_interval");
if (csv_ti && csv_ti->type == tools_cJSON_String && csv_ti->valuestring != NULL) {
g_arguments->csv_ts_interval = csv_ti->valuestring;
} else {
g_arguments->csv_ts_interval = "1d";
}
code = 0;