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

feat: print csv export mode

This commit is contained in:
Yaming Pei 2025-03-03 17:59:58 +08:00
parent 7eef6659ac
commit cd958c9f17
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/taos-tools/inc/benchCsv.h
View File

@ -57,7 +57,8 @@ typedef struct {
typedef struct { typedef struct {
CsvNamingType naming_type; CsvNamingType naming_type;
size_t total_threads; size_t total_threads;
char thread_formatter[TINY_BUFF_LEN]; char mode[MIDDLE_BUFF_LEN];
char thread_formatter[SMALL_BUFF_LEN];
char csv_header[LARGE_BUFF_LEN]; char csv_header[LARGE_BUFF_LEN];
int csv_header_length; int csv_header_length;
SDataBase* db; SDataBase* db;

6
tools/taos-tools/src/benchCsv.c
View File

@ -479,18 +479,22 @@ static int csvInitWriteMeta(SDataBase* db, SSuperTable* stb, CsvWriteMeta* write
switch (write_meta->naming_type) { switch (write_meta->naming_type) {
case CSV_NAMING_I_SINGLE: { case CSV_NAMING_I_SINGLE: {
(void)snprintf(write_meta->mode, sizeof(write_meta->mode), "interlace|no-time-slice");
break; break;
} }
case CSV_NAMING_I_TIME_SLICE: { case CSV_NAMING_I_TIME_SLICE: {
(void)snprintf(write_meta->mode, sizeof(write_meta->mode), "interlace|time-slice");
csvCalcTimestampStep(write_meta); csvCalcTimestampStep(write_meta);
break; break;
} }
case CSV_NAMING_B_THREAD: { case CSV_NAMING_B_THREAD: {
(void)snprintf(write_meta->mode, sizeof(write_meta->mode), "batch|no-time-slice");
write_meta->total_threads = MIN(g_arguments->nthreads, stb->childTblCount); write_meta->total_threads = MIN(g_arguments->nthreads, stb->childTblCount);
csvGenThreadFormatter(write_meta); csvGenThreadFormatter(write_meta);
break; break;
} }
case CSV_NAMING_B_THREAD_TIME_SLICE: { case CSV_NAMING_B_THREAD_TIME_SLICE: {
(void)snprintf(write_meta->mode, sizeof(write_meta->mode), "batch|time-slice");
write_meta->total_threads = MIN(g_arguments->nthreads, stb->childTblCount); write_meta->total_threads = MIN(g_arguments->nthreads, stb->childTblCount);
csvGenThreadFormatter(write_meta); csvGenThreadFormatter(write_meta);
csvCalcTimestampStep(write_meta); csvCalcTimestampStep(write_meta);
@ -1056,6 +1060,8 @@ static int csvGenStbProcess(SDataBase* db, SSuperTable* stb) {
goto end; goto end;
} }
infoPrint("export csv mode: %s.\n", write_meta->mode);
args = benchCalloc(write_meta->total_threads, sizeof(CsvThreadArgs), false); args = benchCalloc(write_meta->total_threads, sizeof(CsvThreadArgs), false);
if (!args) { if (!args) {
ret = -1; ret = -1;