taosdemo.c 445.0 KB
Newer Older
S
TD-1057  
Shengliang Guan 已提交
1
/*
H
hzcheng 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

16 17 18

/*
   when in some thread query return error, thread don't exit, but return, otherwise coredump in other thread.
19
   */
20

21
#include <stdint.h>
22
#include <taos.h>
23
#include <taoserror.h>
S
slguan 已提交
24
#define _GNU_SOURCE
25 26
#define CURL_STATICLIB

S
TD-1057  
Shengliang Guan 已提交
27
#ifdef LINUX
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
#include <argp.h>
#include <inttypes.h>
#ifndef _ALPINE
#include <error.h>
#endif
#include <pthread.h>
#include <semaphore.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <sys/time.h>
#include <time.h>
#include <unistd.h>
#include <wordexp.h>
#include <regex.h>
43
#else
44 45
#include <regex.h>
#include <stdio.h>
46
#endif
H
hzcheng 已提交
47

48
#include <assert.h>
49
#include <stdlib.h>
50 51
#include "cJSON.h"

52
#include "os.h"
H
hzcheng 已提交
53
#include "taos.h"
54
#include "taoserror.h"
H
Hui Li 已提交
55
#include "tutil.h"
S
slguan 已提交
56

57 58 59
#define REQ_EXTRA_BUF_LEN   1024
#define RESP_BUF_LEN        4096

S
slguan 已提交
60 61
extern char configDir[];

62 63
#define STR_INSERT_INTO     "INSERT INTO "

64 65
#define MAX_RECORDS_PER_REQ     32766

66
#define HEAD_BUFF_LEN       TSDB_MAX_COLUMNS*24  // 16*MAX_COLUMNS + (192+32)*2 + insert into ..
67

68
#define BUFFER_SIZE         TSDB_MAX_ALLOWED_SQL_LEN
69
#define COND_BUF_LEN        (BUFFER_SIZE - 30)
70
#define COL_BUFFER_LEN      ((TSDB_COL_NAME_LEN + 15) * TSDB_MAX_COLUMNS)
71

72
#define MAX_USERNAME_SIZE  64
73
#define MAX_HOSTNAME_SIZE  253      // https://man7.org/linux/man-pages/man7/hostname.7.html
74
#define MAX_TB_NAME_SIZE   64
75
#define MAX_DATA_SIZE      (16*TSDB_MAX_COLUMNS)+20     // max record len: 16*MAX_COLUMNS, timestamp string and ,('') need extra space
76
#define OPT_ABORT          1 /* –abort */
77
#define MAX_FILE_NAME_LEN  256              // max file name length on linux is 255.
78

79 80
#define DEFAULT_START_TIME 1500000000000

81
#define MAX_PREPARED_RAND  1000000
82
#define INT_BUFF_LEN            12
83
#define BIGINT_BUFF_LEN         21
84 85
#define SMALLINT_BUFF_LEN       7
#define TINYINT_BUFF_LEN        5
86 87 88 89 90
#define BOOL_BUFF_LEN           6
#define FLOAT_BUFF_LEN          22
#define DOUBLE_BUFF_LEN         42
#define TIMESTAMP_BUFF_LEN      21

91
#define MAX_SAMPLES             10000
92
#define MAX_NUM_COLUMNS        (TSDB_MAX_COLUMNS - 1)      // exclude first column timestamp
93

94 95
#define MAX_DB_COUNT            8
#define MAX_SUPER_TABLE_COUNT   200
96

97
#define MAX_QUERY_SQL_COUNT     100
98

99 100
#define MAX_DATABASE_COUNT      256
#define INPUT_BUF_LEN           256
101

102 103 104 105 106
#define TBNAME_PREFIX_LEN       (TSDB_TABLE_NAME_LEN - 20) // 20 characters reserved for seq
#define SMALL_BUFF_LEN          8
#define DATATYPE_BUFF_LEN       (SMALL_BUFF_LEN*3)
#define NOTE_BUFF_LEN           (SMALL_BUFF_LEN*16)

107
#define DEFAULT_NTHREADS        8
108
#define DEFAULT_TIMESTAMP_STEP  1
109
#define DEFAULT_INTERLACE_ROWS  0
110
#define DEFAULT_DATATYPE_NUM    1
111
#define DEFAULT_CHILDTABLES     10000
112

113
#define STMT_BIND_PARAM_BATCH   1
114 115 116 117 118 119 120

char* g_sampleDataBuf = NULL;
#if STMT_BIND_PARAM_BATCH == 1
    // bind param batch
char* g_sampleBindBatchArray = NULL;
#endif

121 122 123 124 125 126 127
enum TEST_MODE {
    INSERT_TEST,            // 0
    QUERY_TEST,             // 1
    SUBSCRIBE_TEST,         // 2
    INVAID_TEST
};

128
typedef enum CREATE_SUB_TABLE_MOD_EN {
129 130 131
    PRE_CREATE_SUBTBL,
    AUTO_CREATE_SUBTBL,
    NO_CREATE_SUBTBL
132
} CREATE_SUB_TABLE_MOD_EN;
133

134
typedef enum TABLE_EXISTS_EN {
135 136 137
    TBL_NO_EXISTS,
    TBL_ALREADY_EXISTS,
    TBL_EXISTS_BUTT
138
} TABLE_EXISTS_EN;
139

140
enum enumSYNC_MODE {
141 142 143
    SYNC_MODE,
    ASYNC_MODE,
    MODE_BUT
144
};
145

S
Shuduo Sang 已提交
146
enum enum_TAOS_INTERFACE {
147 148 149
    TAOSC_IFACE,
    REST_IFACE,
    STMT_IFACE,
150 151 152
    INTERFACE_BUT
};

153 154 155 156 157 158
typedef enum enumQUERY_CLASS {
    SPECIFIED_CLASS,
    STABLE_CLASS,
    CLASS_BUT
} QUERY_CLASS;

S
Shuduo Sang 已提交
159
typedef enum enum_PROGRESSIVE_OR_INTERLACE {
160 161 162
    PROGRESSIVE_INSERT_MODE,
    INTERLACE_INSERT_MODE,
    INVALID_INSERT_MODE
163
} PROG_OR_INTERLACE_MODE;
164

165
typedef enum enumQUERY_TYPE {
166 167 168
    NO_INSERT_TYPE,
    INSERT_TYPE,
    QUERY_TYPE_BUT
169
} QUERY_TYPE;
170 171

enum _show_db_index {
172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
    TSDB_SHOW_DB_NAME_INDEX,
    TSDB_SHOW_DB_CREATED_TIME_INDEX,
    TSDB_SHOW_DB_NTABLES_INDEX,
    TSDB_SHOW_DB_VGROUPS_INDEX,
    TSDB_SHOW_DB_REPLICA_INDEX,
    TSDB_SHOW_DB_QUORUM_INDEX,
    TSDB_SHOW_DB_DAYS_INDEX,
    TSDB_SHOW_DB_KEEP_INDEX,
    TSDB_SHOW_DB_CACHE_INDEX,
    TSDB_SHOW_DB_BLOCKS_INDEX,
    TSDB_SHOW_DB_MINROWS_INDEX,
    TSDB_SHOW_DB_MAXROWS_INDEX,
    TSDB_SHOW_DB_WALLEVEL_INDEX,
    TSDB_SHOW_DB_FSYNC_INDEX,
    TSDB_SHOW_DB_COMP_INDEX,
    TSDB_SHOW_DB_CACHELAST_INDEX,
    TSDB_SHOW_DB_PRECISION_INDEX,
    TSDB_SHOW_DB_UPDATE_INDEX,
    TSDB_SHOW_DB_STATUS_INDEX,
    TSDB_MAX_SHOW_DB
192 193 194 195
};

// -----------------------------------------SHOW TABLES CONFIGURE -------------------------------------
enum _show_stables_index {
196 197 198 199 200 201 202 203
    TSDB_SHOW_STABLES_NAME_INDEX,
    TSDB_SHOW_STABLES_CREATED_TIME_INDEX,
    TSDB_SHOW_STABLES_COLUMNS_INDEX,
    TSDB_SHOW_STABLES_METRIC_INDEX,
    TSDB_SHOW_STABLES_UID_INDEX,
    TSDB_SHOW_STABLES_TID_INDEX,
    TSDB_SHOW_STABLES_VGID_INDEX,
    TSDB_MAX_SHOW_STABLES
204 205
};

206
enum _describe_table_index {
207 208 209 210 211
    TSDB_DESCRIBE_METRIC_FIELD_INDEX,
    TSDB_DESCRIBE_METRIC_TYPE_INDEX,
    TSDB_DESCRIBE_METRIC_LENGTH_INDEX,
    TSDB_DESCRIBE_METRIC_NOTE_INDEX,
    TSDB_MAX_DESCRIBE_METRIC
212 213
};

H
hzcheng 已提交
214
/* Used by main to communicate with parse_opt. */
215 216
static char *g_dupstr = NULL;

217
typedef struct SArguments_S {
218
    char     *metaFile;
219
    uint32_t test_mode;
220
    char     *host;
221 222 223
    uint16_t port;
    uint16_t iface;
    char *   user;
224
    char     password[SHELL_MAX_PASSWORD_LEN];
225 226 227 228 229 230
    char *   database;
    int      replica;
    char *   tb_prefix;
    char *   sqlFile;
    bool     use_metric;
    bool     drop_database;
231
    bool     aggr_func;
232 233 234 235 236 237
    bool     answer_yes;
    bool     debug_print;
    bool     verbose_print;
    bool     performance_print;
    char *   output_file;
    bool     async_mode;
238 239
    char     data_type[MAX_NUM_COLUMNS+1];
    char     *dataType[MAX_NUM_COLUMNS+1];
240
    uint32_t binwidth;
241 242 243
    uint32_t columnCount;
    uint64_t lenOfOneRow;
    uint32_t nthreads;
244
    uint64_t insert_interval;
245
    uint64_t timestamp_step;
246
    int64_t  query_times;
247
    uint32_t interlaceRows;
248
    uint32_t reqPerReq;                  // num_of_records_per_req
249
    uint64_t max_sql_len;
250 251
    int64_t  ntables;
    int64_t  insertRows;
252 253
    int      abort;
    uint32_t disorderRatio;               // 0: no disorder, >0: x%
254
    int      disorderRange;               // ms, us or ns. according to database precision
255 256 257 258
    uint32_t method_of_delete;
    uint64_t totalInsertRows;
    uint64_t totalAffectedRows;
    bool     demo_mode;                  // use default column name and semi-random data
259 260 261
} SArguments;

typedef struct SColumn_S {
262 263 264 265 266
    char        field[TSDB_COL_NAME_LEN];
    char        data_type;
    char        dataType[DATATYPE_BUFF_LEN];
    uint32_t    dataLen;
    char        note[NOTE_BUFF_LEN];
267 268 269
} StrColumn;

typedef struct SSuperTable_S {
270
    char         stbName[TSDB_TABLE_NAME_LEN];
271 272
    char         dataSource[SMALL_BUFF_LEN];  // rand_gen or sample
    char         childTblPrefix[TBNAME_PREFIX_LEN];
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290
    uint16_t     childTblExists;
    int64_t      childTblCount;
    uint64_t     batchCreateTableNum;     // 0: no batch,  > 0: batch table number in one sql
    uint8_t      autoCreateTable;         // 0: create sub table, 1: auto create sub table
    uint16_t     iface;                   // 0: taosc, 1: rest, 2: stmt
    int64_t      childTblLimit;
    uint64_t     childTblOffset;

    //  int          multiThreadWriteOneTbl;  // 0: no, 1: yes
    uint32_t     interlaceRows;           //
    int          disorderRatio;           // 0: no disorder, >0: x%
    int          disorderRange;           // ms, us or ns. according to database precision
    uint64_t     maxSqlLen;               //

    uint64_t     insertInterval;          // insert interval, will override global insert interval
    int64_t      insertRows;
    int64_t      timeStampStep;
    char         startTimestamp[MAX_TB_NAME_SIZE];
291
    char         sampleFormat[SMALL_BUFF_LEN];  // csv, json
292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
    char         sampleFile[MAX_FILE_NAME_LEN];
    char         tagsFile[MAX_FILE_NAME_LEN];

    uint32_t     columnCount;
    StrColumn    columns[TSDB_MAX_COLUMNS];
    uint32_t     tagCount;
    StrColumn    tags[TSDB_MAX_TAGS];

    char*        childTblName;
    char*        colsOfCreateChildTable;
    uint64_t     lenOfOneRow;
    uint64_t     lenOfTagOfOneRow;

    char*        sampleDataBuf;

    uint32_t     tagSource;    // 0: rand, 1: tag sample
    char*        tagDataBuf;
    uint32_t     tagSampleCount;
    uint32_t     tagUsePos;

312 313 314 315
#if STMT_BIND_PARAM_BATCH == 1
    // bind param batch
    char        *sampleBindBatchArray;
#endif
316 317 318
    // statistics
    uint64_t     totalInsertRows;
    uint64_t     totalAffectedRows;
319 320 321
} SSuperTable;

typedef struct {
322 323 324 325 326 327 328
    char     name[TSDB_DB_NAME_LEN];
    char     create_time[32];
    int64_t  ntables;
    int32_t  vgroups;
    int16_t  replica;
    int16_t  quorum;
    int16_t  days;
329
    char     keeplist[64];
330 331 332 333 334 335 336 337
    int32_t  cache; //MB
    int32_t  blocks;
    int32_t  minrows;
    int32_t  maxrows;
    int8_t   wallevel;
    int32_t  fsync;
    int8_t   comp;
    int8_t   cachelast;
338
    char     precision[SMALL_BUFF_LEN];   // time resolution
339 340
    int8_t   update;
    char     status[16];
341 342
} SDbInfo;

343
typedef struct SDbCfg_S {
344 345 346 347 348 349 350 351 352 353 354 355 356 357
    //  int       maxtablesPerVnode;
    uint32_t  minRows;        // 0 means default
    uint32_t  maxRows;        // 0 means default
    int       comp;
    int       walLevel;
    int       cacheLast;
    int       fsync;
    int       replica;
    int       update;
    int       keep;
    int       days;
    int       cache;
    int       blocks;
    int       quorum;
358
    char      precision[SMALL_BUFF_LEN];
359 360 361
} SDbCfg;

typedef struct SDataBase_S {
362 363 364 365 366
    char         dbName[TSDB_DB_NAME_LEN];
    bool         drop;  // 0: use exists, 1: if exists, drop then new create
    SDbCfg       dbCfg;
    uint64_t     superTblCount;
    SSuperTable  superTbls[MAX_SUPER_TABLE_COUNT];
367 368 369
} SDataBase;

typedef struct SDbs_S {
370 371 372 373 374 375
    char        cfgDir[MAX_FILE_NAME_LEN];
    char        host[MAX_HOSTNAME_SIZE];
    struct      sockaddr_in serv_addr;

    uint16_t    port;
    char        user[MAX_USERNAME_SIZE];
376
    char        password[SHELL_MAX_PASSWORD_LEN];
377 378
    char        resultFile[MAX_FILE_NAME_LEN];
    bool        use_metric;
379
    bool        aggr_func;
380 381 382
    bool        asyncMode;

    uint32_t    threadCount;
383
    uint32_t    threadCountForCreateTbl;
384 385 386 387 388 389
    uint32_t    dbCount;
    SDataBase   db[MAX_DB_COUNT];

    // statistics
    uint64_t    totalInsertRows;
    uint64_t    totalAffectedRows;
390

391 392
} SDbs;

393
typedef struct SpecifiedQueryInfo_S {
394
    uint64_t     queryInterval;  // 0: unlimited  > 0   loop/s
395 396 397 398 399 400 401
    uint32_t     concurrent;
    int          sqlCount;
    uint32_t     asyncMode; // 0: sync, 1: async
    uint64_t     subscribeInterval; // ms
    uint64_t     queryTimes;
    bool         subscribeRestart;
    int          subscribeKeepProgress;
402
    char         sql[MAX_QUERY_SQL_COUNT][BUFFER_SIZE+1];
403 404 405 406 407 408 409 410
    char         result[MAX_QUERY_SQL_COUNT][MAX_FILE_NAME_LEN];
    int          resubAfterConsume[MAX_QUERY_SQL_COUNT];
    int          endAfterConsume[MAX_QUERY_SQL_COUNT];
    TAOS_SUB*    tsub[MAX_QUERY_SQL_COUNT];
    char         topic[MAX_QUERY_SQL_COUNT][32];
    int          consumed[MAX_QUERY_SQL_COUNT];
    TAOS_RES*    res[MAX_QUERY_SQL_COUNT];
    uint64_t     totalQueried;
411
} SpecifiedQueryInfo;
412

413
typedef struct SuperQueryInfo_S {
414
    char         stbName[TSDB_TABLE_NAME_LEN];
415
    uint64_t     queryInterval;  // 0: unlimited  > 0   loop/s
416 417 418 419 420 421 422
    uint32_t     threadCnt;
    uint32_t     asyncMode; // 0: sync, 1: async
    uint64_t     subscribeInterval; // ms
    bool         subscribeRestart;
    int          subscribeKeepProgress;
    uint64_t     queryTimes;
    int64_t      childTblCount;
423
    char         childTblPrefix[TBNAME_PREFIX_LEN];    // 20 characters reserved for seq
424
    int          sqlCount;
425
    char         sql[MAX_QUERY_SQL_COUNT][BUFFER_SIZE+1];
426 427 428 429 430 431 432
    char         result[MAX_QUERY_SQL_COUNT][MAX_FILE_NAME_LEN];
    int          resubAfterConsume;
    int          endAfterConsume;
    TAOS_SUB*    tsub[MAX_QUERY_SQL_COUNT];

    char*        childTblName;
    uint64_t     totalQueried;
433
} SuperQueryInfo;
434 435

typedef struct SQueryMetaInfo_S {
436 437 438 439 440
    char         cfgDir[MAX_FILE_NAME_LEN];
    char         host[MAX_HOSTNAME_SIZE];
    uint16_t     port;
    struct       sockaddr_in serv_addr;
    char         user[MAX_USERNAME_SIZE];
441
    char         password[SHELL_MAX_PASSWORD_LEN];
442
    char         dbName[TSDB_DB_NAME_LEN];
443
    char         queryMode[SMALL_BUFF_LEN];  // taosc, rest
444 445 446 447

    SpecifiedQueryInfo  specifiedQueryInfo;
    SuperQueryInfo      superQueryInfo;
    uint64_t     totalQueried;
448 449 450
} SQueryMetaInfo;

typedef struct SThreadInfo_S {
451 452
    TAOS *    taos;
    TAOS_STMT *stmt;
453
    int64_t     *bind_ts;
454 455 456 457 458 459 460 461 462

#if STMT_BIND_PARAM_BATCH == 1
    int64_t     *bind_ts_array;
    char        *bindParams;
    char        *is_null;
#else
    char*       sampleBindArray;
#endif

463 464 465 466 467 468 469 470 471
    int       threadID;
    char      db_name[TSDB_DB_NAME_LEN];
    uint32_t  time_precision;
    char      filePath[4096];
    FILE      *fp;
    char      tb_prefix[TSDB_TABLE_NAME_LEN];
    uint64_t  start_table_from;
    uint64_t  end_table_to;
    int64_t   ntables;
472
    int64_t   tables_created;
473 474 475 476
    uint64_t  data_of_rate;
    int64_t   start_time;
    char*     cols;
    bool      use_metric;
477
    SSuperTable* stbInfo;
478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502
    char      *buffer;    // sql cmd buffer

    // for async insert
    tsem_t    lock_sem;
    int64_t   counter;
    uint64_t  st;
    uint64_t  et;
    uint64_t  lastTs;

    // sample data
    int64_t   samplePos;
    // statistics
    uint64_t  totalInsertRows;
    uint64_t  totalAffectedRows;

    // insert delay statistics
    uint64_t  cntDelay;
    uint64_t  totalDelay;
    uint64_t  avgDelay;
    uint64_t  maxDelay;
    uint64_t  minDelay;

    // seq of query or subscribe
    uint64_t  querySeq;   // sequence number of sql command
    TAOS_SUB*  tsub;
503

504 505
} threadInfo;

506
#ifdef WINDOWS
507 508
#define _CRT_RAND_S

509
#include <windows.h>
510 511 512 513 514
#include <winsock2.h>

typedef unsigned __int32 uint32_t;

#pragma comment ( lib, "ws2_32.lib" )
515 516
// Some old MinGW/CYGWIN distributions don't define this:
#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
517
#define ENABLE_VIRTUAL_TERMINAL_PROCESSING  0x0004
518
#endif // ENABLE_VIRTUAL_TERMINAL_PROCESSING
519 520 521 522

static HANDLE g_stdoutHandle;
static DWORD g_consoleMode;

523
static void setupForAnsiEscape(void) {
524 525
    DWORD mode = 0;
    g_stdoutHandle = GetStdHandle(STD_OUTPUT_HANDLE);
526

527 528 529
    if(g_stdoutHandle == INVALID_HANDLE_VALUE) {
        exit(GetLastError());
    }
530

531 532 533
    if(!GetConsoleMode(g_stdoutHandle, &mode)) {
        exit(GetLastError());
    }
534

535
    g_consoleMode = mode;
536

537 538
    // Enable ANSI escape codes
    mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
539

540 541 542
    if(!SetConsoleMode(g_stdoutHandle, mode)) {
        exit(GetLastError());
    }
543 544
}

545
static void resetAfterAnsiEscape(void) {
546 547
    // Reset colors
    printf("\x1b[0m");
548

549 550 551 552
    // Reset console mode
    if(!SetConsoleMode(g_stdoutHandle, g_consoleMode)) {
        exit(GetLastError());
    }
553
}
554

555
static int taosRandom()
556 557 558 559 560 561
{
    int number;
    rand_s(&number);

    return number;
}
562
#else   // Not windows
563
static void setupForAnsiEscape(void) {}
564

565
static void resetAfterAnsiEscape(void) {
566 567
    // Reset colors
    printf("\x1b[0m");
568
}
569

570 571
#include <time.h>

572
static int taosRandom()
573
{
574
    return rand();
575 576
}

577
#endif // ifdef Windows
578

579
static void prompt();
580
static int createDatabasesAndStables();
581
static void createChildTables();
582
static int queryDbExec(TAOS *taos, char *command, QUERY_TYPE type, bool quiet);
583
static int postProceSql(char *host, struct sockaddr_in *pServAddr,
584
        uint16_t port, char* sqlstr, threadInfo *pThreadInfo);
585 586
static int64_t getTSRandTail(int64_t timeStampStep, int32_t seq,
        int disorderRatio, int disorderRange);
587 588
static bool getInfoFromJsonFile(char* file);
static void init_rand_data();
589 590 591

/* ************ Global variables ************  */

592
int32_t  g_randint[MAX_PREPARED_RAND];
593
uint32_t  g_randuint[MAX_PREPARED_RAND];
594
int64_t  g_randbigint[MAX_PREPARED_RAND];
595
uint64_t  g_randubigint[MAX_PREPARED_RAND];
596 597 598 599 600
float    g_randfloat[MAX_PREPARED_RAND];
double   g_randdouble[MAX_PREPARED_RAND];

char    *g_randbool_buff = NULL;
char    *g_randint_buff = NULL;
601
char    *g_randuint_buff = NULL;
602 603
char    *g_rand_voltage_buff = NULL;
char    *g_randbigint_buff = NULL;
604
char    *g_randubigint_buff = NULL;
605
char    *g_randsmallint_buff = NULL;
606
char    *g_randusmallint_buff = NULL;
607
char    *g_randtinyint_buff = NULL;
608
char    *g_randutinyint_buff = NULL;
609 610 611 612 613
char    *g_randfloat_buff = NULL;
char    *g_rand_current_buff = NULL;
char    *g_rand_phase_buff = NULL;
char    *g_randdouble_buff = NULL;

614 615 616
char    *g_aggreFuncDemo[] = {"*", "count(*)", "avg(current)", "sum(current)",
    "max(current)", "min(current)", "first(current)", "last(current)"};

617 618
char    *g_aggreFunc[] = {"*", "count(*)", "avg(C0)", "sum(C0)",
    "max(C0)", "min(C0)", "first(C0)", "last(C0)"};
619

620
SArguments g_args = {
621 622 623 624 625 626
    NULL,           // metaFile
    0,              // test_mode
    "localhost",    // host
    6030,           // port
    INTERFACE_BUT,  // iface
    "root",         // user
627 628 629
#ifdef _TD_POWER_
    "powerdb",      // password
#elif (_TD_TQ_ == true)
630
    "tqueue",       // password
631 632
#elif (_TD_PRO_ == true)
    "prodb",       // password
633
#else
634
    "taosdata",     // password
635
#endif
636 637
    "test",         // database
    1,              // replica
638 639 640 641
    "d",             // tb_prefix
    NULL,            // sqlFile
    true,            // use_metric
    true,            // drop_database
642
    false,           // aggr_func
643 644 645 646 647 648
    false,           // debug_print
    false,           // verbose_print
    false,           // performance statistic print
    false,           // answer_yes;
    "./output.txt",  // output_file
    0,               // mode : sync or async
649 650 651
    {TSDB_DATA_TYPE_FLOAT,
    TSDB_DATA_TYPE_INT,
    TSDB_DATA_TYPE_FLOAT},
652
    {
653 654 655
        "FLOAT",         // dataType
        "INT",           // dataType
        "FLOAT",         // dataType. demo mode has 3 columns
656
    },
657
    64,              // binwidth
658 659
    4,               // columnCount, timestamp + float + int + float
    20 + FLOAT_BUFF_LEN + INT_BUFF_LEN + FLOAT_BUFF_LEN, // lenOfOneRow
660
    DEFAULT_NTHREADS,// nthreads
661
    0,               // insert_interval
662
    DEFAULT_TIMESTAMP_STEP, // timestamp_step
663
    1,               // query_times
664
    DEFAULT_INTERLACE_ROWS, // interlaceRows;
665
    30000,           // reqPerReq
666
    (1024*1024),     // max_sql_len
667 668
    DEFAULT_CHILDTABLES,    // ntables
    10000,           // insertRows
669 670 671 672 673 674 675
    0,               // abort
    0,               // disorderRatio
    1000,            // disorderRange
    1,               // method_of_delete
    0,               // totalInsertRows;
    0,               // totalAffectedRows;
    true,            // demo_mode;
676 677 678
};

static SDbs            g_Dbs;
679
static int64_t         g_totalChildTables = DEFAULT_CHILDTABLES;
680
static int64_t         g_actualChildTables = 0;
681 682 683
static SQueryMetaInfo  g_queryInfo;
static FILE *          g_fpOfInsertResult = NULL;

684 685 686 687
#if _MSC_VER <= 1900
#define __func__ __FUNCTION__
#endif

688
#define debugPrint(fmt, ...) \
689
    do { if (g_args.debug_print || g_args.verbose_print) \
690
        fprintf(stderr, "DEBG: "fmt, __VA_ARGS__); } while(0)
691

692
#define verbosePrint(fmt, ...) \
693 694
    do { if (g_args.verbose_print) \
        fprintf(stderr, "VERB: "fmt, __VA_ARGS__); } while(0)
695

696 697
#define performancePrint(fmt, ...) \
    do { if (g_args.performance_print) \
698
        fprintf(stderr, "PERF: "fmt, __VA_ARGS__); } while(0)
699

700
#define errorPrint(fmt, ...) \
701 702 703 704 705 706 707
    do {\
        fprintf(stderr, " \033[31m");\
        fprintf(stderr, "ERROR: "fmt, __VA_ARGS__);\
        fprintf(stderr, " \033[0m");\
    } while(0)

#define errorPrint2(fmt, ...) \
708 709 710 711 712 713 714 715 716 717 718 719 720
    do {\
        struct tm      Tm, *ptm;\
        struct timeval timeSecs; \
        time_t         curTime;\
        gettimeofday(&timeSecs, NULL); \
        curTime = timeSecs.tv_sec;\
        ptm = localtime_r(&curTime, &Tm);\
        fprintf(stderr, " \033[31m");\
        fprintf(stderr, "%02d/%02d %02d:%02d:%02d.%06d %08" PRId64 " ",\
                ptm->tm_mon + 1, ptm->tm_mday, ptm->tm_hour,\
                ptm->tm_min, ptm->tm_sec, (int32_t)timeSecs.tv_usec,\
                taosGetSelfPthreadId());\
        fprintf(stderr, " \033[0m");\
721
        errorPrint(fmt, __VA_ARGS__);\
722
    } while(0)
723

724 725 726
// for strncpy buffer overflow
#define min(a, b) (((a) < (b)) ? (a) : (b))

727

728
///////////////////////////////////////////////////
H
hzcheng 已提交
729

730
static void ERROR_EXIT(const char *msg) { errorPrint("%s", msg); exit(-1); }
731

732 733 734 735 736 737 738 739
#ifndef TAOSDEMO_COMMIT_SHA1
#define TAOSDEMO_COMMIT_SHA1 "unknown"
#endif

#ifndef TD_VERNUMBER
#define TD_VERNUMBER    "unknown"
#endif

740
#ifndef TAOSDEMO_STATUS
741 742 743 744 745 746 747 748 749
#define TAOSDEMO_STATUS "unknown"
#endif

static void printVersion() {
    char tdengine_ver[] = TD_VERNUMBER;
    char taosdemo_ver[] = TAOSDEMO_COMMIT_SHA1;
    char taosdemo_status[] = TAOSDEMO_STATUS;

    if (strlen(taosdemo_status) == 0) {
750
        printf("taosdemo version %s-%s\n",
751 752
                tdengine_ver, taosdemo_ver);
    } else {
753
        printf("taosdemo version %s-%s, status:%s\n",
754 755 756 757
                tdengine_ver, taosdemo_ver, taosdemo_status);
    }
}

758
static void printHelp() {
759 760 761
    char indent[10] = "  ";
    printf("%s\n\n", "Usage: taosdemo [OPTION...]");
    printf("%s%s%s%s\n", indent, "-f, --file=FILE", "\t\t",
762
            "The meta file to the execution procedure.");
763 764
    printf("%s%s%s%s\n", indent, "-u, --user=USER", "\t\t",
            "The user name to use when connecting to the server.");
765
#ifdef _TD_POWER_
766
    printf("%s%s%s%s\n", indent, "-p, --password", "\t\t",
767
            "The password to use when connecting to the server. By default is 'powerdb'");
768
    printf("%s%s%s%s\n", indent, "-c, --config-dir=CONFIG_DIR", "\t",
769
            "Configuration directory. By default is '/etc/power/'.");
770
#elif (_TD_TQ_ == true)
771
    printf("%s%s%s%s\n", indent, "-p, --password", "\t\t",
772
            "The password to use when connecting to the server. By default is 'tqueue'");
773
    printf("%s%s%s%s\n", indent, "-c, --config-dir=CONFIG_DIR", "\t",
774
            "Configuration directory. By default is '/etc/tq/'.");
775 776 777 778 779
#elif (_TD_PRO_ == true)
    printf("%s%s%s%s\n", indent, "-p, --password", "\t\t",
            "The password to use when connecting to the server. By default is 'prodb'");
    printf("%s%s%s%s\n", indent, "-c, --config-dir=CONFIG_DIR", "\t",
            "Configuration directory. By default is '/etc/ProDB/'.");
780
#else
781 782 783 784
    printf("%s%s%s%s\n", indent, "-p, --password", "\t\t",
            "The password to use when connecting to the server.");
    printf("%s%s%s%s\n", indent, "-c, --config-dir=CONFIG_DIR", "\t",
            "Configuration directory.");
785
#endif
786 787 788 789 790
    printf("%s%s%s%s\n", indent, "-h, --host=HOST", "\t\t",
            "TDengine server FQDN to connect. The default host is localhost.");
    printf("%s%s%s%s\n", indent, "-P, --port=PORT", "\t\t",
            "The TCP/IP port number to use for the connection.");
    printf("%s%s%s%s\n", indent, "-I, --interface=INTERFACE", "\t",
791
            "The interface (taosc, rest, and stmt) taosdemo uses. By default use 'taosc'.");
792
    printf("%s%s%s%s\n", indent, "-d, --database=DATABASE", "\t",
793
            "Destination database. By default is 'test'.");
794
    printf("%s%s%s%s\n", indent, "-a, --replica=REPLICA", "\t\t",
795
            "Set the replica parameters of the database, By default use 1, min: 1, max: 3.");
796
    printf("%s%s%s%s\n", indent, "-m, --table-prefix=TABLEPREFIX", "\t",
797
            "Table prefix name. By default use 'd'.");
798 799
    printf("%s%s%s%s\n", indent, "-s, --sql-file=FILE", "\t\t",
            "The select sql file.");
800 801
    printf("%s%s%s%s\n", indent, "-N, --normal-table", "\t\t", "Use normal table flag.");
    printf("%s%s%s%s\n", indent, "-o, --output=FILE", "\t\t",
802
            "Direct output to the named file. By default use './output.txt'.");
803
    printf("%s%s%s%s\n", indent, "-q, --query-mode=MODE", "\t\t",
804
            "Query mode -- 0: SYNC, 1: ASYNC. By default use SYNC.");
805
    printf("%s%s%s%s\n", indent, "-b, --data-type=DATATYPE", "\t",
806
            "The data_type of columns, By default use: FLOAT, INT, FLOAT.");
807
    printf("%s%s%s%s%d\n", indent, "-w, --binwidth=WIDTH", "\t\t",
808
            "The width of data_type 'BINARY' or 'NCHAR'. By default use ",
809 810
            g_args.binwidth);
    printf("%s%s%s%s%d%s%d\n", indent, "-l, --columns=COLUMNS", "\t\t",
811
            "The number of columns per record. Demo mode by default is ",
812
            DEFAULT_DATATYPE_NUM,
813
            " (float, int, float). Max values is ",
814 815
            MAX_NUM_COLUMNS);
    printf("%s%s%s%s\n", indent, indent, indent,
816
            "\t\t\t\tAll of the new column(s) type is INT. If use -b to specify column type, -l will be ignored.");
817 818
    printf("%s%s%s%s%d.\n", indent, "-T, --threads=NUMBER", "\t\t",
            "The number of threads. By default use ", DEFAULT_NTHREADS);
819
    printf("%s%s%s%s\n", indent, "-i, --insert-interval=NUMBER", "\t",
820
            "The sleep time (ms) between insertion. By default is 0.");
821
    printf("%s%s%s%s%d.\n", indent, "-S, --time-step=TIME_STEP", "\t",
822
            "The timestamp step between insertion. By default is ",
823
            DEFAULT_TIMESTAMP_STEP);
824
    printf("%s%s%s%s%d.\n", indent, "-B, --interlace-rows=NUMBER", "\t",
825
            "The interlace rows of insertion. By default is ",
826 827
            DEFAULT_INTERLACE_ROWS);
    printf("%s%s%s%s\n", indent, "-r, --rec-per-req=NUMBER", "\t",
828
            "The number of records per request. By default is 30000.");
829
    printf("%s%s%s%s\n", indent, "-t, --tables=NUMBER", "\t\t",
830
            "The number of tables. By default is 10000.");
831
    printf("%s%s%s%s\n", indent, "-n, --records=NUMBER", "\t\t",
832
            "The number of records per table. By default is 10000.");
833
    printf("%s%s%s%s\n", indent, "-M, --random", "\t\t\t",
834
            "The value of records generated are totally random.");
835 836 837 838
    printf("%s\n", "\t\t\t\tBy default to simulate power equipment scenario.");
    printf("%s%s%s%s\n", indent, "-x, --aggr-func", "\t\t",
            "Test aggregation functions after insertion.");
    printf("%s%s%s%s\n", indent, "-y, --answer-yes", "\t\t", "Input yes for prompt.");
839
    printf("%s%s%s%s\n", indent, "-O, --disorder=NUMBER", "\t\t",
840
            "Insert order mode--0: In order, 1 ~ 50: disorder ratio. By default is in order.");
841
    printf("%s%s%s%s\n", indent, "-R, --disorder-range=NUMBER", "\t",
842
            "Out of order data's range. Unit is ms. By default is 1000.");
843
    printf("%s%s%s%s\n", indent, "-g, --debug", "\t\t\t",
844
            "Print debug info.");
845 846 847 848
    printf("%s%s%s%s\n", indent, "-?, --help\t", "\t\t",
            "Give this help list");
    printf("%s%s%s%s\n", indent, "    --usage\t", "\t\t",
            "Give a short usage message");
849
    printf("%s%s\n", indent, "-V, --version\t\t\tPrint program version.");
850 851
    /*    printf("%s%s%s%s\n", indent, "-D", indent,
          "Delete database if exists. 0: no, 1: yes, default is 1");
852
          */
853 854 855 856
    printf("\nMandatory or optional arguments to long options are also mandatory or optional\n\
for any corresponding short options.\n\
\n\
Report bugs to <support@taosdata.com>.\n");
857 858
}

859 860
static bool isStringNumber(char *input)
{
861 862 863 864
    int len = strlen(input);
    if (0 == len) {
        return false;
    }
865

866 867 868 869
    for (int i = 0; i < len; i++) {
        if (!isdigit(input[i]))
            return false;
    }
870

871
    return true;
872 873
}

874 875 876 877 878 879
static void errorWrongValue(char *program, char *wrong_arg, char *wrong_value)
{
    fprintf(stderr, "%s %s: %s is an invalid value\n", program, wrong_arg, wrong_value);
    fprintf(stderr, "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
}

880
static void errorUnrecognized(char *program, char *wrong_arg)
881 882 883 884 885
{
    fprintf(stderr, "%s: unrecognized options '%s'\n", program, wrong_arg);
    fprintf(stderr, "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
}

886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912
static void errorPrintReqArg(char *program, char *wrong_arg)
{
    fprintf(stderr,
            "%s: option requires an argument -- '%s'\n",
            program, wrong_arg);
    fprintf(stderr,
            "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
}

static void errorPrintReqArg2(char *program, char *wrong_arg)
{
    fprintf(stderr,
            "%s: option requires a number argument '-%s'\n",
            program, wrong_arg);
    fprintf(stderr,
            "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
}

static void errorPrintReqArg3(char *program, char *wrong_arg)
{
    fprintf(stderr,
            "%s: option '%s' requires an argument\n",
            program, wrong_arg);
    fprintf(stderr,
            "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
}

913
static void parse_args(int argc, char *argv[], SArguments *arguments) {
914

915
    for (int i = 1; i < argc; i++) {
916
        if ((0 == strncmp(argv[i], "-f", strlen("-f")))
917
                || (0 == strncmp(argv[i], "--file", strlen("--file")))) {
918
            arguments->demo_mode = false;
919

920 921 922
            if (2 == strlen(argv[i])) {
                if (i+1 == argc) {
                    errorPrintReqArg(argv[0], "f");
923 924 925
                    exit(EXIT_FAILURE);
                }
                arguments->metaFile = argv[++i];
926 927
            } else if (0 == strncmp(argv[i], "-f", strlen("-f"))) {
                arguments->metaFile = (char *)(argv[i] + strlen("-f"));
928
            } else if (strlen("--file") == strlen(argv[i])) {
929 930 931 932 933
                if (i+1 == argc) {
                    errorPrintReqArg3(argv[0], "--file");
                    exit(EXIT_FAILURE);
                }
                arguments->metaFile = argv[++i];
934 935
            } else if (0 == strncmp(argv[i], "--file=", strlen("--file="))) {
                arguments->metaFile = (char *)(argv[i] + strlen("--file="));
936
            } else {
937
                errorUnrecognized(argv[0], argv[i]);
938
                exit(EXIT_FAILURE);
939
            }
940
        } else if ((0 == strncmp(argv[i], "-c", strlen("-c")))
941 942 943 944 945 946 947
                || (0 == strncmp(argv[i], "--config-dir", strlen("--config-dir")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "c");
                    exit(EXIT_FAILURE);
                }
                tstrncpy(configDir, argv[++i], TSDB_FILENAME_LEN);
948
            } else if (0 == strncmp(argv[i], "-c", strlen("-c"))) {
949
                tstrncpy(configDir, (char *)(argv[i] + strlen("-c")), TSDB_FILENAME_LEN);
950
            } else if (strlen("--config-dir") == strlen(argv[i])) {
951 952 953 954 955
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--config-dir");
                    exit(EXIT_FAILURE);
                }
                tstrncpy(configDir, argv[++i], TSDB_FILENAME_LEN);
956 957
            } else if (0 == strncmp(argv[i], "--config-dir=", strlen("--config-dir="))) {
                tstrncpy(configDir, (char *)(argv[i] + strlen("--config-dir=")), TSDB_FILENAME_LEN);
958
            } else {
959
                errorUnrecognized(argv[0], argv[i]);
960
                exit(EXIT_FAILURE);
961
            }
962
        } else if ((0 == strncmp(argv[i], "-h", strlen("-h")))
963 964 965 966 967 968 969
                || (0 == strncmp(argv[i], "--host", strlen("--host")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "h");
                    exit(EXIT_FAILURE);
                }
                arguments->host = argv[++i];
970 971
            } else if (0 == strncmp(argv[i], "-h", strlen("-h"))) {
                arguments->host = (char *)(argv[i] + strlen("-h"));
972
            } else if (strlen("--host") == strlen(argv[i])) {
973 974 975 976 977
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--host");
                    exit(EXIT_FAILURE);
                }
                arguments->host = argv[++i];
978 979
            } else if (0 == strncmp(argv[i], "--host=", strlen("--host="))) {
                arguments->host = (char *)(argv[i] + strlen("--host="));
980
            } else {
981
                errorUnrecognized(argv[0], argv[i]);
982
                exit(EXIT_FAILURE);
983
            }
984 985 986
        } else if (strcmp(argv[i], "-PP") == 0) {
            arguments->performance_print = true;
        } else if ((0 == strncmp(argv[i], "-P", strlen("-P")))
987 988 989 990 991 992 993 994 995 996 997 998 999 1000
                || (0 == strncmp(argv[i], "--port", strlen("--port")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "P");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "P");
                    exit(EXIT_FAILURE);
                }
                arguments->port = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--port=", strlen("--port="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--port=")))) {
                    arguments->port = atoi((char *)(argv[i]+strlen("--port=")));
                }
1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014
            } else if (0 == strncmp(argv[i], "-P", strlen("-P"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-P")))) {
                    arguments->port = atoi((char *)(argv[i]+strlen("-P")));
                }
            } else if (strlen("--port") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--port");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--port");
                    exit(EXIT_FAILURE);
                }
                arguments->port = atoi(argv[++i]);
            } else {
1015
                errorUnrecognized(argv[0], argv[i]);
1016
                exit(EXIT_FAILURE);
1017
            }
1018
        } else if ((0 == strncmp(argv[i], "-I", strlen("-I")))
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031
                || (0 == strncmp(argv[i], "--interface", strlen("--interface")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "I");
                    exit(EXIT_FAILURE);
                }
                if (0 == strcasecmp(argv[i+1], "taosc")) {
                    arguments->iface = TAOSC_IFACE;
                } else if (0 == strcasecmp(argv[i+1], "rest")) {
                    arguments->iface = REST_IFACE;
                } else if (0 == strcasecmp(argv[i+1], "stmt")) {
                    arguments->iface = STMT_IFACE;
                } else {
1032
                    errorWrongValue(argv[0], "-I", argv[i+1]);
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046
                    exit(EXIT_FAILURE);
                }
                i++;
            } else if (0 == strncmp(argv[i], "--interface=", strlen("--interface="))) {
                if (0 == strcasecmp((char *)(argv[i] + strlen("--interface=")), "taosc")) {
                    arguments->iface = TAOSC_IFACE;
                } else if (0 == strcasecmp((char *)(argv[i] + strlen("--interface=")), "rest")) {
                    arguments->iface = REST_IFACE;
                } else if (0 == strcasecmp((char *)(argv[i] + strlen("--interface=")), "stmt")) {
                    arguments->iface = STMT_IFACE;
                } else {
                    errorPrintReqArg3(argv[0], "--interface");
                    exit(EXIT_FAILURE);
                }
1047 1048 1049 1050 1051 1052 1053 1054
            } else if (0 == strncmp(argv[i], "-I", strlen("-I"))) {
                if (0 == strcasecmp((char *)(argv[i] + strlen("-I")), "taosc")) {
                    arguments->iface = TAOSC_IFACE;
                } else if (0 == strcasecmp((char *)(argv[i] + strlen("-I")), "rest")) {
                    arguments->iface = REST_IFACE;
                } else if (0 == strcasecmp((char *)(argv[i] + strlen("-I")), "stmt")) {
                    arguments->iface = STMT_IFACE;
                } else {
1055 1056
                    errorWrongValue(argv[0], "-I",
                            (char *)(argv[i] + strlen("-I")));
1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--interface") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--interface");
                    exit(EXIT_FAILURE);
                }
                if (0 == strcasecmp(argv[i+1], "taosc")) {
                    arguments->iface = TAOSC_IFACE;
                } else if (0 == strcasecmp(argv[i+1], "rest")) {
                    arguments->iface = REST_IFACE;
                } else if (0 == strcasecmp(argv[i+1], "stmt")) {
                    arguments->iface = STMT_IFACE;
                } else {
1071
                    errorWrongValue(argv[0], "--interface", argv[i+1]);
1072 1073 1074 1075
                    exit(EXIT_FAILURE);
                }
                i++;
            } else {
1076
                errorUnrecognized(argv[0], argv[i]);
1077
                exit(EXIT_FAILURE);
1078
            }
1079
        } else if ((0 == strncmp(argv[i], "-u", strlen("-u")))
1080 1081 1082 1083 1084 1085 1086
                || (0 == strncmp(argv[i], "--user", strlen("--user")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "u");
                    exit(EXIT_FAILURE);
                }
                arguments->user = argv[++i];
1087 1088
            } else if (0 == strncmp(argv[i], "-u", strlen("-u"))) {
                arguments->user = (char *)(argv[i++] + strlen("-u"));
1089 1090
            } else if (0 == strncmp(argv[i], "--user=", strlen("--user="))) {
                arguments->user = (char *)(argv[i++] + strlen("--user="));
1091 1092 1093 1094 1095 1096 1097
            } else if (strlen("--user") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--user");
                    exit(EXIT_FAILURE);
                }
                arguments->user = argv[++i];
            } else {
1098
                errorUnrecognized(argv[0], argv[i]);
1099
                exit(EXIT_FAILURE);
1100
            }
1101
        } else if ((0 == strncmp(argv[i], "-p", strlen("-p")))
1102 1103
                || (0 == strcmp(argv[i], "--password"))) {
            if ((strlen(argv[i]) == 2) || (0 == strcmp(argv[i], "--password"))) {
1104 1105
                printf("Enter password: ");
                taosSetConsoleEcho(false);
1106 1107 1108
                if (scanf("%s", arguments->password) > 1) {
                    fprintf(stderr, "password read error!\n");
                }
1109
                taosSetConsoleEcho(true);
1110
            } else {
1111
                tstrncpy(arguments->password, (char *)(argv[i] + 2), SHELL_MAX_PASSWORD_LEN);
1112
            }
1113
        } else if ((0 == strncmp(argv[i], "-o", strlen("-o")))
1114 1115 1116
                || (0 == strncmp(argv[i], "--output", strlen("--output")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
1117
                    errorPrintReqArg3(argv[0], "--output");
1118 1119 1120 1121 1122
                    exit(EXIT_FAILURE);
                }
                arguments->output_file = argv[++i];
            } else if (0 == strncmp(argv[i], "--output=", strlen("--output="))) {
                arguments->output_file = (char *)(argv[i++] + strlen("--output="));
1123 1124 1125 1126 1127 1128 1129 1130 1131
            } else if (0 == strncmp(argv[i], "-o", strlen("-o"))) {
                arguments->output_file = (char *)(argv[i++] + strlen("-o"));
            } else if (strlen("--output") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--output");
                    exit(EXIT_FAILURE);
                }
                arguments->output_file = argv[++i];
            } else {
1132
                errorUnrecognized(argv[0], argv[i]);
1133
                exit(EXIT_FAILURE);
1134
            }
1135
        } else if ((0 == strncmp(argv[i], "-s", strlen("-s")))
1136 1137 1138 1139 1140 1141 1142 1143
                || (0 == strncmp(argv[i], "--sql-file", strlen("--sql-file")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "s");
                    exit(EXIT_FAILURE);
                }
                arguments->sqlFile = argv[++i];
            } else if (0 == strncmp(argv[i], "--sql-file=", strlen("--sql-file="))) {
1144
                arguments->sqlFile = (char *)(argv[i++] + strlen("--sql-file="));
1145
            } else if (0 == strncmp(argv[i], "-s", strlen("-s"))) {
1146
                arguments->sqlFile = (char *)(argv[i++] + strlen("-s"));
1147 1148 1149 1150 1151 1152 1153
            } else if (strlen("--sql-file") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--sql-file");
                    exit(EXIT_FAILURE);
                }
                arguments->sqlFile = argv[++i];
            } else {
1154
                errorUnrecognized(argv[0], argv[i]);
1155
                exit(EXIT_FAILURE);
1156
            }
1157
        } else if ((0 == strncmp(argv[i], "-q", strlen("-q")))
1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174
                || (0 == strncmp(argv[i], "--query-mode", strlen("--query-mode")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "q");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "q");
                    exit(EXIT_FAILURE);
                }
                arguments->async_mode = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--query-mode=", strlen("--query-mode="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--query-mode=")))) {
                    arguments->async_mode = atoi((char *)(argv[i]+strlen("--query-mode=")));
                } else {
                    errorPrintReqArg2(argv[0], "--query-mode");
                    exit(EXIT_FAILURE);
                }
1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191
            } else if (0 == strncmp(argv[i], "-q", strlen("-q"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-q")))) {
                    arguments->async_mode = atoi((char *)(argv[i]+strlen("-q")));
                } else {
                    errorPrintReqArg2(argv[0], "-q");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--query-mode") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--query-mode");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--query-mode");
                    exit(EXIT_FAILURE);
                }
                arguments->async_mode = atoi(argv[++i]);
            } else {
1192
                errorUnrecognized(argv[0], argv[i]);
1193
                exit(EXIT_FAILURE);
1194
            }
1195
        } else if ((0 == strncmp(argv[i], "-T", strlen("-T")))
1196 1197 1198 1199 1200 1201 1202 1203 1204
                || (0 == strncmp(argv[i], "--threads", strlen("--threads")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "T");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "T");
                    exit(EXIT_FAILURE);
                }
1205
                arguments->nthreads = atoi(argv[++i]);
1206 1207
            } else if (0 == strncmp(argv[i], "--threads=", strlen("--threads="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--threads=")))) {
1208
                    arguments->nthreads = atoi((char *)(argv[i]+strlen("--threads=")));
1209 1210 1211 1212
                } else {
                    errorPrintReqArg2(argv[0], "--threads");
                    exit(EXIT_FAILURE);
                }
1213 1214
            } else if (0 == strncmp(argv[i], "-T", strlen("-T"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-T")))) {
1215
                    arguments->nthreads = atoi((char *)(argv[i]+strlen("-T")));
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227
                } else {
                    errorPrintReqArg2(argv[0], "-T");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--threads") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--threads");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--threads");
                    exit(EXIT_FAILURE);
                }
1228
                arguments->nthreads = atoi(argv[++i]);
1229
            } else {
1230
                errorUnrecognized(argv[0], argv[i]);
1231
                exit(EXIT_FAILURE);
1232
            }
1233
        } else if ((0 == strncmp(argv[i], "-i", strlen("-i")))
1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244
                || (0 == strncmp(argv[i], "--insert-interval", strlen("--insert-interval")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "i");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "i");
                    exit(EXIT_FAILURE);
                }
                arguments->insert_interval = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--insert-interval=", strlen("--insert-interval="))) {
1245
                if (isStringNumber((char *)(argv[i] + strlen("--insert-interval=")))) {
1246 1247 1248 1249 1250
                    arguments->insert_interval = atoi((char *)(argv[i]+strlen("--insert-interval=")));
                } else {
                    errorPrintReqArg3(argv[0], "--insert-innterval");
                    exit(EXIT_FAILURE);
                }
1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267
            } else if (0 == strncmp(argv[i], "-i", strlen("-i"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-i")))) {
                    arguments->insert_interval = atoi((char *)(argv[i]+strlen("-i")));
                } else {
                    errorPrintReqArg3(argv[0], "-i");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--insert-interval")== strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--insert-interval");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--insert-interval");
                    exit(EXIT_FAILURE);
                }
                arguments->insert_interval = atoi(argv[++i]);
            } else {
1268
                errorUnrecognized(argv[0], argv[i]);
1269 1270
                exit(EXIT_FAILURE);
            }
1271
        } else if ((0 == strncmp(argv[i], "-S", strlen("-S")))
1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288
                || (0 == strncmp(argv[i], "--time-step", strlen("--time-step")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "S");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "S");
                    exit(EXIT_FAILURE);
                }
                arguments->async_mode = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--time-step=", strlen("--time-step="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--time-step=")))) {
                    arguments->async_mode = atoi((char *)(argv[i]+strlen("--time-step=")));
                } else {
                    errorPrintReqArg2(argv[0], "--time-step");
                    exit(EXIT_FAILURE);
                }
1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
            } else if (0 == strncmp(argv[i], "-S", strlen("-S"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-S")))) {
                    arguments->async_mode = atoi((char *)(argv[i]+strlen("-S")));
                } else {
                    errorPrintReqArg2(argv[0], "-S");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--time-step") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--time-step");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--time-step");
                    exit(EXIT_FAILURE);
                }
                arguments->async_mode = atoi(argv[++i]);
            } else {
1306
                errorUnrecognized(argv[0], argv[i]);
1307
                exit(EXIT_FAILURE);
1308
            }
1309 1310 1311 1312 1313 1314 1315 1316
        } else if (strcmp(argv[i], "-qt") == 0) {
            if ((argc == i+1)
                    || (!isStringNumber(argv[i+1]))) {
                printHelp();
                errorPrint("%s", "\n\t-qt need a number following!\n");
                exit(EXIT_FAILURE);
            }
            arguments->query_times = atoi(argv[++i]);
1317
        } else if ((0 == strncmp(argv[i], "-B", strlen("-B")))
1318
                || (0 == strncmp(argv[i], "--interlace-rows", strlen("--interlace-rows")))) {
1319
            if (strlen("-B") == strlen(argv[i])) {
1320 1321 1322 1323 1324 1325 1326
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "B");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "B");
                    exit(EXIT_FAILURE);
                }
1327
                arguments->interlaceRows = atoi(argv[++i]);
1328 1329
            } else if (0 == strncmp(argv[i], "--interlace-rows=", strlen("--interlace-rows="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--interlace-rows=")))) {
1330
                    arguments->interlaceRows = atoi((char *)(argv[i]+strlen("--interlace-rows=")));
1331 1332 1333 1334
                } else {
                    errorPrintReqArg2(argv[0], "--interlace-rows");
                    exit(EXIT_FAILURE);
                }
1335 1336
            } else if (0 == strncmp(argv[i], "-B", strlen("-B"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-B")))) {
1337
                    arguments->interlaceRows = atoi((char *)(argv[i]+strlen("-B")));
1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349
                } else {
                    errorPrintReqArg2(argv[0], "-B");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--interlace-rows")== strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--interlace-rows");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--interlace-rows");
                    exit(EXIT_FAILURE);
                }
1350
                arguments->interlaceRows = atoi(argv[++i]);
1351
            } else {
1352
                errorUnrecognized(argv[0], argv[i]);
1353
                exit(EXIT_FAILURE);
1354
            }
1355
        } else if ((0 == strncmp(argv[i], "-r", strlen("-r")))
1356
                || (0 == strncmp(argv[i], "--rec-per-req", 13))) {
1357
            if (strlen("-r") == strlen(argv[i])) {
1358 1359 1360 1361 1362 1363 1364
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "r");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "r");
                    exit(EXIT_FAILURE);
                }
1365
                arguments->reqPerReq = atoi(argv[++i]);
1366 1367
            } else if (0 == strncmp(argv[i], "--rec-per-req=", strlen("--rec-per-req="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--rec-per-req=")))) {
1368
                    arguments->reqPerReq = atoi((char *)(argv[i]+strlen("--rec-per-req=")));
1369 1370 1371 1372
                } else {
                    errorPrintReqArg2(argv[0], "--rec-per-req");
                    exit(EXIT_FAILURE);
                }
1373 1374
            } else if (0 == strncmp(argv[i], "-r", strlen("-r"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-r")))) {
1375
                    arguments->reqPerReq = atoi((char *)(argv[i]+strlen("-r")));
1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
                } else {
                    errorPrintReqArg2(argv[0], "-r");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--rec-per-req")== strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--rec-per-req");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--rec-per-req");
                    exit(EXIT_FAILURE);
                }
1388
                arguments->reqPerReq = atoi(argv[++i]);
1389
            } else {
1390
                errorUnrecognized(argv[0], argv[i]);
1391
                exit(EXIT_FAILURE);
1392
            }
1393
        } else if ((0 == strncmp(argv[i], "-t", strlen("-t")))
1394 1395 1396 1397 1398 1399 1400 1401 1402
                || (0 == strncmp(argv[i], "--tables", strlen("--tables")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "t");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "t");
                    exit(EXIT_FAILURE);
                }
1403
                arguments->ntables = atoi(argv[++i]);
1404 1405
            } else if (0 == strncmp(argv[i], "--tables=", strlen("--tables="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--tables=")))) {
1406
                    arguments->ntables = atoi((char *)(argv[i]+strlen("--tables=")));
1407 1408 1409 1410
                } else {
                    errorPrintReqArg2(argv[0], "--tables");
                    exit(EXIT_FAILURE);
                }
1411 1412
            } else if (0 == strncmp(argv[i], "-t", strlen("-t"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-t")))) {
1413
                    arguments->ntables = atoi((char *)(argv[i]+strlen("-t")));
1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425
                } else {
                    errorPrintReqArg2(argv[0], "-t");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--tables") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--tables");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--tables");
                    exit(EXIT_FAILURE);
                }
1426
                arguments->ntables = atoi(argv[++i]);
1427
            } else {
1428
                errorUnrecognized(argv[0], argv[i]);
1429
                exit(EXIT_FAILURE);
1430
            }
1431

1432
            g_totalChildTables = arguments->ntables;
1433
        } else if ((0 == strncmp(argv[i], "-n", strlen("-n")))
1434 1435 1436 1437 1438 1439 1440 1441 1442
                || (0 == strncmp(argv[i], "--records", strlen("--records")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "n");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "n");
                    exit(EXIT_FAILURE);
                }
1443
                arguments->insertRows = atoi(argv[++i]);
1444 1445
            } else if (0 == strncmp(argv[i], "--records=", strlen("--records="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--records=")))) {
1446
                    arguments->insertRows = atoi((char *)(argv[i]+strlen("--records=")));
1447 1448 1449 1450
                } else {
                    errorPrintReqArg2(argv[0], "--records");
                    exit(EXIT_FAILURE);
                }
1451 1452
            } else if (0 == strncmp(argv[i], "-n", strlen("-n"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-n")))) {
1453
                    arguments->insertRows = atoi((char *)(argv[i]+strlen("-n")));
1454
                } else {
1455
                    errorPrintReqArg2(argv[0], "-n");
1456 1457 1458 1459 1460 1461 1462 1463 1464 1465
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--records") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--records");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--records");
                    exit(EXIT_FAILURE);
                }
1466
                arguments->insertRows = atoi(argv[++i]);
1467
            } else {
1468
                errorUnrecognized(argv[0], argv[i]);
1469
                exit(EXIT_FAILURE);
1470
            }
1471
        } else if ((0 == strncmp(argv[i], "-d", strlen("-d")))
1472 1473 1474
                || (0 == strncmp(argv[i], "--database", strlen("--database")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
1475
                    errorPrintReqArg(argv[0], "d");
1476 1477 1478 1479 1480
                    exit(EXIT_FAILURE);
                }
                arguments->database = argv[++i];
            } else if (0 == strncmp(argv[i], "--database=", strlen("--database="))) {
                arguments->output_file = (char *)(argv[i] + strlen("--database="));
1481 1482 1483 1484 1485 1486 1487 1488 1489
            } else if (0 == strncmp(argv[i], "-d", strlen("-d"))) {
                arguments->output_file = (char *)(argv[i] + strlen("-d"));
            } else if (strlen("--database") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--database");
                    exit(EXIT_FAILURE);
                }
                arguments->database = argv[++i];
            } else {
1490
                errorUnrecognized(argv[0], argv[i]);
1491
                exit(EXIT_FAILURE);
1492
            }
1493
        } else if ((0 == strncmp(argv[i], "-l", strlen("-l")))
1494
                || (0 == strncmp(argv[i], "--columns", strlen("--columns")))) {
1495
            arguments->demo_mode = false;
1496 1497 1498 1499 1500 1501 1502 1503
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "l");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "l");
                    exit(EXIT_FAILURE);
                }
1504
                arguments->columnCount = atoi(argv[++i]);
1505 1506
            } else if (0 == strncmp(argv[i], "--columns=", strlen("--columns="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--columns=")))) {
1507
                    arguments->columnCount = atoi((char *)(argv[i]+strlen("--columns=")));
1508 1509 1510 1511 1512 1513
                } else {
                    errorPrintReqArg2(argv[0], "--columns");
                    exit(EXIT_FAILURE);
                }
            } else if (0 == strncmp(argv[i], "-l", strlen("-l"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-l")))) {
1514
                    arguments->columnCount = atoi((char *)(argv[i]+strlen("-l")));
1515
                } else {
1516 1517 1518 1519 1520 1521 1522 1523
                    errorPrintReqArg2(argv[0], "-l");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--columns")== strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--columns");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
1524
                    errorPrintReqArg2(argv[0], "--columns");
1525 1526
                    exit(EXIT_FAILURE);
                }
1527
                arguments->columnCount = atoi(argv[++i]);
1528
            } else {
1529
                errorUnrecognized(argv[0], argv[i]);
1530
                exit(EXIT_FAILURE);
1531
            }
1532

1533
            if (arguments->columnCount > MAX_NUM_COLUMNS) {
1534
                printf("WARNING: max acceptable columns count is %d\n", MAX_NUM_COLUMNS);
1535
                prompt();
1536
                arguments->columnCount = MAX_NUM_COLUMNS;
1537
            }
1538

1539 1540 1541
            for (int col = DEFAULT_DATATYPE_NUM; col < arguments->columnCount; col ++) {
                arguments->dataType[col] = "INT";
                arguments->data_type[col] = TSDB_DATA_TYPE_INT;
1542
            }
1543 1544 1545
            for (int col = arguments->columnCount; col < MAX_NUM_COLUMNS; col++) {
                arguments->dataType[col] = NULL;
                arguments->data_type[col] = TSDB_DATA_TYPE_NULL;
1546
            }
1547
        } else if ((0 == strncmp(argv[i], "-b", strlen("-b")))
1548
                || (0 == strncmp(argv[i], "--data-type", strlen("--data-type")))) {
1549
            arguments->demo_mode = false;
1550 1551 1552 1553 1554 1555 1556 1557 1558 1559

            char *dataType;
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "b");
                    exit(EXIT_FAILURE);
                }
                dataType = argv[++i];
            } else if (0 == strncmp(argv[i], "--data-type=", strlen("--data-type="))) {
                dataType = (char *)(argv[i] + strlen("--data-type="));
1560 1561 1562 1563 1564 1565 1566 1567 1568
            } else if (0 == strncmp(argv[i], "-b", strlen("-b"))) {
                dataType = (char *)(argv[i] + strlen("-b"));
            } else if (strlen("--data-type") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--data-type");
                    exit(EXIT_FAILURE);
                }
                dataType = argv[++i];
            } else {
1569
                errorUnrecognized(argv[0], argv[i]);
1570
                exit(EXIT_FAILURE);
1571
            }
1572 1573

            if (strstr(dataType, ",") == NULL) {
1574
                // only one col
1575 1576 1577 1578 1579 1580 1581 1582 1583
                if (strcasecmp(dataType, "INT")
                        && strcasecmp(dataType, "FLOAT")
                        && strcasecmp(dataType, "TINYINT")
                        && strcasecmp(dataType, "BOOL")
                        && strcasecmp(dataType, "SMALLINT")
                        && strcasecmp(dataType, "BIGINT")
                        && strcasecmp(dataType, "DOUBLE")
                        && strcasecmp(dataType, "BINARY")
                        && strcasecmp(dataType, "TIMESTAMP")
1584 1585 1586 1587 1588
                        && strcasecmp(dataType, "NCHAR")
                        && strcasecmp(dataType, "UTINYINT")
                        && strcasecmp(dataType, "USMALLINT")
                        && strcasecmp(dataType, "UINT")
                        && strcasecmp(dataType, "UBIGINT")) {
1589 1590 1591 1592
                    printHelp();
                    errorPrint("%s", "-b: Invalid data_type!\n");
                    exit(EXIT_FAILURE);
                }
1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613
                arguments->dataType[0] = dataType;
                if (0 == strcasecmp(dataType, "INT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_INT;
                } else if (0 == strcasecmp(dataType, "TINYINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_TINYINT;
                } else if (0 == strcasecmp(dataType, "SMALLINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_SMALLINT;
                } else if (0 == strcasecmp(dataType, "BIGINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_BIGINT;
                } else if (0 == strcasecmp(dataType, "FLOAT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_FLOAT;
                } else if (0 == strcasecmp(dataType, "DOUBLE")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_DOUBLE;
                } else if (0 == strcasecmp(dataType, "BINARY")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_BINARY;
                } else if (0 == strcasecmp(dataType, "NCHAR")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_NCHAR;
                } else if (0 == strcasecmp(dataType, "BOOL")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_BOOL;
                } else if (0 == strcasecmp(dataType, "TIMESTAMP")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_TIMESTAMP;
1614 1615 1616 1617 1618 1619 1620 1621
                } else if (0 == strcasecmp(dataType, "UTINYINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_UTINYINT;
                } else if (0 == strcasecmp(dataType, "USMALLINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_USMALLINT;
                } else if (0 == strcasecmp(dataType, "UINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_UINT;
                } else if (0 == strcasecmp(dataType, "UBIGINT")) {
                    arguments->data_type[0] = TSDB_DATA_TYPE_UBIGINT;
1622 1623 1624 1625 1626
                } else {
                    arguments->data_type[0] = TSDB_DATA_TYPE_NULL;
                }
                arguments->dataType[1] = NULL;
                arguments->data_type[1] = TSDB_DATA_TYPE_NULL;
1627 1628 1629
            } else {
                // more than one col
                int index = 0;
1630
                g_dupstr = strdup(dataType);
1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642
                char *running = g_dupstr;
                char *token = strsep(&running, ",");
                while(token != NULL) {
                    if (strcasecmp(token, "INT")
                            && strcasecmp(token, "FLOAT")
                            && strcasecmp(token, "TINYINT")
                            && strcasecmp(token, "BOOL")
                            && strcasecmp(token, "SMALLINT")
                            && strcasecmp(token, "BIGINT")
                            && strcasecmp(token, "DOUBLE")
                            && strcasecmp(token, "BINARY")
                            && strcasecmp(token, "TIMESTAMP")
1643 1644 1645 1646 1647
                            && strcasecmp(token, "NCHAR")
                            && strcasecmp(token, "UTINYINT")
                            && strcasecmp(token, "USMALLINT")
                            && strcasecmp(token, "UINT")
                            && strcasecmp(token, "UBIGINT")) {
1648 1649 1650 1651 1652
                        printHelp();
                        free(g_dupstr);
                        errorPrint("%s", "-b: Invalid data_type!\n");
                        exit(EXIT_FAILURE);
                    }
1653 1654 1655 1656 1657 1658 1659 1660 1661 1662

                    if (0 == strcasecmp(token, "INT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_INT;
                    } else if (0 == strcasecmp(token, "FLOAT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_FLOAT;
                    } else if (0 == strcasecmp(token, "SMALLINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_SMALLINT;
                    } else if (0 == strcasecmp(token, "BIGINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_BIGINT;
                    } else if (0 == strcasecmp(token, "DOUBLE")) {
1663
                        arguments->data_type[index] = TSDB_DATA_TYPE_DOUBLE;
1664 1665 1666 1667 1668 1669 1670 1671 1672 1673
                    } else if (0 == strcasecmp(token, "TINYINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_TINYINT;
                    } else if (0 == strcasecmp(token, "BINARY")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_BINARY;
                    } else if (0 == strcasecmp(token, "NCHAR")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_NCHAR;
                    } else if (0 == strcasecmp(token, "BOOL")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_BOOL;
                    } else if (0 == strcasecmp(token, "TIMESTAMP")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_TIMESTAMP;
1674 1675 1676 1677 1678 1679 1680 1681
                    } else if (0 == strcasecmp(token, "UTINYINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_UTINYINT;
                    } else if (0 == strcasecmp(token, "USMALLINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_USMALLINT;
                    } else if (0 == strcasecmp(token, "UINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_UINT;
                    } else if (0 == strcasecmp(token, "UBIGINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_UBIGINT;
1682 1683 1684 1685 1686
                    } else {
                        arguments->data_type[index] = TSDB_DATA_TYPE_NULL;
                    }
                    arguments->dataType[index] = token;
                    index ++;
1687
                    token = strsep(&running, ",");
1688
                    if (index >= MAX_NUM_COLUMNS) break;
1689
                }
1690 1691
                arguments->dataType[index] = NULL;
                arguments->data_type[index] = TSDB_DATA_TYPE_NULL;
1692
            }
1693
        } else if ((0 == strncmp(argv[i], "-w", strlen("-w")))
1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710
                || (0 == strncmp(argv[i], "--binwidth", strlen("--binwidth")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "w");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "w");
                    exit(EXIT_FAILURE);
                }
                arguments->binwidth = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--binwidth=", strlen("--binwidth="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--binwidth=")))) {
                    arguments->binwidth = atoi((char *)(argv[i]+strlen("--binwidth=")));
                } else {
                    errorPrintReqArg2(argv[0], "--binwidth");
                    exit(EXIT_FAILURE);
                }
1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727
            } else if (0 == strncmp(argv[i], "-w", strlen("-w"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-w")))) {
                    arguments->binwidth = atoi((char *)(argv[i]+strlen("-w")));
                } else {
                    errorPrintReqArg2(argv[0], "-w");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--binwidth") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--binwidth");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--binwidth");
                    exit(EXIT_FAILURE);
                }
                arguments->binwidth = atoi(argv[++i]);
            } else {
1728
                errorUnrecognized(argv[0], argv[i]);
1729
                exit(EXIT_FAILURE);
1730
            }
1731
        } else if ((0 == strncmp(argv[i], "-m", strlen("-m")))
1732 1733 1734
                || (0 == strncmp(argv[i], "--table-prefix", strlen("--table-prefix")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
1735
                    errorPrintReqArg(argv[0], "m");
1736 1737 1738 1739 1740
                    exit(EXIT_FAILURE);
                }
                arguments->tb_prefix = argv[++i];
            } else if (0 == strncmp(argv[i], "--table-prefix=", strlen("--table-prefix="))) {
                arguments->tb_prefix = (char *)(argv[i] + strlen("--table-prefix="));
1741 1742 1743 1744 1745 1746 1747 1748 1749
            } else if (0 == strncmp(argv[i], "-m", strlen("-m"))) {
                arguments->tb_prefix = (char *)(argv[i] + strlen("-m"));
            } else if (strlen("--table-prefix") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--table-prefix");
                    exit(EXIT_FAILURE);
                }
                arguments->tb_prefix = argv[++i];
            } else {
1750
                errorUnrecognized(argv[0], argv[i]);
1751
                exit(EXIT_FAILURE);
1752
            }
1753 1754
        } else if ((strcmp(argv[i], "-N") == 0)
                || (0 == strcmp(argv[i], "--normal-table"))) {
1755
            arguments->demo_mode = false;
1756
            arguments->use_metric = false;
1757 1758
        } else if ((strcmp(argv[i], "-M") == 0)
                || (0 == strcmp(argv[i], "--random"))) {
1759
            arguments->demo_mode = false;
1760
        } else if ((strcmp(argv[i], "-x") == 0)
1761 1762
                || (0 == strcmp(argv[i], "--aggr-func"))) {
            arguments->aggr_func = true;
1763 1764
        } else if ((strcmp(argv[i], "-y") == 0)
                || (0 == strcmp(argv[i], "--answer-yes"))) {
1765
            arguments->answer_yes = true;
1766 1767
        } else if ((strcmp(argv[i], "-g") == 0)
                || (0 == strcmp(argv[i], "--debug"))) {
1768 1769 1770
            arguments->debug_print = true;
        } else if (strcmp(argv[i], "-gg") == 0) {
            arguments->verbose_print = true;
1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815
        } else if ((0 == strncmp(argv[i], "-R", strlen("-R")))
                || (0 == strncmp(argv[i], "--disorder-range",
                        strlen("--disorder-range")))) {
            if (strlen("-R") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "R");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "R");
                    exit(EXIT_FAILURE);
                }
                arguments->disorderRange = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--disorder-range=",
                        strlen("--disorder-range="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--disorder-range=")))) {
                    arguments->disorderRange =
                        atoi((char *)(argv[i]+strlen("--disorder-range=")));
                } else {
                    errorPrintReqArg2(argv[0], "--disorder-range");
                    exit(EXIT_FAILURE);
                }
            } else if (0 == strncmp(argv[i], "-R", strlen("-R"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-R")))) {
                    arguments->disorderRange =
                        atoi((char *)(argv[i]+strlen("-R")));
                } else {
                    errorPrintReqArg2(argv[0], "-R");
                    exit(EXIT_FAILURE);
                }

                if (arguments->disorderRange < 0) {
                    errorPrint("Invalid disorder range %d, will be set to %d\n",
                            arguments->disorderRange, 1000);
                    arguments->disorderRange = 1000;
                }
            } else if (strlen("--disorder-range") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--disorder-range");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--disorder-range");
                    exit(EXIT_FAILURE);
                }
                arguments->disorderRange = atoi(argv[++i]);
            } else {
1816
                errorUnrecognized(argv[0], argv[i]);
1817 1818
                exit(EXIT_FAILURE);
            }
1819
        } else if ((0 == strncmp(argv[i], "-O", strlen("-O")))
1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836
                || (0 == strncmp(argv[i], "--disorder", strlen("--disorder")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "O");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "O");
                    exit(EXIT_FAILURE);
                }
                arguments->disorderRatio = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--disorder=", strlen("--disorder="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--disorder=")))) {
                    arguments->disorderRatio = atoi((char *)(argv[i]+strlen("--disorder=")));
                } else {
                    errorPrintReqArg2(argv[0], "--disorder");
                    exit(EXIT_FAILURE);
                }
1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853
            } else if (0 == strncmp(argv[i], "-O", strlen("-O"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-O")))) {
                    arguments->disorderRatio = atoi((char *)(argv[i]+strlen("-O")));
                } else {
                    errorPrintReqArg2(argv[0], "-O");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--disorder") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--disorder");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--disorder");
                    exit(EXIT_FAILURE);
                }
                arguments->disorderRatio = atoi(argv[++i]);
            } else {
1854
                errorUnrecognized(argv[0], argv[i]);
1855
                exit(EXIT_FAILURE);
1856
            }
1857

1858
            if (arguments->disorderRatio > 50) {
1859 1860
                errorPrint("Invalid disorder ratio %d, will be set to %d\n",
                        arguments->disorderRatio, 50);
1861 1862
                arguments->disorderRatio = 50;
            }
1863

1864
            if (arguments->disorderRatio < 0) {
1865 1866
                errorPrint("Invalid disorder ratio %d, will be set to %d\n",
                        arguments->disorderRatio, 0);
1867 1868
                arguments->disorderRatio = 0;
            }
1869
        } else if ((0 == strncmp(argv[i], "-a", strlen("-a")))
1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889
                || (0 == strncmp(argv[i], "--replica",
                        strlen("--replica")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "a");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "a");
                    exit(EXIT_FAILURE);
                }
                arguments->replica = atoi(argv[++i]);
            } else if (0 == strncmp(argv[i], "--replica=",
                        strlen("--replica="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--replica=")))) {
                    arguments->replica =
                        atoi((char *)(argv[i]+strlen("--replica=")));
                } else {
                    errorPrintReqArg2(argv[0], "--replica");
                    exit(EXIT_FAILURE);
                }
1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907
            } else if (0 == strncmp(argv[i], "-a", strlen("-a"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-a")))) {
                    arguments->replica =
                        atoi((char *)(argv[i]+strlen("-a")));
                } else {
                    errorPrintReqArg2(argv[0], "-a");
                    exit(EXIT_FAILURE);
                }
            } else if (strlen("--replica") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--replica");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "--replica");
                    exit(EXIT_FAILURE);
                }
                arguments->replica = atoi(argv[++i]);
            } else {
1908
                errorUnrecognized(argv[0], argv[i]);
1909
                exit(EXIT_FAILURE);
1910
            }
1911

1912
            if (arguments->replica > 3 || arguments->replica < 1) {
1913 1914
                errorPrint("Invalid replica value %d, will be set to %d\n",
                        arguments->replica, 1);
1915 1916 1917 1918 1919
                arguments->replica = 1;
            }
        } else if (strcmp(argv[i], "-D") == 0) {
            arguments->method_of_delete = atoi(argv[++i]);
            if (arguments->method_of_delete > 3) {
1920
                errorPrint("%s", "\n\t-D need a value (0~3) number following!\n");
1921 1922
                exit(EXIT_FAILURE);
            }
1923 1924
        } else if ((strcmp(argv[i], "--version") == 0)
                || (strcmp(argv[i], "-V") == 0)) {
1925 1926
            printVersion();
            exit(0);
1927 1928
        } else if ((strcmp(argv[i], "--help") == 0)
                || (strcmp(argv[i], "-?") == 0)) {
1929 1930
            printHelp();
            exit(0);
1931 1932 1933 1934
        } else if (strcmp(argv[i], "--usage") == 0) {
            printf("    Usage: taosdemo [-f JSONFILE] [-u USER] [-p PASSWORD] [-c CONFIG_DIR]\n\
                    [-h HOST] [-P PORT] [-I INTERFACE] [-d DATABASE] [-a REPLICA]\n\
                    [-m TABLEPREFIX] [-s SQLFILE] [-N] [-o OUTPUTFILE] [-q QUERYMODE]\n\
1935
                    [-b DATATYPES] [-w WIDTH_OF_BINARY] [-l COLUMNS] [-T THREADNUMBER]\n\
1936 1937 1938 1939
                    [-i SLEEPTIME] [-S TIME_STEP] [-B INTERLACE_ROWS] [-t TABLES]\n\
                    [-n RECORDS] [-M] [-x] [-y] [-O ORDERMODE] [-R RANGE] [-a REPLIcA][-g]\n\
                    [--help] [--usage] [--version]\n");
            exit(0);
1940
        } else {
1941 1942
            // to simulate argp_option output
            if (strlen(argv[i]) > 2) {
1943 1944 1945 1946 1947 1948 1949 1950 1951
                if (0 == strncmp(argv[i], "--", 2)) {
                    fprintf(stderr, "%s: unrecognized options '%s'\n", argv[0], argv[i]);
                } else if (0 == strncmp(argv[i], "-", 1)) {
                    char tmp[2] = {0};
                    tstrncpy(tmp, argv[i]+1, 2);
                    fprintf(stderr, "%s: invalid options -- '%s'\n", argv[0], tmp);
                } else {
                    fprintf(stderr, "%s: Too many arguments\n", argv[0]);
                }
1952 1953 1954 1955 1956
            } else {
                fprintf(stderr, "%s invalid options -- '%s'\n", argv[0],
                        (char *)((char *)argv[i])+1);
            }
            fprintf(stderr, "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
1957 1958
            exit(EXIT_FAILURE);
        }
S
TD-1057  
Shengliang Guan 已提交
1959
    }
1960

1961
    int columnCount;
1962
    for (columnCount = 0; columnCount < MAX_NUM_COLUMNS; columnCount ++) {
1963
        if (g_args.dataType[columnCount] == NULL) {
1964 1965
            break;
        }
1966 1967
    }

1968
    if (0 == columnCount) {
1969
        ERROR_EXIT("data type error!");
1970
    }
1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984
    g_args.columnCount = columnCount;

    g_args.lenOfOneRow = 20; // timestamp
    for (int c = 0; c < g_args.columnCount; c++) {
        switch(g_args.data_type[c]) {
            case TSDB_DATA_TYPE_BINARY:
                g_args.lenOfOneRow += g_args.binwidth + 3;
                break;

            case TSDB_DATA_TYPE_NCHAR:
                g_args.lenOfOneRow += g_args.binwidth + 3;
                break;

            case TSDB_DATA_TYPE_INT:
1985
            case TSDB_DATA_TYPE_UINT:
1986 1987 1988 1989
                g_args.lenOfOneRow += INT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BIGINT:
1990
            case TSDB_DATA_TYPE_UBIGINT:
1991 1992 1993 1994
                g_args.lenOfOneRow += BIGINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
1995
            case TSDB_DATA_TYPE_USMALLINT:
1996 1997 1998 1999
                g_args.lenOfOneRow += SMALLINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TINYINT:
2000
            case TSDB_DATA_TYPE_UTINYINT:
2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024
                g_args.lenOfOneRow += TINYINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BOOL:
                g_args.lenOfOneRow += BOOL_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_FLOAT:
                g_args.lenOfOneRow += FLOAT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_DOUBLE:
                g_args.lenOfOneRow += DOUBLE_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TIMESTAMP:
                g_args.lenOfOneRow += TIMESTAMP_BUFF_LEN;
                break;

            default:
                errorPrint2("get error data type : %s\n", g_args.dataType[c]);
                exit(EXIT_FAILURE);
        }
    }
2025

2026
    if (((arguments->debug_print) && (NULL != arguments->metaFile))
2027 2028 2029 2030 2031 2032 2033 2034 2035 2036
            || arguments->verbose_print) {
        printf("###################################################################\n");
        printf("# meta file:                         %s\n", arguments->metaFile);
        printf("# Server IP:                         %s:%hu\n",
                arguments->host == NULL ? "localhost" : arguments->host,
                arguments->port );
        printf("# User:                              %s\n", arguments->user);
        printf("# Password:                          %s\n", arguments->password);
        printf("# Use metric:                        %s\n",
                arguments->use_metric ? "true" : "false");
2037
        if (*(arguments->dataType)) {
2038
            printf("# Specified data type:               ");
2039
            for (int c = 0; c < MAX_NUM_COLUMNS; c++)
2040 2041
                if (arguments->dataType[c])
                    printf("%s,", arguments->dataType[c]);
2042 2043 2044 2045 2046 2047 2048
                else
                    break;
            printf("\n");
        }
        printf("# Insertion interval:                %"PRIu64"\n",
                arguments->insert_interval);
        printf("# Number of records per req:         %u\n",
2049
                arguments->reqPerReq);
2050 2051
        printf("# Max SQL length:                    %"PRIu64"\n",
                arguments->max_sql_len);
2052
        printf("# Length of Binary:                  %d\n", arguments->binwidth);
2053
        printf("# Number of Threads:                 %d\n", arguments->nthreads);
2054
        printf("# Number of Tables:                  %"PRId64"\n",
2055
                arguments->ntables);
2056
        printf("# Number of Data per Table:          %"PRId64"\n",
2057
                arguments->insertRows);
2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070
        printf("# Database name:                     %s\n", arguments->database);
        printf("# Table prefix:                      %s\n", arguments->tb_prefix);
        if (arguments->disorderRatio) {
            printf("# Data order:                        %d\n", arguments->disorderRatio);
            printf("# Data out of order rate:            %d\n", arguments->disorderRange);
        }
        printf("# Delete method:                     %d\n", arguments->method_of_delete);
        printf("# Answer yes when prompt:            %d\n", arguments->answer_yes);
        printf("# Print debug info:                  %d\n", arguments->debug_print);
        printf("# Print verbose info:                %d\n", arguments->verbose_print);
        printf("###################################################################\n");

        prompt();
2071
    }
2072
}
H
hzcheng 已提交
2073

2074
static void tmfclose(FILE *fp) {
2075 2076 2077
    if (NULL != fp) {
        fclose(fp);
    }
2078
}
2079

2080
static void tmfree(char *buf) {
2081 2082
    if (NULL != buf) {
        free(buf);
2083
        buf = NULL;
2084
    }
2085
}
H
Hui Li 已提交
2086

2087
static int queryDbExec(TAOS *taos, char *command, QUERY_TYPE type, bool quiet) {
2088

2089
    verbosePrint("%s() LN%d - command: %s\n", __func__, __LINE__, command);
2090

2091 2092
    TAOS_RES *res = taos_query(taos, command);
    int32_t code = taos_errno(res);
2093

2094 2095
    if (code != 0) {
        if (!quiet) {
2096
            errorPrint2("Failed to execute <%s>, reason: %s\n",
2097 2098 2099 2100 2101
                    command, taos_errstr(res));
        }
        taos_free_result(res);
        //taos_close(taos);
        return -1;
2102
    }
H
hzcheng 已提交
2103

2104 2105 2106 2107
    if (INSERT_TYPE == type) {
        int affectedRows = taos_affected_rows(res);
        taos_free_result(res);
        return affectedRows;
2108
    }
H
hzcheng 已提交
2109

2110
    taos_free_result(res);
2111
    return 0;
2112
}
H
hzcheng 已提交
2113

2114
static void appendResultBufToFile(char *resultBuf, threadInfo *pThreadInfo)
2115
{
2116 2117
    pThreadInfo->fp = fopen(pThreadInfo->filePath, "at");
    if (pThreadInfo->fp == NULL) {
2118
        errorPrint2(
2119 2120 2121 2122
                "%s() LN%d, failed to open result file: %s, result will not save to file\n",
                __func__, __LINE__, pThreadInfo->filePath);
        return;
    }
2123

2124 2125 2126
    fprintf(pThreadInfo->fp, "%s", resultBuf);
    tmfclose(pThreadInfo->fp);
    pThreadInfo->fp = NULL;
2127 2128
}

2129
static void fetchResult(TAOS_RES *res, threadInfo* pThreadInfo) {
2130 2131 2132 2133 2134 2135 2136
    TAOS_ROW    row = NULL;
    int         num_rows = 0;
    int         num_fields = taos_field_count(res);
    TAOS_FIELD *fields     = taos_fetch_fields(res);

    char* databuf = (char*) calloc(1, 100*1024*1024);
    if (databuf == NULL) {
2137
        errorPrint2("%s() LN%d, failed to malloc, warning: save result to file slowly!\n",
2138 2139 2140
                __func__, __LINE__);
        return ;
    }
H
hzcheng 已提交
2141

2142
    int64_t   totalLen = 0;
H
hzcheng 已提交
2143

2144 2145
    // fetch the records row by row
    while((row = taos_fetch_row(res))) {
2146
        if (totalLen >= (100*1024*1024 - HEAD_BUFF_LEN*2)) {
2147 2148 2149 2150 2151 2152
            if (strlen(pThreadInfo->filePath) > 0)
                appendResultBufToFile(databuf, pThreadInfo);
            totalLen = 0;
            memset(databuf, 0, 100*1024*1024);
        }
        num_rows++;
2153
        char  temp[HEAD_BUFF_LEN] = {0};
2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168
        int len = taos_print_row(temp, row, fields, num_fields);
        len += sprintf(temp + len, "\n");
        //printf("query result:%s\n", temp);
        memcpy(databuf + totalLen, temp, len);
        totalLen += len;
        verbosePrint("%s() LN%d, totalLen: %"PRId64"\n",
                __func__, __LINE__, totalLen);
    }

    verbosePrint("%s() LN%d, databuf=%s resultFile=%s\n",
            __func__, __LINE__, databuf, pThreadInfo->filePath);
    if (strlen(pThreadInfo->filePath) > 0) {
        appendResultBufToFile(databuf, pThreadInfo);
    }
    free(databuf);
2169
}
H
hzcheng 已提交
2170

2171
static void selectAndGetResult(
2172 2173
        threadInfo *pThreadInfo, char *command)
{
2174 2175 2176
    if (0 == strncasecmp(g_queryInfo.queryMode, "taosc", strlen("taosc"))) {
        TAOS_RES *res = taos_query(pThreadInfo->taos, command);
        if (res == NULL || taos_errno(res) != 0) {
2177
            errorPrint2("%s() LN%d, failed to execute sql:%s, reason:%s\n",
2178 2179 2180 2181
                    __func__, __LINE__, command, taos_errstr(res));
            taos_free_result(res);
            return;
        }
2182

2183 2184
        fetchResult(res, pThreadInfo);
        taos_free_result(res);
2185

2186 2187 2188 2189 2190 2191 2192 2193
    } else if (0 == strncasecmp(g_queryInfo.queryMode, "rest", strlen("rest"))) {
        int retCode = postProceSql(
                g_queryInfo.host, &(g_queryInfo.serv_addr), g_queryInfo.port,
                command,
                pThreadInfo);
        if (0 != retCode) {
            printf("====restful return fail, threadID[%d]\n", pThreadInfo->threadID);
        }
2194

2195
    } else {
2196
        errorPrint2("%s() LN%d, unknown query mode: %s\n",
2197 2198
                __func__, __LINE__, g_queryInfo.queryMode);
    }
2199
}
H
hzcheng 已提交
2200

2201
static char *rand_bool_str() {
2202 2203 2204
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2205
    return g_randbool_buff + ((cursor % MAX_PREPARED_RAND) * BOOL_BUFF_LEN);
2206 2207
}

2208
static int32_t rand_bool() {
2209 2210
    static int cursor;
    cursor++;
2211 2212
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randint[cursor % MAX_PREPARED_RAND] % 2;
2213 2214
}

2215 2216 2217 2218 2219
static char *rand_tinyint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2220 2221
    return g_randtinyint_buff +
        ((cursor % MAX_PREPARED_RAND) * TINYINT_BUFF_LEN);
2222 2223 2224 2225
}

static int32_t rand_tinyint()
{
2226 2227
    static int cursor;
    cursor++;
2228 2229
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randint[cursor % MAX_PREPARED_RAND] % 128;
2230 2231
}

2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248
static char *rand_utinyint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randutinyint_buff +
        ((cursor % MAX_PREPARED_RAND) * TINYINT_BUFF_LEN);
}

static int32_t rand_utinyint()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randuint[cursor % MAX_PREPARED_RAND] % 255;
}

2249 2250 2251 2252 2253
static char *rand_smallint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2254 2255
    return g_randsmallint_buff +
        ((cursor % MAX_PREPARED_RAND) * SMALLINT_BUFF_LEN);
2256 2257
}

2258 2259
static int32_t rand_smallint()
{
2260 2261
    static int cursor;
    cursor++;
2262
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280
    return g_randint[cursor % MAX_PREPARED_RAND] % 32768;
}

static char *rand_usmallint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randusmallint_buff +
        ((cursor % MAX_PREPARED_RAND) * SMALLINT_BUFF_LEN);
}

static int32_t rand_usmallint()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randuint[cursor % MAX_PREPARED_RAND] % 65535;
2281 2282
}

2283 2284 2285 2286 2287
static char *rand_int_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2288
    return g_randint_buff + ((cursor % MAX_PREPARED_RAND) * INT_BUFF_LEN);
2289 2290 2291 2292
}

static int32_t rand_int()
{
2293 2294
    static int cursor;
    cursor++;
2295 2296
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randint[cursor % MAX_PREPARED_RAND];
2297 2298
}

2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314
static char *rand_uint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randuint_buff + ((cursor % MAX_PREPARED_RAND) * INT_BUFF_LEN);
}

static int32_t rand_uint()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randuint[cursor % MAX_PREPARED_RAND];
}

2315 2316 2317 2318 2319
static char *rand_bigint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2320 2321
    return g_randbigint_buff +
        ((cursor % MAX_PREPARED_RAND) * BIGINT_BUFF_LEN);
2322 2323
}

2324 2325
static int64_t rand_bigint()
{
2326 2327
    static int cursor;
    cursor++;
2328 2329
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randbigint[cursor % MAX_PREPARED_RAND];
2330 2331
}

2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348
static char *rand_ubigint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randubigint_buff +
        ((cursor % MAX_PREPARED_RAND) * BIGINT_BUFF_LEN);
}

static int64_t rand_ubigint()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randubigint[cursor % MAX_PREPARED_RAND];
}

2349 2350 2351 2352 2353
static char *rand_float_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2354
    return g_randfloat_buff + ((cursor % MAX_PREPARED_RAND) * FLOAT_BUFF_LEN);
2355 2356
}

S
Shengliang Guan 已提交
2357

2358 2359
static float rand_float()
{
2360 2361
    static int cursor;
    cursor++;
2362 2363
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randfloat[cursor % MAX_PREPARED_RAND];
2364 2365
}

2366 2367 2368 2369 2370
static char *demo_current_float_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2371 2372
    return g_rand_current_buff +
        ((cursor % MAX_PREPARED_RAND) * FLOAT_BUFF_LEN);
2373 2374 2375 2376
}

static float UNUSED_FUNC demo_current_float()
{
2377 2378
    static int cursor;
    cursor++;
2379 2380 2381
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return (float)(9.8 + 0.04 * (g_randint[cursor % MAX_PREPARED_RAND] % 10)
            + g_randfloat[cursor % MAX_PREPARED_RAND]/1000000000);
2382 2383 2384 2385 2386 2387 2388
}

static char *demo_voltage_int_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2389 2390
    return g_rand_voltage_buff +
        ((cursor % MAX_PREPARED_RAND) * INT_BUFF_LEN);
2391 2392
}

2393 2394
static int32_t UNUSED_FUNC demo_voltage_int()
{
2395 2396
    static int cursor;
    cursor++;
2397 2398
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return 215 + g_randint[cursor % MAX_PREPARED_RAND] % 10;
2399 2400 2401 2402 2403 2404
}

static char *demo_phase_float_str() {
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2405
    return g_rand_phase_buff + ((cursor % MAX_PREPARED_RAND) * FLOAT_BUFF_LEN);
2406 2407
}

2408
static float UNUSED_FUNC demo_phase_float() {
2409 2410
    static int cursor;
    cursor++;
2411 2412 2413
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return (float)((115 + g_randint[cursor % MAX_PREPARED_RAND] % 10
                + g_randfloat[cursor % MAX_PREPARED_RAND]/1000000000)/360);
2414 2415
}

2416 2417 2418 2419 2420 2421
#if 0
static const char charNum[] = "0123456789";

static void nonrand_string(char *, int) __attribute__ ((unused));   // reserve for debugging purpose
static void nonrand_string(char *str, int size)
{
2422 2423 2424 2425 2426 2427 2428 2429
    str[0] = 0;
    if (size > 0) {
        int n;
        for (n = 0; n < size; n++) {
            str[n] = charNum[n % 10];
        }
        str[n] = 0;
    }
2430 2431 2432
}
#endif

2433
static const char charset[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890";
2434

2435
static void rand_string(char *str, int size) {
2436 2437 2438 2439 2440 2441 2442 2443 2444
    str[0] = 0;
    if (size > 0) {
        //--size;
        int n;
        for (n = 0; n < size; n++) {
            int key = abs(rand_tinyint()) % (int)(sizeof(charset) - 1);
            str[n] = charset[key];
        }
        str[n] = 0;
2445 2446 2447
    }
}

2448 2449 2450 2451 2452 2453 2454 2455 2456 2457
static char *rand_double_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randdouble_buff + (cursor * DOUBLE_BUFF_LEN);
}

static double rand_double()
{
2458 2459 2460
    static int cursor;
    cursor++;
    cursor = cursor % MAX_PREPARED_RAND;
2461
    return g_randdouble[cursor];
2462 2463 2464
}

static void init_rand_data() {
2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485

    g_randint_buff = calloc(1, INT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randint_buff);
    g_rand_voltage_buff = calloc(1, INT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_rand_voltage_buff);
    g_randbigint_buff = calloc(1, BIGINT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randbigint_buff);
    g_randsmallint_buff = calloc(1, SMALLINT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randsmallint_buff);
    g_randtinyint_buff = calloc(1, TINYINT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randtinyint_buff);
    g_randbool_buff = calloc(1, BOOL_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randbool_buff);
    g_randfloat_buff = calloc(1, FLOAT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randfloat_buff);
    g_rand_current_buff = calloc(1, FLOAT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_rand_current_buff);
    g_rand_phase_buff = calloc(1, FLOAT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_rand_phase_buff);
    g_randdouble_buff = calloc(1, DOUBLE_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randdouble_buff);
2486 2487 2488 2489 2490 2491 2492 2493
    g_randuint_buff = calloc(1, INT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randuint_buff);
    g_randutinyint_buff = calloc(1, TINYINT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randutinyint_buff);
    g_randusmallint_buff = calloc(1, SMALLINT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randusmallint_buff);
    g_randubigint_buff = calloc(1, BIGINT_BUFF_LEN * MAX_PREPARED_RAND);
    assert(g_randubigint_buff);
2494

2495
    for (int i = 0; i < MAX_PREPARED_RAND; i++) {
2496 2497
        g_randint[i] = (int)(taosRandom() % RAND_MAX - (RAND_MAX >> 1));
        g_randuint[i] = (int)(taosRandom());
2498 2499 2500 2501 2502 2503 2504 2505
        sprintf(g_randint_buff + i * INT_BUFF_LEN, "%d",
                g_randint[i]);
        sprintf(g_rand_voltage_buff + i * INT_BUFF_LEN, "%d",
                215 + g_randint[i] % 10);

        sprintf(g_randbool_buff + i * BOOL_BUFF_LEN, "%s",
                ((g_randint[i] % 2) & 1)?"true":"false");
        sprintf(g_randsmallint_buff + i * SMALLINT_BUFF_LEN, "%d",
2506
                g_randint[i] % 32768);
2507 2508
        sprintf(g_randtinyint_buff + i * TINYINT_BUFF_LEN, "%d",
                g_randint[i] % 128);
2509 2510 2511 2512 2513 2514 2515 2516 2517
        sprintf(g_randuint_buff + i * INT_BUFF_LEN, "%d",
                g_randuint[i]);
        sprintf(g_randusmallint_buff + i * SMALLINT_BUFF_LEN, "%d",
                g_randuint[i] % 65535);
        sprintf(g_randutinyint_buff + i * TINYINT_BUFF_LEN, "%d",
                g_randuint[i] % 255);

        g_randbigint[i] = (int64_t)(taosRandom() % RAND_MAX - (RAND_MAX >> 1));
        g_randubigint[i] = (uint64_t)(taosRandom());
2518 2519
        sprintf(g_randbigint_buff + i * BIGINT_BUFF_LEN, "%"PRId64"",
                g_randbigint[i]);
2520 2521
        sprintf(g_randubigint_buff + i * BIGINT_BUFF_LEN, "%"PRId64"",
                g_randubigint[i]);
2522

2523
        g_randfloat[i] = (float)(taosRandom() / 1000.0) * (taosRandom() % 2 > 0.5 ? 1 : -1);
2524 2525 2526 2527 2528 2529 2530 2531 2532
        sprintf(g_randfloat_buff + i * FLOAT_BUFF_LEN, "%f",
                g_randfloat[i]);
        sprintf(g_rand_current_buff + i * FLOAT_BUFF_LEN, "%f",
                (float)(9.8 + 0.04 * (g_randint[i] % 10)
                    + g_randfloat[i]/1000000000));
        sprintf(g_rand_phase_buff + i * FLOAT_BUFF_LEN, "%f",
                (float)((115 + g_randint[i] % 10
                        + g_randfloat[i]/1000000000)/360));

2533
        g_randdouble[i] = (double)(taosRandom() / 1000000.0) * (taosRandom() % 2 > 0.5 ? 1 : -1);
2534 2535
        sprintf(g_randdouble_buff + i * DOUBLE_BUFF_LEN, "%f",
                g_randdouble[i]);
2536
    }
2537 2538
}

2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558
#define SHOW_PARSE_RESULT_START()   \
    do { if (g_args.metaFile)  \
        printf("\033[1m\033[40;32m================ %s parse result START ================\033[0m\n", \
                g_args.metaFile); } while(0)

#define SHOW_PARSE_RESULT_END() \
    do { if (g_args.metaFile)   \
        printf("\033[1m\033[40;32m================ %s parse result END================\033[0m\n", \
                g_args.metaFile); } while(0)

#define SHOW_PARSE_RESULT_START_TO_FILE(fp)   \
    do { if (g_args.metaFile)  \
        fprintf(fp, "\033[1m\033[40;32m================ %s parse result START ================\033[0m\n", \
                g_args.metaFile); } while(0)

#define SHOW_PARSE_RESULT_END_TO_FILE(fp) \
    do { if (g_args.metaFile)   \
        fprintf(fp, "\033[1m\033[40;32m================ %s parse result END================\033[0m\n", \
                g_args.metaFile); } while(0)

2559
static int printfInsertMeta() {
2560
    SHOW_PARSE_RESULT_START();
2561

2562 2563 2564
    if (g_args.demo_mode) {
        printf("\ntaosdemo is simulating data generated by power equipment monitoring...\n\n");
    } else {
2565
        printf("\ntaosdemo is simulating random data as you request..\n\n");
2566
    }
2567

2568 2569 2570
    if (g_args.iface != INTERFACE_BUT) {
        // first time if no iface specified
        printf("interface:                  \033[33m%s\033[0m\n",
2571 2572
                (g_args.iface==TAOSC_IFACE)?"taosc":
                (g_args.iface==REST_IFACE)?"rest":"stmt");
2573 2574
    }

2575 2576 2577 2578 2579 2580 2581 2582
    printf("host:                       \033[33m%s:%u\033[0m\n",
            g_Dbs.host, g_Dbs.port);
    printf("user:                       \033[33m%s\033[0m\n", g_Dbs.user);
    printf("password:                   \033[33m%s\033[0m\n", g_Dbs.password);
    printf("configDir:                  \033[33m%s\033[0m\n", configDir);
    printf("resultFile:                 \033[33m%s\033[0m\n", g_Dbs.resultFile);
    printf("thread num of insert data:  \033[33m%d\033[0m\n", g_Dbs.threadCount);
    printf("thread num of create table: \033[33m%d\033[0m\n",
2583
            g_Dbs.threadCountForCreateTbl);
2584 2585 2586
    printf("top insert interval:        \033[33m%"PRIu64"\033[0m\n",
            g_args.insert_interval);
    printf("number of records per req:  \033[33m%u\033[0m\n",
2587
            g_args.reqPerReq);
2588 2589 2590 2591
    printf("max sql length:             \033[33m%"PRIu64"\033[0m\n",
            g_args.max_sql_len);

    printf("database count:             \033[33m%d\033[0m\n", g_Dbs.dbCount);
2592

2593 2594 2595 2596 2597
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        printf("database[\033[33m%d\033[0m]:\n", i);
        printf("  database[%d] name:      \033[33m%s\033[0m\n",
                i, g_Dbs.db[i].dbName);
        if (0 == g_Dbs.db[i].drop) {
2598
            printf("  drop:                  \033[33m no\033[0m\n");
2599
        } else {
2600
            printf("  drop:                  \033[33m yes\033[0m\n");
2601
        }
2602

2603 2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651
        if (g_Dbs.db[i].dbCfg.blocks > 0) {
            printf("  blocks:                \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.blocks);
        }
        if (g_Dbs.db[i].dbCfg.cache > 0) {
            printf("  cache:                 \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.cache);
        }
        if (g_Dbs.db[i].dbCfg.days > 0) {
            printf("  days:                  \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.days);
        }
        if (g_Dbs.db[i].dbCfg.keep > 0) {
            printf("  keep:                  \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.keep);
        }
        if (g_Dbs.db[i].dbCfg.replica > 0) {
            printf("  replica:               \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.replica);
        }
        if (g_Dbs.db[i].dbCfg.update > 0) {
            printf("  update:                \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.update);
        }
        if (g_Dbs.db[i].dbCfg.minRows > 0) {
            printf("  minRows:               \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.minRows);
        }
        if (g_Dbs.db[i].dbCfg.maxRows > 0) {
            printf("  maxRows:               \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.maxRows);
        }
        if (g_Dbs.db[i].dbCfg.comp > 0) {
            printf("  comp:                  \033[33m%d\033[0m\n", g_Dbs.db[i].dbCfg.comp);
        }
        if (g_Dbs.db[i].dbCfg.walLevel > 0) {
            printf("  walLevel:              \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.walLevel);
        }
        if (g_Dbs.db[i].dbCfg.fsync > 0) {
            printf("  fsync:                 \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.fsync);
        }
        if (g_Dbs.db[i].dbCfg.quorum > 0) {
            printf("  quorum:                \033[33m%d\033[0m\n",
                    g_Dbs.db[i].dbCfg.quorum);
        }
        if (g_Dbs.db[i].dbCfg.precision[0] != 0) {
            if ((0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ms", 2))
2652 2653
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "us", 2))
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ns", 2))) {
2654 2655 2656 2657 2658 2659 2660 2661
                printf("  precision:             \033[33m%s\033[0m\n",
                        g_Dbs.db[i].dbCfg.precision);
            } else {
                printf("\033[1m\033[40;31m  precision error:       %s\033[0m\n",
                        g_Dbs.db[i].dbCfg.precision);
                return -1;
            }
        }
2662

2663 2664 2665 2666 2667 2668
        printf("  super table count:     \033[33m%"PRIu64"\033[0m\n",
                g_Dbs.db[i].superTblCount);
        for (uint64_t j = 0; j < g_Dbs.db[i].superTblCount; j++) {
            printf("  super table[\033[33m%"PRIu64"\033[0m]:\n", j);

            printf("      stbName:           \033[33m%s\033[0m\n",
2669
                    g_Dbs.db[i].superTbls[j].stbName);
2670 2671 2672 2673 2674 2675 2676 2677 2678

            if (PRE_CREATE_SUBTBL == g_Dbs.db[i].superTbls[j].autoCreateTable) {
                printf("      autoCreateTable:   \033[33m%s\033[0m\n",  "no");
            } else if (AUTO_CREATE_SUBTBL ==
                    g_Dbs.db[i].superTbls[j].autoCreateTable) {
                printf("      autoCreateTable:   \033[33m%s\033[0m\n",  "yes");
            } else {
                printf("      autoCreateTable:   \033[33m%s\033[0m\n",  "error");
            }
2679

2680 2681 2682 2683 2684 2685 2686
            if (TBL_NO_EXISTS == g_Dbs.db[i].superTbls[j].childTblExists) {
                printf("      childTblExists:    \033[33m%s\033[0m\n",  "no");
            } else if (TBL_ALREADY_EXISTS == g_Dbs.db[i].superTbls[j].childTblExists) {
                printf("      childTblExists:    \033[33m%s\033[0m\n",  "yes");
            } else {
                printf("      childTblExists:    \033[33m%s\033[0m\n",  "error");
            }
2687

2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708
            printf("      childTblCount:     \033[33m%"PRId64"\033[0m\n",
                    g_Dbs.db[i].superTbls[j].childTblCount);
            printf("      childTblPrefix:    \033[33m%s\033[0m\n",
                    g_Dbs.db[i].superTbls[j].childTblPrefix);
            printf("      dataSource:        \033[33m%s\033[0m\n",
                    g_Dbs.db[i].superTbls[j].dataSource);
            printf("      iface:             \033[33m%s\033[0m\n",
                    (g_Dbs.db[i].superTbls[j].iface==TAOSC_IFACE)?"taosc":
                    (g_Dbs.db[i].superTbls[j].iface==REST_IFACE)?"rest":"stmt");
            if (g_Dbs.db[i].superTbls[j].childTblLimit > 0) {
                printf("      childTblLimit:     \033[33m%"PRId64"\033[0m\n",
                        g_Dbs.db[i].superTbls[j].childTblLimit);
            }
            if (g_Dbs.db[i].superTbls[j].childTblOffset > 0) {
                printf("      childTblOffset:    \033[33m%"PRIu64"\033[0m\n",
                        g_Dbs.db[i].superTbls[j].childTblOffset);
            }
            printf("      insertRows:        \033[33m%"PRId64"\033[0m\n",
                    g_Dbs.db[i].superTbls[j].insertRows);
            /*
               if (0 == g_Dbs.db[i].superTbls[j].multiThreadWriteOneTbl) {
2709
               printf("      multiThreadWriteOneTbl:  \033[33m no\033[0m\n");
2710
               }else {
2711
               printf("      multiThreadWriteOneTbl:  \033[33m yes\033[0m\n");
2712 2713 2714 2715 2716 2717 2718 2719 2720
               }
               */
            printf("      interlaceRows:     \033[33m%u\033[0m\n",
                    g_Dbs.db[i].superTbls[j].interlaceRows);

            if (g_Dbs.db[i].superTbls[j].interlaceRows > 0) {
                printf("      stable insert interval:   \033[33m%"PRIu64"\033[0m\n",
                        g_Dbs.db[i].superTbls[j].insertInterval);
            }
2721

2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772
            printf("      disorderRange:     \033[33m%d\033[0m\n",
                    g_Dbs.db[i].superTbls[j].disorderRange);
            printf("      disorderRatio:     \033[33m%d\033[0m\n",
                    g_Dbs.db[i].superTbls[j].disorderRatio);
            printf("      maxSqlLen:         \033[33m%"PRIu64"\033[0m\n",
                    g_Dbs.db[i].superTbls[j].maxSqlLen);
            printf("      timeStampStep:     \033[33m%"PRId64"\033[0m\n",
                    g_Dbs.db[i].superTbls[j].timeStampStep);
            printf("      startTimestamp:    \033[33m%s\033[0m\n",
                    g_Dbs.db[i].superTbls[j].startTimestamp);
            printf("      sampleFormat:      \033[33m%s\033[0m\n",
                    g_Dbs.db[i].superTbls[j].sampleFormat);
            printf("      sampleFile:        \033[33m%s\033[0m\n",
                    g_Dbs.db[i].superTbls[j].sampleFile);
            printf("      tagsFile:          \033[33m%s\033[0m\n",
                    g_Dbs.db[i].superTbls[j].tagsFile);
            printf("      columnCount:       \033[33m%d\033[0m\n",
                    g_Dbs.db[i].superTbls[j].columnCount);
            for (int k = 0; k < g_Dbs.db[i].superTbls[j].columnCount; k++) {
                //printf("dataType:%s, dataLen:%d\t", g_Dbs.db[i].superTbls[j].columns[k].dataType, g_Dbs.db[i].superTbls[j].columns[k].dataLen);
                if ((0 == strncasecmp(g_Dbs.db[i].superTbls[j].columns[k].dataType,
                                "binary", 6))
                        || (0 == strncasecmp(g_Dbs.db[i].superTbls[j].columns[k].dataType,
                                "nchar", 5))) {
                    printf("column[\033[33m%d\033[0m]:\033[33m%s(%d)\033[0m ", k,
                            g_Dbs.db[i].superTbls[j].columns[k].dataType,
                            g_Dbs.db[i].superTbls[j].columns[k].dataLen);
                } else {
                    printf("column[%d]:\033[33m%s\033[0m ", k,
                            g_Dbs.db[i].superTbls[j].columns[k].dataType);
                }
            }
            printf("\n");

            printf("      tagCount:            \033[33m%d\033[0m\n        ",
                    g_Dbs.db[i].superTbls[j].tagCount);
            for (int k = 0; k < g_Dbs.db[i].superTbls[j].tagCount; k++) {
                //printf("dataType:%s, dataLen:%d\t", g_Dbs.db[i].superTbls[j].tags[k].dataType, g_Dbs.db[i].superTbls[j].tags[k].dataLen);
                if ((0 == strncasecmp(g_Dbs.db[i].superTbls[j].tags[k].dataType,
                                "binary", strlen("binary")))
                        || (0 == strncasecmp(g_Dbs.db[i].superTbls[j].tags[k].dataType,
                                "nchar", strlen("nchar")))) {
                    printf("tag[%d]:\033[33m%s(%d)\033[0m ", k,
                            g_Dbs.db[i].superTbls[j].tags[k].dataType,
                            g_Dbs.db[i].superTbls[j].tags[k].dataLen);
                } else {
                    printf("tag[%d]:\033[33m%s\033[0m ", k,
                            g_Dbs.db[i].superTbls[j].tags[k].dataType);
                }
            }
            printf("\n");
2773
        }
2774
        printf("\n");
2775
    }
2776

2777
    SHOW_PARSE_RESULT_END();
2778

2779
    return 0;
2780 2781 2782
}

static void printfInsertMetaToFile(FILE* fp) {
2783

2784
    SHOW_PARSE_RESULT_START_TO_FILE(fp);
2785

2786 2787 2788 2789 2790
    fprintf(fp, "host:                       %s:%u\n", g_Dbs.host, g_Dbs.port);
    fprintf(fp, "user:                       %s\n", g_Dbs.user);
    fprintf(fp, "configDir:                  %s\n", configDir);
    fprintf(fp, "resultFile:                 %s\n", g_Dbs.resultFile);
    fprintf(fp, "thread num of insert data:  %d\n", g_Dbs.threadCount);
2791
    fprintf(fp, "thread num of create table: %d\n", g_Dbs.threadCountForCreateTbl);
2792
    fprintf(fp, "number of records per req:  %u\n", g_args.reqPerReq);
2793 2794
    fprintf(fp, "max sql length:             %"PRIu64"\n", g_args.max_sql_len);
    fprintf(fp, "database count:          %d\n", g_Dbs.dbCount);
2795

2796 2797 2798 2799 2800 2801 2802 2803
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        fprintf(fp, "database[%d]:\n", i);
        fprintf(fp, "  database[%d] name:       %s\n", i, g_Dbs.db[i].dbName);
        if (0 == g_Dbs.db[i].drop) {
            fprintf(fp, "  drop:                  no\n");
        }else {
            fprintf(fp, "  drop:                  yes\n");
        }
2804

2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842
        if (g_Dbs.db[i].dbCfg.blocks > 0) {
            fprintf(fp, "  blocks:                %d\n", g_Dbs.db[i].dbCfg.blocks);
        }
        if (g_Dbs.db[i].dbCfg.cache > 0) {
            fprintf(fp, "  cache:                 %d\n", g_Dbs.db[i].dbCfg.cache);
        }
        if (g_Dbs.db[i].dbCfg.days > 0) {
            fprintf(fp, "  days:                  %d\n", g_Dbs.db[i].dbCfg.days);
        }
        if (g_Dbs.db[i].dbCfg.keep > 0) {
            fprintf(fp, "  keep:                  %d\n", g_Dbs.db[i].dbCfg.keep);
        }
        if (g_Dbs.db[i].dbCfg.replica > 0) {
            fprintf(fp, "  replica:               %d\n", g_Dbs.db[i].dbCfg.replica);
        }
        if (g_Dbs.db[i].dbCfg.update > 0) {
            fprintf(fp, "  update:                %d\n", g_Dbs.db[i].dbCfg.update);
        }
        if (g_Dbs.db[i].dbCfg.minRows > 0) {
            fprintf(fp, "  minRows:               %d\n", g_Dbs.db[i].dbCfg.minRows);
        }
        if (g_Dbs.db[i].dbCfg.maxRows > 0) {
            fprintf(fp, "  maxRows:               %d\n", g_Dbs.db[i].dbCfg.maxRows);
        }
        if (g_Dbs.db[i].dbCfg.comp > 0) {
            fprintf(fp, "  comp:                  %d\n", g_Dbs.db[i].dbCfg.comp);
        }
        if (g_Dbs.db[i].dbCfg.walLevel > 0) {
            fprintf(fp, "  walLevel:              %d\n", g_Dbs.db[i].dbCfg.walLevel);
        }
        if (g_Dbs.db[i].dbCfg.fsync > 0) {
            fprintf(fp, "  fsync:                 %d\n", g_Dbs.db[i].dbCfg.fsync);
        }
        if (g_Dbs.db[i].dbCfg.quorum > 0) {
            fprintf(fp, "  quorum:                %d\n", g_Dbs.db[i].dbCfg.quorum);
        }
        if (g_Dbs.db[i].dbCfg.precision[0] != 0) {
            if ((0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ms", 2))
2843
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ns", 2))
2844 2845 2846 2847 2848 2849 2850 2851
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "us", 2))) {
                fprintf(fp, "  precision:             %s\n",
                        g_Dbs.db[i].dbCfg.precision);
            } else {
                fprintf(fp, "  precision error:       %s\n",
                        g_Dbs.db[i].dbCfg.precision);
            }
        }
2852

2853 2854 2855 2856 2857 2858
        fprintf(fp, "  super table count:     %"PRIu64"\n",
                g_Dbs.db[i].superTblCount);
        for (int j = 0; j < g_Dbs.db[i].superTblCount; j++) {
            fprintf(fp, "  super table[%d]:\n", j);

            fprintf(fp, "      stbName:           %s\n",
2859
                    g_Dbs.db[i].superTbls[j].stbName);
2860 2861 2862 2863 2864 2865 2866 2867 2868

            if (PRE_CREATE_SUBTBL == g_Dbs.db[i].superTbls[j].autoCreateTable) {
                fprintf(fp, "      autoCreateTable:   %s\n",  "no");
            } else if (AUTO_CREATE_SUBTBL
                    == g_Dbs.db[i].superTbls[j].autoCreateTable) {
                fprintf(fp, "      autoCreateTable:   %s\n",  "yes");
            } else {
                fprintf(fp, "      autoCreateTable:   %s\n",  "error");
            }
2869

2870 2871 2872 2873 2874 2875 2876 2877
            if (TBL_NO_EXISTS == g_Dbs.db[i].superTbls[j].childTblExists) {
                fprintf(fp, "      childTblExists:    %s\n",  "no");
            } else if (TBL_ALREADY_EXISTS
                    == g_Dbs.db[i].superTbls[j].childTblExists) {
                fprintf(fp, "      childTblExists:    %s\n",  "yes");
            } else {
                fprintf(fp, "      childTblExists:    %s\n",  "error");
            }
2878

2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958
            fprintf(fp, "      childTblCount:     %"PRId64"\n",
                    g_Dbs.db[i].superTbls[j].childTblCount);
            fprintf(fp, "      childTblPrefix:    %s\n",
                    g_Dbs.db[i].superTbls[j].childTblPrefix);
            fprintf(fp, "      dataSource:        %s\n",
                    g_Dbs.db[i].superTbls[j].dataSource);
            fprintf(fp, "      iface:             %s\n",
                    (g_Dbs.db[i].superTbls[j].iface==TAOSC_IFACE)?"taosc":
                    (g_Dbs.db[i].superTbls[j].iface==REST_IFACE)?"rest":"stmt");
            fprintf(fp, "      insertRows:        %"PRId64"\n",
                    g_Dbs.db[i].superTbls[j].insertRows);
            fprintf(fp, "      interlace rows:    %u\n",
                    g_Dbs.db[i].superTbls[j].interlaceRows);
            if (g_Dbs.db[i].superTbls[j].interlaceRows > 0) {
                fprintf(fp, "      stable insert interval:   %"PRIu64"\n",
                        g_Dbs.db[i].superTbls[j].insertInterval);
            }
            /*
               if (0 == g_Dbs.db[i].superTbls[j].multiThreadWriteOneTbl) {
               fprintf(fp, "      multiThreadWriteOneTbl:  no\n");
               }else {
               fprintf(fp, "      multiThreadWriteOneTbl:  yes\n");
               }
               */
            fprintf(fp, "      interlaceRows:     %u\n",
                    g_Dbs.db[i].superTbls[j].interlaceRows);
            fprintf(fp, "      disorderRange:     %d\n",
                    g_Dbs.db[i].superTbls[j].disorderRange);
            fprintf(fp, "      disorderRatio:     %d\n",
                    g_Dbs.db[i].superTbls[j].disorderRatio);
            fprintf(fp, "      maxSqlLen:         %"PRIu64"\n",
                    g_Dbs.db[i].superTbls[j].maxSqlLen);

            fprintf(fp, "      timeStampStep:     %"PRId64"\n",
                    g_Dbs.db[i].superTbls[j].timeStampStep);
            fprintf(fp, "      startTimestamp:    %s\n",
                    g_Dbs.db[i].superTbls[j].startTimestamp);
            fprintf(fp, "      sampleFormat:      %s\n",
                    g_Dbs.db[i].superTbls[j].sampleFormat);
            fprintf(fp, "      sampleFile:        %s\n",
                    g_Dbs.db[i].superTbls[j].sampleFile);
            fprintf(fp, "      tagsFile:          %s\n",
                    g_Dbs.db[i].superTbls[j].tagsFile);

            fprintf(fp, "      columnCount:       %d\n        ",
                    g_Dbs.db[i].superTbls[j].columnCount);
            for (int k = 0; k < g_Dbs.db[i].superTbls[j].columnCount; k++) {
                //printf("dataType:%s, dataLen:%d\t", g_Dbs.db[i].superTbls[j].columns[k].dataType, g_Dbs.db[i].superTbls[j].columns[k].dataLen);
                if ((0 == strncasecmp(
                                g_Dbs.db[i].superTbls[j].columns[k].dataType,
                                "binary", strlen("binary")))
                        || (0 == strncasecmp(
                                g_Dbs.db[i].superTbls[j].columns[k].dataType,
                                "nchar", strlen("nchar")))) {
                    fprintf(fp, "column[%d]:%s(%d) ", k,
                            g_Dbs.db[i].superTbls[j].columns[k].dataType,
                            g_Dbs.db[i].superTbls[j].columns[k].dataLen);
                } else {
                    fprintf(fp, "column[%d]:%s ",
                            k, g_Dbs.db[i].superTbls[j].columns[k].dataType);
                }
            }
            fprintf(fp, "\n");

            fprintf(fp, "      tagCount:            %d\n        ",
                    g_Dbs.db[i].superTbls[j].tagCount);
            for (int k = 0; k < g_Dbs.db[i].superTbls[j].tagCount; k++) {
                //printf("dataType:%s, dataLen:%d\t", g_Dbs.db[i].superTbls[j].tags[k].dataType, g_Dbs.db[i].superTbls[j].tags[k].dataLen);
                if ((0 == strncasecmp(g_Dbs.db[i].superTbls[j].tags[k].dataType,
                                "binary", strlen("binary")))
                        || (0 == strncasecmp(g_Dbs.db[i].superTbls[j].tags[k].dataType,
                                "nchar", strlen("nchar")))) {
                    fprintf(fp, "tag[%d]:%s(%d) ",
                            k, g_Dbs.db[i].superTbls[j].tags[k].dataType,
                            g_Dbs.db[i].superTbls[j].tags[k].dataLen);
                } else {
                    fprintf(fp, "tag[%d]:%s ", k, g_Dbs.db[i].superTbls[j].tags[k].dataType);
                }
            }
            fprintf(fp, "\n");
2959
        }
2960
        fprintf(fp, "\n");
2961
    }
2962

2963
    SHOW_PARSE_RESULT_END_TO_FILE(fp);
2964 2965 2966
}

static void printfQueryMeta() {
2967

2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017
    SHOW_PARSE_RESULT_START();

    printf("host:                    \033[33m%s:%u\033[0m\n",
            g_queryInfo.host, g_queryInfo.port);
    printf("user:                    \033[33m%s\033[0m\n", g_queryInfo.user);
    printf("database name:           \033[33m%s\033[0m\n", g_queryInfo.dbName);

    printf("\n");

    if ((SUBSCRIBE_TEST == g_args.test_mode) || (QUERY_TEST == g_args.test_mode)) {
        printf("specified table query info:                   \n");
        printf("sqlCount:       \033[33m%d\033[0m\n",
                g_queryInfo.specifiedQueryInfo.sqlCount);
        if (g_queryInfo.specifiedQueryInfo.sqlCount > 0) {
            printf("specified tbl query times:\n");
            printf("                \033[33m%"PRIu64"\033[0m\n",
                    g_queryInfo.specifiedQueryInfo.queryTimes);
            printf("query interval: \033[33m%"PRIu64" ms\033[0m\n",
                    g_queryInfo.specifiedQueryInfo.queryInterval);
            printf("top query times:\033[33m%"PRIu64"\033[0m\n", g_args.query_times);
            printf("concurrent:     \033[33m%d\033[0m\n",
                    g_queryInfo.specifiedQueryInfo.concurrent);
            printf("mod:            \033[33m%s\033[0m\n",
                    (g_queryInfo.specifiedQueryInfo.asyncMode)?"async":"sync");
            printf("interval:       \033[33m%"PRIu64"\033[0m\n",
                    g_queryInfo.specifiedQueryInfo.subscribeInterval);
            printf("restart:        \033[33m%d\033[0m\n",
                    g_queryInfo.specifiedQueryInfo.subscribeRestart);
            printf("keepProgress:   \033[33m%d\033[0m\n",
                    g_queryInfo.specifiedQueryInfo.subscribeKeepProgress);

            for (int i = 0; i < g_queryInfo.specifiedQueryInfo.sqlCount; i++) {
                printf("  sql[%d]: \033[33m%s\033[0m\n",
                        i, g_queryInfo.specifiedQueryInfo.sql[i]);
            }
            printf("\n");
        }

        printf("super table query info:\n");
        printf("sqlCount:       \033[33m%d\033[0m\n",
                g_queryInfo.superQueryInfo.sqlCount);

        if (g_queryInfo.superQueryInfo.sqlCount > 0) {
            printf("query interval: \033[33m%"PRIu64"\033[0m\n",
                    g_queryInfo.superQueryInfo.queryInterval);
            printf("threadCnt:      \033[33m%d\033[0m\n",
                    g_queryInfo.superQueryInfo.threadCnt);
            printf("childTblCount:  \033[33m%"PRId64"\033[0m\n",
                    g_queryInfo.superQueryInfo.childTblCount);
            printf("stable name:    \033[33m%s\033[0m\n",
3018
                    g_queryInfo.superQueryInfo.stbName);
3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036
            printf("stb query times:\033[33m%"PRIu64"\033[0m\n",
                    g_queryInfo.superQueryInfo.queryTimes);

            printf("mod:            \033[33m%s\033[0m\n",
                    (g_queryInfo.superQueryInfo.asyncMode)?"async":"sync");
            printf("interval:       \033[33m%"PRIu64"\033[0m\n",
                    g_queryInfo.superQueryInfo.subscribeInterval);
            printf("restart:        \033[33m%d\033[0m\n",
                    g_queryInfo.superQueryInfo.subscribeRestart);
            printf("keepProgress:   \033[33m%d\033[0m\n",
                    g_queryInfo.superQueryInfo.subscribeKeepProgress);

            for (int i = 0; i < g_queryInfo.superQueryInfo.sqlCount; i++) {
                printf("  sql[%d]: \033[33m%s\033[0m\n",
                        i, g_queryInfo.superQueryInfo.sql[i]);
            }
            printf("\n");
        }
3037
    }
3038

3039
    SHOW_PARSE_RESULT_END();
3040 3041
}

3042
static char* formatTimestamp(char* buf, int64_t val, int precision) {
3043
    time_t tt;
3044
    if (precision == TSDB_TIME_PRECISION_MICRO) {
3045
        tt = (time_t)(val / 1000000);
3046 3047
    } if (precision == TSDB_TIME_PRECISION_NANO) {
        tt = (time_t)(val / 1000000000);
3048 3049 3050
    } else {
        tt = (time_t)(val / 1000);
    }
3051

3052 3053 3054 3055 3056 3057 3058
    /* comment out as it make testcases like select_with_tags.sim fail.
       but in windows, this may cause the call to localtime crash if tt < 0,
       need to find a better solution.
       if (tt < 0) {
       tt = 0;
       }
       */
3059 3060

#ifdef WINDOWS
3061
    if (tt < 0) tt = 0;
3062 3063
#endif

3064 3065
    struct tm* ptm = localtime(&tt);
    size_t pos = strftime(buf, 32, "%Y-%m-%d %H:%M:%S", ptm);
3066

3067
    if (precision == TSDB_TIME_PRECISION_MICRO) {
3068
        sprintf(buf + pos, ".%06d", (int)(val % 1000000));
3069 3070
    } else if (precision == TSDB_TIME_PRECISION_NANO) {
        sprintf(buf + pos, ".%09d", (int)(val % 1000000000));
3071 3072 3073
    } else {
        sprintf(buf + pos, ".%03d", (int)(val % 1000));
    }
3074

3075
    return buf;
3076 3077
}

3078
static void xDumpFieldToFile(FILE* fp, const char* val,
3079
        TAOS_FIELD* field, int32_t length, int precision) {
3080

3081 3082 3083 3084
    if (val == NULL) {
        fprintf(fp, "%s", TSDB_DATA_NULL_STR);
        return;
    }
3085

3086 3087 3088
    char buf[TSDB_MAX_BYTES_PER_ROW];
    switch (field->type) {
        case TSDB_DATA_TYPE_BOOL:
3089
            fprintf(fp, "%d", ((((int32_t)(*((int8_t*)val))) == 1) ? 1 : 0));
3090
            break;
3091

3092 3093 3094
        case TSDB_DATA_TYPE_TINYINT:
            fprintf(fp, "%d", *((int8_t *)val));
            break;
3095

3096 3097 3098 3099
        case TSDB_DATA_TYPE_UTINYINT:
            fprintf(fp, "%d", *((uint8_t *)val));
            break;

3100 3101 3102
        case TSDB_DATA_TYPE_SMALLINT:
            fprintf(fp, "%d", *((int16_t *)val));
            break;
3103

3104 3105 3106 3107
        case TSDB_DATA_TYPE_USMALLINT:
            fprintf(fp, "%d", *((uint16_t *)val));
            break;

3108 3109 3110
        case TSDB_DATA_TYPE_INT:
            fprintf(fp, "%d", *((int32_t *)val));
            break;
3111

3112 3113 3114 3115
        case TSDB_DATA_TYPE_UINT:
            fprintf(fp, "%d", *((uint32_t *)val));
            break;

3116
        case TSDB_DATA_TYPE_BIGINT:
3117
            fprintf(fp, "%"PRId64"", *((int64_t *)val));
3118
            break;
3119

3120 3121 3122 3123
        case TSDB_DATA_TYPE_UBIGINT:
            fprintf(fp, "%"PRId64"", *((uint64_t *)val));
            break;

3124 3125 3126
        case TSDB_DATA_TYPE_FLOAT:
            fprintf(fp, "%.5f", GET_FLOAT_VAL(val));
            break;
3127

3128 3129 3130
        case TSDB_DATA_TYPE_DOUBLE:
            fprintf(fp, "%.9f", GET_DOUBLE_VAL(val));
            break;
3131

3132 3133 3134 3135 3136 3137
        case TSDB_DATA_TYPE_BINARY:
        case TSDB_DATA_TYPE_NCHAR:
            memcpy(buf, val, length);
            buf[length] = 0;
            fprintf(fp, "\'%s\'", buf);
            break;
3138

3139 3140 3141 3142
        case TSDB_DATA_TYPE_TIMESTAMP:
            formatTimestamp(buf, *(int64_t*)val, precision);
            fprintf(fp, "'%s'", buf);
            break;
3143

3144 3145 3146
        default:
            break;
    }
3147 3148 3149
}

static int xDumpResultToFile(const char* fname, TAOS_RES* tres) {
3150 3151 3152 3153
    TAOS_ROW row = taos_fetch_row(tres);
    if (row == NULL) {
        return 0;
    }
3154

3155 3156
    FILE* fp = fopen(fname, "at");
    if (fp == NULL) {
3157
        errorPrint2("%s() LN%d, failed to open file: %s\n",
3158 3159 3160
                __func__, __LINE__, fname);
        return -1;
    }
3161

3162 3163 3164
    int num_fields = taos_num_fields(tres);
    TAOS_FIELD *fields = taos_fetch_fields(tres);
    int precision = taos_result_precision(tres);
3165

3166 3167 3168 3169 3170
    for (int col = 0; col < num_fields; col++) {
        if (col > 0) {
            fprintf(fp, ",");
        }
        fprintf(fp, "%s", fields[col].name);
3171 3172 3173
    }
    fputc('\n', fp);

3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188
    int numOfRows = 0;
    do {
        int32_t* length = taos_fetch_lengths(tres);
        for (int i = 0; i < num_fields; i++) {
            if (i > 0) {
                fputc(',', fp);
            }
            xDumpFieldToFile(fp,
                    (const char*)row[i], fields +i, length[i], precision);
        }
        fputc('\n', fp);

        numOfRows++;
        row = taos_fetch_row(tres);
    } while( row != NULL);
3189

3190
    fclose(fp);
3191

3192
    return numOfRows;
3193 3194
}

3195
static int getDbFromServer(TAOS * taos, SDbInfo** dbInfos) {
3196 3197 3198
    TAOS_RES * res;
    TAOS_ROW row = NULL;
    int count = 0;
3199

3200 3201
    res = taos_query(taos, "show databases;");
    int32_t code = taos_errno(res);
3202

3203
    if (code != 0) {
3204
        errorPrint2("failed to run <show databases>, reason: %s\n",
3205 3206
                taos_errstr(res));
        return -1;
3207
    }
3208

3209
    TAOS_FIELD *fields = taos_fetch_fields(res);
3210

3211 3212 3213 3214 3215 3216
    while((row = taos_fetch_row(res)) != NULL) {
        // sys database name : 'log'
        if (strncasecmp(row[TSDB_SHOW_DB_NAME_INDEX], "log",
                    fields[TSDB_SHOW_DB_NAME_INDEX].bytes) == 0) {
            continue;
        }
3217

3218 3219
        dbInfos[count] = (SDbInfo *)calloc(1, sizeof(SDbInfo));
        if (dbInfos[count] == NULL) {
3220
            errorPrint2("failed to allocate memory for some dbInfo[%d]\n", count);
3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259
            return -1;
        }

        tstrncpy(dbInfos[count]->name, (char *)row[TSDB_SHOW_DB_NAME_INDEX],
                fields[TSDB_SHOW_DB_NAME_INDEX].bytes);
        formatTimestamp(dbInfos[count]->create_time,
                *(int64_t*)row[TSDB_SHOW_DB_CREATED_TIME_INDEX],
                TSDB_TIME_PRECISION_MILLI);
        dbInfos[count]->ntables = *((int64_t *)row[TSDB_SHOW_DB_NTABLES_INDEX]);
        dbInfos[count]->vgroups = *((int32_t *)row[TSDB_SHOW_DB_VGROUPS_INDEX]);
        dbInfos[count]->replica = *((int16_t *)row[TSDB_SHOW_DB_REPLICA_INDEX]);
        dbInfos[count]->quorum = *((int16_t *)row[TSDB_SHOW_DB_QUORUM_INDEX]);
        dbInfos[count]->days = *((int16_t *)row[TSDB_SHOW_DB_DAYS_INDEX]);

        tstrncpy(dbInfos[count]->keeplist, (char *)row[TSDB_SHOW_DB_KEEP_INDEX],
                fields[TSDB_SHOW_DB_KEEP_INDEX].bytes);
        dbInfos[count]->cache = *((int32_t *)row[TSDB_SHOW_DB_CACHE_INDEX]);
        dbInfos[count]->blocks = *((int32_t *)row[TSDB_SHOW_DB_BLOCKS_INDEX]);
        dbInfos[count]->minrows = *((int32_t *)row[TSDB_SHOW_DB_MINROWS_INDEX]);
        dbInfos[count]->maxrows = *((int32_t *)row[TSDB_SHOW_DB_MAXROWS_INDEX]);
        dbInfos[count]->wallevel = *((int8_t *)row[TSDB_SHOW_DB_WALLEVEL_INDEX]);
        dbInfos[count]->fsync = *((int32_t *)row[TSDB_SHOW_DB_FSYNC_INDEX]);
        dbInfos[count]->comp = (int8_t)(*((int8_t *)row[TSDB_SHOW_DB_COMP_INDEX]));
        dbInfos[count]->cachelast =
            (int8_t)(*((int8_t *)row[TSDB_SHOW_DB_CACHELAST_INDEX]));

        tstrncpy(dbInfos[count]->precision,
                (char *)row[TSDB_SHOW_DB_PRECISION_INDEX],
                fields[TSDB_SHOW_DB_PRECISION_INDEX].bytes);
        dbInfos[count]->update = *((int8_t *)row[TSDB_SHOW_DB_UPDATE_INDEX]);
        tstrncpy(dbInfos[count]->status, (char *)row[TSDB_SHOW_DB_STATUS_INDEX],
                fields[TSDB_SHOW_DB_STATUS_INDEX].bytes);

        count++;
        if (count > MAX_DATABASE_COUNT) {
            errorPrint("%s() LN%d, The database count overflow than %d\n",
                    __func__, __LINE__, MAX_DATABASE_COUNT);
            break;
        }
3260 3261
    }

3262
    return count;
3263 3264
}

3265
static void printfDbInfoForQueryToFile(
3266
        char* filename, SDbInfo* dbInfos, int index) {
3267

3268 3269
    if (filename[0] == 0)
        return;
3270

3271 3272 3273 3274 3275
    FILE *fp = fopen(filename, "at");
    if (fp == NULL) {
        errorPrint( "failed to open file: %s\n", filename);
        return;
    }
3276

3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297
    fprintf(fp, "================ database[%d] ================\n", index);
    fprintf(fp, "name: %s\n", dbInfos->name);
    fprintf(fp, "created_time: %s\n", dbInfos->create_time);
    fprintf(fp, "ntables: %"PRId64"\n", dbInfos->ntables);
    fprintf(fp, "vgroups: %d\n", dbInfos->vgroups);
    fprintf(fp, "replica: %d\n", dbInfos->replica);
    fprintf(fp, "quorum: %d\n", dbInfos->quorum);
    fprintf(fp, "days: %d\n", dbInfos->days);
    fprintf(fp, "keep0,keep1,keep(D): %s\n", dbInfos->keeplist);
    fprintf(fp, "cache(MB): %d\n", dbInfos->cache);
    fprintf(fp, "blocks: %d\n", dbInfos->blocks);
    fprintf(fp, "minrows: %d\n", dbInfos->minrows);
    fprintf(fp, "maxrows: %d\n", dbInfos->maxrows);
    fprintf(fp, "wallevel: %d\n", dbInfos->wallevel);
    fprintf(fp, "fsync: %d\n", dbInfos->fsync);
    fprintf(fp, "comp: %d\n", dbInfos->comp);
    fprintf(fp, "cachelast: %d\n", dbInfos->cachelast);
    fprintf(fp, "precision: %s\n", dbInfos->precision);
    fprintf(fp, "update: %d\n", dbInfos->update);
    fprintf(fp, "status: %s\n", dbInfos->status);
    fprintf(fp, "\n");
3298

3299
    fclose(fp);
3300 3301 3302
}

static void printfQuerySystemInfo(TAOS * taos) {
3303 3304
    char filename[MAX_FILE_NAME_LEN] = {0};
    char buffer[1024] = {0};
3305 3306 3307 3308 3309 3310
    TAOS_RES* res;

    time_t t;
    struct tm* lt;
    time(&t);
    lt = localtime(&t);
3311
    snprintf(filename, MAX_FILE_NAME_LEN, "querySystemInfo-%d-%d-%d %d:%d:%d",
3312 3313 3314 3315 3316 3317
            lt->tm_year+1900, lt->tm_mon, lt->tm_mday, lt->tm_hour, lt->tm_min,
            lt->tm_sec);

    // show variables
    res = taos_query(taos, "show variables;");
    //fetchResult(res, filename);
3318
    xDumpResultToFile(filename, res);
3319

3320 3321
    // show dnodes
    res = taos_query(taos, "show dnodes;");
3322
    xDumpResultToFile(filename, res);
3323
    //fetchResult(res, filename);
3324

3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341 3342
    // show databases
    res = taos_query(taos, "show databases;");
    SDbInfo** dbInfos = (SDbInfo **)calloc(MAX_DATABASE_COUNT, sizeof(SDbInfo *));
    if (dbInfos == NULL) {
        errorPrint("%s() LN%d, failed to allocate memory\n", __func__, __LINE__);
        return;
    }
    int dbCount = getDbFromServer(taos, dbInfos);
    if (dbCount <= 0) {
        free(dbInfos);
        return;
    }

    for (int i = 0; i < dbCount; i++) {
        // printf database info
        printfDbInfoForQueryToFile(filename, dbInfos[i], i);

        // show db.vgroups
3343
        snprintf(buffer, 1024, "show %s.vgroups;", dbInfos[i]->name);
3344 3345 3346 3347
        res = taos_query(taos, buffer);
        xDumpResultToFile(filename, res);

        // show db.stables
3348
        snprintf(buffer, 1024, "show %s.stables;", dbInfos[i]->name);
3349 3350 3351 3352
        res = taos_query(taos, buffer);
        xDumpResultToFile(filename, res);
        free(dbInfos[i]);
    }
3353

3354
    free(dbInfos);
3355 3356
}

3357
static int postProceSql(char *host, struct sockaddr_in *pServAddr, uint16_t port,
3358
        char* sqlstr, threadInfo *pThreadInfo)
3359
{
3360
    char *req_fmt = "POST %s HTTP/1.1\r\nHost: %s:%d\r\nAccept: */*\r\nAuthorization: Basic %s\r\nContent-Length: %d\r\nContent-Type: application/x-www-form-urlencoded\r\n\r\n%s";
3361

3362
    char *url = "/rest/sql";
3363

3364 3365 3366 3367
    int bytes, sent, received, req_str_len, resp_len;
    char *request_buf;
    char response_buf[RESP_BUF_LEN];
    uint16_t rest_port = port + TSDB_PORT_HTTP;
3368

3369
    int req_buf_len = strlen(sqlstr) + REQ_EXTRA_BUF_LEN;
3370

3371 3372
    request_buf = malloc(req_buf_len);
    if (NULL == request_buf) {
3373
        errorPrint("%s", "cannot allocate memory.\n");
3374
        exit(EXIT_FAILURE);
3375
    }
3376

3377 3378
    char userpass_buf[INPUT_BUF_LEN];
    int mod_table[] = {0, 2, 1};
3379

3380
    static char base64[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
3381 3382 3383 3384 3385 3386 3387
        'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
        'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X',
        'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f',
        'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
        'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
        'w', 'x', 'y', 'z', '0', '1', '2', '3',
        '4', '5', '6', '7', '8', '9', '+', '/'};
3388

3389
    snprintf(userpass_buf, INPUT_BUF_LEN, "%s:%s",
3390
            g_Dbs.user, g_Dbs.password);
3391 3392
    size_t userpass_buf_len = strlen(userpass_buf);
    size_t encoded_len = 4 * ((userpass_buf_len +2) / 3);
3393

3394
    char base64_buf[INPUT_BUF_LEN];
3395
#ifdef WINDOWS
3396
    WSADATA wsaData;
3397
    WSAStartup(MAKEWORD(2, 2), &wsaData);
3398 3399
    SOCKET sockfd;
#else
3400
    int sockfd;
3401
#endif
3402 3403
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) {
3404
#ifdef WINDOWS
3405
        errorPrint( "Could not create socket : %d" , WSAGetLastError());
3406
#endif
3407 3408
        debugPrint("%s() LN%d, sockfd=%d\n", __func__, __LINE__, sockfd);
        free(request_buf);
3409
        ERROR_EXIT("opening socket");
3410
    }
3411

3412
    int retConn = connect(sockfd, (struct sockaddr *)pServAddr, sizeof(struct sockaddr));
3413 3414 3415
    debugPrint("%s() LN%d connect() return %d\n", __func__, __LINE__, retConn);
    if (retConn < 0) {
        free(request_buf);
3416
        ERROR_EXIT("connecting");
3417
    }
3418

3419
    memset(base64_buf, 0, INPUT_BUF_LEN);
3420

3421
    for (int n = 0, m = 0; n < userpass_buf_len;) {
3422 3423 3424 3425 3426 3427 3428
        uint32_t oct_a = n < userpass_buf_len ?
            (unsigned char) userpass_buf[n++]:0;
        uint32_t oct_b = n < userpass_buf_len ?
            (unsigned char) userpass_buf[n++]:0;
        uint32_t oct_c = n < userpass_buf_len ?
            (unsigned char) userpass_buf[n++]:0;
        uint32_t triple = (oct_a << 0x10) + (oct_b << 0x08) + oct_c;
3429

3430 3431 3432 3433
        base64_buf[m++] = base64[(triple >> 3* 6) & 0x3f];
        base64_buf[m++] = base64[(triple >> 2* 6) & 0x3f];
        base64_buf[m++] = base64[(triple >> 1* 6) & 0x3f];
        base64_buf[m++] = base64[(triple >> 0* 6) & 0x3f];
3434
    }
3435

3436
    for (int l = 0; l < mod_table[userpass_buf_len % 3]; l++)
3437
        base64_buf[encoded_len - 1 - l] = '=';
3438

3439 3440 3441
    debugPrint("%s() LN%d: auth string base64 encoded: %s\n",
            __func__, __LINE__, base64_buf);
    char *auth = base64_buf;
3442

3443 3444 3445 3446 3447 3448
    int r = snprintf(request_buf,
            req_buf_len,
            req_fmt, url, host, rest_port,
            auth, strlen(sqlstr), sqlstr);
    if (r >= req_buf_len) {
        free(request_buf);
3449
        ERROR_EXIT("too long request");
3450 3451
    }
    verbosePrint("%s() LN%d: Request:\n%s\n", __func__, __LINE__, request_buf);
3452

3453 3454 3455
    req_str_len = strlen(request_buf);
    sent = 0;
    do {
3456
#ifdef WINDOWS
3457
        bytes = send(sockfd, request_buf + sent, req_str_len - sent, 0);
3458
#else
3459
        bytes = write(sockfd, request_buf + sent, req_str_len - sent);
3460
#endif
3461
        if (bytes < 0)
3462
            ERROR_EXIT("writing message to socket");
3463 3464 3465 3466
        if (bytes == 0)
            break;
        sent+=bytes;
    } while(sent < req_str_len);
3467

3468 3469 3470 3471
    memset(response_buf, 0, RESP_BUF_LEN);
    resp_len = sizeof(response_buf) - 1;
    received = 0;
    do {
3472
#ifdef WINDOWS
3473
        bytes = recv(sockfd, response_buf + received, resp_len - received, 0);
3474
#else
3475
        bytes = read(sockfd, response_buf + received, resp_len - received);
3476
#endif
3477 3478
        if (bytes < 0) {
            free(request_buf);
3479
            ERROR_EXIT("reading response from socket");
3480 3481 3482 3483 3484
        }
        if (bytes == 0)
            break;
        received += bytes;
    } while(received < resp_len);
3485

3486 3487
    if (received == resp_len) {
        free(request_buf);
3488
        ERROR_EXIT("storing complete response from socket");
3489
    }
H
Haojun Liao 已提交
3490

3491 3492
    response_buf[RESP_BUF_LEN - 1] = '\0';
    printf("Response:\n%s\n", response_buf);
H
Haojun Liao 已提交
3493

3494
    if (strlen(pThreadInfo->filePath) > 0) {
3495
        appendResultBufToFile(response_buf, pThreadInfo);
3496 3497
    }

3498
    free(request_buf);
3499
#ifdef WINDOWS
3500
    closesocket(sockfd);
3501 3502
    WSACleanup();
#else
3503
    close(sockfd);
3504
#endif
3505

3506
    return 0;
3507 3508
}

3509
static char* getTagValueFromTagSample(SSuperTable* stbInfo, int tagUsePos) {
3510 3511
    char*  dataBuf = (char*)calloc(TSDB_MAX_SQL_LEN+1, 1);
    if (NULL == dataBuf) {
3512
        errorPrint2("%s() LN%d, calloc failed! size:%d\n",
3513 3514 3515
                __func__, __LINE__, TSDB_MAX_SQL_LEN+1);
        return NULL;
    }
3516

3517 3518 3519
    int    dataLen = 0;
    dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
            "(%s)", stbInfo->tagDataBuf + stbInfo->lenOfTagOfOneRow * tagUsePos);
3520

3521
    return dataBuf;
3522 3523
}

3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542
static char *generateBinaryNCharTagValues(int64_t tableSeq, uint32_t len)
{
    char* buf = (char*)calloc(len, 1);
    if (NULL == buf) {
        printf("calloc failed! size:%d\n", len);
        return NULL;
    }

    if (tableSeq % 2) {
        tstrncpy(buf, "beijing", len);
    } else {
        tstrncpy(buf, "shanghai", len);
    }
    //rand_string(buf, stbInfo->tags[i].dataLen);

    return buf;
}

static char* generateTagValuesForStb(SSuperTable* stbInfo, int64_t tableSeq) {
3543 3544 3545
    char*  dataBuf = (char*)calloc(TSDB_MAX_SQL_LEN+1, 1);
    if (NULL == dataBuf) {
        printf("calloc failed! size:%d\n", TSDB_MAX_SQL_LEN+1);
3546
        return NULL;
3547
    }
3548

3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561
    int    dataLen = 0;
    dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen, "(");
    for (int i = 0; i < stbInfo->tagCount; i++) {
        if ((0 == strncasecmp(stbInfo->tags[i].dataType,
                        "binary", strlen("binary")))
                || (0 == strncasecmp(stbInfo->tags[i].dataType,
                        "nchar", strlen("nchar")))) {
            if (stbInfo->tags[i].dataLen > TSDB_MAX_BINARY_LEN) {
                printf("binary or nchar length overflow, max size:%u\n",
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                tmfree(dataBuf);
                return NULL;
            }
3562

3563 3564
            int32_t tagBufLen = stbInfo->tags[i].dataLen + 1;
            char *buf = generateBinaryNCharTagValues(tableSeq, tagBufLen);
3565 3566 3567 3568 3569
            if (NULL == buf) {
                tmfree(dataBuf);
                return NULL;
            }
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3570
                    "\'%s\',", buf);
3571 3572 3573
            tmfree(buf);
        } else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "int", strlen("int"))) {
3574 3575 3576
            if ((g_args.demo_mode) && (i == 0)) {
                dataLen += snprintf(dataBuf + dataLen,
                        TSDB_MAX_SQL_LEN - dataLen,
3577
                        "%"PRId64",", (tableSeq % 10) + 1);
3578 3579 3580
            } else {
                dataLen += snprintf(dataBuf + dataLen,
                        TSDB_MAX_SQL_LEN - dataLen,
3581
                        "%"PRId64",", tableSeq);
3582
            }
3583 3584 3585
        } else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "bigint", strlen("bigint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3586
                    "%"PRId64",", rand_bigint());
3587 3588 3589
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "float", strlen("float"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3590
                    "%f,", rand_float());
3591 3592 3593
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "double", strlen("double"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3594
                    "%f,", rand_double());
3595 3596 3597
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "smallint", strlen("smallint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3598
                    "%d,", rand_smallint());
3599 3600 3601
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "tinyint", strlen("tinyint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3602
                    "%d,", rand_tinyint());
3603 3604 3605
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "bool", strlen("bool"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3606
                    "%d,", rand_bool());
3607 3608 3609
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "timestamp", strlen("timestamp"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626
                    "%"PRId64",", rand_ubigint());
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "utinyint", strlen("utinyint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
                    "%d,", rand_utinyint());
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "usmallint", strlen("usmallint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
                    "%d,", rand_usmallint());
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "uint", strlen("uint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
                    "%d,", rand_uint());
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "ubigint", strlen("ubigint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
                    "%"PRId64",", rand_ubigint());
3627
        }  else {
3628
            errorPrint2("No support data type: %s\n", stbInfo->tags[i].dataType);
3629 3630 3631
            tmfree(dataBuf);
            return NULL;
        }
3632
    }
3633

3634
    dataLen -= 1;
3635 3636
    dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen, ")");
    return dataBuf;
3637 3638
}

3639
static int calcRowLen(SSuperTable*  superTbls) {
3640 3641 3642 3643 3644 3645
    int colIndex;
    int  lenOfOneRow = 0;

    for (colIndex = 0; colIndex < superTbls->columnCount; colIndex++) {
        char* dataType = superTbls->columns[colIndex].dataType;

3646 3647 3648 3649 3650 3651 3652 3653 3654 3655
        switch(superTbls->columns[colIndex].data_type) {
            case TSDB_DATA_TYPE_BINARY:
                lenOfOneRow += superTbls->columns[colIndex].dataLen + 3;
                break;

            case TSDB_DATA_TYPE_NCHAR:
                lenOfOneRow += superTbls->columns[colIndex].dataLen + 3;
                break;

            case TSDB_DATA_TYPE_INT:
3656
            case TSDB_DATA_TYPE_UINT:
3657 3658 3659 3660
                lenOfOneRow += INT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BIGINT:
3661
            case TSDB_DATA_TYPE_UBIGINT:
3662 3663 3664 3665
                lenOfOneRow += BIGINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
3666
            case TSDB_DATA_TYPE_USMALLINT:
3667 3668 3669 3670
                lenOfOneRow += SMALLINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TINYINT:
3671
            case TSDB_DATA_TYPE_UTINYINT:
3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693
                lenOfOneRow += TINYINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BOOL:
                lenOfOneRow += BOOL_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_FLOAT:
                lenOfOneRow += FLOAT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_DOUBLE:
                lenOfOneRow += DOUBLE_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TIMESTAMP:
                lenOfOneRow += TIMESTAMP_BUFF_LEN;
                break;

            default:
                errorPrint2("get error data type : %s\n", dataType);
                exit(EXIT_FAILURE);
3694 3695 3696 3697 3698 3699 3700 3701
        }
    }

    superTbls->lenOfOneRow = lenOfOneRow + 20; // timestamp

    int tagIndex;
    int lenOfTagOfOneRow = 0;
    for (tagIndex = 0; tagIndex < superTbls->tagCount; tagIndex++) {
3702 3703 3704 3705
        char * dataType = superTbls->tags[tagIndex].dataType;
        switch (superTbls->tags[tagIndex].data_type)
        {
        case TSDB_DATA_TYPE_BINARY:
3706
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + 3;
3707 3708
            break;
        case TSDB_DATA_TYPE_NCHAR:
3709
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + 3;
3710 3711 3712
            break;
        case TSDB_DATA_TYPE_INT:
        case TSDB_DATA_TYPE_UINT:
3713
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + INT_BUFF_LEN;
3714 3715 3716
            break;
        case TSDB_DATA_TYPE_BIGINT:
        case TSDB_DATA_TYPE_UBIGINT:
3717
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + BIGINT_BUFF_LEN;
3718 3719 3720
            break;
        case TSDB_DATA_TYPE_SMALLINT:
        case TSDB_DATA_TYPE_USMALLINT:
3721
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + SMALLINT_BUFF_LEN;
3722 3723 3724
            break;
        case TSDB_DATA_TYPE_TINYINT:
        case TSDB_DATA_TYPE_UTINYINT:
3725
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + TINYINT_BUFF_LEN;
3726 3727
            break;
        case TSDB_DATA_TYPE_BOOL:
3728
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + BOOL_BUFF_LEN;
3729 3730
            break;
        case TSDB_DATA_TYPE_FLOAT:
3731
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + FLOAT_BUFF_LEN;
3732 3733
            break;
        case TSDB_DATA_TYPE_DOUBLE:
3734
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + DOUBLE_BUFF_LEN;
3735 3736
            break;
        default:
3737
            errorPrint2("get error tag type : %s\n", dataType);
3738
            exit(EXIT_FAILURE);
3739
        }
3740 3741
    }

3742
    superTbls->lenOfTagOfOneRow = lenOfTagOfOneRow;
3743

3744
    return 0;
3745 3746
}

3747
static int getChildNameOfSuperTableWithLimitAndOffset(TAOS * taos,
3748
        char* dbName, char* stbName, char** childTblNameOfSuperTbl,
3749
        int64_t* childTblCountOfSuperTbl, int64_t limit, uint64_t offset) {
3750

3751
    char command[1024] = "\0";
3752
    char limitBuf[100] = "\0";
3753

3754 3755
    TAOS_RES * res;
    TAOS_ROW row = NULL;
3756

3757
    char* childTblName = *childTblNameOfSuperTbl;
3758

3759 3760
    snprintf(limitBuf, 100, " limit %"PRId64" offset %"PRIu64"",
        limit, offset);
3761

3762
    //get all child table name use cmd: select tbname from superTblName;
3763
    snprintf(command, 1024, "select tbname from %s.%s %s",
3764
            dbName, stbName, limitBuf);
3765

3766 3767 3768 3769
    res = taos_query(taos, command);
    int32_t code = taos_errno(res);
    if (code != 0) {
        taos_free_result(res);
3770
        taos_close(taos);
3771
        errorPrint2("%s() LN%d, failed to run command %s\n",
3772
                __func__, __LINE__, command);
3773
        exit(EXIT_FAILURE);
3774
    }
3775

3776 3777 3778 3779 3780 3781 3782
    int64_t childTblCount = (limit < 0)?10000:limit;
    int64_t count = 0;
    if (childTblName == NULL) {
        childTblName = (char*)calloc(1, childTblCount * TSDB_TABLE_NAME_LEN);
        if (NULL ==  childTblName) {
            taos_free_result(res);
            taos_close(taos);
3783
            errorPrint2("%s() LN%d, failed to allocate memory!\n", __func__, __LINE__);
3784
            exit(EXIT_FAILURE);
3785
        }
3786 3787
    }

3788 3789 3790 3791 3792
    char* pTblName = childTblName;
    while((row = taos_fetch_row(res)) != NULL) {
        int32_t* len = taos_fetch_lengths(res);

        if (0 == strlen((char *)row[0])) {
3793
            errorPrint2("%s() LN%d, No.%"PRId64" table return empty name\n",
3794
                    __func__, __LINE__, count);
3795
            exit(EXIT_FAILURE);
3796 3797 3798 3799 3800 3801 3802 3803 3804 3805 3806 3807 3808 3809 3810 3811 3812 3813
        }

        tstrncpy(pTblName, (char *)row[0], len[0]+1);
        //printf("==== sub table name: %s\n", pTblName);
        count++;
        if (count >= childTblCount - 1) {
            char *tmp = realloc(childTblName,
                    (size_t)childTblCount*1.5*TSDB_TABLE_NAME_LEN+1);
            if (tmp != NULL) {
                childTblName = tmp;
                childTblCount = (int)(childTblCount*1.5);
                memset(childTblName + count*TSDB_TABLE_NAME_LEN, 0,
                        (size_t)((childTblCount-count)*TSDB_TABLE_NAME_LEN));
            } else {
                // exit, if allocate more memory failed
                tmfree(childTblName);
                taos_free_result(res);
                taos_close(taos);
3814
                errorPrint2("%s() LN%d, realloc fail for save child table name of %s.%s\n",
3815
                        __func__, __LINE__, dbName, stbName);
3816
                exit(EXIT_FAILURE);
3817 3818 3819
            }
        }
        pTblName = childTblName + count * TSDB_TABLE_NAME_LEN;
3820
    }
3821

3822 3823
    *childTblCountOfSuperTbl = count;
    *childTblNameOfSuperTbl  = childTblName;
3824

3825 3826
    taos_free_result(res);
    return 0;
3827 3828
}

3829
static int getAllChildNameOfSuperTable(TAOS * taos, char* dbName,
3830
        char* stbName, char** childTblNameOfSuperTbl,
3831
        int64_t* childTblCountOfSuperTbl) {
3832

3833
    return getChildNameOfSuperTableWithLimitAndOffset(taos, dbName, stbName,
3834
            childTblNameOfSuperTbl, childTblCountOfSuperTbl,
3835
            -1, 0);
3836 3837
}

3838
static int getSuperTableFromServer(TAOS * taos, char* dbName,
3839
        SSuperTable*  superTbls) {
3840

3841
    char command[1024] = "\0";
3842 3843 3844 3845 3846
    TAOS_RES * res;
    TAOS_ROW row = NULL;
    int count = 0;

    //get schema use cmd: describe superTblName;
3847
    snprintf(command, 1024, "describe %s.%s", dbName, superTbls->stbName);
3848 3849 3850 3851 3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866 3867 3868
    res = taos_query(taos, command);
    int32_t code = taos_errno(res);
    if (code != 0) {
        printf("failed to run command %s\n", command);
        taos_free_result(res);
        return -1;
    }

    int tagIndex = 0;
    int columnIndex = 0;
    TAOS_FIELD *fields = taos_fetch_fields(res);
    while((row = taos_fetch_row(res)) != NULL) {
        if (0 == count) {
            count++;
            continue;
        }

        if (strcmp((char *)row[TSDB_DESCRIBE_METRIC_NOTE_INDEX], "TAG") == 0) {
            tstrncpy(superTbls->tags[tagIndex].field,
                    (char *)row[TSDB_DESCRIBE_METRIC_FIELD_INDEX],
                    fields[TSDB_DESCRIBE_METRIC_FIELD_INDEX].bytes);
3869
            if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3870 3871
                        "INT", strlen("INT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_INT;
3872
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3873 3874
                        "TINYINT", strlen("TINYINT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_TINYINT;
3875
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3876 3877
                        "SMALLINT", strlen("SMALLINT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_SMALLINT;
3878
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3879 3880
                        "BIGINT", strlen("BIGINT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_BIGINT;
3881
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3882 3883
                        "FLOAT", strlen("FLOAT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_FLOAT;
3884
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3885 3886
                        "DOUBLE", strlen("DOUBLE"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_DOUBLE;
3887
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3888 3889
                        "BINARY", strlen("BINARY"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_BINARY;
3890
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3891 3892
                        "NCHAR", strlen("NCHAR"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_NCHAR;
3893
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3894 3895
                        "BOOL", strlen("BOOL"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_BOOL;
3896
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3897 3898
                        "TIMESTAMP", strlen("TIMESTAMP"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_TIMESTAMP;
3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909 3910 3911 3912 3913 3914 3915 3916 3917 3918 3919 3920 3921 3922
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "TINYINT UNSIGNED", strlen("TINYINT UNSIGNED"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_UTINYINT;
                tstrncpy(superTbls->tags[tagIndex].dataType,"UTINYINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "SMALLINT UNSIGNED", strlen("SMALLINT UNSIGNED"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_USMALLINT;
                tstrncpy(superTbls->tags[tagIndex].dataType,"USMALLINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "INT UNSIGNED", strlen("INT UNSIGNED"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_UINT;
                tstrncpy(superTbls->tags[tagIndex].dataType,"UINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            }else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "BIGINT UNSIGNED", strlen("BIGINT UNSIGNED"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_UBIGINT;
                tstrncpy(superTbls->tags[tagIndex].dataType,"UBIGINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
3923 3924 3925
            } else {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_NULL;
            }
3926 3927 3928 3929
            superTbls->tags[tagIndex].dataLen =
                *((int *)row[TSDB_DESCRIBE_METRIC_LENGTH_INDEX]);
            tstrncpy(superTbls->tags[tagIndex].note,
                    (char *)row[TSDB_DESCRIBE_METRIC_NOTE_INDEX],
3930 3931
                    min(NOTE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_NOTE_INDEX].bytes) + 1);
3932 3933 3934 3935 3936 3937 3938
            if (strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL)
            {
                tstrncpy(superTbls->tags[tagIndex].dataType,
                    (char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            }
3939 3940 3941 3942 3943
            tagIndex++;
        } else {
            tstrncpy(superTbls->columns[columnIndex].field,
                    (char *)row[TSDB_DESCRIBE_METRIC_FIELD_INDEX],
                    fields[TSDB_DESCRIBE_METRIC_FIELD_INDEX].bytes);
3944

3945 3946 3947 3948
            
            if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "INT", strlen("INT")) && 
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3949
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_INT;
3950 3951 3952
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "TINYINT", strlen("TINYINT")) && 
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3953
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_TINYINT;
3954 3955 3956
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "SMALLINT", strlen("SMALLINT")) && 
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3957
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_SMALLINT;
3958 3959 3960
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "BIGINT", strlen("BIGINT")) && 
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3961
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_BIGINT;
3962
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3963 3964
                        "FLOAT", strlen("FLOAT"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_FLOAT;
3965
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3966 3967
                        "DOUBLE", strlen("DOUBLE"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_DOUBLE;
3968
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3969 3970
                        "BINARY", strlen("BINARY"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_BINARY;
3971
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3972 3973
                        "NCHAR", strlen("NCHAR"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_NCHAR;
3974
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3975 3976
                        "BOOL", strlen("BOOL"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_BOOL;
3977
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3978 3979
                        "TIMESTAMP", strlen("TIMESTAMP"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_TIMESTAMP;
3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "TINYINT UNSIGNED", strlen("TINYINT UNSIGNED"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_UTINYINT;
                tstrncpy(superTbls->columns[columnIndex].dataType,"UTINYINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "SMALLINT UNSIGNED", strlen("SMALLINT UNSIGNED"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_USMALLINT;
                tstrncpy(superTbls->columns[columnIndex].dataType,"USMALLINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "INT UNSIGNED", strlen("INT UNSIGNED"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_UINT;
                tstrncpy(superTbls->columns[columnIndex].dataType,"UINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                        "BIGINT UNSIGNED", strlen("BIGINT UNSIGNED"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_UBIGINT;
                tstrncpy(superTbls->columns[columnIndex].dataType,"UBIGINT",
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
4004 4005 4006
            } else {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_NULL;
            }
4007 4008 4009 4010
            superTbls->columns[columnIndex].dataLen =
                *((int *)row[TSDB_DESCRIBE_METRIC_LENGTH_INDEX]);
            tstrncpy(superTbls->columns[columnIndex].note,
                    (char *)row[TSDB_DESCRIBE_METRIC_NOTE_INDEX],
4011 4012
                    min(NOTE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_NOTE_INDEX].bytes) + 1);
4013 4014 4015 4016 4017 4018 4019
            
            if (strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
                tstrncpy(superTbls->columns[columnIndex].dataType,
                    (char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
                    min(DATATYPE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_TYPE_INDEX].bytes) + 1);
            }
4020

4021 4022 4023 4024 4025 4026 4027
            columnIndex++;
        }
        count++;
    }

    superTbls->columnCount = columnIndex;
    superTbls->tagCount    = tagIndex;
4028
    taos_free_result(res);
4029

4030 4031 4032 4033
    calcRowLen(superTbls);

    /*
       if (TBL_ALREADY_EXISTS == superTbls->childTblExists) {
4034
    //get all child table name use cmd: select tbname from superTblName;
4035 4036 4037
    int childTblCount = 10000;
    superTbls->childTblName = (char*)calloc(1, childTblCount * TSDB_TABLE_NAME_LEN);
    if (superTbls->childTblName == NULL) {
4038
    errorPrint2("%s() LN%d, alloc memory failed!\n", __func__, __LINE__);
4039
    return -1;
4040
    }
4041
    getAllChildNameOfSuperTable(taos, dbName,
4042
    superTbls->stbName,
4043 4044 4045 4046 4047
    &superTbls->childTblName,
    &superTbls->childTblCount);
    }
    */
    return 0;
4048 4049
}

H
Haojun Liao 已提交
4050
static int createSuperTable(
4051 4052
        TAOS * taos, char* dbName,
        SSuperTable*  superTbl) {
H
Haojun Liao 已提交
4053

4054 4055
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);
4056

4057
    char cols[COL_BUFFER_LEN] = "\0";
4058
    int len = 0;
4059

4060
    int  lenOfOneRow = 0;
4061

4062
    if (superTbl->columnCount == 0) {
4063
        errorPrint2("%s() LN%d, super table column count is %d\n",
4064
                __func__, __LINE__, superTbl->columnCount);
4065
        free(command);
4066 4067
        return -1;
    }
4068

4069
    for (int colIndex = 0; colIndex < superTbl->columnCount; colIndex++) {
4070

4071 4072
        switch(superTbl->columns[colIndex].data_type) {
            case TSDB_DATA_TYPE_BINARY:
4073
                len += snprintf(cols + len, COL_BUFFER_LEN - len,
4074 4075 4076 4077 4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091
                        ",C%d %s(%d)", colIndex, "BINARY",
                        superTbl->columns[colIndex].dataLen);
                lenOfOneRow += superTbl->columns[colIndex].dataLen + 3;
                break;

            case TSDB_DATA_TYPE_NCHAR:
                len += snprintf(cols + len, COL_BUFFER_LEN - len,
                        ",C%d %s(%d)", colIndex, "NCHAR",
                        superTbl->columns[colIndex].dataLen);
                lenOfOneRow += superTbl->columns[colIndex].dataLen + 3;
                break;

            case TSDB_DATA_TYPE_INT:
                if ((g_args.demo_mode) && (colIndex == 1)) {
                    len += snprintf(cols + len, COL_BUFFER_LEN - len,
                            ", VOLTAGE INT");
                } else {
                    len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s", colIndex, "INT");
4092
                }
4093 4094
                lenOfOneRow += INT_BUFF_LEN;
                break;
4095

4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143
            case TSDB_DATA_TYPE_BIGINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "BIGINT");
                lenOfOneRow += BIGINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "SMALLINT");
                lenOfOneRow += SMALLINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TINYINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s", colIndex, "TINYINT");
                lenOfOneRow += TINYINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BOOL:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s", colIndex, "BOOL");
                lenOfOneRow += BOOL_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_FLOAT:
                if (g_args.demo_mode) {
                    if (colIndex == 0) {
                        len += snprintf(cols + len, COL_BUFFER_LEN - len, ", CURRENT FLOAT");
                    } else if (colIndex == 2) {
                        len += snprintf(cols + len, COL_BUFFER_LEN - len, ", PHASE FLOAT");
                    }
                } else {
                    len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s", colIndex, "FLOAT");
                }

                lenOfOneRow += FLOAT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_DOUBLE:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "DOUBLE");
                lenOfOneRow += DOUBLE_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TIMESTAMP:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "TIMESTAMP");
                lenOfOneRow += TIMESTAMP_BUFF_LEN;
                break;

4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163 4164 4165 4166 4167
            case TSDB_DATA_TYPE_UTINYINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "TINYINT UNSIGNED");
                lenOfOneRow += TINYINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_USMALLINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "SMALLINT UNSIGNED");
                lenOfOneRow += SMALLINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_UINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "INT UNSIGNED");
                lenOfOneRow += INT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_UBIGINT:
                len += snprintf(cols + len, COL_BUFFER_LEN - len, ",C%d %s",
                        colIndex, "BIGINT UNSIGNED");
                lenOfOneRow += BIGINT_BUFF_LEN;
                break;

4168 4169 4170 4171 4172 4173
            default:
                taos_close(taos);
                free(command);
                errorPrint2("%s() LN%d, config error data type : %s\n",
                        __func__, __LINE__, superTbl->columns[colIndex].dataType);
                exit(EXIT_FAILURE);
4174
        }
4175 4176
    }

4177
    superTbl->lenOfOneRow = lenOfOneRow + 20; // timestamp
4178

4179 4180 4181 4182
    // save for creating child table
    superTbl->colsOfCreateChildTable = (char*)calloc(len+20, 1);
    if (NULL == superTbl->colsOfCreateChildTable) {
        taos_close(taos);
4183
        free(command);
4184
        errorPrint2("%s() LN%d, Failed when calloc, size:%d",
4185 4186
                __func__, __LINE__, len+1);
        exit(EXIT_FAILURE);
4187
    }
4188

4189 4190 4191
    snprintf(superTbl->colsOfCreateChildTable, len+20, "(ts timestamp%s)", cols);
    verbosePrint("%s() LN%d: %s\n",
            __func__, __LINE__, superTbl->colsOfCreateChildTable);
4192

4193
    if (superTbl->tagCount == 0) {
4194
        errorPrint2("%s() LN%d, super table tag count is %d\n",
4195
                __func__, __LINE__, superTbl->tagCount);
4196
        free(command);
4197 4198
        return -1;
    }
4199

4200
    char tags[TSDB_MAX_TAGS_LEN] = "\0";
4201 4202
    int tagIndex;
    len = 0;
4203

4204
    int lenOfTagOfOneRow = 0;
4205
    len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len, "(");
4206 4207 4208 4209
    for (tagIndex = 0; tagIndex < superTbl->tagCount; tagIndex++) {
        char* dataType = superTbl->tags[tagIndex].dataType;

        if (strcasecmp(dataType, "BINARY") == 0) {
4210
            if ((g_args.demo_mode) && (tagIndex == 1)) {
4211
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4212
                        "location BINARY(%d),",
4213 4214
                        superTbl->tags[tagIndex].dataLen);
            } else {
4215
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4216
                        "T%d %s(%d),", tagIndex, "BINARY",
4217
                        superTbl->tags[tagIndex].dataLen);
4218
            }
4219 4220
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + 3;
        } else if (strcasecmp(dataType, "NCHAR") == 0) {
4221
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4222
                    "T%d %s(%d),", tagIndex,
4223 4224 4225
                    "NCHAR", superTbl->tags[tagIndex].dataLen);
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + 3;
        } else if (strcasecmp(dataType, "INT") == 0)  {
4226
            if ((g_args.demo_mode) && (tagIndex == 0)) {
4227 4228
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                        "groupId INT, ");
4229
            } else {
4230
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4231
                        "T%d %s,", tagIndex, "INT");
4232
            }
4233
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + INT_BUFF_LEN;
4234
        } else if (strcasecmp(dataType, "BIGINT") == 0)  {
4235
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4236
                    "T%d %s,", tagIndex, "BIGINT");
4237
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + BIGINT_BUFF_LEN;
4238
        } else if (strcasecmp(dataType, "SMALLINT") == 0)  {
4239
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4240
                    "T%d %s,", tagIndex, "SMALLINT");
4241
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + SMALLINT_BUFF_LEN;
4242
        } else if (strcasecmp(dataType, "TINYINT") == 0)  {
4243
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4244
                    "T%d %s,", tagIndex, "TINYINT");
4245
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + TINYINT_BUFF_LEN;
4246
        } else if (strcasecmp(dataType, "BOOL") == 0)  {
4247
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4248
                    "T%d %s,", tagIndex, "BOOL");
4249
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + BOOL_BUFF_LEN;
4250
        } else if (strcasecmp(dataType, "FLOAT") == 0) {
4251
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4252
                    "T%d %s,", tagIndex, "FLOAT");
4253
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + FLOAT_BUFF_LEN;
4254
        } else if (strcasecmp(dataType, "DOUBLE") == 0) {
4255
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4256
                    "T%d %s,", tagIndex, "DOUBLE");
4257
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + DOUBLE_BUFF_LEN;
4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272 4273
        } else if (strcasecmp(dataType, "UTINYINT") == 0) {
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                    "T%d %s,", tagIndex, "TINYINT UNSIGNED");
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + TINYINT_BUFF_LEN;
        } else if (strcasecmp(dataType, "USMALLINT") == 0) {
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                    "T%d %s,", tagIndex, "SMALLINT UNSIGNED");
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + SMALLINT_BUFF_LEN;
        } else if (strcasecmp(dataType, "UINT") == 0) {
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                    "T%d %s,", tagIndex, "INT UNSIGNED");
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + INT_BUFF_LEN;
        } else if (strcasecmp(dataType, "UBIGINT") == 0) {
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                    "T%d %s,", tagIndex, "BIGINT UNSIGNED");
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + BIGINT_BUFF_LEN;
4274 4275 4276 4277
        } else if (strcasecmp(dataType, "TIMESTAMP") == 0) {
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                    "T%d %s,", tagIndex, "TIMESTAMP");
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + TIMESTAMP_BUFF_LEN;
4278 4279
        } else {
            taos_close(taos);
4280
            free(command);
4281
            errorPrint2("%s() LN%d, config error tag type : %s\n",
4282
                    __func__, __LINE__, dataType);
4283
            exit(EXIT_FAILURE);
4284
        }
4285
    }
4286

4287
    len -= 1;
4288
    len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len, ")");
4289

4290
    superTbl->lenOfTagOfOneRow = lenOfTagOfOneRow;
4291

4292
    snprintf(command, BUFFER_SIZE,
4293 4294
            "CREATE TABLE IF NOT EXISTS %s.%s (ts TIMESTAMP%s) TAGS %s",
            dbName, superTbl->stbName, cols, tags);
4295
    if (0 != queryDbExec(taos, command, NO_INSERT_TYPE, false)) {
4296
        errorPrint2("create supertable %s failed!\n\n",
4297
                superTbl->stbName);
4298
        free(command);
4299 4300
        return -1;
    }
4301

4302
    debugPrint("create supertable %s success!\n\n", superTbl->stbName);
4303
    free(command);
4304
    return 0;
4305 4306
}

4307
int createDatabasesAndStables(char *command) {
4308 4309 4310 4311
    TAOS * taos = NULL;
    int    ret = 0;
    taos = taos_connect(g_Dbs.host, g_Dbs.user, g_Dbs.password, NULL, g_Dbs.port);
    if (taos == NULL) {
4312
        errorPrint2("Failed to connect to TDengine, reason:%s\n", taos_errstr(NULL));
4313
        return -1;
4314
    }
4315

4316 4317 4318 4319 4320 4321 4322
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        if (g_Dbs.db[i].drop) {
            sprintf(command, "drop database if exists %s;", g_Dbs.db[i].dbName);
            if (0 != queryDbExec(taos, command, NO_INSERT_TYPE, false)) {
                taos_close(taos);
                return -1;
            }
4323

4324 4325
            int dataLen = 0;
            dataLen += snprintf(command + dataLen,
4326
                    BUFFER_SIZE - dataLen, "CREATE DATABASE IF NOT EXISTS %s",
4327
                    g_Dbs.db[i].dbName);
4328

4329 4330
            if (g_Dbs.db[i].dbCfg.blocks > 0) {
                dataLen += snprintf(command + dataLen,
4331
                        BUFFER_SIZE - dataLen, " BLOCKS %d",
4332
                        g_Dbs.db[i].dbCfg.blocks);
4333 4334 4335
            }
            if (g_Dbs.db[i].dbCfg.cache > 0) {
                dataLen += snprintf(command + dataLen,
4336
                        BUFFER_SIZE - dataLen, " CACHE %d",
4337
                        g_Dbs.db[i].dbCfg.cache);
4338 4339 4340
            }
            if (g_Dbs.db[i].dbCfg.days > 0) {
                dataLen += snprintf(command + dataLen,
4341
                        BUFFER_SIZE - dataLen, " DAYS %d",
4342
                        g_Dbs.db[i].dbCfg.days);
4343 4344 4345
            }
            if (g_Dbs.db[i].dbCfg.keep > 0) {
                dataLen += snprintf(command + dataLen,
4346
                        BUFFER_SIZE - dataLen, " KEEP %d",
4347
                        g_Dbs.db[i].dbCfg.keep);
4348 4349 4350
            }
            if (g_Dbs.db[i].dbCfg.quorum > 1) {
                dataLen += snprintf(command + dataLen,
4351
                        BUFFER_SIZE - dataLen, " QUORUM %d",
4352
                        g_Dbs.db[i].dbCfg.quorum);
4353 4354 4355
            }
            if (g_Dbs.db[i].dbCfg.replica > 0) {
                dataLen += snprintf(command + dataLen,
4356
                        BUFFER_SIZE - dataLen, " REPLICA %d",
4357
                        g_Dbs.db[i].dbCfg.replica);
4358 4359 4360
            }
            if (g_Dbs.db[i].dbCfg.update > 0) {
                dataLen += snprintf(command + dataLen,
4361
                        BUFFER_SIZE - dataLen, " UPDATE %d",
4362
                        g_Dbs.db[i].dbCfg.update);
4363 4364 4365 4366 4367 4368 4369
            }
            //if (g_Dbs.db[i].dbCfg.maxtablesPerVnode > 0) {
            //  dataLen += snprintf(command + dataLen,
            //  BUFFER_SIZE - dataLen, "tables %d ", g_Dbs.db[i].dbCfg.maxtablesPerVnode);
            //}
            if (g_Dbs.db[i].dbCfg.minRows > 0) {
                dataLen += snprintf(command + dataLen,
4370
                        BUFFER_SIZE - dataLen, " MINROWS %d",
4371
                        g_Dbs.db[i].dbCfg.minRows);
4372 4373 4374
            }
            if (g_Dbs.db[i].dbCfg.maxRows > 0) {
                dataLen += snprintf(command + dataLen,
4375
                        BUFFER_SIZE - dataLen, " MAXROWS %d",
4376
                        g_Dbs.db[i].dbCfg.maxRows);
4377 4378 4379
            }
            if (g_Dbs.db[i].dbCfg.comp > 0) {
                dataLen += snprintf(command + dataLen,
4380
                        BUFFER_SIZE - dataLen, " COMP %d",
4381
                        g_Dbs.db[i].dbCfg.comp);
4382 4383 4384
            }
            if (g_Dbs.db[i].dbCfg.walLevel > 0) {
                dataLen += snprintf(command + dataLen,
4385 4386
                        BUFFER_SIZE - dataLen, " wal %d",
                        g_Dbs.db[i].dbCfg.walLevel);
4387 4388 4389
            }
            if (g_Dbs.db[i].dbCfg.cacheLast > 0) {
                dataLen += snprintf(command + dataLen,
4390
                        BUFFER_SIZE - dataLen, " CACHELAST %d",
4391
                        g_Dbs.db[i].dbCfg.cacheLast);
4392 4393 4394
            }
            if (g_Dbs.db[i].dbCfg.fsync > 0) {
                dataLen += snprintf(command + dataLen, BUFFER_SIZE - dataLen,
4395
                        " FSYNC %d", g_Dbs.db[i].dbCfg.fsync);
4396
            }
4397
            if ((0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ms", 2))
4398
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision,
4399
                            "ns", 2))
4400
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision,
4401
                            "us", 2))) {
4402 4403 4404
                dataLen += snprintf(command + dataLen, BUFFER_SIZE - dataLen,
                        " precision \'%s\';", g_Dbs.db[i].dbCfg.precision);
            }
4405

4406 4407
            if (0 != queryDbExec(taos, command, NO_INSERT_TYPE, false)) {
                taos_close(taos);
4408
                errorPrint("\ncreate database %s failed!\n\n",
4409
                        g_Dbs.db[i].dbName);
4410 4411 4412 4413
                return -1;
            }
            printf("\ncreate database %s success!\n\n", g_Dbs.db[i].dbName);
        }
4414

4415 4416
        debugPrint("%s() LN%d supertbl count:%"PRIu64"\n",
                __func__, __LINE__, g_Dbs.db[i].superTblCount);
4417

4418
        int validStbCount = 0;
4419

4420 4421
        for (uint64_t j = 0; j < g_Dbs.db[i].superTblCount; j++) {
            sprintf(command, "describe %s.%s;", g_Dbs.db[i].dbName,
4422
                    g_Dbs.db[i].superTbls[j].stbName);
4423
            ret = queryDbExec(taos, command, NO_INSERT_TYPE, true);
4424

4425 4426 4427
            if ((ret != 0) || (g_Dbs.db[i].drop)) {
                ret = createSuperTable(taos, g_Dbs.db[i].dbName,
                        &g_Dbs.db[i].superTbls[j]);
4428

4429 4430 4431 4432
                if (0 != ret) {
                    errorPrint("create super table %"PRIu64" failed!\n\n", j);
                    continue;
                }
4433 4434
            } else {
                ret = getSuperTableFromServer(taos, g_Dbs.db[i].dbName,
4435
                    &g_Dbs.db[i].superTbls[j]);
4436 4437 4438 4439 4440
                if (0 != ret) {
                    errorPrint2("\nget super table %s.%s info failed!\n\n",
                            g_Dbs.db[i].dbName, g_Dbs.db[i].superTbls[j].stbName);
                    continue;
                }
4441 4442 4443 4444 4445 4446 4447 4448
            }
            validStbCount ++;
        }
        g_Dbs.db[i].superTblCount = validStbCount;
    }

    taos_close(taos);
    return 0;
4449 4450
}

4451 4452
static void* createTable(void *sarg)
{
4453
    threadInfo *pThreadInfo = (threadInfo *)sarg;
4454
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
4455

4456 4457
    setThreadName("createTable");

4458
    uint64_t  lastPrintTime = taosGetTimestampMs();
4459

4460
    int buff_len = BUFFER_SIZE;
4461

4462 4463
    pThreadInfo->buffer = calloc(buff_len, 1);
    if (pThreadInfo->buffer == NULL) {
4464
        errorPrint2("%s() LN%d, Memory allocated failed!\n", __func__, __LINE__);
4465
        exit(EXIT_FAILURE);
4466 4467 4468 4469 4470 4471 4472 4473 4474 4475 4476 4477 4478
    }

    int len = 0;
    int batchNum = 0;

    verbosePrint("%s() LN%d: Creating table from %"PRIu64" to %"PRIu64"\n",
            __func__, __LINE__,
            pThreadInfo->start_table_from, pThreadInfo->end_table_to);

    for (uint64_t i = pThreadInfo->start_table_from;
            i <= pThreadInfo->end_table_to; i++) {
        if (0 == g_Dbs.use_metric) {
            snprintf(pThreadInfo->buffer, buff_len,
4479
                    "CREATE TABLE IF NOT EXISTS %s.%s%"PRIu64" %s;",
4480 4481 4482
                    pThreadInfo->db_name,
                    g_args.tb_prefix, i,
                    pThreadInfo->cols);
4483
            batchNum ++;
4484
        } else {
4485 4486
            if (stbInfo == NULL) {
                free(pThreadInfo->buffer);
4487
                errorPrint2("%s() LN%d, use metric, but super table info is NULL\n",
4488
                        __func__, __LINE__);
4489
                exit(EXIT_FAILURE);
4490 4491 4492 4493 4494
            } else {
                if (0 == len) {
                    batchNum = 0;
                    memset(pThreadInfo->buffer, 0, buff_len);
                    len += snprintf(pThreadInfo->buffer + len,
4495
                            buff_len - len, "CREATE TABLE ");
4496
                }
4497

4498
                char* tagsValBuf = NULL;
4499 4500
                if (0 == stbInfo->tagSource) {
                    tagsValBuf = generateTagValuesForStb(stbInfo, i);
4501
                } else {
4502 4503 4504 4505
                    if (0 == stbInfo->tagSampleCount) {
                        free(pThreadInfo->buffer);
                        ERROR_EXIT("use sample file for tag, but has no content!\n");
                    }
4506
                    tagsValBuf = getTagValueFromTagSample(
4507 4508
                            stbInfo,
                            i % stbInfo->tagSampleCount);
4509
                }
4510

4511 4512
                if (NULL == tagsValBuf) {
                    free(pThreadInfo->buffer);
4513
                    ERROR_EXIT("use metric, but tag buffer is NULL\n");
4514 4515 4516 4517
                }
                len += snprintf(pThreadInfo->buffer + len,
                        buff_len - len,
                        "if not exists %s.%s%"PRIu64" using %s.%s tags %s ",
4518
                        pThreadInfo->db_name, stbInfo->childTblPrefix,
4519
                        i, pThreadInfo->db_name,
4520
                        stbInfo->stbName, tagsValBuf);
4521 4522
                free(tagsValBuf);
                batchNum++;
4523
                if ((batchNum < stbInfo->batchCreateTableNum)
4524
                        && ((buff_len - len)
4525
                            >= (stbInfo->lenOfTagOfOneRow + 256))) {
4526 4527 4528
                    continue;
                }
            }
4529
        }
4530

4531
        len = 0;
S
Shengliang Guan 已提交
4532

4533
        if (0 != queryDbExec(pThreadInfo->taos, pThreadInfo->buffer,
4534
                    NO_INSERT_TYPE, false)) {
4535
            errorPrint2("queryDbExec() failed. buffer:\n%s\n", pThreadInfo->buffer);
4536 4537 4538
            free(pThreadInfo->buffer);
            return NULL;
        }
4539
        pThreadInfo->tables_created += batchNum;
4540

4541
        uint64_t currentPrintTime = taosGetTimestampMs();
4542 4543 4544 4545 4546
        if (currentPrintTime - lastPrintTime > 30*1000) {
            printf("thread[%d] already create %"PRIu64" - %"PRIu64" tables\n",
                    pThreadInfo->threadID, pThreadInfo->start_table_from, i);
            lastPrintTime = currentPrintTime;
        }
4547
    }
4548

4549 4550 4551
    if (0 != len) {
        if (0 != queryDbExec(pThreadInfo->taos, pThreadInfo->buffer,
                    NO_INSERT_TYPE, false)) {
4552
            errorPrint2("queryDbExec() failed. buffer:\n%s\n", pThreadInfo->buffer);
4553
        }
4554
    }
4555

4556 4557
    free(pThreadInfo->buffer);
    return NULL;
4558 4559
}

4560
static int startMultiThreadCreateChildTable(
4561
        char* cols, int threads, uint64_t tableFrom, int64_t ntables,
4562
        char* db_name, SSuperTable* stbInfo) {
4563

4564 4565
    pthread_t *pids = calloc(1, threads * sizeof(pthread_t));
    threadInfo *infos = calloc(1, threads * sizeof(threadInfo));
4566

4567
    if ((NULL == pids) || (NULL == infos)) {
4568
        ERROR_EXIT("createChildTable malloc failed\n");
4569
    }
4570

4571 4572 4573
    if (threads < 1) {
        threads = 1;
    }
4574

4575 4576 4577 4578
    int64_t a = ntables / threads;
    if (a < 1) {
        threads = ntables;
        a = 1;
4579
    }
4580

4581 4582
    int64_t b = 0;
    b = ntables % threads;
4583

4584 4585 4586 4587
    for (int64_t i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
        pThreadInfo->threadID = i;
        tstrncpy(pThreadInfo->db_name, db_name, TSDB_DB_NAME_LEN);
4588
        pThreadInfo->stbInfo = stbInfo;
4589 4590 4591 4592 4593 4594 4595 4596
        verbosePrint("%s() %d db_name: %s\n", __func__, __LINE__, db_name);
        pThreadInfo->taos = taos_connect(
                g_Dbs.host,
                g_Dbs.user,
                g_Dbs.password,
                db_name,
                g_Dbs.port);
        if (pThreadInfo->taos == NULL) {
4597
            errorPrint2("%s() LN%d, Failed to connect to TDengine, reason:%s\n",
4598 4599 4600 4601 4602
                    __func__, __LINE__, taos_errstr(NULL));
            free(pids);
            free(infos);
            return -1;
        }
4603

4604 4605 4606 4607 4608 4609 4610
        pThreadInfo->start_table_from = tableFrom;
        pThreadInfo->ntables = i<b?a+1:a;
        pThreadInfo->end_table_to = i < b ? tableFrom + a : tableFrom + a - 1;
        tableFrom = pThreadInfo->end_table_to + 1;
        pThreadInfo->use_metric = true;
        pThreadInfo->cols = cols;
        pThreadInfo->minDelay = UINT64_MAX;
4611
        pThreadInfo->tables_created = 0;
4612 4613
        pthread_create(pids + i, NULL, createTable, pThreadInfo);
    }
4614

4615 4616 4617
    for (int i = 0; i < threads; i++) {
        pthread_join(pids[i], NULL);
    }
4618

4619 4620 4621
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
        taos_close(pThreadInfo->taos);
4622 4623

        g_actualChildTables += pThreadInfo->tables_created;
4624 4625 4626 4627 4628 4629
    }

    free(pids);
    free(infos);

    return 0;
4630 4631
}

4632
static void createChildTables() {
4633
    char tblColsBuf[TSDB_MAX_BYTES_PER_ROW];
4634
    int len;
4635

4636 4637 4638 4639 4640
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        if (g_Dbs.use_metric) {
            if (g_Dbs.db[i].superTblCount > 0) {
                // with super table
                for (int j = 0; j < g_Dbs.db[i].superTblCount; j++) {
4641 4642 4643 4644
                    if ((AUTO_CREATE_SUBTBL
                                == g_Dbs.db[i].superTbls[j].autoCreateTable)
                            || (TBL_ALREADY_EXISTS
                                == g_Dbs.db[i].superTbls[j].childTblExists)) {
4645 4646 4647
                        continue;
                    }
                    verbosePrint("%s() LN%d: %s\n", __func__, __LINE__,
4648
                            g_Dbs.db[i].superTbls[j].colsOfCreateChildTable);
4649 4650 4651 4652 4653 4654 4655
                    uint64_t startFrom = 0;

                    verbosePrint("%s() LN%d: create %"PRId64" child tables from %"PRIu64"\n",
                            __func__, __LINE__, g_totalChildTables, startFrom);

                    startMultiThreadCreateChildTable(
                            g_Dbs.db[i].superTbls[j].colsOfCreateChildTable,
4656
                            g_Dbs.threadCountForCreateTbl,
4657 4658 4659 4660 4661 4662 4663
                            startFrom,
                            g_Dbs.db[i].superTbls[j].childTblCount,
                            g_Dbs.db[i].dbName, &(g_Dbs.db[i].superTbls[j]));
                }
            }
        } else {
            // normal table
4664
            len = snprintf(tblColsBuf, TSDB_MAX_BYTES_PER_ROW, "(TS TIMESTAMP");
4665 4666 4667
            for (int j = 0; j < g_args.columnCount; j++) {
                if ((strncasecmp(g_args.dataType[j], "BINARY", strlen("BINARY")) == 0)
                        || (strncasecmp(g_args.dataType[j],
4668
                                "NCHAR", strlen("NCHAR")) == 0)) {
4669
                    snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len,
4670
                            ",C%d %s(%d)", j, g_args.dataType[j], g_args.binwidth);
4671
                } else {
4672
                    snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len,
4673
                            ",C%d %s", j, g_args.dataType[j]);
4674 4675 4676
                }
                len = strlen(tblColsBuf);
            }
4677

4678
            snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len, ")");
4679 4680 4681

            verbosePrint("%s() LN%d: dbName: %s num of tb: %"PRId64" schema: %s\n",
                    __func__, __LINE__,
4682
                    g_Dbs.db[i].dbName, g_args.ntables, tblColsBuf);
4683 4684
            startMultiThreadCreateChildTable(
                    tblColsBuf,
4685
                    g_Dbs.threadCountForCreateTbl,
4686
                    0,
4687
                    g_args.ntables,
4688 4689
                    g_Dbs.db[i].dbName,
                    NULL);
4690
        }
4691
    }
4692 4693 4694
}

/*
4695 4696
   Read 10000 lines at most. If more than 10000 lines, continue to read after using
   */
4697
static int readTagFromCsvFileToMem(SSuperTable  * stbInfo) {
4698 4699 4700
    size_t  n = 0;
    ssize_t readLen = 0;
    char *  line = NULL;
4701

4702
    FILE *fp = fopen(stbInfo->tagsFile, "r");
4703 4704
    if (fp == NULL) {
        printf("Failed to open tags file: %s, reason:%s\n",
4705
                stbInfo->tagsFile, strerror(errno));
4706
        return -1;
4707 4708
    }

4709 4710 4711
    if (stbInfo->tagDataBuf) {
        free(stbInfo->tagDataBuf);
        stbInfo->tagDataBuf = NULL;
4712 4713
    }

4714 4715
    int tagCount = 10000;
    int count = 0;
4716
    char* tagDataBuf = calloc(1, stbInfo->lenOfTagOfOneRow * tagCount);
4717 4718
    if (tagDataBuf == NULL) {
        printf("Failed to calloc, reason:%s\n", strerror(errno));
4719 4720 4721 4722
        fclose(fp);
        return -1;
    }

4723 4724 4725 4726 4727 4728 4729 4730 4731
    while((readLen = tgetline(&line, &n, fp)) != -1) {
        if (('\r' == line[readLen - 1]) || ('\n' == line[readLen - 1])) {
            line[--readLen] = 0;
        }

        if (readLen == 0) {
            continue;
        }

4732
        memcpy(tagDataBuf + count * stbInfo->lenOfTagOfOneRow, line, readLen);
4733 4734 4735 4736
        count++;

        if (count >= tagCount - 1) {
            char *tmp = realloc(tagDataBuf,
4737
                    (size_t)tagCount*1.5*stbInfo->lenOfTagOfOneRow);
4738 4739 4740
            if (tmp != NULL) {
                tagDataBuf = tmp;
                tagCount = (int)(tagCount*1.5);
4741 4742
                memset(tagDataBuf + count*stbInfo->lenOfTagOfOneRow,
                        0, (size_t)((tagCount-count)*stbInfo->lenOfTagOfOneRow));
4743 4744
            } else {
                // exit, if allocate more memory failed
4745
                printf("realloc fail for save tag val from %s\n", stbInfo->tagsFile);
4746 4747 4748 4749 4750 4751 4752
                tmfree(tagDataBuf);
                free(line);
                fclose(fp);
                return -1;
            }
        }
    }
4753

4754 4755
    stbInfo->tagDataBuf = tagDataBuf;
    stbInfo->tagSampleCount = count;
4756 4757 4758 4759

    free(line);
    fclose(fp);
    return 0;
4760 4761 4762
}

/*
4763 4764
   Read 10000 lines at most. If more than 10000 lines, continue to read after using
   */
4765
static int generateSampleFromCsvForStb(
4766
        SSuperTable* stbInfo) {
4767 4768 4769 4770 4771
    size_t  n = 0;
    ssize_t readLen = 0;
    char *  line = NULL;
    int getRows = 0;

4772
    FILE*  fp = fopen(stbInfo->sampleFile, "r");
4773
    if (fp == NULL) {
4774
        errorPrint("Failed to open sample file: %s, reason:%s\n",
4775
                stbInfo->sampleFile, strerror(errno));
4776 4777
        return -1;
    }
4778

4779 4780
    assert(stbInfo->sampleDataBuf);
    memset(stbInfo->sampleDataBuf, 0,
4781
            MAX_SAMPLES * stbInfo->lenOfOneRow);
4782 4783 4784 4785
    while(1) {
        readLen = tgetline(&line, &n, fp);
        if (-1 == readLen) {
            if(0 != fseek(fp, 0, SEEK_SET)) {
4786
                errorPrint("Failed to fseek file: %s, reason:%s\n",
4787
                        stbInfo->sampleFile, strerror(errno));
4788 4789 4790 4791 4792
                fclose(fp);
                return -1;
            }
            continue;
        }
4793

4794 4795 4796
        if (('\r' == line[readLen - 1]) || ('\n' == line[readLen - 1])) {
            line[--readLen] = 0;
        }
4797

4798 4799 4800
        if (readLen == 0) {
            continue;
        }
4801

4802
        if (readLen > stbInfo->lenOfOneRow) {
4803
            printf("sample row len[%d] overflow define schema len[%"PRIu64"], so discard this row\n",
4804
                    (int32_t)readLen, stbInfo->lenOfOneRow);
4805 4806
            continue;
        }
4807

4808
        memcpy(stbInfo->sampleDataBuf + getRows * stbInfo->lenOfOneRow,
4809 4810 4811
                line, readLen);
        getRows++;

4812
        if (getRows == MAX_SAMPLES) {
4813 4814
            break;
        }
4815 4816
    }

4817 4818 4819
    fclose(fp);
    tmfree(line);
    return 0;
4820 4821
}

4822
static bool getColumnAndTagTypeFromInsertJsonFile(
4823
        cJSON* stbInfo, SSuperTable* superTbls) {
4824 4825 4826 4827 4828
    bool  ret = false;

    // columns
    cJSON *columns = cJSON_GetObjectItem(stbInfo, "columns");
    if (columns && columns->type != cJSON_Array) {
4829
        errorPrint("%s", "failed to read json, columns not found\n");
4830 4831 4832 4833 4834 4835
        goto PARSE_OVER;
    } else if (NULL == columns) {
        superTbls->columnCount = 0;
        superTbls->tagCount    = 0;
        return true;
    }
4836

4837 4838
    int columnSize = cJSON_GetArraySize(columns);
    if ((columnSize + 1/* ts */) > TSDB_MAX_COLUMNS) {
4839 4840
        errorPrint("failed to read json, column size overflow, max column size is %d\n",
                TSDB_MAX_COLUMNS);
4841 4842
        goto PARSE_OVER;
    }
4843

4844 4845 4846
    int count = 1;
    int index = 0;
    StrColumn    columnCase;
4847

4848 4849 4850 4851
    //superTbls->columnCount = columnSize;
    for (int k = 0; k < columnSize; ++k) {
        cJSON* column = cJSON_GetArrayItem(columns, k);
        if (column == NULL) continue;
4852

4853 4854 4855 4856 4857
        count = 1;
        cJSON* countObj = cJSON_GetObjectItem(column, "count");
        if (countObj && countObj->type == cJSON_Number) {
            count = countObj->valueint;
        } else if (countObj && countObj->type != cJSON_Number) {
4858
            errorPrint("%s", "failed to read json, column count not found\n");
4859 4860 4861 4862 4863 4864 4865 4866 4867 4868
            goto PARSE_OVER;
        } else {
            count = 1;
        }

        // column info
        memset(&columnCase, 0, sizeof(StrColumn));
        cJSON *dataType = cJSON_GetObjectItem(column, "type");
        if (!dataType || dataType->type != cJSON_String
                || dataType->valuestring == NULL) {
4869
            errorPrint("%s", "failed to read json, column type not found\n");
4870 4871
            goto PARSE_OVER;
        }
4872 4873 4874
        //tstrncpy(superTbls->columns[k].dataType, dataType->valuestring, DATATYPE_BUFF_LEN);
        tstrncpy(columnCase.dataType, dataType->valuestring,
                min(DATATYPE_BUFF_LEN, strlen(dataType->valuestring) + 1));
4875 4876 4877 4878 4879 4880 4881 4882 4883

        cJSON* dataLen = cJSON_GetObjectItem(column, "len");
        if (dataLen && dataLen->type == cJSON_Number) {
            columnCase.dataLen = dataLen->valueint;
        } else if (dataLen && dataLen->type != cJSON_Number) {
            debugPrint("%s() LN%d: failed to read json, column len not found\n",
                    __func__, __LINE__);
            goto PARSE_OVER;
        } else {
4884
            columnCase.dataLen = SMALL_BUFF_LEN;
4885 4886 4887 4888
        }

        for (int n = 0; n < count; ++n) {
            tstrncpy(superTbls->columns[index].dataType,
4889 4890
                    columnCase.dataType,
                    min(DATATYPE_BUFF_LEN, strlen(columnCase.dataType) + 1));
4891

4892 4893 4894 4895 4896 4897
            superTbls->columns[index].dataLen = columnCase.dataLen;
            index++;
        }
    }

    if ((index + 1 /* ts */) > MAX_NUM_COLUMNS) {
4898 4899
        errorPrint("failed to read json, column size overflow, allowed max column size is %d\n",
                MAX_NUM_COLUMNS);
4900 4901 4902 4903
        goto PARSE_OVER;
    }

    superTbls->columnCount = index;
4904

4905 4906 4907 4908 4909 4910 4911 4912 4913 4914 4915 4916 4917 4918 4919 4920 4921 4922 4923 4924 4925 4926 4927 4928 4929 4930 4931 4932 4933 4934 4935
    for (int c = 0; c < superTbls->columnCount; c++) {
        if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "INT", strlen("INT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_INT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "TINYINT", strlen("TINYINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_TINYINT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "SMALLINT", strlen("SMALLINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_SMALLINT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "BIGINT", strlen("BIGINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_BIGINT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "FLOAT", strlen("FLOAT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_FLOAT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "DOUBLE", strlen("DOUBLE"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_DOUBLE;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "BINARY", strlen("BINARY"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_BINARY;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "NCHAR", strlen("NCHAR"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_NCHAR;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "BOOL", strlen("BOOL"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_BOOL;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "TIMESTAMP", strlen("TIMESTAMP"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_TIMESTAMP;
4936 4937 4938 4939 4940 4941 4942 4943 4944 4945 4946 4947
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "UTINYINT", strlen("UTINYINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_UTINYINT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "USMALLINT", strlen("USMALLINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_USMALLINT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "UINT", strlen("UINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_UINT;
        } else if (0 == strncasecmp(superTbls->columns[c].dataType,
                    "UBIGINT", strlen("UBIGINT"))) {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_UBIGINT;
4948 4949 4950 4951 4952
        } else {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_NULL;
        }
    }

4953
    count = 1;
4954 4955 4956 4957
    index = 0;
    // tags
    cJSON *tags = cJSON_GetObjectItem(stbInfo, "tags");
    if (!tags || tags->type != cJSON_Array) {
4958
        errorPrint("%s", "failed to read json, tags not found\n");
4959 4960 4961 4962 4963
        goto PARSE_OVER;
    }

    int tagSize = cJSON_GetArraySize(tags);
    if (tagSize > TSDB_MAX_TAGS) {
4964 4965
        errorPrint("failed to read json, tags size overflow, max tag size is %d\n",
                TSDB_MAX_TAGS);
4966
        goto PARSE_OVER;
4967 4968
    }

4969 4970 4971 4972 4973 4974 4975 4976 4977 4978
    //superTbls->tagCount = tagSize;
    for (int k = 0; k < tagSize; ++k) {
        cJSON* tag = cJSON_GetArrayItem(tags, k);
        if (tag == NULL) continue;

        count = 1;
        cJSON* countObj = cJSON_GetObjectItem(tag, "count");
        if (countObj && countObj->type == cJSON_Number) {
            count = countObj->valueint;
        } else if (countObj && countObj->type != cJSON_Number) {
4979
            errorPrint("%s", "failed to read json, column count not found\n");
4980 4981 4982 4983 4984 4985 4986 4987 4988 4989
            goto PARSE_OVER;
        } else {
            count = 1;
        }

        // column info
        memset(&columnCase, 0, sizeof(StrColumn));
        cJSON *dataType = cJSON_GetObjectItem(tag, "type");
        if (!dataType || dataType->type != cJSON_String
                || dataType->valuestring == NULL) {
4990
            errorPrint("%s", "failed to read json, tag type not found\n");
4991 4992
            goto PARSE_OVER;
        }
4993 4994
        tstrncpy(columnCase.dataType, dataType->valuestring,
                min(DATATYPE_BUFF_LEN, strlen(dataType->valuestring) + 1));
4995 4996 4997 4998 4999

        cJSON* dataLen = cJSON_GetObjectItem(tag, "len");
        if (dataLen && dataLen->type == cJSON_Number) {
            columnCase.dataLen = dataLen->valueint;
        } else if (dataLen && dataLen->type != cJSON_Number) {
5000
            errorPrint("%s", "failed to read json, column len not found\n");
5001 5002 5003 5004 5005 5006 5007
            goto PARSE_OVER;
        } else {
            columnCase.dataLen = 0;
        }

        for (int n = 0; n < count; ++n) {
            tstrncpy(superTbls->tags[index].dataType, columnCase.dataType,
5008
                    min(DATATYPE_BUFF_LEN, strlen(columnCase.dataType) + 1));
5009 5010 5011
            superTbls->tags[index].dataLen = columnCase.dataLen;
            index++;
        }
5012
    }
5013

5014
    if (index > TSDB_MAX_TAGS) {
5015 5016
        errorPrint("failed to read json, tags size overflow, allowed max tag count is %d\n",
                TSDB_MAX_TAGS);
5017 5018
        goto PARSE_OVER;
    }
5019

5020
    superTbls->tagCount = index;
5021

5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033 5034 5035 5036 5037 5038 5039 5040 5041 5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052
    for (int t = 0; t < superTbls->tagCount; t++) {
        if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "INT", strlen("INT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_INT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "TINYINT", strlen("TINYINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_TINYINT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "SMALLINT", strlen("SMALLINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_SMALLINT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "BIGINT", strlen("BIGINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_BIGINT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "FLOAT", strlen("FLOAT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_FLOAT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "DOUBLE", strlen("DOUBLE"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_DOUBLE;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "BINARY", strlen("BINARY"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_BINARY;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "NCHAR", strlen("NCHAR"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_NCHAR;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "BOOL", strlen("BOOL"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_BOOL;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "TIMESTAMP", strlen("TIMESTAMP"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_TIMESTAMP;
5053 5054 5055 5056 5057 5058 5059 5060 5061 5062 5063 5064
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "UTINYINT", strlen("UTINYINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_UTINYINT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "USMALLINT", strlen("USMALLINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_USMALLINT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "UINT", strlen("UINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_UINT;
        } else if (0 == strncasecmp(superTbls->tags[t].dataType,
                    "UBIGINT", strlen("UBIGINT"))) {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_UBIGINT;
5065 5066 5067 5068 5069
        } else {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_NULL;
        }
    }

5070
    if ((superTbls->columnCount + superTbls->tagCount + 1 /* ts */) > TSDB_MAX_COLUMNS) {
5071 5072
        errorPrint("columns + tags is more than allowed max columns count: %d\n",
                TSDB_MAX_COLUMNS);
5073 5074 5075
        goto PARSE_OVER;
    }
    ret = true;
5076

5077
PARSE_OVER:
5078
    return ret;
5079 5080 5081
}

static bool getMetaFromInsertJsonFile(cJSON* root) {
5082
    bool  ret = false;
5083

5084 5085 5086
    cJSON* cfgdir = cJSON_GetObjectItem(root, "cfgdir");
    if (cfgdir && cfgdir->type == cJSON_String && cfgdir->valuestring != NULL) {
        tstrncpy(g_Dbs.cfgDir, cfgdir->valuestring, MAX_FILE_NAME_LEN);
5087
    }
5088

5089 5090 5091 5092 5093
    cJSON* host = cJSON_GetObjectItem(root, "host");
    if (host && host->type == cJSON_String && host->valuestring != NULL) {
        tstrncpy(g_Dbs.host, host->valuestring, MAX_HOSTNAME_SIZE);
    } else if (!host) {
        tstrncpy(g_Dbs.host, "127.0.0.1", MAX_HOSTNAME_SIZE);
5094
    } else {
5095
        errorPrint("%s", "failed to read json, host not found\n");
5096
        goto PARSE_OVER;
5097 5098
    }

5099 5100 5101 5102 5103
    cJSON* port = cJSON_GetObjectItem(root, "port");
    if (port && port->type == cJSON_Number) {
        g_Dbs.port = port->valueint;
    } else if (!port) {
        g_Dbs.port = 6030;
5104
    }
5105

5106 5107 5108 5109 5110
    cJSON* user = cJSON_GetObjectItem(root, "user");
    if (user && user->type == cJSON_String && user->valuestring != NULL) {
        tstrncpy(g_Dbs.user, user->valuestring, MAX_USERNAME_SIZE);
    } else if (!user) {
        tstrncpy(g_Dbs.user, "root", MAX_USERNAME_SIZE);
5111 5112
    }

5113 5114
    cJSON* password = cJSON_GetObjectItem(root, "password");
    if (password && password->type == cJSON_String && password->valuestring != NULL) {
5115
        tstrncpy(g_Dbs.password, password->valuestring, SHELL_MAX_PASSWORD_LEN);
5116
    } else if (!password) {
5117
        tstrncpy(g_Dbs.password, "taosdata", SHELL_MAX_PASSWORD_LEN);
5118 5119
    }

5120 5121 5122 5123 5124
    cJSON* resultfile = cJSON_GetObjectItem(root, "result_file");
    if (resultfile && resultfile->type == cJSON_String && resultfile->valuestring != NULL) {
        tstrncpy(g_Dbs.resultFile, resultfile->valuestring, MAX_FILE_NAME_LEN);
    } else if (!resultfile) {
        tstrncpy(g_Dbs.resultFile, "./insert_res.txt", MAX_FILE_NAME_LEN);
5125 5126
    }

5127 5128 5129 5130 5131
    cJSON* threads = cJSON_GetObjectItem(root, "thread_count");
    if (threads && threads->type == cJSON_Number) {
        g_Dbs.threadCount = threads->valueint;
    } else if (!threads) {
        g_Dbs.threadCount = 1;
5132
    } else {
5133
        errorPrint("%s", "failed to read json, threads not found\n");
5134
        goto PARSE_OVER;
5135
    }
5136

5137 5138
    cJSON* threads2 = cJSON_GetObjectItem(root, "thread_count_create_tbl");
    if (threads2 && threads2->type == cJSON_Number) {
5139
        g_Dbs.threadCountForCreateTbl = threads2->valueint;
5140
    } else if (!threads2) {
5141
        g_Dbs.threadCountForCreateTbl = 1;
5142
    } else {
5143
        errorPrint("%s", "failed to read json, threads2 not found\n");
5144
        goto PARSE_OVER;
5145
    }
5146

5147 5148 5149
    cJSON* gInsertInterval = cJSON_GetObjectItem(root, "insert_interval");
    if (gInsertInterval && gInsertInterval->type == cJSON_Number) {
        if (gInsertInterval->valueint <0) {
5150
            errorPrint("%s", "failed to read json, insert interval input mistake\n");
5151 5152 5153 5154 5155
            goto PARSE_OVER;
        }
        g_args.insert_interval = gInsertInterval->valueint;
    } else if (!gInsertInterval) {
        g_args.insert_interval = 0;
5156
    } else {
5157
        errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5158
        goto PARSE_OVER;
5159
    }
5160

5161 5162 5163
    cJSON* interlaceRows = cJSON_GetObjectItem(root, "interlace_rows");
    if (interlaceRows && interlaceRows->type == cJSON_Number) {
        if (interlaceRows->valueint < 0) {
5164
            errorPrint("%s", "failed to read json, interlaceRows input mistake\n");
5165
            goto PARSE_OVER;
5166

5167
        }
5168
        g_args.interlaceRows = interlaceRows->valueint;
5169
    } else if (!interlaceRows) {
5170
        g_args.interlaceRows = 0; // 0 means progressive mode, > 0 mean interlace mode. max value is less or equ num_of_records_per_req
5171
    } else {
5172
        errorPrint("%s", "failed to read json, interlaceRows input mistake\n");
5173
        goto PARSE_OVER;
5174 5175
    }

5176 5177 5178 5179 5180 5181 5182 5183 5184 5185
    cJSON* maxSqlLen = cJSON_GetObjectItem(root, "max_sql_len");
    if (maxSqlLen && maxSqlLen->type == cJSON_Number) {
        if (maxSqlLen->valueint < 0) {
            errorPrint("%s() LN%d, failed to read json, max_sql_len input mistake\n",
                    __func__, __LINE__);
            goto PARSE_OVER;
        }
        g_args.max_sql_len = maxSqlLen->valueint;
    } else if (!maxSqlLen) {
        g_args.max_sql_len = (1024*1024);
5186
    } else {
5187 5188 5189
        errorPrint("%s() LN%d, failed to read json, max_sql_len input mistake\n",
                __func__, __LINE__);
        goto PARSE_OVER;
5190 5191
    }

5192 5193 5194 5195 5196 5197 5198 5199 5200 5201 5202 5203 5204 5205
    cJSON* numRecPerReq = cJSON_GetObjectItem(root, "num_of_records_per_req");
    if (numRecPerReq && numRecPerReq->type == cJSON_Number) {
        if (numRecPerReq->valueint <= 0) {
            errorPrint("%s() LN%d, failed to read json, num_of_records_per_req input mistake\n",
                    __func__, __LINE__);
            goto PARSE_OVER;
        } else if (numRecPerReq->valueint > MAX_RECORDS_PER_REQ) {
            printf("NOTICE: number of records per request value %"PRIu64" > %d\n\n",
                    numRecPerReq->valueint, MAX_RECORDS_PER_REQ);
            printf("        number of records per request value will be set to %d\n\n",
                    MAX_RECORDS_PER_REQ);
            prompt();
            numRecPerReq->valueint = MAX_RECORDS_PER_REQ;
        }
5206
        g_args.reqPerReq = numRecPerReq->valueint;
5207
    } else if (!numRecPerReq) {
5208
        g_args.reqPerReq = MAX_RECORDS_PER_REQ;
5209
    } else {
5210 5211 5212
        errorPrint("%s() LN%d, failed to read json, num_of_records_per_req not found\n",
                __func__, __LINE__);
        goto PARSE_OVER;
5213 5214
    }

5215 5216 5217 5218 5219 5220 5221 5222 5223 5224 5225 5226 5227
    cJSON *answerPrompt = cJSON_GetObjectItem(root, "confirm_parameter_prompt"); // yes, no,
    if (answerPrompt
            && answerPrompt->type == cJSON_String
            && answerPrompt->valuestring != NULL) {
        if (0 == strncasecmp(answerPrompt->valuestring, "yes", 3)) {
            g_args.answer_yes = false;
        } else if (0 == strncasecmp(answerPrompt->valuestring, "no", 2)) {
            g_args.answer_yes = true;
        } else {
            g_args.answer_yes = false;
        }
    } else if (!answerPrompt) {
        g_args.answer_yes = true;   // default is no, mean answer_yes.
5228
    } else {
5229 5230
        errorPrint("%s", "failed to read json, confirm_parameter_prompt input mistake\n");
        goto PARSE_OVER;
5231 5232
    }

5233
    // rows per table need be less than insert batch
5234
    if (g_args.interlaceRows > g_args.reqPerReq) {
5235
        printf("NOTICE: interlace rows value %u > num_of_records_per_req %u\n\n",
5236
                g_args.interlaceRows, g_args.reqPerReq);
5237
        printf("        interlace rows value will be set to num_of_records_per_req %u\n\n",
5238
                g_args.reqPerReq);
5239
        prompt();
5240
        g_args.interlaceRows = g_args.reqPerReq;
5241 5242
    }

5243 5244
    cJSON* dbs = cJSON_GetObjectItem(root, "databases");
    if (!dbs || dbs->type != cJSON_Array) {
5245
        errorPrint("%s", "failed to read json, databases not found\n");
5246
        goto PARSE_OVER;
5247
    }
5248

5249 5250 5251
    int dbSize = cJSON_GetArraySize(dbs);
    if (dbSize > MAX_DB_COUNT) {
        errorPrint(
5252
                "failed to read json, databases size overflow, max database is %d\n",
5253 5254
                MAX_DB_COUNT);
        goto PARSE_OVER;
5255 5256
    }

5257 5258 5259 5260
    g_Dbs.dbCount = dbSize;
    for (int i = 0; i < dbSize; ++i) {
        cJSON* dbinfos = cJSON_GetArrayItem(dbs, i);
        if (dbinfos == NULL) continue;
5261

5262 5263 5264
        // dbinfo
        cJSON *dbinfo = cJSON_GetObjectItem(dbinfos, "dbinfo");
        if (!dbinfo || dbinfo->type != cJSON_Object) {
5265
            errorPrint("%s", "failed to read json, dbinfo not found\n");
5266 5267
            goto PARSE_OVER;
        }
5268

5269 5270
        cJSON *dbName = cJSON_GetObjectItem(dbinfo, "name");
        if (!dbName || dbName->type != cJSON_String || dbName->valuestring == NULL) {
5271
            errorPrint("%s", "failed to read json, db name not found\n");
5272 5273 5274
            goto PARSE_OVER;
        }
        tstrncpy(g_Dbs.db[i].dbName, dbName->valuestring, TSDB_DB_NAME_LEN);
5275

5276 5277 5278 5279 5280 5281 5282 5283 5284 5285
        cJSON *drop = cJSON_GetObjectItem(dbinfo, "drop");
        if (drop && drop->type == cJSON_String && drop->valuestring != NULL) {
            if (0 == strncasecmp(drop->valuestring, "yes", strlen("yes"))) {
                g_Dbs.db[i].drop = true;
            } else {
                g_Dbs.db[i].drop = false;
            }
        } else if (!drop) {
            g_Dbs.db[i].drop = g_args.drop_database;
        } else {
5286
            errorPrint("%s", "failed to read json, drop input mistake\n");
5287 5288
            goto PARSE_OVER;
        }
5289

5290 5291 5292 5293
        cJSON *precision = cJSON_GetObjectItem(dbinfo, "precision");
        if (precision && precision->type == cJSON_String
                && precision->valuestring != NULL) {
            tstrncpy(g_Dbs.db[i].dbCfg.precision, precision->valuestring,
5294
                    SMALL_BUFF_LEN);
5295
        } else if (!precision) {
5296
            memset(g_Dbs.db[i].dbCfg.precision, 0, SMALL_BUFF_LEN);
5297
        } else {
5298
            errorPrint("%s", "failed to read json, precision not found\n");
5299 5300
            goto PARSE_OVER;
        }
5301

5302 5303 5304 5305 5306
        cJSON* update = cJSON_GetObjectItem(dbinfo, "update");
        if (update && update->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.update = update->valueint;
        } else if (!update) {
            g_Dbs.db[i].dbCfg.update = -1;
5307
        } else {
5308
            errorPrint("%s", "failed to read json, update not found\n");
5309
            goto PARSE_OVER;
5310
        }
5311

5312 5313 5314 5315 5316 5317
        cJSON* replica = cJSON_GetObjectItem(dbinfo, "replica");
        if (replica && replica->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.replica = replica->valueint;
        } else if (!replica) {
            g_Dbs.db[i].dbCfg.replica = -1;
        } else {
5318
            errorPrint("%s", "failed to read json, replica not found\n");
5319 5320
            goto PARSE_OVER;
        }
5321

5322 5323 5324 5325 5326 5327
        cJSON* keep = cJSON_GetObjectItem(dbinfo, "keep");
        if (keep && keep->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.keep = keep->valueint;
        } else if (!keep) {
            g_Dbs.db[i].dbCfg.keep = -1;
        } else {
5328
            errorPrint("%s", "failed to read json, keep not found\n");
5329 5330
            goto PARSE_OVER;
        }
5331

5332 5333 5334 5335 5336
        cJSON* days = cJSON_GetObjectItem(dbinfo, "days");
        if (days && days->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.days = days->valueint;
        } else if (!days) {
            g_Dbs.db[i].dbCfg.days = -1;
5337
        } else {
5338
            errorPrint("%s", "failed to read json, days not found\n");
5339 5340
            goto PARSE_OVER;
        }
5341

5342 5343 5344 5345 5346 5347
        cJSON* cache = cJSON_GetObjectItem(dbinfo, "cache");
        if (cache && cache->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.cache = cache->valueint;
        } else if (!cache) {
            g_Dbs.db[i].dbCfg.cache = -1;
        } else {
5348
            errorPrint("%s", "failed to read json, cache not found\n");
5349
            goto PARSE_OVER;
5350 5351
        }

5352 5353 5354 5355 5356 5357
        cJSON* blocks= cJSON_GetObjectItem(dbinfo, "blocks");
        if (blocks && blocks->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.blocks = blocks->valueint;
        } else if (!blocks) {
            g_Dbs.db[i].dbCfg.blocks = -1;
        } else {
5358
            errorPrint("%s", "failed to read json, block not found\n");
5359
            goto PARSE_OVER;
5360 5361
        }

5362 5363 5364 5365 5366 5367 5368 5369 5370 5371 5372 5373 5374 5375 5376 5377
        //cJSON* maxtablesPerVnode= cJSON_GetObjectItem(dbinfo, "maxtablesPerVnode");
        //if (maxtablesPerVnode && maxtablesPerVnode->type == cJSON_Number) {
        //  g_Dbs.db[i].dbCfg.maxtablesPerVnode = maxtablesPerVnode->valueint;
        //} else if (!maxtablesPerVnode) {
        //  g_Dbs.db[i].dbCfg.maxtablesPerVnode = TSDB_DEFAULT_TABLES;
        //} else {
        // printf("failed to read json, maxtablesPerVnode not found");
        // goto PARSE_OVER;
        //}

        cJSON* minRows= cJSON_GetObjectItem(dbinfo, "minRows");
        if (minRows && minRows->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.minRows = minRows->valueint;
        } else if (!minRows) {
            g_Dbs.db[i].dbCfg.minRows = 0;    // 0 means default
        } else {
5378
            errorPrint("%s", "failed to read json, minRows not found\n");
5379 5380
            goto PARSE_OVER;
        }
5381

5382 5383 5384 5385 5386 5387
        cJSON* maxRows= cJSON_GetObjectItem(dbinfo, "maxRows");
        if (maxRows && maxRows->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.maxRows = maxRows->valueint;
        } else if (!maxRows) {
            g_Dbs.db[i].dbCfg.maxRows = 0;    // 0 means default
        } else {
5388
            errorPrint("%s", "failed to read json, maxRows not found\n");
5389 5390
            goto PARSE_OVER;
        }
5391

5392 5393 5394 5395 5396 5397
        cJSON* comp= cJSON_GetObjectItem(dbinfo, "comp");
        if (comp && comp->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.comp = comp->valueint;
        } else if (!comp) {
            g_Dbs.db[i].dbCfg.comp = -1;
        } else {
5398
            errorPrint("%s", "failed to read json, comp not found\n");
5399 5400
            goto PARSE_OVER;
        }
5401

5402 5403 5404 5405 5406 5407
        cJSON* walLevel= cJSON_GetObjectItem(dbinfo, "walLevel");
        if (walLevel && walLevel->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.walLevel = walLevel->valueint;
        } else if (!walLevel) {
            g_Dbs.db[i].dbCfg.walLevel = -1;
        } else {
5408
            errorPrint("%s", "failed to read json, walLevel not found\n");
5409 5410
            goto PARSE_OVER;
        }
5411

5412 5413 5414 5415 5416
        cJSON* cacheLast= cJSON_GetObjectItem(dbinfo, "cachelast");
        if (cacheLast && cacheLast->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.cacheLast = cacheLast->valueint;
        } else if (!cacheLast) {
            g_Dbs.db[i].dbCfg.cacheLast = -1;
5417
        } else {
5418
            errorPrint("%s", "failed to read json, cacheLast not found\n");
5419
            goto PARSE_OVER;
5420
        }
5421

5422 5423 5424 5425 5426
        cJSON* quorum= cJSON_GetObjectItem(dbinfo, "quorum");
        if (quorum && quorum->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.quorum = quorum->valueint;
        } else if (!quorum) {
            g_Dbs.db[i].dbCfg.quorum = 1;
5427
        } else {
5428 5429
            printf("failed to read json, quorum input mistake");
            goto PARSE_OVER;
5430
        }
5431 5432 5433 5434 5435 5436 5437

        cJSON* fsync= cJSON_GetObjectItem(dbinfo, "fsync");
        if (fsync && fsync->type == cJSON_Number) {
            g_Dbs.db[i].dbCfg.fsync = fsync->valueint;
        } else if (!fsync) {
            g_Dbs.db[i].dbCfg.fsync = -1;
        } else {
5438
            errorPrint("%s", "failed to read json, fsync input mistake\n");
5439
            goto PARSE_OVER;
5440 5441
        }

5442
        // super_tables
5443 5444
        cJSON *stables = cJSON_GetObjectItem(dbinfos, "super_tables");
        if (!stables || stables->type != cJSON_Array) {
5445
            errorPrint("%s", "failed to read json, super_tables not found\n");
5446
            goto PARSE_OVER;
5447
        }
5448

5449 5450 5451
        int stbSize = cJSON_GetArraySize(stables);
        if (stbSize > MAX_SUPER_TABLE_COUNT) {
            errorPrint(
5452 5453
                    "failed to read json, supertable size overflow, max supertable is %d\n",
                    MAX_SUPER_TABLE_COUNT);
5454
            goto PARSE_OVER;
5455
        }
5456

5457 5458 5459 5460
        g_Dbs.db[i].superTblCount = stbSize;
        for (int j = 0; j < stbSize; ++j) {
            cJSON* stbInfo = cJSON_GetArrayItem(stables, j);
            if (stbInfo == NULL) continue;
5461

5462 5463 5464 5465
            // dbinfo
            cJSON *stbName = cJSON_GetObjectItem(stbInfo, "name");
            if (!stbName || stbName->type != cJSON_String
                    || stbName->valuestring == NULL) {
5466
                errorPrint("%s", "failed to read json, stb name not found\n");
5467 5468
                goto PARSE_OVER;
            }
5469
            tstrncpy(g_Dbs.db[i].superTbls[j].stbName, stbName->valuestring,
5470
                    TSDB_TABLE_NAME_LEN);
5471

5472 5473
            cJSON *prefix = cJSON_GetObjectItem(stbInfo, "childtable_prefix");
            if (!prefix || prefix->type != cJSON_String || prefix->valuestring == NULL) {
5474
                errorPrint("%s", "failed to read json, childtable_prefix not found\n");
5475 5476 5477
                goto PARSE_OVER;
            }
            tstrncpy(g_Dbs.db[i].superTbls[j].childTblPrefix, prefix->valuestring,
5478
                    TBNAME_PREFIX_LEN);
5479 5480 5481 5482 5483 5484 5485 5486 5487 5488 5489 5490 5491 5492 5493 5494

            cJSON *autoCreateTbl = cJSON_GetObjectItem(stbInfo, "auto_create_table");
            if (autoCreateTbl
                    && autoCreateTbl->type == cJSON_String
                    && autoCreateTbl->valuestring != NULL) {
                if ((0 == strncasecmp(autoCreateTbl->valuestring, "yes", 3))
                        && (TBL_ALREADY_EXISTS != g_Dbs.db[i].superTbls[j].childTblExists)) {
                    g_Dbs.db[i].superTbls[j].autoCreateTable = AUTO_CREATE_SUBTBL;
                } else if (0 == strncasecmp(autoCreateTbl->valuestring, "no", 2)) {
                    g_Dbs.db[i].superTbls[j].autoCreateTable = PRE_CREATE_SUBTBL;
                } else {
                    g_Dbs.db[i].superTbls[j].autoCreateTable = PRE_CREATE_SUBTBL;
                }
            } else if (!autoCreateTbl) {
                g_Dbs.db[i].superTbls[j].autoCreateTable = PRE_CREATE_SUBTBL;
            } else {
5495
                errorPrint("%s", "failed to read json, auto_create_table not found\n");
5496 5497
                goto PARSE_OVER;
            }
5498

5499 5500 5501 5502
            cJSON* batchCreateTbl = cJSON_GetObjectItem(stbInfo, "batch_create_tbl_num");
            if (batchCreateTbl && batchCreateTbl->type == cJSON_Number) {
                g_Dbs.db[i].superTbls[j].batchCreateTableNum = batchCreateTbl->valueint;
            } else if (!batchCreateTbl) {
5503
                g_Dbs.db[i].superTbls[j].batchCreateTableNum = 10;
5504
            } else {
5505
                errorPrint("%s", "failed to read json, batch_create_tbl_num not found\n");
5506 5507 5508 5509 5510 5511 5512 5513 5514 5515 5516 5517 5518 5519 5520 5521 5522 5523 5524
                goto PARSE_OVER;
            }

            cJSON *childTblExists = cJSON_GetObjectItem(stbInfo, "child_table_exists"); // yes, no
            if (childTblExists
                    && childTblExists->type == cJSON_String
                    && childTblExists->valuestring != NULL) {
                if ((0 == strncasecmp(childTblExists->valuestring, "yes", 3))
                        && (g_Dbs.db[i].drop == false)) {
                    g_Dbs.db[i].superTbls[j].childTblExists = TBL_ALREADY_EXISTS;
                } else if ((0 == strncasecmp(childTblExists->valuestring, "no", 2)
                            || (g_Dbs.db[i].drop == true))) {
                    g_Dbs.db[i].superTbls[j].childTblExists = TBL_NO_EXISTS;
                } else {
                    g_Dbs.db[i].superTbls[j].childTblExists = TBL_NO_EXISTS;
                }
            } else if (!childTblExists) {
                g_Dbs.db[i].superTbls[j].childTblExists = TBL_NO_EXISTS;
            } else {
5525 5526
                errorPrint("%s",
                        "failed to read json, child_table_exists not found\n");
5527 5528 5529 5530 5531 5532 5533 5534 5535
                goto PARSE_OVER;
            }

            if (TBL_ALREADY_EXISTS == g_Dbs.db[i].superTbls[j].childTblExists) {
                g_Dbs.db[i].superTbls[j].autoCreateTable = PRE_CREATE_SUBTBL;
            }

            cJSON* count = cJSON_GetObjectItem(stbInfo, "childtable_count");
            if (!count || count->type != cJSON_Number || 0 >= count->valueint) {
5536 5537
                errorPrint("%s",
                        "failed to read json, childtable_count input mistake\n");
5538 5539 5540
                goto PARSE_OVER;
            }
            g_Dbs.db[i].superTbls[j].childTblCount = count->valueint;
5541
            g_totalChildTables += g_Dbs.db[i].superTbls[j].childTblCount;
5542 5543 5544 5545 5546

            cJSON *dataSource = cJSON_GetObjectItem(stbInfo, "data_source");
            if (dataSource && dataSource->type == cJSON_String
                    && dataSource->valuestring != NULL) {
                tstrncpy(g_Dbs.db[i].superTbls[j].dataSource,
5547 5548
                        dataSource->valuestring,
                        min(SMALL_BUFF_LEN, strlen(dataSource->valuestring) + 1));
5549
            } else if (!dataSource) {
5550 5551
                tstrncpy(g_Dbs.db[i].superTbls[j].dataSource, "rand",
                        min(SMALL_BUFF_LEN, strlen("rand") + 1));
5552
            } else {
5553
                errorPrint("%s", "failed to read json, data_source not found\n");
5554 5555 5556 5557 5558 5559 5560 5561 5562 5563 5564 5565 5566
                goto PARSE_OVER;
            }

            cJSON *stbIface = cJSON_GetObjectItem(stbInfo, "insert_mode"); // taosc , rest, stmt
            if (stbIface && stbIface->type == cJSON_String
                    && stbIface->valuestring != NULL) {
                if (0 == strcasecmp(stbIface->valuestring, "taosc")) {
                    g_Dbs.db[i].superTbls[j].iface= TAOSC_IFACE;
                } else if (0 == strcasecmp(stbIface->valuestring, "rest")) {
                    g_Dbs.db[i].superTbls[j].iface= REST_IFACE;
                } else if (0 == strcasecmp(stbIface->valuestring, "stmt")) {
                    g_Dbs.db[i].superTbls[j].iface= STMT_IFACE;
                } else {
5567 5568
                    errorPrint("failed to read json, insert_mode %s not recognized\n",
                            stbIface->valuestring);
5569 5570 5571 5572 5573 5574 5575 5576 5577 5578 5579 5580 5581
                    goto PARSE_OVER;
                }
            } else if (!stbIface) {
                g_Dbs.db[i].superTbls[j].iface = TAOSC_IFACE;
            } else {
                errorPrint("%s", "failed to read json, insert_mode not found\n");
                goto PARSE_OVER;
            }

            cJSON* childTbl_limit = cJSON_GetObjectItem(stbInfo, "childtable_limit");
            if ((childTbl_limit) && (g_Dbs.db[i].drop != true)
                    && (g_Dbs.db[i].superTbls[j].childTblExists == TBL_ALREADY_EXISTS)) {
                if (childTbl_limit->type != cJSON_Number) {
5582
                    errorPrint("%s", "failed to read json, childtable_limit\n");
5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594
                    goto PARSE_OVER;
                }
                g_Dbs.db[i].superTbls[j].childTblLimit = childTbl_limit->valueint;
            } else {
                g_Dbs.db[i].superTbls[j].childTblLimit = -1;    // select ... limit -1 means all query result, drop = yes mean all table need recreate, limit value is invalid.
            }

            cJSON* childTbl_offset = cJSON_GetObjectItem(stbInfo, "childtable_offset");
            if ((childTbl_offset) && (g_Dbs.db[i].drop != true)
                    && (g_Dbs.db[i].superTbls[j].childTblExists == TBL_ALREADY_EXISTS)) {
                if ((childTbl_offset->type != cJSON_Number)
                        || (0 > childTbl_offset->valueint)) {
5595
                    errorPrint("%s", "failed to read json, childtable_offset\n");
5596 5597 5598 5599 5600 5601 5602 5603 5604 5605 5606 5607 5608 5609 5610
                    goto PARSE_OVER;
                }
                g_Dbs.db[i].superTbls[j].childTblOffset = childTbl_offset->valueint;
            } else {
                g_Dbs.db[i].superTbls[j].childTblOffset = 0;
            }

            cJSON *ts = cJSON_GetObjectItem(stbInfo, "start_timestamp");
            if (ts && ts->type == cJSON_String && ts->valuestring != NULL) {
                tstrncpy(g_Dbs.db[i].superTbls[j].startTimestamp,
                        ts->valuestring, TSDB_DB_NAME_LEN);
            } else if (!ts) {
                tstrncpy(g_Dbs.db[i].superTbls[j].startTimestamp,
                        "now", TSDB_DB_NAME_LEN);
            } else {
5611
                errorPrint("%s", "failed to read json, start_timestamp not found\n");
5612 5613 5614 5615 5616 5617 5618
                goto PARSE_OVER;
            }

            cJSON* timestampStep = cJSON_GetObjectItem(stbInfo, "timestamp_step");
            if (timestampStep && timestampStep->type == cJSON_Number) {
                g_Dbs.db[i].superTbls[j].timeStampStep = timestampStep->valueint;
            } else if (!timestampStep) {
5619
                g_Dbs.db[i].superTbls[j].timeStampStep = g_args.timestamp_step;
5620
            } else {
5621
                errorPrint("%s", "failed to read json, timestamp_step not found\n");
5622 5623 5624 5625 5626 5627 5628
                goto PARSE_OVER;
            }

            cJSON *sampleFormat = cJSON_GetObjectItem(stbInfo, "sample_format");
            if (sampleFormat && sampleFormat->type
                    == cJSON_String && sampleFormat->valuestring != NULL) {
                tstrncpy(g_Dbs.db[i].superTbls[j].sampleFormat,
5629 5630 5631
                        sampleFormat->valuestring,
                        min(SMALL_BUFF_LEN,
                            strlen(sampleFormat->valuestring) + 1));
5632
            } else if (!sampleFormat) {
5633 5634
                tstrncpy(g_Dbs.db[i].superTbls[j].sampleFormat, "csv",
                        SMALL_BUFF_LEN);
5635
            } else {
5636
                errorPrint("%s", "failed to read json, sample_format not found\n");
5637 5638 5639 5640 5641 5642 5643
                goto PARSE_OVER;
            }

            cJSON *sampleFile = cJSON_GetObjectItem(stbInfo, "sample_file");
            if (sampleFile && sampleFile->type == cJSON_String
                    && sampleFile->valuestring != NULL) {
                tstrncpy(g_Dbs.db[i].superTbls[j].sampleFile,
5644 5645 5646
                        sampleFile->valuestring,
                        min(MAX_FILE_NAME_LEN,
                            strlen(sampleFile->valuestring) + 1));
5647
            } else if (!sampleFile) {
5648 5649
                memset(g_Dbs.db[i].superTbls[j].sampleFile, 0,
                        MAX_FILE_NAME_LEN);
5650
            } else {
5651
                errorPrint("%s", "failed to read json, sample_file not found\n");
5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663 5664 5665 5666 5667 5668
                goto PARSE_OVER;
            }

            cJSON *tagsFile = cJSON_GetObjectItem(stbInfo, "tags_file");
            if ((tagsFile && tagsFile->type == cJSON_String)
                    && (tagsFile->valuestring != NULL)) {
                tstrncpy(g_Dbs.db[i].superTbls[j].tagsFile,
                        tagsFile->valuestring, MAX_FILE_NAME_LEN);
                if (0 == g_Dbs.db[i].superTbls[j].tagsFile[0]) {
                    g_Dbs.db[i].superTbls[j].tagSource = 0;
                } else {
                    g_Dbs.db[i].superTbls[j].tagSource = 1;
                }
            } else if (!tagsFile) {
                memset(g_Dbs.db[i].superTbls[j].tagsFile, 0, MAX_FILE_NAME_LEN);
                g_Dbs.db[i].superTbls[j].tagSource = 0;
            } else {
5669
                errorPrint("%s", "failed to read json, tags_file not found\n");
5670 5671 5672 5673 5674 5675 5676 5677 5678 5679 5680 5681 5682 5683 5684
                goto PARSE_OVER;
            }

            cJSON* stbMaxSqlLen = cJSON_GetObjectItem(stbInfo, "max_sql_len");
            if (stbMaxSqlLen && stbMaxSqlLen->type == cJSON_Number) {
                int32_t len = stbMaxSqlLen->valueint;
                if (len > TSDB_MAX_ALLOWED_SQL_LEN) {
                    len = TSDB_MAX_ALLOWED_SQL_LEN;
                } else if (len < 5) {
                    len = 5;
                }
                g_Dbs.db[i].superTbls[j].maxSqlLen = len;
            } else if (!maxSqlLen) {
                g_Dbs.db[i].superTbls[j].maxSqlLen = g_args.max_sql_len;
            } else {
5685
                errorPrint("%s", "failed to read json, stbMaxSqlLen input mistake\n");
5686 5687 5688 5689 5690 5691 5692 5693 5694 5695 5696 5697 5698 5699 5700 5701
                goto PARSE_OVER;
            }
            /*
               cJSON *multiThreadWriteOneTbl =
               cJSON_GetObjectItem(stbInfo, "multi_thread_write_one_tbl"); // no , yes
               if (multiThreadWriteOneTbl
               && multiThreadWriteOneTbl->type == cJSON_String
               && multiThreadWriteOneTbl->valuestring != NULL) {
               if (0 == strncasecmp(multiThreadWriteOneTbl->valuestring, "yes", 3)) {
               g_Dbs.db[i].superTbls[j].multiThreadWriteOneTbl = 1;
               } else {
               g_Dbs.db[i].superTbls[j].multiThreadWriteOneTbl = 0;
               }
               } else if (!multiThreadWriteOneTbl) {
               g_Dbs.db[i].superTbls[j].multiThreadWriteOneTbl = 0;
               } else {
5702
               errorPrint("%s", "failed to read json, multiThreadWriteOneTbl not found\n");
5703 5704 5705 5706 5707 5708
               goto PARSE_OVER;
               }
               */
            cJSON* insertRows = cJSON_GetObjectItem(stbInfo, "insert_rows");
            if (insertRows && insertRows->type == cJSON_Number) {
                if (insertRows->valueint < 0) {
5709
                    errorPrint("%s", "failed to read json, insert_rows input mistake\n");
5710 5711 5712 5713 5714 5715
                    goto PARSE_OVER;
                }
                g_Dbs.db[i].superTbls[j].insertRows = insertRows->valueint;
            } else if (!insertRows) {
                g_Dbs.db[i].superTbls[j].insertRows = 0x7FFFFFFFFFFFFFFF;
            } else {
5716
                errorPrint("%s", "failed to read json, insert_rows input mistake\n");
5717 5718 5719 5720 5721 5722
                goto PARSE_OVER;
            }

            cJSON* stbInterlaceRows = cJSON_GetObjectItem(stbInfo, "interlace_rows");
            if (stbInterlaceRows && stbInterlaceRows->type == cJSON_Number) {
                if (stbInterlaceRows->valueint < 0) {
5723
                    errorPrint("%s", "failed to read json, interlace rows input mistake\n");
5724 5725 5726 5727 5728 5729 5730 5731 5732 5733 5734 5735 5736 5737
                    goto PARSE_OVER;
                }
                g_Dbs.db[i].superTbls[j].interlaceRows = stbInterlaceRows->valueint;

                if (g_Dbs.db[i].superTbls[j].interlaceRows > g_Dbs.db[i].superTbls[j].insertRows) {
                    printf("NOTICE: db[%d].superTbl[%d]'s interlace rows value %u > insert_rows %"PRId64"\n\n",
                            i, j, g_Dbs.db[i].superTbls[j].interlaceRows,
                            g_Dbs.db[i].superTbls[j].insertRows);
                    printf("        interlace rows value will be set to insert_rows %"PRId64"\n\n",
                            g_Dbs.db[i].superTbls[j].insertRows);
                    prompt();
                    g_Dbs.db[i].superTbls[j].interlaceRows = g_Dbs.db[i].superTbls[j].insertRows;
                }
            } else if (!stbInterlaceRows) {
5738
                g_Dbs.db[i].superTbls[j].interlaceRows = g_args.interlaceRows; // 0 means progressive mode, > 0 mean interlace mode. max value is less or equ num_of_records_per_req
5739 5740
            } else {
                errorPrint(
5741
                        "%s", "failed to read json, interlace rows input mistake\n");
5742 5743 5744 5745 5746 5747 5748 5749 5750 5751 5752 5753 5754 5755 5756
                goto PARSE_OVER;
            }

            cJSON* disorderRatio = cJSON_GetObjectItem(stbInfo, "disorder_ratio");
            if (disorderRatio && disorderRatio->type == cJSON_Number) {
                if (disorderRatio->valueint > 50)
                    disorderRatio->valueint = 50;

                if (disorderRatio->valueint < 0)
                    disorderRatio->valueint = 0;

                g_Dbs.db[i].superTbls[j].disorderRatio = disorderRatio->valueint;
            } else if (!disorderRatio) {
                g_Dbs.db[i].superTbls[j].disorderRatio = 0;
            } else {
5757
                errorPrint("%s", "failed to read json, disorderRatio not found\n");
5758 5759 5760 5761 5762 5763 5764 5765 5766
                goto PARSE_OVER;
            }

            cJSON* disorderRange = cJSON_GetObjectItem(stbInfo, "disorder_range");
            if (disorderRange && disorderRange->type == cJSON_Number) {
                g_Dbs.db[i].superTbls[j].disorderRange = disorderRange->valueint;
            } else if (!disorderRange) {
                g_Dbs.db[i].superTbls[j].disorderRange = 1000;
            } else {
5767
                errorPrint("%s", "failed to read json, disorderRange not found\n");
5768 5769 5770 5771 5772 5773 5774
                goto PARSE_OVER;
            }

            cJSON* insertInterval = cJSON_GetObjectItem(stbInfo, "insert_interval");
            if (insertInterval && insertInterval->type == cJSON_Number) {
                g_Dbs.db[i].superTbls[j].insertInterval = insertInterval->valueint;
                if (insertInterval->valueint < 0) {
5775
                    errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5776 5777 5778
                    goto PARSE_OVER;
                }
            } else if (!insertInterval) {
5779
                verbosePrint("%s() LN%d: stable insert interval be overrode by global %"PRIu64".\n",
5780 5781 5782
                        __func__, __LINE__, g_args.insert_interval);
                g_Dbs.db[i].superTbls[j].insertInterval = g_args.insert_interval;
            } else {
5783
                errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5784 5785
                goto PARSE_OVER;
            }
5786

5787 5788 5789 5790 5791
            int retVal = getColumnAndTagTypeFromInsertJsonFile(
                    stbInfo, &g_Dbs.db[i].superTbls[j]);
            if (false == retVal) {
                goto PARSE_OVER;
            }
5792
        }
5793
    }
5794

5795
    ret = true;
5796

5797
PARSE_OVER:
5798
    return ret;
5799 5800 5801
}

static bool getMetaFromQueryJsonFile(cJSON* root) {
5802
    bool  ret = false;
5803

5804 5805 5806 5807
    cJSON* cfgdir = cJSON_GetObjectItem(root, "cfgdir");
    if (cfgdir && cfgdir->type == cJSON_String && cfgdir->valuestring != NULL) {
        tstrncpy(g_queryInfo.cfgDir, cfgdir->valuestring, MAX_FILE_NAME_LEN);
    }
H
hzcheng 已提交
5808

5809 5810 5811 5812 5813
    cJSON* host = cJSON_GetObjectItem(root, "host");
    if (host && host->type == cJSON_String && host->valuestring != NULL) {
        tstrncpy(g_queryInfo.host, host->valuestring, MAX_HOSTNAME_SIZE);
    } else if (!host) {
        tstrncpy(g_queryInfo.host, "127.0.0.1", MAX_HOSTNAME_SIZE);
H
hzcheng 已提交
5814
    } else {
5815
        errorPrint("%s", "failed to read json, host not found\n");
5816
        goto PARSE_OVER;
H
hzcheng 已提交
5817
    }
5818

5819 5820 5821 5822 5823
    cJSON* port = cJSON_GetObjectItem(root, "port");
    if (port && port->type == cJSON_Number) {
        g_queryInfo.port = port->valueint;
    } else if (!port) {
        g_queryInfo.port = 6030;
5824
    }
5825

5826 5827 5828 5829 5830
    cJSON* user = cJSON_GetObjectItem(root, "user");
    if (user && user->type == cJSON_String && user->valuestring != NULL) {
        tstrncpy(g_queryInfo.user, user->valuestring, MAX_USERNAME_SIZE);
    } else if (!user) {
        tstrncpy(g_queryInfo.user, "root", MAX_USERNAME_SIZE); ;
5831 5832
    }

5833 5834
    cJSON* password = cJSON_GetObjectItem(root, "password");
    if (password && password->type == cJSON_String && password->valuestring != NULL) {
5835
        tstrncpy(g_queryInfo.password, password->valuestring, SHELL_MAX_PASSWORD_LEN);
5836
    } else if (!password) {
5837
        tstrncpy(g_queryInfo.password, "taosdata", SHELL_MAX_PASSWORD_LEN);;
5838
    }
5839

5840 5841 5842 5843 5844 5845 5846 5847 5848 5849 5850 5851
    cJSON *answerPrompt = cJSON_GetObjectItem(root, "confirm_parameter_prompt"); // yes, no,
    if (answerPrompt && answerPrompt->type == cJSON_String
            && answerPrompt->valuestring != NULL) {
        if (0 == strncasecmp(answerPrompt->valuestring, "yes", 3)) {
            g_args.answer_yes = false;
        } else if (0 == strncasecmp(answerPrompt->valuestring, "no", 2)) {
            g_args.answer_yes = true;
        } else {
            g_args.answer_yes = false;
        }
    } else if (!answerPrompt) {
        g_args.answer_yes = false;
5852
    } else {
5853
        errorPrint("%s", "failed to read json, confirm_parameter_prompt not found\n");
5854
        goto PARSE_OVER;
5855
    }
5856

5857 5858 5859
    cJSON* gQueryTimes = cJSON_GetObjectItem(root, "query_times");
    if (gQueryTimes && gQueryTimes->type == cJSON_Number) {
        if (gQueryTimes->valueint <= 0) {
5860
            errorPrint("%s()", "failed to read json, query_times input mistake\n");
5861 5862 5863 5864 5865
            goto PARSE_OVER;
        }
        g_args.query_times = gQueryTimes->valueint;
    } else if (!gQueryTimes) {
        g_args.query_times = 1;
5866
    } else {
5867
        errorPrint("%s", "failed to read json, query_times input mistake\n");
5868
        goto PARSE_OVER;
5869
    }
5870

5871 5872 5873 5874
    cJSON* dbs = cJSON_GetObjectItem(root, "databases");
    if (dbs && dbs->type == cJSON_String && dbs->valuestring != NULL) {
        tstrncpy(g_queryInfo.dbName, dbs->valuestring, TSDB_DB_NAME_LEN);
    } else if (!dbs) {
5875
        errorPrint("%s", "failed to read json, databases not found\n");
5876
        goto PARSE_OVER;
5877
    }
5878

5879
    cJSON* queryMode = cJSON_GetObjectItem(root, "query_mode");
5880 5881 5882 5883 5884
    if (queryMode
            && queryMode->type == cJSON_String
            && queryMode->valuestring != NULL) {
        tstrncpy(g_queryInfo.queryMode, queryMode->valuestring,
                min(SMALL_BUFF_LEN, strlen(queryMode->valuestring) + 1));
5885
    } else if (!queryMode) {
5886 5887
        tstrncpy(g_queryInfo.queryMode, "taosc",
                min(SMALL_BUFF_LEN, strlen("taosc") + 1));
5888
    } else {
5889
        errorPrint("%s", "failed to read json, query_mode not found\n");
5890
        goto PARSE_OVER;
5891
    }
5892

5893 5894 5895 5896 5897 5898
    // specified_table_query
    cJSON *specifiedQuery = cJSON_GetObjectItem(root, "specified_table_query");
    if (!specifiedQuery) {
        g_queryInfo.specifiedQueryInfo.concurrent = 1;
        g_queryInfo.specifiedQueryInfo.sqlCount = 0;
    } else if (specifiedQuery->type != cJSON_Object) {
5899
        errorPrint("%s", "failed to read json, super_table_query not found\n");
5900 5901
        goto PARSE_OVER;
    } else {
5902 5903 5904 5905 5906 5907
        cJSON* queryInterval = cJSON_GetObjectItem(specifiedQuery, "query_interval");
        if (queryInterval && queryInterval->type == cJSON_Number) {
            g_queryInfo.specifiedQueryInfo.queryInterval = queryInterval->valueint;
        } else if (!queryInterval) {
            g_queryInfo.specifiedQueryInfo.queryInterval = 0;
        }
5908

5909 5910 5911 5912 5913
        cJSON* specifiedQueryTimes = cJSON_GetObjectItem(specifiedQuery,
                "query_times");
        if (specifiedQueryTimes && specifiedQueryTimes->type == cJSON_Number) {
            if (specifiedQueryTimes->valueint <= 0) {
                errorPrint(
5914 5915
                        "failed to read json, query_times: %"PRId64", need be a valid (>0) number\n",
                        specifiedQueryTimes->valueint);
5916
                goto PARSE_OVER;
5917

5918 5919 5920 5921
            }
            g_queryInfo.specifiedQueryInfo.queryTimes = specifiedQueryTimes->valueint;
        } else if (!specifiedQueryTimes) {
            g_queryInfo.specifiedQueryInfo.queryTimes = g_args.query_times;
5922
        } else {
5923 5924 5925
            errorPrint("%s() LN%d, failed to read json, query_times input mistake\n",
                    __func__, __LINE__);
            goto PARSE_OVER;
5926
        }
5927

5928 5929 5930 5931
        cJSON* concurrent = cJSON_GetObjectItem(specifiedQuery, "concurrent");
        if (concurrent && concurrent->type == cJSON_Number) {
            if (concurrent->valueint <= 0) {
                errorPrint(
5932
                        "query sqlCount %d or concurrent %d is not correct.\n",
5933 5934 5935 5936 5937 5938 5939 5940
                        g_queryInfo.specifiedQueryInfo.sqlCount,
                        g_queryInfo.specifiedQueryInfo.concurrent);
                goto PARSE_OVER;
            }
            g_queryInfo.specifiedQueryInfo.concurrent = concurrent->valueint;
        } else if (!concurrent) {
            g_queryInfo.specifiedQueryInfo.concurrent = 1;
        }
5941

5942 5943 5944 5945 5946 5947 5948 5949
        cJSON* specifiedAsyncMode = cJSON_GetObjectItem(specifiedQuery, "mode");
        if (specifiedAsyncMode && specifiedAsyncMode->type == cJSON_String
                && specifiedAsyncMode->valuestring != NULL) {
            if (0 == strcmp("sync", specifiedAsyncMode->valuestring)) {
                g_queryInfo.specifiedQueryInfo.asyncMode = SYNC_MODE;
            } else if (0 == strcmp("async", specifiedAsyncMode->valuestring)) {
                g_queryInfo.specifiedQueryInfo.asyncMode = ASYNC_MODE;
            } else {
5950
                errorPrint("%s", "failed to read json, async mode input error\n");
5951 5952 5953 5954 5955
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.asyncMode = SYNC_MODE;
        }
5956

5957 5958 5959 5960 5961 5962 5963 5964
        cJSON* interval = cJSON_GetObjectItem(specifiedQuery, "interval");
        if (interval && interval->type == cJSON_Number) {
            g_queryInfo.specifiedQueryInfo.subscribeInterval = interval->valueint;
        } else if (!interval) {
            //printf("failed to read json, subscribe interval no found\n");
            //goto PARSE_OVER;
            g_queryInfo.specifiedQueryInfo.subscribeInterval = 10000;
        }
5965

5966 5967 5968 5969 5970 5971 5972
        cJSON* restart = cJSON_GetObjectItem(specifiedQuery, "restart");
        if (restart && restart->type == cJSON_String && restart->valuestring != NULL) {
            if (0 == strcmp("yes", restart->valuestring)) {
                g_queryInfo.specifiedQueryInfo.subscribeRestart = true;
            } else if (0 == strcmp("no", restart->valuestring)) {
                g_queryInfo.specifiedQueryInfo.subscribeRestart = false;
            } else {
5973
                errorPrint("%s", "failed to read json, subscribe restart error\n");
5974 5975 5976 5977 5978
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.subscribeRestart = true;
        }
5979

5980 5981 5982 5983 5984 5985 5986 5987 5988
        cJSON* keepProgress = cJSON_GetObjectItem(specifiedQuery, "keepProgress");
        if (keepProgress
                && keepProgress->type == cJSON_String
                && keepProgress->valuestring != NULL) {
            if (0 == strcmp("yes", keepProgress->valuestring)) {
                g_queryInfo.specifiedQueryInfo.subscribeKeepProgress = 1;
            } else if (0 == strcmp("no", keepProgress->valuestring)) {
                g_queryInfo.specifiedQueryInfo.subscribeKeepProgress = 0;
            } else {
5989
                errorPrint("%s", "failed to read json, subscribe keepProgress error\n");
5990 5991 5992 5993 5994
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.subscribeKeepProgress = 0;
        }
5995

5996 5997 5998 5999 6000
        // sqls
        cJSON* specifiedSqls = cJSON_GetObjectItem(specifiedQuery, "sqls");
        if (!specifiedSqls) {
            g_queryInfo.specifiedQueryInfo.sqlCount = 0;
        } else if (specifiedSqls->type != cJSON_Array) {
6001
            errorPrint("%s", "failed to read json, super sqls not found\n");
6002 6003 6004 6005 6006
            goto PARSE_OVER;
        } else {
            int superSqlSize = cJSON_GetArraySize(specifiedSqls);
            if (superSqlSize * g_queryInfo.specifiedQueryInfo.concurrent
                    > MAX_QUERY_SQL_COUNT) {
6007
                errorPrint("failed to read json, query sql(%d) * concurrent(%d) overflow, max is %d\n",
6008 6009 6010 6011 6012
                        superSqlSize,
                        g_queryInfo.specifiedQueryInfo.concurrent,
                        MAX_QUERY_SQL_COUNT);
                goto PARSE_OVER;
            }
6013

6014 6015 6016 6017 6018 6019 6020
            g_queryInfo.specifiedQueryInfo.sqlCount = superSqlSize;
            for (int j = 0; j < superSqlSize; ++j) {
                cJSON* sql = cJSON_GetArrayItem(specifiedSqls, j);
                if (sql == NULL) continue;

                cJSON *sqlStr = cJSON_GetObjectItem(sql, "sql");
                if (!sqlStr || sqlStr->type != cJSON_String || sqlStr->valuestring == NULL) {
6021
                    errorPrint("%s", "failed to read json, sql not found\n");
6022 6023 6024
                    goto PARSE_OVER;
                }
                tstrncpy(g_queryInfo.specifiedQueryInfo.sql[j],
6025
                        sqlStr->valuestring, BUFFER_SIZE);
6026 6027 6028 6029 6030 6031 6032 6033 6034 6035 6036 6037 6038 6039 6040 6041 6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057 6058 6059 6060

                // default value is -1, which mean infinite loop
                g_queryInfo.specifiedQueryInfo.endAfterConsume[j] = -1;
                cJSON* endAfterConsume =
                    cJSON_GetObjectItem(specifiedQuery, "endAfterConsume");
                if (endAfterConsume
                        && endAfterConsume->type == cJSON_Number) {
                    g_queryInfo.specifiedQueryInfo.endAfterConsume[j]
                        = endAfterConsume->valueint;
                }
                if (g_queryInfo.specifiedQueryInfo.endAfterConsume[j] < -1)
                    g_queryInfo.specifiedQueryInfo.endAfterConsume[j] = -1;

                g_queryInfo.specifiedQueryInfo.resubAfterConsume[j] = -1;
                cJSON* resubAfterConsume =
                    cJSON_GetObjectItem(specifiedQuery, "resubAfterConsume");
                if ((resubAfterConsume)
                        && (resubAfterConsume->type == cJSON_Number)
                        && (resubAfterConsume->valueint >= 0)) {
                    g_queryInfo.specifiedQueryInfo.resubAfterConsume[j]
                        = resubAfterConsume->valueint;
                }

                if (g_queryInfo.specifiedQueryInfo.resubAfterConsume[j] < -1)
                    g_queryInfo.specifiedQueryInfo.resubAfterConsume[j] = -1;

                cJSON *result = cJSON_GetObjectItem(sql, "result");
                if ((NULL != result) && (result->type == cJSON_String)
                        && (result->valuestring != NULL)) {
                    tstrncpy(g_queryInfo.specifiedQueryInfo.result[j],
                            result->valuestring, MAX_FILE_NAME_LEN);
                } else if (NULL == result) {
                    memset(g_queryInfo.specifiedQueryInfo.result[j],
                            0, MAX_FILE_NAME_LEN);
                } else {
6061 6062
                    errorPrint("%s",
                            "failed to read json, super query result file not found\n");
6063 6064 6065 6066
                    goto PARSE_OVER;
                }
            }
        }
6067
    }
6068

6069 6070 6071 6072 6073 6074
    // super_table_query
    cJSON *superQuery = cJSON_GetObjectItem(root, "super_table_query");
    if (!superQuery) {
        g_queryInfo.superQueryInfo.threadCnt = 1;
        g_queryInfo.superQueryInfo.sqlCount = 0;
    } else if (superQuery->type != cJSON_Object) {
6075
        errorPrint("%s", "failed to read json, sub_table_query not found\n");
6076
        ret = true;
6077 6078
        goto PARSE_OVER;
    } else {
6079 6080 6081 6082 6083 6084 6085 6086 6087 6088
        cJSON* subrate = cJSON_GetObjectItem(superQuery, "query_interval");
        if (subrate && subrate->type == cJSON_Number) {
            g_queryInfo.superQueryInfo.queryInterval = subrate->valueint;
        } else if (!subrate) {
            g_queryInfo.superQueryInfo.queryInterval = 0;
        }

        cJSON* superQueryTimes = cJSON_GetObjectItem(superQuery, "query_times");
        if (superQueryTimes && superQueryTimes->type == cJSON_Number) {
            if (superQueryTimes->valueint <= 0) {
6089 6090
                errorPrint("failed to read json, query_times: %"PRId64", need be a valid (>0) number\n",
                        superQueryTimes->valueint);
6091 6092 6093 6094 6095 6096
                goto PARSE_OVER;
            }
            g_queryInfo.superQueryInfo.queryTimes = superQueryTimes->valueint;
        } else if (!superQueryTimes) {
            g_queryInfo.superQueryInfo.queryTimes = g_args.query_times;
        } else {
6097
            errorPrint("%s", "failed to read json, query_times input mistake\n");
6098 6099
            goto PARSE_OVER;
        }
6100

6101 6102 6103
        cJSON* threads = cJSON_GetObjectItem(superQuery, "threads");
        if (threads && threads->type == cJSON_Number) {
            if (threads->valueint <= 0) {
6104
                errorPrint("%s", "failed to read json, threads input mistake\n");
6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117 6118 6119 6120 6121 6122
                goto PARSE_OVER;

            }
            g_queryInfo.superQueryInfo.threadCnt = threads->valueint;
        } else if (!threads) {
            g_queryInfo.superQueryInfo.threadCnt = 1;
        }

        //cJSON* subTblCnt = cJSON_GetObjectItem(superQuery, "childtable_count");
        //if (subTblCnt && subTblCnt->type == cJSON_Number) {
        //  g_queryInfo.superQueryInfo.childTblCount = subTblCnt->valueint;
        //} else if (!subTblCnt) {
        //  g_queryInfo.superQueryInfo.childTblCount = 0;
        //}

        cJSON* stblname = cJSON_GetObjectItem(superQuery, "stblname");
        if (stblname && stblname->type == cJSON_String
                && stblname->valuestring != NULL) {
6123
            tstrncpy(g_queryInfo.superQueryInfo.stbName, stblname->valuestring,
6124 6125
                    TSDB_TABLE_NAME_LEN);
        } else {
6126
            errorPrint("%s", "failed to read json, super table name input error\n");
6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137
            goto PARSE_OVER;
        }

        cJSON* superAsyncMode = cJSON_GetObjectItem(superQuery, "mode");
        if (superAsyncMode && superAsyncMode->type == cJSON_String
                && superAsyncMode->valuestring != NULL) {
            if (0 == strcmp("sync", superAsyncMode->valuestring)) {
                g_queryInfo.superQueryInfo.asyncMode = SYNC_MODE;
            } else if (0 == strcmp("async", superAsyncMode->valuestring)) {
                g_queryInfo.superQueryInfo.asyncMode = ASYNC_MODE;
            } else {
6138
                errorPrint("%s", "failed to read json, async mode input error\n");
6139 6140 6141 6142 6143 6144 6145 6146 6147
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.superQueryInfo.asyncMode = SYNC_MODE;
        }

        cJSON* superInterval = cJSON_GetObjectItem(superQuery, "interval");
        if (superInterval && superInterval->type == cJSON_Number) {
            if (superInterval->valueint < 0) {
6148
                errorPrint("%s", "failed to read json, interval input mistake\n");
6149 6150 6151 6152 6153 6154 6155 6156 6157 6158 6159 6160 6161 6162 6163 6164 6165
                goto PARSE_OVER;
            }
            g_queryInfo.superQueryInfo.subscribeInterval = superInterval->valueint;
        } else if (!superInterval) {
            //printf("failed to read json, subscribe interval no found\n");
            //goto PARSE_OVER;
            g_queryInfo.superQueryInfo.subscribeInterval = 10000;
        }

        cJSON* subrestart = cJSON_GetObjectItem(superQuery, "restart");
        if (subrestart && subrestart->type == cJSON_String
                && subrestart->valuestring != NULL) {
            if (0 == strcmp("yes", subrestart->valuestring)) {
                g_queryInfo.superQueryInfo.subscribeRestart = true;
            } else if (0 == strcmp("no", subrestart->valuestring)) {
                g_queryInfo.superQueryInfo.subscribeRestart = false;
            } else {
6166
                errorPrint("%s", "failed to read json, subscribe restart error\n");
6167 6168 6169 6170 6171 6172 6173 6174 6175 6176 6177 6178 6179 6180 6181
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.superQueryInfo.subscribeRestart = true;
        }

        cJSON* superkeepProgress = cJSON_GetObjectItem(superQuery, "keepProgress");
        if (superkeepProgress &&
                superkeepProgress->type == cJSON_String
                && superkeepProgress->valuestring != NULL) {
            if (0 == strcmp("yes", superkeepProgress->valuestring)) {
                g_queryInfo.superQueryInfo.subscribeKeepProgress = 1;
            } else if (0 == strcmp("no", superkeepProgress->valuestring)) {
                g_queryInfo.superQueryInfo.subscribeKeepProgress = 0;
            } else {
6182 6183
                errorPrint("%s",
                        "failed to read json, subscribe super table keepProgress error\n");
6184 6185 6186 6187 6188 6189 6190
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.superQueryInfo.subscribeKeepProgress = 0;
        }

        // default value is -1, which mean do not resub
6191
        g_queryInfo.superQueryInfo.endAfterConsume = -1;
6192 6193 6194 6195 6196 6197 6198 6199 6200
        cJSON* superEndAfterConsume =
            cJSON_GetObjectItem(superQuery, "endAfterConsume");
        if (superEndAfterConsume
                && superEndAfterConsume->type == cJSON_Number) {
            g_queryInfo.superQueryInfo.endAfterConsume =
                superEndAfterConsume->valueint;
        }
        if (g_queryInfo.superQueryInfo.endAfterConsume < -1)
            g_queryInfo.superQueryInfo.endAfterConsume = -1;
6201

6202
        // default value is -1, which mean do not resub
6203
        g_queryInfo.superQueryInfo.resubAfterConsume = -1;
6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218 6219
        cJSON* superResubAfterConsume =
            cJSON_GetObjectItem(superQuery, "resubAfterConsume");
        if ((superResubAfterConsume)
                && (superResubAfterConsume->type == cJSON_Number)
                && (superResubAfterConsume->valueint >= 0)) {
            g_queryInfo.superQueryInfo.resubAfterConsume =
                superResubAfterConsume->valueint;
        }
        if (g_queryInfo.superQueryInfo.resubAfterConsume < -1)
            g_queryInfo.superQueryInfo.resubAfterConsume = -1;

        // supert table sqls
        cJSON* superSqls = cJSON_GetObjectItem(superQuery, "sqls");
        if (!superSqls) {
            g_queryInfo.superQueryInfo.sqlCount = 0;
        } else if (superSqls->type != cJSON_Array) {
6220
            errorPrint("%s", "failed to read json, super sqls not found\n");
6221 6222 6223 6224
            goto PARSE_OVER;
        } else {
            int superSqlSize = cJSON_GetArraySize(superSqls);
            if (superSqlSize > MAX_QUERY_SQL_COUNT) {
6225 6226
                errorPrint("failed to read json, query sql size overflow, max is %d\n",
                        MAX_QUERY_SQL_COUNT);
6227 6228
                goto PARSE_OVER;
            }
6229

6230 6231 6232 6233
            g_queryInfo.superQueryInfo.sqlCount = superSqlSize;
            for (int j = 0; j < superSqlSize; ++j) {
                cJSON* sql = cJSON_GetArrayItem(superSqls, j);
                if (sql == NULL) continue;
6234

6235 6236 6237
                cJSON *sqlStr = cJSON_GetObjectItem(sql, "sql");
                if (!sqlStr || sqlStr->type != cJSON_String
                        || sqlStr->valuestring == NULL) {
6238
                    errorPrint("%s", "failed to read json, sql not found\n");
6239 6240 6241
                    goto PARSE_OVER;
                }
                tstrncpy(g_queryInfo.superQueryInfo.sql[j], sqlStr->valuestring,
6242
                        BUFFER_SIZE);
6243 6244 6245

                cJSON *result = cJSON_GetObjectItem(sql, "result");
                if (result != NULL && result->type == cJSON_String
6246
                        && result->valuestring != NULL) {
6247 6248 6249 6250 6251
                    tstrncpy(g_queryInfo.superQueryInfo.result[j],
                            result->valuestring, MAX_FILE_NAME_LEN);
                } else if (NULL == result) {
                    memset(g_queryInfo.superQueryInfo.result[j], 0, MAX_FILE_NAME_LEN);
                }  else {
6252
                    errorPrint("%s", "failed to read json, sub query result file not found\n");
6253 6254 6255
                    goto PARSE_OVER;
                }
            }
6256
        }
6257
    }
H
hzcheng 已提交
6258

6259
    ret = true;
H
hzcheng 已提交
6260

6261
PARSE_OVER:
6262
    return ret;
6263
}
H
hzcheng 已提交
6264

6265
static bool getInfoFromJsonFile(char* file) {
6266
    debugPrint("%s %d %s\n", __func__, __LINE__, file);
6267

6268 6269
    FILE *fp = fopen(file, "r");
    if (!fp) {
6270
        errorPrint("failed to read %s, reason:%s\n", file, strerror(errno));
6271 6272
        return false;
    }
H
Hui Li 已提交
6273

6274 6275 6276 6277 6278 6279 6280
    bool  ret = false;
    int   maxLen = 6400000;
    char *content = calloc(1, maxLen + 1);
    int   len = fread(content, 1, maxLen, fp);
    if (len <= 0) {
        free(content);
        fclose(fp);
6281
        errorPrint("failed to read %s, content is null", file);
6282 6283
        return false;
    }
H
Hui Li 已提交
6284

6285 6286 6287
    content[len] = 0;
    cJSON* root = cJSON_Parse(content);
    if (root == NULL) {
6288
        errorPrint("failed to cjson parse %s, invalid json format\n", file);
6289 6290
        goto PARSE_OVER;
    }
H
Hui Li 已提交
6291

6292 6293 6294 6295 6296 6297 6298 6299 6300
    cJSON* filetype = cJSON_GetObjectItem(root, "filetype");
    if (filetype && filetype->type == cJSON_String && filetype->valuestring != NULL) {
        if (0 == strcasecmp("insert", filetype->valuestring)) {
            g_args.test_mode = INSERT_TEST;
        } else if (0 == strcasecmp("query", filetype->valuestring)) {
            g_args.test_mode = QUERY_TEST;
        } else if (0 == strcasecmp("subscribe", filetype->valuestring)) {
            g_args.test_mode = SUBSCRIBE_TEST;
        } else {
6301
            errorPrint("%s", "failed to read json, filetype not support\n");
6302 6303 6304 6305
            goto PARSE_OVER;
        }
    } else if (!filetype) {
        g_args.test_mode = INSERT_TEST;
6306
    } else {
6307
        errorPrint("%s", "failed to read json, filetype not found\n");
6308
        goto PARSE_OVER;
6309
    }
H
hzcheng 已提交
6310

6311 6312 6313 6314 6315 6316
    if (INSERT_TEST == g_args.test_mode) {
        ret = getMetaFromInsertJsonFile(root);
    } else if ((QUERY_TEST == g_args.test_mode)
            || (SUBSCRIBE_TEST == g_args.test_mode)) {
        ret = getMetaFromQueryJsonFile(root);
    } else {
6317 6318
        errorPrint("%s",
                "input json file type error! please input correct file type: insert or query or subscribe\n");
6319 6320
        goto PARSE_OVER;
    }
6321

6322
PARSE_OVER:
6323 6324 6325 6326
    free(content);
    cJSON_Delete(root);
    fclose(fp);
    return ret;
6327
}
H
hzcheng 已提交
6328

6329
static int prepareSampleData() {
6330 6331 6332 6333 6334 6335 6336
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        for (int j = 0; j < g_Dbs.db[i].superTblCount; j++) {
            if (g_Dbs.db[i].superTbls[j].tagsFile[0] != 0) {
                if (readTagFromCsvFileToMem(&g_Dbs.db[i].superTbls[j]) != 0) {
                    return -1;
                }
            }
6337
        }
6338
    }
6339

6340
    return 0;
6341 6342
}

6343
static void postFreeResource() {
6344
    tmfclose(g_fpOfInsertResult);
6345

6346 6347 6348
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        for (uint64_t j = 0; j < g_Dbs.db[i].superTblCount; j++) {
            if (0 != g_Dbs.db[i].superTbls[j].colsOfCreateChildTable) {
6349
                tmfree(g_Dbs.db[i].superTbls[j].colsOfCreateChildTable);
6350 6351 6352
                g_Dbs.db[i].superTbls[j].colsOfCreateChildTable = NULL;
            }
            if (0 != g_Dbs.db[i].superTbls[j].sampleDataBuf) {
6353
                tmfree(g_Dbs.db[i].superTbls[j].sampleDataBuf);
6354 6355
                g_Dbs.db[i].superTbls[j].sampleDataBuf = NULL;
            }
6356

6357 6358 6359 6360 6361 6362 6363 6364 6365 6366 6367 6368 6369
#if STMT_BIND_PARAM_BATCH == 1
            for (int c = 0;
                    c < g_Dbs.db[i].superTbls[j].columnCount; c ++) {

                if (g_Dbs.db[i].superTbls[j].sampleBindBatchArray) {

                    tmfree((char *)((uintptr_t)*(uintptr_t*)(
                                    g_Dbs.db[i].superTbls[j].sampleBindBatchArray
                                    + sizeof(char*) * c)));
                }
            }
            tmfree(g_Dbs.db[i].superTbls[j].sampleBindBatchArray);
#endif
6370
            if (0 != g_Dbs.db[i].superTbls[j].tagDataBuf) {
6371
                tmfree(g_Dbs.db[i].superTbls[j].tagDataBuf);
6372 6373 6374
                g_Dbs.db[i].superTbls[j].tagDataBuf = NULL;
            }
            if (0 != g_Dbs.db[i].superTbls[j].childTblName) {
6375
                tmfree(g_Dbs.db[i].superTbls[j].childTblName);
6376 6377 6378
                g_Dbs.db[i].superTbls[j].childTblName = NULL;
            }
        }
S
Shuaiqiang Chang 已提交
6379
    }
6380 6381 6382 6383 6384 6385 6386 6387 6388 6389

    tmfree(g_randbool_buff);
    tmfree(g_randint_buff);
    tmfree(g_rand_voltage_buff);
    tmfree(g_randbigint_buff);
    tmfree(g_randsmallint_buff);
    tmfree(g_randtinyint_buff);
    tmfree(g_randfloat_buff);
    tmfree(g_rand_current_buff);
    tmfree(g_rand_phase_buff);
6390

6391 6392 6393 6394 6395 6396 6397 6398 6399 6400 6401 6402 6403
    tmfree(g_sampleDataBuf);

#if STMT_BIND_PARAM_BATCH == 1
    for (int l = 0;
             l < g_args.columnCount; l ++) {
        if (g_sampleBindBatchArray) {
            tmfree((char *)((uintptr_t)*(uintptr_t*)(
                            g_sampleBindBatchArray
                            + sizeof(char*) * l)));
        }
    }
    tmfree(g_sampleBindBatchArray);
#endif
6404
}
S
Shuaiqiang Chang 已提交
6405

6406 6407
static int getRowDataFromSample(
        char* dataBuf, int64_t maxLen, int64_t timestamp,
6408
        SSuperTable* stbInfo, int64_t* sampleUsePos)
6409
{
6410
    if ((*sampleUsePos) == MAX_SAMPLES) {
6411 6412 6413 6414 6415 6416 6417 6418 6419
        *sampleUsePos = 0;
    }

    int    dataLen = 0;

    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
            "(%" PRId64 ", ", timestamp);
    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
            "%s",
6420 6421
            stbInfo->sampleDataBuf
            + stbInfo->lenOfOneRow * (*sampleUsePos));
6422 6423 6424 6425 6426
    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen, ")");

    (*sampleUsePos)++;

    return dataLen;
6427
}
S
Shuaiqiang Chang 已提交
6428

6429 6430
static int64_t generateStbRowData(
        SSuperTable* stbInfo,
6431 6432 6433
        char* recBuf,
        int64_t remainderBufLen,
        int64_t timestamp)
6434
{
6435 6436 6437
    int64_t   dataLen = 0;
    char  *pstr = recBuf;
    int64_t maxLen = MAX_DATA_SIZE;
6438
    int tmpLen;
6439 6440

    dataLen += snprintf(pstr + dataLen, maxLen - dataLen,
6441
            "(%" PRId64 "", timestamp);
6442 6443

    for (int i = 0; i < stbInfo->columnCount; i++) {
6444 6445 6446 6447 6448
        tstrncpy(pstr + dataLen, ",", 2);
        dataLen += 1;

        if ((stbInfo->columns[i].data_type == TSDB_DATA_TYPE_BINARY)
                || (stbInfo->columns[i].data_type == TSDB_DATA_TYPE_NCHAR)) {
6449
            if (stbInfo->columns[i].dataLen > TSDB_MAX_BINARY_LEN) {
6450
                errorPrint2("binary or nchar length overflow, max size:%u\n",
6451 6452 6453
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
6454

6455 6456 6457 6458 6459
            if ((stbInfo->columns[i].dataLen + 1) >
                    /* need count 3 extra chars \', \', and , */
                    (remainderBufLen - dataLen - 3)) {
                return 0;
            }
6460 6461
            char* buf = (char*)calloc(stbInfo->columns[i].dataLen+1, 1);
            if (NULL == buf) {
6462
                errorPrint2("calloc failed! size:%d\n", stbInfo->columns[i].dataLen);
6463 6464 6465
                return -1;
            }
            rand_string(buf, stbInfo->columns[i].dataLen);
6466
            dataLen += snprintf(pstr + dataLen, maxLen - dataLen, "\'%s\'", buf);
6467
            tmfree(buf);
6468

6469
        } else {
6470 6471 6472 6473 6474 6475 6476 6477 6478 6479 6480
            char *tmp = NULL;
            switch(stbInfo->columns[i].data_type) {
                case TSDB_DATA_TYPE_INT:
                    if ((g_args.demo_mode) && (i == 1)) {
                        tmp = demo_voltage_int_str();
                    } else {
                        tmp = rand_int_str();
                    }
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, INT_BUFF_LEN));
                    break;
6481

6482 6483 6484 6485 6486 6487
                case TSDB_DATA_TYPE_UINT:
                    tmp = rand_uint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, INT_BUFF_LEN));
                    break;

6488 6489
                case TSDB_DATA_TYPE_BIGINT:
                    tmp = rand_bigint_str();
6490 6491 6492 6493 6494 6495 6496 6497
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
                    break;
                
                case TSDB_DATA_TYPE_UBIGINT:
                    tmp = rand_ubigint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
6498 6499 6500 6501 6502 6503 6504 6505 6506
                    break;

                case TSDB_DATA_TYPE_FLOAT:
                    if (g_args.demo_mode) {
                        if (i == 0) {
                            tmp = demo_current_float_str();
                        } else {
                            tmp = demo_phase_float_str();
                        }
6507
                    } else {
6508
                        tmp = rand_float_str();
6509
                    }
6510
                    tmpLen = strlen(tmp);
6511
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, FLOAT_BUFF_LEN));
6512 6513 6514 6515 6516
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    tmp = rand_double_str();
                    tmpLen = strlen(tmp);
6517
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, DOUBLE_BUFF_LEN));
6518 6519 6520 6521 6522
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
                    tmp = rand_smallint_str();
                    tmpLen = strlen(tmp);
6523 6524 6525 6526 6527 6528 6529
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, SMALLINT_BUFF_LEN));
                    break;

                case TSDB_DATA_TYPE_USMALLINT:
                    tmp = rand_usmallint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, SMALLINT_BUFF_LEN));
6530 6531 6532 6533 6534
                    break;

                case TSDB_DATA_TYPE_TINYINT:
                    tmp = rand_tinyint_str();
                    tmpLen = strlen(tmp);
6535 6536 6537 6538 6539 6540 6541
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, TINYINT_BUFF_LEN));
                    break;

                case TSDB_DATA_TYPE_UTINYINT:
                    tmp = rand_utinyint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, TINYINT_BUFF_LEN));
6542 6543 6544 6545 6546
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    tmp = rand_bool_str();
                    tmpLen = strlen(tmp);
6547
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BOOL_BUFF_LEN));
6548 6549 6550 6551 6552
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    tmp = rand_bigint_str();
                    tmpLen = strlen(tmp);
6553
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
6554 6555 6556 6557 6558 6559 6560 6561 6562
                    break;

                case TSDB_DATA_TYPE_NULL:
                    break;

                default:
                    errorPrint2("Not support data type: %s\n",
                            stbInfo->columns[i].dataType);
                    exit(EXIT_FAILURE);
6563
            }
6564
            if (tmp) {
6565
                dataLen += tmpLen;
6566
            }
6567
        }
6568

6569
        if (dataLen > (remainderBufLen - (128)))
6570
            return 0;
6571
    }
6572

6573
    dataLen += snprintf(pstr + dataLen, 2, ")");
6574

6575 6576
    verbosePrint("%s() LN%d, dataLen:%"PRId64"\n", __func__, __LINE__, dataLen);
    verbosePrint("%s() LN%d, recBuf:\n\t%s\n", __func__, __LINE__, recBuf);
6577

6578
    return strlen(recBuf);
6579
}
S
Shuaiqiang Chang 已提交
6580

6581
static int64_t generateData(char *recBuf, char *data_type,
6582
        int64_t timestamp, int lenOfBinary) {
6583 6584
    memset(recBuf, 0, MAX_DATA_SIZE);
    char *pstr = recBuf;
6585
    pstr += sprintf(pstr, "(%"PRId64"", timestamp);
6586

6587
    int columnCount = g_args.columnCount;
6588

6589 6590
    bool b;
    char *s;
6591
    for (int i = 0; i < columnCount; i++) {
6592 6593 6594 6595 6596 6597 6598 6599 6600 6601 6602 6603 6604 6605 6606 6607 6608 6609 6610 6611 6612 6613 6614 6615 6616 6617 6618 6619 6620 6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631 6632 6633 6634 6635 6636 6637 6638 6639 6640 6641 6642 6643 6644 6645 6646 6647 6648 6649
        switch (data_type[i]) {
            case TSDB_DATA_TYPE_TINYINT:
                pstr += sprintf(pstr, ",%d", rand_tinyint() );
                break;

            case TSDB_DATA_TYPE_SMALLINT:
                pstr += sprintf(pstr, ",%d", rand_smallint());
                break;

            case TSDB_DATA_TYPE_INT:
                pstr += sprintf(pstr, ",%d", rand_int());
                break;

            case TSDB_DATA_TYPE_BIGINT:
                pstr += sprintf(pstr, ",%"PRId64"", rand_bigint());
                break;

            case TSDB_DATA_TYPE_TIMESTAMP:
                pstr += sprintf(pstr, ",%"PRId64"", rand_bigint());
                break;

            case TSDB_DATA_TYPE_FLOAT:
                pstr += sprintf(pstr, ",%10.4f", rand_float());
                break;

            case TSDB_DATA_TYPE_DOUBLE:
                pstr += sprintf(pstr, ",%20.8f", rand_double());
                break;

            case TSDB_DATA_TYPE_BOOL:
                b = rand_bool() & 1;
                pstr += sprintf(pstr, ",%s", b ? "true" : "false");
                break;

            case TSDB_DATA_TYPE_BINARY:
                s = malloc(lenOfBinary + 1);
                if (s == NULL) {
                    errorPrint2("%s() LN%d, memory allocation %d bytes failed\n",
                            __func__, __LINE__, lenOfBinary + 1);
                    exit(EXIT_FAILURE);
                }
                rand_string(s, lenOfBinary);
                pstr += sprintf(pstr, ",\"%s\"", s);
                free(s);
                break;

            case TSDB_DATA_TYPE_NCHAR:
                s = malloc(lenOfBinary + 1);
                if (s == NULL) {
                    errorPrint2("%s() LN%d, memory allocation %d bytes failed\n",
                            __func__, __LINE__, lenOfBinary + 1);
                    exit(EXIT_FAILURE);
                }
                rand_string(s, lenOfBinary);
                pstr += sprintf(pstr, ",\"%s\"", s);
                free(s);
                break;

6650 6651 6652 6653 6654 6655 6656 6657 6658 6659 6660 6661 6662 6663 6664 6665
            case TSDB_DATA_TYPE_UTINYINT:
                pstr += sprintf(pstr, ",%d", rand_utinyint() );
                break;

            case TSDB_DATA_TYPE_USMALLINT:
                pstr += sprintf(pstr, ",%d", rand_usmallint());
                break;

            case TSDB_DATA_TYPE_UINT:
                pstr += sprintf(pstr, ",%d", rand_uint());
                break;

            case TSDB_DATA_TYPE_UBIGINT:
                pstr += sprintf(pstr, ",%"PRId64"", rand_ubigint());
                break;

6666 6667 6668 6669 6670 6671 6672
            case TSDB_DATA_TYPE_NULL:
                break;

            default:
                errorPrint2("%s() LN%d, Unknown data type %d\n",
                        __func__, __LINE__,
                        data_type[i]);
6673
                exit(EXIT_FAILURE);
6674
        }
6675

6676
        if (strlen(recBuf) > MAX_DATA_SIZE) {
6677
            ERROR_EXIT("column length too long, abort");
6678
        }
6679 6680
    }

6681
    pstr += sprintf(pstr, ")");
6682

6683
    verbosePrint("%s() LN%d, recBuf:\n\t%s\n", __func__, __LINE__, recBuf);
6684

6685
    return (int32_t)strlen(recBuf);
6686 6687
}

6688 6689 6690 6691 6692 6693
static int generateSampleFromRand(
        char *sampleDataBuf,
        uint64_t lenOfOneRow,
        int columnCount,
        StrColumn *columns
        )
6694 6695 6696 6697
{
    char data[MAX_DATA_SIZE];
    memset(data, 0, MAX_DATA_SIZE);

6698
    char *buff = malloc(lenOfOneRow);
6699
    if (NULL == buff) {
6700 6701
        errorPrint2("%s() LN%d, memory allocation %"PRIu64" bytes failed\n",
                __func__, __LINE__, lenOfOneRow);
6702 6703 6704
        exit(EXIT_FAILURE);
    }

6705
    for (int i=0; i < MAX_SAMPLES; i++) {
6706
        uint64_t pos = 0;
6707
        memset(buff, 0, lenOfOneRow);
6708

6709 6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722 6723
        for (int c = 0; c < columnCount; c++) {
            char *tmp = NULL;

            uint32_t dataLen;
            char data_type = (columns)?(columns[c].data_type):g_args.data_type[c];

            switch(data_type) {
                case TSDB_DATA_TYPE_BINARY:
                    dataLen = (columns)?columns[c].dataLen:g_args.binwidth;
                    rand_string(data, dataLen);
                    pos += sprintf(buff + pos, "%s,", data);
                    break;

                case TSDB_DATA_TYPE_NCHAR:
                    dataLen = (columns)?columns[c].dataLen:g_args.binwidth;
6724
                    rand_string(data, dataLen - 1);
6725 6726 6727 6728 6729 6730
                    pos += sprintf(buff + pos, "%s,", data);
                    break;

                case TSDB_DATA_TYPE_INT:
                    if ((g_args.demo_mode) && (c == 1)) {
                        tmp = demo_voltage_int_str();
6731
                    } else {
6732
                        tmp = rand_int_str();
6733
                    }
6734 6735 6736
                    pos += sprintf(buff + pos, "%s,", tmp);
                    break;

6737 6738 6739 6740
                case TSDB_DATA_TYPE_UINT:
                    pos += sprintf(buff + pos, "%s,", rand_uint_str());
                    break;

6741 6742 6743 6744
                case TSDB_DATA_TYPE_BIGINT:
                    pos += sprintf(buff + pos, "%s,", rand_bigint_str());
                    break;

6745 6746 6747 6748
                case TSDB_DATA_TYPE_UBIGINT:
                    pos += sprintf(buff + pos, "%s,", rand_ubigint_str());
                    break;

6749 6750 6751 6752 6753 6754 6755 6756 6757 6758 6759 6760 6761 6762 6763 6764 6765 6766 6767 6768 6769
                case TSDB_DATA_TYPE_FLOAT:
                    if (g_args.demo_mode) {
                        if (c == 0) {
                            tmp = demo_current_float_str();
                        } else {
                            tmp = demo_phase_float_str();
                        }
                    } else {
                        tmp = rand_float_str();
                    }
                    pos += sprintf(buff + pos, "%s,", tmp);
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    pos += sprintf(buff + pos, "%s,", rand_double_str());
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
                    pos += sprintf(buff + pos, "%s,", rand_smallint_str());
                    break;

6770 6771 6772 6773
                case TSDB_DATA_TYPE_USMALLINT:
                    pos += sprintf(buff + pos, "%s,", rand_usmallint_str());
                    break;

6774 6775 6776 6777
                case TSDB_DATA_TYPE_TINYINT:
                    pos += sprintf(buff + pos, "%s,", rand_tinyint_str());
                    break;

6778 6779 6780 6781
                case TSDB_DATA_TYPE_UTINYINT:
                    pos += sprintf(buff + pos, "%s,", rand_utinyint_str());
                    break;

6782 6783 6784 6785 6786 6787 6788 6789 6790 6791 6792 6793 6794 6795 6796 6797
                case TSDB_DATA_TYPE_BOOL:
                    pos += sprintf(buff + pos, "%s,", rand_bool_str());
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    pos += sprintf(buff + pos, "%s,", rand_bigint_str());
                    break;

                case TSDB_DATA_TYPE_NULL:
                    break;

                default:
                    errorPrint2("%s() LN%d, Unknown data type %s\n",
                            __func__, __LINE__,
                            (columns)?(columns[c].dataType):g_args.dataType[c]);
                    exit(EXIT_FAILURE);
6798 6799
            }
        }
6800

6801
        *(buff + pos - 1) = 0;
6802
        memcpy(sampleDataBuf + i * lenOfOneRow, buff, pos);
6803 6804 6805 6806 6807 6808
    }

    free(buff);
    return 0;
}

6809 6810 6811 6812 6813 6814 6815 6816 6817 6818 6819 6820 6821 6822 6823 6824 6825 6826 6827
static int generateSampleFromRandForNtb()
{
    return generateSampleFromRand(
            g_sampleDataBuf,
            g_args.lenOfOneRow,
            g_args.columnCount,
            NULL);
}

static int generateSampleFromRandForStb(SSuperTable *stbInfo)
{
    return generateSampleFromRand(
            stbInfo->sampleDataBuf,
            stbInfo->lenOfOneRow,
            stbInfo->columnCount,
            stbInfo->columns);
}

static int prepareSampleForNtb() {
6828
    g_sampleDataBuf = calloc(g_args.lenOfOneRow * MAX_SAMPLES, 1);
6829 6830 6831
    if (NULL == g_sampleDataBuf) {
        errorPrint2("%s() LN%d, Failed to calloc %"PRIu64" Bytes, reason:%s\n",
                __func__, __LINE__,
6832
                g_args.lenOfOneRow * MAX_SAMPLES,
6833 6834 6835 6836 6837 6838 6839 6840
                strerror(errno));
        return -1;
    }

    return generateSampleFromRandForNtb();
}

static int prepareSampleForStb(SSuperTable *stbInfo) {
6841

6842
    stbInfo->sampleDataBuf = calloc(
6843
            stbInfo->lenOfOneRow * MAX_SAMPLES, 1);
6844
    if (NULL == stbInfo->sampleDataBuf) {
6845
        errorPrint2("%s() LN%d, Failed to calloc %"PRIu64" Bytes, reason:%s\n",
6846
                __func__, __LINE__,
6847
                stbInfo->lenOfOneRow * MAX_SAMPLES,
6848 6849 6850
                strerror(errno));
        return -1;
    }
6851

6852
    int ret;
6853 6854 6855 6856 6857
    if (0 == strncasecmp(stbInfo->dataSource, "sample", strlen("sample"))) {
        ret = generateSampleFromCsvForStb(stbInfo);
    } else {
        ret = generateSampleFromRandForStb(stbInfo);
    }
6858

6859
    if (0 != ret) {
6860
        errorPrint2("%s() LN%d, read sample from csv file failed.\n",
6861
                __func__, __LINE__);
6862
        tmfree(stbInfo->sampleDataBuf);
6863
        stbInfo->sampleDataBuf = NULL;
6864 6865
        return -1;
    }
6866

6867
    return 0;
6868 6869
}

6870
static int32_t execInsert(threadInfo *pThreadInfo, uint32_t k)
6871
{
6872
    int32_t affectedRows;
6873
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
6874

6875
    uint16_t iface;
6876 6877
    if (stbInfo)
        iface = stbInfo->iface;
6878 6879 6880 6881 6882 6883
    else {
        if (g_args.iface == INTERFACE_BUT)
            iface = TAOSC_IFACE;
        else
            iface = g_args.iface;
    }
6884 6885 6886

    debugPrint("[%d] %s() LN%d %s\n", pThreadInfo->threadID,
            __func__, __LINE__,
6887 6888
            (iface==TAOSC_IFACE)?
            "taosc":(iface==REST_IFACE)?"rest":"stmt");
6889 6890 6891

    switch(iface) {
        case TAOSC_IFACE:
6892 6893 6894
            verbosePrint("[%d] %s() LN%d %s\n", pThreadInfo->threadID,
                    __func__, __LINE__, pThreadInfo->buffer);

6895 6896 6897 6898 6899 6900
            affectedRows = queryDbExec(
                    pThreadInfo->taos,
                    pThreadInfo->buffer, INSERT_TYPE, false);
            break;

        case REST_IFACE:
6901 6902 6903
            verbosePrint("[%d] %s() LN%d %s\n", pThreadInfo->threadID,
                    __func__, __LINE__, pThreadInfo->buffer);

6904
            if (0 != postProceSql(g_Dbs.host, &g_Dbs.serv_addr, g_Dbs.port,
6905
                        pThreadInfo->buffer, pThreadInfo)) {
6906 6907 6908 6909 6910 6911 6912 6913 6914
                affectedRows = -1;
                printf("========restful return fail, threadID[%d]\n",
                        pThreadInfo->threadID);
            } else {
                affectedRows = k;
            }
            break;

        case STMT_IFACE:
6915 6916
            debugPrint("%s() LN%d, stmt=%p",
                    __func__, __LINE__, pThreadInfo->stmt);
6917
            if (0 != taos_stmt_execute(pThreadInfo->stmt)) {
6918
                errorPrint2("%s() LN%d, failied to execute insert statement. reason: %s\n",
6919
                        __func__, __LINE__, taos_stmt_errstr(pThreadInfo->stmt));
6920 6921

                fprintf(stderr, "\n\033[31m === Please reduce batch number if WAL size exceeds limit. ===\033[0m\n\n");
6922
                exit(EXIT_FAILURE);
6923 6924 6925 6926 6927
            }
            affectedRows = k;
            break;

        default:
6928
            errorPrint2("%s() LN%d: unknown insert mode: %d\n",
6929
                    __func__, __LINE__, stbInfo->iface);
6930
            affectedRows = 0;
6931
    }
6932

6933
    return affectedRows;
6934 6935
}

6936 6937
static void getTableName(char *pTblName,
        threadInfo* pThreadInfo, uint64_t tableSeq)
6938
{
6939 6940 6941 6942
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    if (stbInfo) {
        if (AUTO_CREATE_SUBTBL != stbInfo->autoCreateTable) {
            if (stbInfo->childTblLimit > 0) {
6943
                snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s",
6944 6945
                        stbInfo->childTblName +
                        (tableSeq - stbInfo->childTblOffset) * TSDB_TABLE_NAME_LEN);
6946 6947 6948 6949 6950 6951
            } else {
                verbosePrint("[%d] %s() LN%d: from=%"PRIu64" count=%"PRId64" seq=%"PRIu64"\n",
                        pThreadInfo->threadID, __func__, __LINE__,
                        pThreadInfo->start_table_from,
                        pThreadInfo->ntables, tableSeq);
                snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s",
6952
                        stbInfo->childTblName + tableSeq * TSDB_TABLE_NAME_LEN);
6953 6954 6955
            }
        } else {
            snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s%"PRIu64"",
6956
                    stbInfo->childTblPrefix, tableSeq);
6957
        }
6958
    } else {
6959 6960
        snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s%"PRIu64"",
                g_args.tb_prefix, tableSeq);
6961 6962 6963
    }
}

6964
static int32_t generateDataTailWithoutStb(
6965
        uint32_t batch, char* buffer,
6966
        int64_t remainderBufLen, int64_t insertRows,
6967
        uint64_t recordFrom, int64_t startTime,
6968
        /* int64_t *pSamplePos, */int64_t *dataLen) {
6969

6970 6971
    uint64_t len = 0;
    char *pstr = buffer;
6972

6973
    verbosePrint("%s() LN%d batch=%d\n", __func__, __LINE__, batch);
6974

6975 6976
    int32_t k = 0;
    for (k = 0; k < batch;) {
6977
        char *data = pstr;
6978
        memset(data, 0, MAX_DATA_SIZE);
6979

6980
        int64_t retLen = 0;
6981

6982
        char *data_type = g_args.data_type;
6983
        int lenOfBinary = g_args.binwidth;
6984

6985 6986 6987
        if (g_args.disorderRatio) {
            retLen = generateData(data, data_type,
                    startTime + getTSRandTail(
6988
                        g_args.timestamp_step, k,
6989 6990 6991 6992 6993
                        g_args.disorderRatio,
                        g_args.disorderRange),
                    lenOfBinary);
        } else {
            retLen = generateData(data, data_type,
6994
                    startTime + g_args.timestamp_step * k,
6995 6996
                    lenOfBinary);
        }
6997

6998 6999
        if (len > remainderBufLen)
            break;
7000

7001
        pstr += retLen;
7002 7003 7004
        k++;
        len += retLen;
        remainderBufLen -= retLen;
7005

7006 7007
        verbosePrint("%s() LN%d len=%"PRIu64" k=%d \nbuffer=%s\n",
                __func__, __LINE__, len, k, buffer);
7008

7009
        recordFrom ++;
7010

7011 7012 7013
        if (recordFrom >= insertRows) {
            break;
        }
7014
    }
7015

7016 7017
    *dataLen = len;
    return k;
7018 7019
}

7020 7021 7022 7023 7024 7025 7026 7027 7028 7029 7030 7031 7032 7033 7034 7035 7036
static int64_t getTSRandTail(int64_t timeStampStep, int32_t seq,
        int disorderRatio, int disorderRange)
{
    int64_t randTail = timeStampStep * seq;
    if (disorderRatio > 0) {
        int rand_num = taosRandom() % 100;
        if(rand_num < disorderRatio) {
            randTail = (randTail +
                    (taosRandom() % disorderRange + 1)) * (-1);
            debugPrint("rand data generated, back %"PRId64"\n", randTail);
        }
    }

    return randTail;
}

static int32_t generateStbDataTail(
7037
        SSuperTable* stbInfo,
7038
        uint32_t batch, char* buffer,
7039
        int64_t remainderBufLen, int64_t insertRows,
7040
        uint64_t recordFrom, int64_t startTime,
7041
        int64_t *pSamplePos, int64_t *dataLen) {
7042
    uint64_t len = 0;
7043

7044
    char *pstr = buffer;
7045

7046
    bool tsRand;
7047
    if (0 == strncasecmp(stbInfo->dataSource, "rand", strlen("rand"))) {
7048 7049 7050 7051 7052 7053
        tsRand = true;
    } else {
        tsRand = false;
    }
    verbosePrint("%s() LN%d batch=%u buflen=%"PRId64"\n",
            __func__, __LINE__, batch, remainderBufLen);
7054

7055 7056
    int32_t k;
    for (k = 0; k < batch;) {
7057
        char *data = pstr;
7058

7059
        int64_t lenOfRow = 0;
7060

7061
        if (tsRand) {
7062 7063
            if (stbInfo->disorderRatio > 0) {
                lenOfRow = generateStbRowData(stbInfo, data,
7064
                        remainderBufLen,
7065
                        startTime + getTSRandTail(
7066 7067 7068
                            stbInfo->timeStampStep, k,
                            stbInfo->disorderRatio,
                            stbInfo->disorderRange)
7069 7070
                        );
            } else {
7071
                lenOfRow = generateStbRowData(stbInfo, data,
7072
                        remainderBufLen,
7073
                        startTime + stbInfo->timeStampStep * k
7074 7075 7076 7077 7078 7079
                        );
            }
        } else {
            lenOfRow = getRowDataFromSample(
                    data,
                    (remainderBufLen < MAX_DATA_SIZE)?remainderBufLen:MAX_DATA_SIZE,
7080 7081
                    startTime + stbInfo->timeStampStep * k,
                    stbInfo,
7082 7083
                    pSamplePos);
        }
7084

7085 7086 7087 7088
        if (lenOfRow == 0) {
            data[0] = '\0';
            break;
        }
7089 7090 7091
        if ((lenOfRow + 1) > remainderBufLen) {
            break;
        }
7092

7093
        pstr += lenOfRow;
7094 7095 7096
        k++;
        len += lenOfRow;
        remainderBufLen -= lenOfRow;
7097

7098 7099
        verbosePrint("%s() LN%d len=%"PRIu64" k=%u \nbuffer=%s\n",
                __func__, __LINE__, len, k, buffer);
7100

7101
        recordFrom ++;
7102

7103 7104 7105
        if (recordFrom >= insertRows) {
            break;
        }
7106
    }
7107

7108 7109
    *dataLen = len;
    return k;
7110
}
7111

7112 7113 7114

static int generateSQLHeadWithoutStb(char *tableName,
        char *dbName,
7115
        char *buffer, int remainderBufLen)
7116
{
7117
    int len;
7118

7119
    char headBuf[HEAD_BUFF_LEN];
7120

7121 7122 7123 7124 7125 7126
    len = snprintf(
            headBuf,
            HEAD_BUFF_LEN,
            "%s.%s values",
            dbName,
            tableName);
7127

7128 7129
    if (len > remainderBufLen)
        return -1;
7130

7131
    tstrncpy(buffer, headBuf, len + 1);
7132

7133
    return len;
7134 7135 7136
}

static int generateStbSQLHead(
7137
        SSuperTable* stbInfo,
7138
        char *tableName, int64_t tableSeq,
7139 7140 7141
        char *dbName,
        char *buffer, int remainderBufLen)
{
7142
    int len;
7143

7144
    char headBuf[HEAD_BUFF_LEN];
7145

7146
    if (AUTO_CREATE_SUBTBL == stbInfo->autoCreateTable) {
7147
        char* tagsValBuf = NULL;
7148 7149
        if (0 == stbInfo->tagSource) {
            tagsValBuf = generateTagValuesForStb(stbInfo, tableSeq);
7150
        } else {
7151
            tagsValBuf = getTagValueFromTagSample(
7152 7153
                    stbInfo,
                    tableSeq % stbInfo->tagSampleCount);
7154 7155
        }
        if (NULL == tagsValBuf) {
7156
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
7157 7158 7159
                    __func__, __LINE__);
            return -1;
        }
7160

7161 7162 7163 7164 7165 7166 7167
        len = snprintf(
                headBuf,
                HEAD_BUFF_LEN,
                "%s.%s using %s.%s TAGS%s values",
                dbName,
                tableName,
                dbName,
7168
                stbInfo->stbName,
7169 7170
                tagsValBuf);
        tmfree(tagsValBuf);
7171
    } else if (TBL_ALREADY_EXISTS == stbInfo->childTblExists) {
7172 7173 7174 7175 7176 7177
        len = snprintf(
                headBuf,
                HEAD_BUFF_LEN,
                "%s.%s values",
                dbName,
                tableName);
7178
    } else {
7179 7180 7181 7182 7183 7184 7185
        len = snprintf(
                headBuf,
                HEAD_BUFF_LEN,
                "%s.%s values",
                dbName,
                tableName);
    }
7186

7187 7188
    if (len > remainderBufLen)
        return -1;
7189

7190
    tstrncpy(buffer, headBuf, len + 1);
7191

7192
    return len;
7193 7194
}

7195
static int32_t generateStbInterlaceData(
7196
        threadInfo *pThreadInfo,
7197 7198 7199
        char *tableName, uint32_t batchPerTbl,
        uint64_t i,
        uint32_t batchPerTblTimes,
7200
        uint64_t tableSeq,
7201
        char *buffer,
7202
        int64_t insertRows,
7203
        int64_t startTime,
7204
        uint64_t *pRemainderBufLen)
7205
{
7206 7207
    assert(buffer);
    char *pstr = buffer;
7208

7209
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
7210
    int headLen = generateStbSQLHead(
7211
            stbInfo,
7212 7213
            tableName, tableSeq, pThreadInfo->db_name,
            pstr, *pRemainderBufLen);
7214

7215 7216 7217 7218 7219
    if (headLen <= 0) {
        return 0;
    }
    // generate data buffer
    verbosePrint("[%d] %s() LN%d i=%"PRIu64" buffer:\n%s\n",
7220
            pThreadInfo->threadID, __func__, __LINE__, i, buffer);
7221

7222 7223
    pstr += headLen;
    *pRemainderBufLen -= headLen;
7224

7225
    int64_t dataLen = 0;
7226

7227
    verbosePrint("[%d] %s() LN%d i=%"PRIu64" batchPerTblTimes=%u batchPerTbl = %u\n",
7228 7229
            pThreadInfo->threadID, __func__, __LINE__,
            i, batchPerTblTimes, batchPerTbl);
7230

7231
    if (0 == strncasecmp(stbInfo->startTimestamp, "now", 3)) {
7232 7233
        startTime = taosGetTimestamp(pThreadInfo->time_precision);
    }
7234

7235
    int32_t k = generateStbDataTail(
7236
            stbInfo,
7237 7238 7239
            batchPerTbl, pstr, *pRemainderBufLen, insertRows, 0,
            startTime,
            &(pThreadInfo->samplePos), &dataLen);
7240

7241 7242 7243 7244 7245 7246 7247 7248 7249 7250
    if (k == batchPerTbl) {
        pstr += dataLen;
        *pRemainderBufLen -= dataLen;
    } else {
        debugPrint("%s() LN%d, generated data tail: %u, not equal batch per table: %u\n",
                __func__, __LINE__, k, batchPerTbl);
        pstr -= headLen;
        pstr[0] = '\0';
        k = 0;
    }
7251

7252
    return k;
7253 7254 7255
}

static int64_t generateInterlaceDataWithoutStb(
7256
        char *tableName, uint32_t batch,
7257 7258 7259
        uint64_t tableSeq,
        char *dbName, char *buffer,
        int64_t insertRows,
7260
        int64_t startTime,
7261 7262
        uint64_t *pRemainderBufLen)
{
7263 7264
    assert(buffer);
    char *pstr = buffer;
7265

7266 7267
    int headLen = generateSQLHeadWithoutStb(
            tableName, dbName,
7268 7269
            pstr, *pRemainderBufLen);

7270 7271 7272
    if (headLen <= 0) {
        return 0;
    }
7273

7274 7275
    pstr += headLen;
    *pRemainderBufLen -= headLen;
7276

7277
    int64_t dataLen = 0;
7278

7279
    int32_t k = generateDataTailWithoutStb(
7280
            batch, pstr, *pRemainderBufLen, insertRows, 0,
7281
            startTime,
7282
            &dataLen);
7283

7284 7285 7286 7287 7288 7289 7290 7291 7292 7293 7294 7295
    if (k == batch) {
        pstr += dataLen;
        *pRemainderBufLen -= dataLen;
    } else {
        debugPrint("%s() LN%d, generated data tail: %d, not equal batch per table: %u\n",
                __func__, __LINE__, k, batch);
        pstr -= headLen;
        pstr[0] = '\0';
        k = 0;
    }

    return k;
7296 7297
}

7298 7299
static int32_t prepareStmtBindArrayByType(
        TAOS_BIND *bind,
7300
        char data_type, int32_t dataLen,
7301 7302 7303
        int32_t timePrec,
        char *value)
{
7304
    int32_t *bind_int;
7305
    uint32_t *bind_uint;
7306
    int64_t *bind_bigint;
7307
    uint64_t *bind_ubigint;
7308 7309 7310 7311 7312
    float   *bind_float;
    double  *bind_double;
    int8_t  *bind_bool;
    int64_t *bind_ts2;
    int16_t *bind_smallint;
7313
    uint16_t *bind_usmallint;
7314
    int8_t  *bind_tinyint;
7315
    uint8_t  *bind_utinyint;
7316 7317 7318 7319 7320 7321 7322 7323 7324

    switch(data_type) {
        case TSDB_DATA_TYPE_BINARY:
            if (dataLen > TSDB_MAX_BINARY_LEN) {
                errorPrint2("binary length overflow, max size:%u\n",
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
            char *bind_binary;
7325

7326 7327 7328 7329 7330 7331 7332 7333 7334 7335
            bind->buffer_type = TSDB_DATA_TYPE_BINARY;
            if (value) {
                bind_binary = calloc(1, strlen(value) + 1);
                strncpy(bind_binary, value, strlen(value));
                bind->buffer_length = strlen(bind_binary);
            } else {
                bind_binary = calloc(1, dataLen + 1);
                rand_string(bind_binary, dataLen);
                bind->buffer_length = dataLen;
            }
7336

7337 7338 7339 7340
            bind->length = &bind->buffer_length;
            bind->buffer = bind_binary;
            bind->is_null = NULL;
            break;
7341

7342 7343 7344 7345 7346 7347 7348
        case TSDB_DATA_TYPE_NCHAR:
            if (dataLen > TSDB_MAX_BINARY_LEN) {
                errorPrint2("nchar length overflow, max size:%u\n",
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
            char *bind_nchar;
7349

7350 7351 7352 7353 7354 7355 7356 7357
            bind->buffer_type = TSDB_DATA_TYPE_NCHAR;
            if (value) {
                bind_nchar = calloc(1, strlen(value) + 1);
                strncpy(bind_nchar, value, strlen(value));
            } else {
                bind_nchar = calloc(1, dataLen + 1);
                rand_string(bind_nchar, dataLen);
            }
7358

7359 7360 7361 7362 7363
            bind->buffer_length = strlen(bind_nchar);
            bind->buffer = bind_nchar;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7364

7365 7366 7367
        case TSDB_DATA_TYPE_INT:
            bind_int = malloc(sizeof(int32_t));
            assert(bind_int);
7368

7369 7370 7371 7372 7373 7374 7375 7376 7377 7378 7379
            if (value) {
                *bind_int = atoi(value);
            } else {
                *bind_int = rand_int();
            }
            bind->buffer_type = TSDB_DATA_TYPE_INT;
            bind->buffer_length = sizeof(int32_t);
            bind->buffer = bind_int;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7380 7381 7382 7383 7384 7385 7386 7387 7388 7389 7390 7391 7392 7393 7394 7395
        
        case TSDB_DATA_TYPE_UINT:
            bind_uint = malloc(sizeof(uint32_t));
            assert(bind_uint);

            if (value) {
                *bind_uint = atoi(value);
            } else {
                *bind_uint = rand_int();
            }
            bind->buffer_type = TSDB_DATA_TYPE_UINT;
            bind->buffer_length = sizeof(uint32_t);
            bind->buffer = bind_uint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7396

7397 7398 7399
        case TSDB_DATA_TYPE_BIGINT:
            bind_bigint = malloc(sizeof(int64_t));
            assert(bind_bigint);
7400

7401 7402 7403 7404 7405 7406 7407 7408 7409 7410 7411
            if (value) {
                *bind_bigint = atoll(value);
            } else {
                *bind_bigint = rand_bigint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_BIGINT;
            bind->buffer_length = sizeof(int64_t);
            bind->buffer = bind_bigint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7412

7413 7414 7415 7416 7417 7418 7419 7420 7421 7422 7423 7424 7425 7426 7427 7428
        case TSDB_DATA_TYPE_UBIGINT:
            bind_ubigint = malloc(sizeof(uint64_t));
            assert(bind_ubigint);

            if (value) {
                *bind_ubigint = atoll(value);
            } else {
                *bind_ubigint = rand_bigint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_UBIGINT;
            bind->buffer_length = sizeof(uint64_t);
            bind->buffer = bind_ubigint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

7429 7430 7431 7432 7433 7434
        case TSDB_DATA_TYPE_FLOAT:
            bind_float = malloc(sizeof(float));
            assert(bind_float);

            if (value) {
                *bind_float = (float)atof(value);
7435
            } else {
7436
                *bind_float = rand_float();
7437
            }
7438 7439 7440 7441 7442 7443 7444 7445 7446 7447 7448 7449 7450 7451 7452 7453 7454 7455 7456 7457 7458 7459 7460 7461 7462 7463 7464 7465 7466 7467 7468 7469 7470 7471 7472 7473 7474 7475 7476
            bind->buffer_type = TSDB_DATA_TYPE_FLOAT;
            bind->buffer_length = sizeof(float);
            bind->buffer = bind_float;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

        case TSDB_DATA_TYPE_DOUBLE:
            bind_double = malloc(sizeof(double));
            assert(bind_double);

            if (value) {
                *bind_double = atof(value);
            } else {
                *bind_double = rand_double();
            }
            bind->buffer_type = TSDB_DATA_TYPE_DOUBLE;
            bind->buffer_length = sizeof(double);
            bind->buffer = bind_double;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

        case TSDB_DATA_TYPE_SMALLINT:
            bind_smallint = malloc(sizeof(int16_t));
            assert(bind_smallint);

            if (value) {
                *bind_smallint = (int16_t)atoi(value);
            } else {
                *bind_smallint = rand_smallint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_SMALLINT;
            bind->buffer_length = sizeof(int16_t);
            bind->buffer = bind_smallint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

7477 7478 7479 7480 7481 7482 7483 7484 7485 7486 7487 7488 7489 7490 7491 7492
        case TSDB_DATA_TYPE_USMALLINT:
            bind_usmallint = malloc(sizeof(uint16_t));
            assert(bind_usmallint);

            if (value) {
                *bind_usmallint = (uint16_t)atoi(value);
            } else {
                *bind_usmallint = rand_smallint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_SMALLINT;
            bind->buffer_length = sizeof(uint16_t);
            bind->buffer = bind_usmallint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

7493 7494 7495 7496 7497 7498 7499 7500 7501 7502 7503 7504 7505 7506 7507 7508
        case TSDB_DATA_TYPE_TINYINT:
            bind_tinyint = malloc(sizeof(int8_t));
            assert(bind_tinyint);

            if (value) {
                *bind_tinyint = (int8_t)atoi(value);
            } else {
                *bind_tinyint = rand_tinyint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_TINYINT;
            bind->buffer_length = sizeof(int8_t);
            bind->buffer = bind_tinyint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

7509 7510 7511 7512 7513 7514 7515 7516 7517 7518 7519 7520 7521 7522 7523 7524
        case TSDB_DATA_TYPE_UTINYINT:
            bind_utinyint = malloc(sizeof(uint8_t));
            assert(bind_utinyint);

            if (value) {
                *bind_utinyint = (int8_t)atoi(value);
            } else {
                *bind_utinyint = rand_tinyint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_UTINYINT;
            bind->buffer_length = sizeof(uint8_t);
            bind->buffer = bind_utinyint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

7525 7526 7527 7528 7529 7530 7531 7532 7533 7534 7535 7536 7537 7538 7539 7540 7541 7542 7543
        case TSDB_DATA_TYPE_BOOL:
            bind_bool = malloc(sizeof(int8_t));
            assert(bind_bool);

            if (value) {
                if (strncasecmp(value, "true", 4)) {
                    *bind_bool = true;
                } else {
                    *bind_bool = false;
                }
            } else {
                *bind_bool = rand_bool();
            }
            bind->buffer_type = TSDB_DATA_TYPE_BOOL;
            bind->buffer_length = sizeof(int8_t);
            bind->buffer = bind_bool;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7544

7545 7546 7547 7548 7549 7550 7551 7552 7553 7554 7555 7556
        case TSDB_DATA_TYPE_TIMESTAMP:
            bind_ts2 = malloc(sizeof(int64_t));
            assert(bind_ts2);

            if (value) {
                if (strchr(value, ':') && strchr(value, '-')) {
                    int i = 0;
                    while(value[i] != '\0') {
                        if (value[i] == '\"' || value[i] == '\'') {
                            value[i] = ' ';
                        }
                        i++;
7557
                    }
7558 7559 7560 7561 7562 7563 7564 7565 7566 7567 7568
                    int64_t tmpEpoch;
                    if (TSDB_CODE_SUCCESS != taosParseTime(
                                value, &tmpEpoch, strlen(value),
                                timePrec, 0)) {
                        free(bind_ts2);
                        errorPrint2("Input %s, time format error!\n", value);
                        return -1;
                    }
                    *bind_ts2 = tmpEpoch;
                } else {
                    *bind_ts2 = atoll(value);
7569 7570
                }
            } else {
7571
                *bind_ts2 = rand_bigint();
7572
            }
7573 7574 7575 7576 7577 7578 7579 7580 7581 7582 7583 7584 7585
            bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
            bind->buffer_length = sizeof(int64_t);
            bind->buffer = bind_ts2;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;

        case TSDB_DATA_TYPE_NULL:
            break;

        default:
            errorPrint2("Not support data type: %d\n", data_type);
            exit(EXIT_FAILURE);
7586 7587 7588 7589 7590 7591 7592
    }

    return 0;
}

static int32_t prepareStmtBindArrayByTypeForRand(
        TAOS_BIND *bind,
7593
        char data_type, int32_t dataLen,
7594 7595 7596
        int32_t timePrec,
        char **ptr,
        char *value)
7597
{
7598
    int32_t *bind_int;
7599
    uint32_t *bind_uint;
7600
    int64_t *bind_bigint;
7601
    uint64_t *bind_ubigint;
7602 7603 7604
    float   *bind_float;
    double  *bind_double;
    int16_t *bind_smallint;
7605
    uint16_t *bind_usmallint;
7606
    int8_t  *bind_tinyint;
7607
    uint8_t  *bind_utinyint;
7608 7609 7610 7611 7612
    int8_t  *bind_bool;
    int64_t *bind_ts2;

    switch(data_type) {
        case TSDB_DATA_TYPE_BINARY:
7613

7614 7615 7616 7617 7618 7619
            if (dataLen > TSDB_MAX_BINARY_LEN) {
                errorPrint2("binary length overflow, max size:%u\n",
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
            char *bind_binary = (char *)*ptr;
7620

7621 7622 7623 7624 7625 7626 7627 7628
            bind->buffer_type = TSDB_DATA_TYPE_BINARY;
            if (value) {
                strncpy(bind_binary, value, strlen(value));
                bind->buffer_length = strlen(bind_binary);
            } else {
                rand_string(bind_binary, dataLen);
                bind->buffer_length = dataLen;
            }
7629

7630 7631 7632
            bind->length = &bind->buffer_length;
            bind->buffer = bind_binary;
            bind->is_null = NULL;
7633

7634 7635
            *ptr += bind->buffer_length;
            break;
7636

7637 7638 7639 7640 7641 7642 7643
        case TSDB_DATA_TYPE_NCHAR:
            if (dataLen > TSDB_MAX_BINARY_LEN) {
                errorPrint2("nchar length overflow, max size: %u\n",
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
            char *bind_nchar = (char *)*ptr;
7644

7645 7646 7647 7648 7649 7650
            bind->buffer_type = TSDB_DATA_TYPE_NCHAR;
            if (value) {
                strncpy(bind_nchar, value, strlen(value));
            } else {
                rand_string(bind_nchar, dataLen);
            }
7651

7652 7653 7654 7655
            bind->buffer_length = strlen(bind_nchar);
            bind->buffer = bind_nchar;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7656

7657 7658
            *ptr += bind->buffer_length;
            break;
7659

7660 7661
        case TSDB_DATA_TYPE_INT:
            bind_int = (int32_t *)*ptr;
7662

7663 7664 7665 7666 7667 7668 7669 7670 7671 7672
            if (value) {
                *bind_int = atoi(value);
            } else {
                *bind_int = rand_int();
            }
            bind->buffer_type = TSDB_DATA_TYPE_INT;
            bind->buffer_length = sizeof(int32_t);
            bind->buffer = bind_int;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7673

7674 7675
            *ptr += bind->buffer_length;
            break;
7676

7677 7678 7679 7680 7681 7682 7683 7684 7685 7686 7687 7688 7689 7690 7691 7692 7693
        case TSDB_DATA_TYPE_UINT:
            bind_uint = (uint32_t *)*ptr;

            if (value) {
                *bind_uint = atoi(value);
            } else {
                *bind_uint = rand_int();
            }
            bind->buffer_type = TSDB_DATA_TYPE_UINT;
            bind->buffer_length = sizeof(uint32_t);
            bind->buffer = bind_uint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

7694 7695
        case TSDB_DATA_TYPE_BIGINT:
            bind_bigint = (int64_t *)*ptr;
7696

7697 7698 7699 7700 7701 7702 7703 7704 7705 7706
            if (value) {
                *bind_bigint = atoll(value);
            } else {
                *bind_bigint = rand_bigint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_BIGINT;
            bind->buffer_length = sizeof(int64_t);
            bind->buffer = bind_bigint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7707

7708 7709
            *ptr += bind->buffer_length;
            break;
7710

7711 7712 7713 7714 7715 7716 7717 7718 7719 7720 7721 7722 7723 7724 7725 7726 7727
        case TSDB_DATA_TYPE_UBIGINT:
            bind_ubigint = (uint64_t *)*ptr;

            if (value) {
                *bind_ubigint = atoll(value);
            } else {
                *bind_ubigint = rand_bigint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_UBIGINT;
            bind->buffer_length = sizeof(uint64_t);
            bind->buffer = bind_ubigint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

7728 7729
        case TSDB_DATA_TYPE_FLOAT:
            bind_float = (float *)*ptr;
7730

7731 7732 7733 7734 7735 7736 7737 7738 7739 7740
            if (value) {
                *bind_float = (float)atof(value);
            } else {
                *bind_float = rand_float();
            }
            bind->buffer_type = TSDB_DATA_TYPE_FLOAT;
            bind->buffer_length = sizeof(float);
            bind->buffer = bind_float;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7741

7742 7743
            *ptr += bind->buffer_length;
            break;
7744

7745 7746
        case TSDB_DATA_TYPE_DOUBLE:
            bind_double = (double *)*ptr;
7747

7748 7749
            if (value) {
                *bind_double = atof(value);
7750
            } else {
7751
                *bind_double = rand_double();
7752
            }
7753 7754 7755 7756 7757 7758 7759 7760 7761 7762 7763 7764 7765 7766 7767 7768 7769 7770 7771 7772 7773 7774 7775 7776 7777 7778
            bind->buffer_type = TSDB_DATA_TYPE_DOUBLE;
            bind->buffer_length = sizeof(double);
            bind->buffer = bind_double;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

        case TSDB_DATA_TYPE_SMALLINT:
            bind_smallint = (int16_t *)*ptr;

            if (value) {
                *bind_smallint = (int16_t)atoi(value);
            } else {
                *bind_smallint = rand_smallint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_SMALLINT;
            bind->buffer_length = sizeof(int16_t);
            bind->buffer = bind_smallint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

7779 7780 7781 7782 7783 7784 7785 7786 7787 7788 7789 7790 7791 7792 7793 7794 7795
        case TSDB_DATA_TYPE_USMALLINT:
            bind_usmallint = (uint16_t *)*ptr;

            if (value) {
                *bind_usmallint = (uint16_t)atoi(value);
            } else {
                *bind_usmallint = rand_smallint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_USMALLINT;
            bind->buffer_length = sizeof(uint16_t);
            bind->buffer = bind_usmallint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

7796 7797 7798 7799 7800 7801 7802 7803 7804 7805 7806 7807 7808 7809 7810 7811 7812
        case TSDB_DATA_TYPE_TINYINT:
            bind_tinyint = (int8_t *)*ptr;

            if (value) {
                *bind_tinyint = (int8_t)atoi(value);
            } else {
                *bind_tinyint = rand_tinyint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_TINYINT;
            bind->buffer_length = sizeof(int8_t);
            bind->buffer = bind_tinyint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

7813 7814 7815 7816 7817 7818 7819 7820 7821 7822 7823 7824 7825 7826 7827 7828 7829
        case TSDB_DATA_TYPE_UTINYINT:
            bind_utinyint = (uint8_t *)*ptr;

            if (value) {
                *bind_utinyint = (uint8_t)atoi(value);
            } else {
                *bind_utinyint = rand_tinyint();
            }
            bind->buffer_type = TSDB_DATA_TYPE_UTINYINT;
            bind->buffer_length = sizeof(uint8_t);
            bind->buffer = bind_utinyint;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

7830 7831 7832 7833 7834 7835 7836 7837
        case TSDB_DATA_TYPE_BOOL:
            bind_bool = (int8_t *)*ptr;

            if (value) {
                if (strncasecmp(value, "true", 4)) {
                    *bind_bool = true;
                } else {
                    *bind_bool = false;
7838
                }
7839 7840 7841 7842 7843 7844 7845 7846 7847 7848 7849 7850 7851 7852 7853 7854 7855 7856 7857 7858 7859 7860 7861 7862 7863 7864 7865 7866 7867 7868 7869 7870 7871 7872
            } else {
                *bind_bool = rand_bool();
            }
            bind->buffer_type = TSDB_DATA_TYPE_BOOL;
            bind->buffer_length = sizeof(int8_t);
            bind->buffer = bind_bool;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            *ptr += bind->buffer_length;
            break;

        case TSDB_DATA_TYPE_TIMESTAMP:
            bind_ts2 = (int64_t *)*ptr;

            if (value) {
                if (strchr(value, ':') && strchr(value, '-')) {
                    int i = 0;
                    while(value[i] != '\0') {
                        if (value[i] == '\"' || value[i] == '\'') {
                            value[i] = ' ';
                        }
                        i++;
                    }
                    int64_t tmpEpoch;
                    if (TSDB_CODE_SUCCESS != taosParseTime(
                                value, &tmpEpoch, strlen(value),
                                timePrec, 0)) {
                        errorPrint2("Input %s, time format error!\n", value);
                        return -1;
                    }
                    *bind_ts2 = tmpEpoch;
                } else {
                    *bind_ts2 = atoll(value);
7873 7874
                }
            } else {
7875
                *bind_ts2 = rand_bigint();
7876
            }
7877 7878 7879 7880 7881
            bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
            bind->buffer_length = sizeof(int64_t);
            bind->buffer = bind_ts2;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7882

7883 7884 7885 7886 7887 7888
            *ptr += bind->buffer_length;
            break;

        default:
            errorPrint2("No support data type: %d\n", data_type);
            return -1;
7889 7890 7891 7892 7893 7894
    }

    return 0;
}

static int32_t prepareStmtWithoutStb(
7895
        threadInfo *pThreadInfo,
7896 7897 7898 7899 7900
        char *tableName,
        uint32_t batch,
        int64_t insertRows,
        int64_t recordFrom,
        int64_t startTime)
7901
{
7902
    TAOS_STMT *stmt = pThreadInfo->stmt;
7903 7904
    int ret = taos_stmt_set_tbname(stmt, tableName);
    if (ret != 0) {
7905
        errorPrint2("failed to execute taos_stmt_set_tbname(%s). return 0x%x. reason: %s\n",
7906
                tableName, ret, taos_stmt_errstr(stmt));
7907 7908 7909
        return ret;
    }

7910
    char *data_type = g_args.data_type;
7911

7912
    char *bindArray = malloc(sizeof(TAOS_BIND) * (g_args.columnCount + 1));
7913
    if (bindArray == NULL) {
7914
        errorPrint2("Failed to allocate %d bind params\n",
7915
                (g_args.columnCount + 1));
7916
        return -1;
7917 7918
    }

7919 7920 7921 7922 7923 7924
    int32_t k = 0;
    for (k = 0; k < batch;) {
        /* columnCount + 1 (ts) */

        TAOS_BIND *bind = (TAOS_BIND *)(bindArray + 0);

7925
        int64_t *bind_ts = pThreadInfo->bind_ts;
7926 7927

        bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
7928 7929 7930

        if (g_args.disorderRatio) {
            *bind_ts = startTime + getTSRandTail(
7931
                    g_args.timestamp_step, k,
7932 7933 7934
                    g_args.disorderRatio,
                    g_args.disorderRange);
        } else {
7935
            *bind_ts = startTime + g_args.timestamp_step * k;
7936
        }
7937 7938 7939 7940 7941
        bind->buffer_length = sizeof(int64_t);
        bind->buffer = bind_ts;
        bind->length = &bind->buffer_length;
        bind->is_null = NULL;

7942
        for (int i = 0; i < g_args.columnCount; i ++) {
7943 7944
            bind = (TAOS_BIND *)((char *)bindArray
                    + (sizeof(TAOS_BIND) * (i + 1)));
7945 7946 7947
            if ( -1 == prepareStmtBindArrayByType(
                        bind,
                        data_type[i],
7948
                        g_args.binwidth,
7949 7950
                        pThreadInfo->time_precision,
                        NULL)) {
7951
                free(bindArray);
7952 7953 7954
                return -1;
            }
        }
7955
        if (0 != taos_stmt_bind_param(stmt, (TAOS_BIND *)bindArray)) {
7956
            errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
7957 7958 7959
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            break;
        }
7960
        // if msg > 3MB, break
7961
        if (0 != taos_stmt_add_batch(stmt)) {
7962
            errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
7963 7964 7965
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            break;
        }
7966 7967 7968 7969 7970 7971 7972 7973

        k++;
        recordFrom ++;
        if (recordFrom >= insertRows) {
            break;
        }
    }

7974
    free(bindArray);
7975 7976 7977
    return k;
}

7978 7979 7980 7981 7982 7983 7984 7985 7986 7987 7988
static int32_t prepareStbStmtBindTag(
        char *bindArray, SSuperTable *stbInfo,
        char *tagsVal,
        int32_t timePrec)
{
    TAOS_BIND *tag;

    for (int t = 0; t < stbInfo->tagCount; t ++) {
        tag = (TAOS_BIND *)((char *)bindArray + (sizeof(TAOS_BIND) * t));
        if ( -1 == prepareStmtBindArrayByType(
                    tag,
7989
                    stbInfo->tags[t].data_type,
7990 7991 7992 7993 7994 7995 7996 7997 7998 7999 8000 8001 8002
                    stbInfo->tags[t].dataLen,
                    timePrec,
                    NULL)) {
            return -1;
        }
    }

    return 0;
}

static int32_t prepareStbStmtBindRand(
        int64_t *ts,
        char *bindArray, SSuperTable *stbInfo,
8003
        int64_t startTime, int32_t recSeq,
8004
        int32_t timePrec)
8005 8006 8007 8008 8009 8010 8011
{
    char data[MAX_DATA_SIZE];
    memset(data, 0, MAX_DATA_SIZE);
    char *ptr = data;

    TAOS_BIND *bind;

8012 8013
    for (int i = 0; i < stbInfo->columnCount + 1; i ++) {
        bind = (TAOS_BIND *)((char *)bindArray + (sizeof(TAOS_BIND) * i));
8014

8015 8016
        if (i == 0) {
            int64_t *bind_ts = ts;
8017

8018 8019 8020 8021 8022 8023 8024 8025 8026 8027 8028 8029 8030 8031 8032 8033 8034
            bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
            if (stbInfo->disorderRatio) {
                *bind_ts = startTime + getTSRandTail(
                        stbInfo->timeStampStep, recSeq,
                        stbInfo->disorderRatio,
                        stbInfo->disorderRange);
            } else {
                *bind_ts = startTime + stbInfo->timeStampStep * recSeq;
            }
            bind->buffer_length = sizeof(int64_t);
            bind->buffer = bind_ts;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;

            ptr += bind->buffer_length;
        } else if ( -1 == prepareStmtBindArrayByTypeForRand(
                    bind,
8035
                    stbInfo->columns[i-1].data_type,
8036 8037 8038 8039 8040 8041 8042
                    stbInfo->columns[i-1].dataLen,
                    timePrec,
                    &ptr,
                    NULL)) {
            return -1;
        }
    }
8043

8044 8045 8046 8047 8048 8049 8050 8051 8052 8053 8054 8055 8056 8057 8058 8059 8060 8061 8062 8063 8064 8065 8066 8067 8068 8069 8070 8071 8072 8073 8074 8075 8076 8077 8078 8079 8080 8081 8082 8083 8084 8085 8086 8087 8088 8089 8090 8091 8092 8093 8094 8095 8096 8097 8098 8099 8100 8101 8102 8103 8104 8105 8106 8107 8108 8109 8110 8111 8112 8113 8114 8115 8116 8117 8118 8119 8120 8121 8122 8123 8124 8125 8126 8127 8128 8129 8130 8131 8132 8133 8134 8135 8136 8137 8138 8139 8140 8141 8142 8143 8144 8145 8146 8147 8148 8149 8150 8151 8152 8153 8154 8155 8156 8157 8158 8159
    return 0;
}

UNUSED_FUNC static int32_t prepareStbStmtRand(
        threadInfo *pThreadInfo,
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
        uint64_t insertRows,
        uint64_t recordFrom,
        int64_t startTime)
{
    int ret;
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;

    if (AUTO_CREATE_SUBTBL == stbInfo->autoCreateTable) {
        char* tagsValBuf = NULL;

        if (0 == stbInfo->tagSource) {
            tagsValBuf = generateTagValuesForStb(stbInfo, tableSeq);
        } else {
            tagsValBuf = getTagValueFromTagSample(
                    stbInfo,
                    tableSeq % stbInfo->tagSampleCount);
        }

        if (NULL == tagsValBuf) {
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
                    __func__, __LINE__);
            return -1;
        }

        char *tagsArray = calloc(1, sizeof(TAOS_BIND) * stbInfo->tagCount);
        if (NULL == tagsArray) {
            tmfree(tagsValBuf);
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
                    __func__, __LINE__);
            return -1;
        }

        if (-1 == prepareStbStmtBindTag(
                    tagsArray, stbInfo, tagsValBuf, pThreadInfo->time_precision
                    /* is tag */)) {
            tmfree(tagsValBuf);
            tmfree(tagsArray);
            return -1;
        }

        ret = taos_stmt_set_tbname_tags(stmt, tableName, (TAOS_BIND *)tagsArray);

        tmfree(tagsValBuf);
        tmfree(tagsArray);

        if (0 != ret) {
            errorPrint2("%s() LN%d, stmt_set_tbname_tags() failed! reason: %s\n",
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    } else {
        ret = taos_stmt_set_tbname(stmt, tableName);
        if (0 != ret) {
            errorPrint2("%s() LN%d, stmt_set_tbname() failed! reason: %s\n",
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    }

    char *bindArray = calloc(1, sizeof(TAOS_BIND) * (stbInfo->columnCount + 1));
    if (bindArray == NULL) {
        errorPrint2("%s() LN%d, Failed to allocate %d bind params\n",
                __func__, __LINE__, (stbInfo->columnCount + 1));
        return -1;
    }

    uint32_t k;
    for (k = 0; k < batch;) {
        /* columnCount + 1 (ts) */
        if (-1 == prepareStbStmtBindRand(
                    pThreadInfo->bind_ts,
                    bindArray, stbInfo,
                    startTime, k,
                    pThreadInfo->time_precision
                    /* is column */)) {
            free(bindArray);
            return -1;
        }
        ret = taos_stmt_bind_param(stmt, (TAOS_BIND *)bindArray);
        if (0 != ret) {
            errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            free(bindArray);
            return -1;
        }
        // if msg > 3MB, break
        ret = taos_stmt_add_batch(stmt);
        if (0 != ret) {
            errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            free(bindArray);
            return -1;
        }

        k++;
        recordFrom ++;

        if (recordFrom >= insertRows) {
            break;
        }
    }

    free(bindArray);
    return k;
}

#if STMT_BIND_PARAM_BATCH == 1
8160
static int execStbBindParamBatch(
8161 8162 8163 8164 8165 8166 8167 8168 8169 8170 8171 8172 8173
        threadInfo *pThreadInfo,
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
        uint64_t insertRows,
        uint64_t recordFrom,
        int64_t startTime,
        int64_t *pSamplePos)
{
    int ret;
    TAOS_STMT *stmt = pThreadInfo->stmt;

    SSuperTable *stbInfo = pThreadInfo->stbInfo;
8174 8175 8176
    assert(stbInfo);

    uint32_t columnCount = pThreadInfo->stbInfo->columnCount;
8177

8178
    uint32_t thisBatch = MAX_SAMPLES - (*pSamplePos);
8179 8180 8181 8182 8183 8184 8185 8186 8187 8188 8189 8190 8191 8192 8193 8194 8195 8196 8197 8198 8199 8200

    if (thisBatch > batch) {
        thisBatch = batch;
    }
    verbosePrint("%s() LN%d, batch=%d pos=%"PRId64" thisBatch=%d\n",
            __func__, __LINE__, batch, *pSamplePos, thisBatch);

    memset(pThreadInfo->bindParams, 0,
            (sizeof(TAOS_MULTI_BIND) * (columnCount + 1)));
    memset(pThreadInfo->is_null, 0, thisBatch);

    for (int c = 0; c < columnCount + 1; c ++) {
        TAOS_MULTI_BIND *param = (TAOS_MULTI_BIND *)(pThreadInfo->bindParams + sizeof(TAOS_MULTI_BIND) * c);

        char data_type;

        if (c == 0) {
            data_type = TSDB_DATA_TYPE_TIMESTAMP;
            param->buffer_length = sizeof(int64_t);
            param->buffer = pThreadInfo->bind_ts_array;

        } else {
8201
            data_type = stbInfo->columns[c-1].data_type;
8202 8203 8204 8205 8206

            char *tmpP;

            switch(data_type) {
                case TSDB_DATA_TYPE_BINARY:
8207 8208 8209 8210 8211 8212 8213 8214 8215 8216 8217 8218 8219 8220
                    param->buffer_length =
                        stbInfo->columns[c-1].dataLen;

                    tmpP =
                        (char *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray
                                    +sizeof(char*)*(c-1)));

                    verbosePrint("%s() LN%d, tmpP=%p pos=%"PRId64" width=%"PRIxPTR" position=%"PRId64"\n",
                            __func__, __LINE__, tmpP, *pSamplePos, param->buffer_length,
                            (*pSamplePos) * param->buffer_length);

                    param->buffer = (void *)(tmpP + *pSamplePos * param->buffer_length);
                    break;

8221 8222
                case TSDB_DATA_TYPE_NCHAR:
                    param->buffer_length =
8223
                        stbInfo->columns[c-1].dataLen;
8224 8225 8226 8227 8228

                    tmpP =
                        (char *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray
                                    +sizeof(char*)*(c-1)));

8229 8230 8231
                    verbosePrint("%s() LN%d, tmpP=%p pos=%"PRId64" width=%"PRIxPTR" position=%"PRId64"\n",
                            __func__, __LINE__, tmpP, *pSamplePos, param->buffer_length,
                            (*pSamplePos) * param->buffer_length);
8232

8233
                    param->buffer = (void *)(tmpP + *pSamplePos * param->buffer_length);
8234 8235 8236
                    break;

                case TSDB_DATA_TYPE_INT:
8237
                case TSDB_DATA_TYPE_UINT:
8238
                    param->buffer_length = sizeof(int32_t);
8239
                    param->buffer =
8240
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8241
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8242 8243 8244
                    break;

                case TSDB_DATA_TYPE_TINYINT:
8245
                case TSDB_DATA_TYPE_UTINYINT:
8246
                    param->buffer_length = sizeof(int8_t);
8247
                    param->buffer =
8248 8249 8250
                        (void *)((uintptr_t)*(uintptr_t*)(
                                    stbInfo->sampleBindBatchArray
                                    +sizeof(char*)*(c-1))
8251
                                + stbInfo->columns[c-1].dataLen*(*pSamplePos));
8252 8253 8254
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
8255
                case TSDB_DATA_TYPE_USMALLINT:
8256
                    param->buffer_length = sizeof(int16_t);
8257
                    param->buffer =
8258
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8259
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8260 8261 8262
                    break;

                case TSDB_DATA_TYPE_BIGINT:
8263
                case TSDB_DATA_TYPE_UBIGINT:
8264
                    param->buffer_length = sizeof(int64_t);
8265
                    param->buffer =
8266
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8267
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8268 8269 8270 8271
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    param->buffer_length = sizeof(int8_t);
8272
                    param->buffer =
8273
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8274
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8275 8276 8277 8278
                    break;

                case TSDB_DATA_TYPE_FLOAT:
                    param->buffer_length = sizeof(float);
8279
                    param->buffer =
8280
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8281
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8282 8283 8284 8285
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    param->buffer_length = sizeof(double);
8286
                    param->buffer =
8287
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8288
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8289 8290 8291 8292
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    param->buffer_length = sizeof(int64_t);
8293
                    param->buffer =
8294
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8295
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8296 8297 8298 8299 8300 8301 8302 8303 8304 8305 8306 8307 8308 8309 8310 8311 8312 8313 8314 8315
                    break;

                default:
                    errorPrint("%s() LN%d, wrong data type: %d\n",
                            __func__,
                            __LINE__,
                            data_type);
                    exit(EXIT_FAILURE);

            }
        }

        param->buffer_type = data_type;
        param->length = malloc(sizeof(int32_t) * thisBatch);
        assert(param->length);

        for (int b = 0; b < thisBatch; b++) {
            if (param->buffer_type == TSDB_DATA_TYPE_NCHAR) {
                param->length[b] = strlen(
                     (char *)param->buffer + b *
8316
                         stbInfo->columns[c].dataLen
8317 8318 8319 8320 8321 8322 8323 8324 8325 8326 8327 8328 8329 8330 8331 8332 8333 8334 8335 8336 8337 8338 8339 8340 8341 8342 8343 8344
                        );
            } else {
                param->length[b] = param->buffer_length;
            }
        }
        param->is_null = pThreadInfo->is_null;
        param->num = thisBatch;
    }

    uint32_t k;
    for (k = 0; k < thisBatch;) {
        /* columnCount + 1 (ts) */
        if (stbInfo->disorderRatio) {
            *(pThreadInfo->bind_ts_array + k) = startTime + getTSRandTail(
                    stbInfo->timeStampStep, k,
                    stbInfo->disorderRatio,
                    stbInfo->disorderRange);
        } else {
            *(pThreadInfo->bind_ts_array + k) = startTime + stbInfo->timeStampStep * k;
        }

        debugPrint("%s() LN%d, k=%d ts=%"PRId64"\n",
                __func__, __LINE__,
                k, *(pThreadInfo->bind_ts_array +k));
        k++;
        recordFrom ++;

        (*pSamplePos) ++;
8345
        if ((*pSamplePos) == MAX_SAMPLES) {
8346 8347 8348 8349 8350 8351 8352 8353 8354 8355 8356 8357 8358 8359 8360 8361 8362 8363 8364 8365 8366 8367 8368 8369 8370 8371 8372 8373 8374 8375 8376 8377 8378 8379 8380 8381 8382 8383 8384 8385 8386 8387 8388 8389 8390 8391 8392 8393 8394 8395 8396 8397 8398 8399
            *pSamplePos = 0;
        }

        if (recordFrom >= insertRows) {
            break;
        }
    }

    ret = taos_stmt_bind_param_batch(stmt, (TAOS_MULTI_BIND *)pThreadInfo->bindParams);
    if (0 != ret) {
        errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
                __func__, __LINE__, taos_stmt_errstr(stmt));
        return -1;
    }

    for (int c = 0; c < stbInfo->columnCount + 1; c ++) {
        TAOS_MULTI_BIND *param = (TAOS_MULTI_BIND *)(pThreadInfo->bindParams + sizeof(TAOS_MULTI_BIND) * c);
        free(param->length);
    }

    // if msg > 3MB, break
    ret = taos_stmt_add_batch(stmt);
    if (0 != ret) {
        errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
                __func__, __LINE__, taos_stmt_errstr(stmt));
        return -1;
    }
    return k;
}

static int parseSamplefileToStmtBatch(
        SSuperTable* stbInfo)
{
    // char *sampleDataBuf = (stbInfo)?
    //    stbInfo->sampleDataBuf:g_sampleDataBuf;
    int32_t columnCount = (stbInfo)?stbInfo->columnCount:g_args.columnCount;
    char *sampleBindBatchArray = NULL;

    if (stbInfo) {
        stbInfo->sampleBindBatchArray = calloc(1, sizeof(uintptr_t *) * columnCount);
        sampleBindBatchArray = stbInfo->sampleBindBatchArray;
    } else {
        g_sampleBindBatchArray = calloc(1, sizeof(uintptr_t *) * columnCount);
        sampleBindBatchArray = g_sampleBindBatchArray;
    }
    assert(sampleBindBatchArray);

    for (int c = 0; c < columnCount; c++) {
        char data_type = (stbInfo)?stbInfo->columns[c].data_type:g_args.data_type[c];

        char *tmpP = NULL;

        switch(data_type) {
            case TSDB_DATA_TYPE_INT:
8400
            case TSDB_DATA_TYPE_UINT:
8401
                tmpP = calloc(1, sizeof(int) * MAX_SAMPLES);
8402 8403 8404 8405 8406
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_TINYINT:
8407
            case TSDB_DATA_TYPE_UTINYINT:
8408
                tmpP = calloc(1, sizeof(int8_t) * MAX_SAMPLES);
8409 8410 8411 8412 8413
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
8414
            case TSDB_DATA_TYPE_USMALLINT:
8415
                tmpP = calloc(1, sizeof(int16_t) * MAX_SAMPLES);
8416 8417 8418 8419 8420
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BIGINT:
8421
            case TSDB_DATA_TYPE_UBIGINT:
8422
                tmpP = calloc(1, sizeof(int64_t) * MAX_SAMPLES);
8423 8424 8425 8426 8427
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BOOL:
8428
                tmpP = calloc(1, sizeof(int8_t) * MAX_SAMPLES);
8429 8430 8431 8432 8433
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_FLOAT:
8434
                tmpP = calloc(1, sizeof(float) * MAX_SAMPLES);
8435 8436 8437 8438 8439
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_DOUBLE:
8440
                tmpP = calloc(1, sizeof(double) * MAX_SAMPLES);
8441 8442 8443 8444 8445 8446
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BINARY:
            case TSDB_DATA_TYPE_NCHAR:
8447
                tmpP = calloc(1, MAX_SAMPLES *
8448 8449 8450 8451 8452 8453
                        (((stbInfo)?stbInfo->columns[c].dataLen:g_args.binwidth)));
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_TIMESTAMP:
8454
                tmpP = calloc(1, sizeof(int64_t) * MAX_SAMPLES);
8455 8456 8457 8458 8459 8460 8461 8462 8463 8464 8465 8466 8467 8468
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            default:
                errorPrint("Unknown data type: %s\n",
                        (stbInfo)?stbInfo->columns[c].dataType:g_args.dataType[c]);
                exit(EXIT_FAILURE);
        }
    }

    char *sampleDataBuf = (stbInfo)?stbInfo->sampleDataBuf:g_sampleDataBuf;
    int64_t lenOfOneRow = (stbInfo)?stbInfo->lenOfOneRow:g_args.lenOfOneRow;

8469
    for (int i=0; i < MAX_SAMPLES; i++) {
8470 8471 8472 8473 8474 8475 8476 8477 8478 8479 8480 8481 8482 8483 8484 8485 8486 8487 8488 8489 8490 8491 8492 8493 8494 8495 8496 8497 8498 8499
        int cursor = 0;

        for (int c = 0; c < columnCount; c++) {
            char data_type = (stbInfo)?
                stbInfo->columns[c].data_type:
                g_args.data_type[c];
            char *restStr = sampleDataBuf
                + lenOfOneRow * i + cursor;
            int lengthOfRest = strlen(restStr);

            int index = 0;
            for (index = 0; index < lengthOfRest; index ++) {
                if (restStr[index] == ',') {
                    break;
                }
            }

            char *tmpStr = calloc(1, index + 1);
            if (NULL == tmpStr) {
                errorPrint2("%s() LN%d, Failed to allocate %d bind buffer\n",
                        __func__, __LINE__, index + 1);
                return -1;
            }

            strncpy(tmpStr, restStr, index);
            cursor += index + 1; // skip ',' too
            char *tmpP;

            switch(data_type) {
                case TSDB_DATA_TYPE_INT:
8500
                case TSDB_DATA_TYPE_UINT:
8501 8502 8503 8504 8505 8506 8507 8508 8509 8510 8511 8512 8513 8514 8515 8516 8517 8518
                    *((int32_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int32_t)*i)) =
                        atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_FLOAT:
                    *(float*)(((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(float)*i)) =
                        (float)atof(tmpStr);
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    *(double*)(((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(double)*i)) =
                        atof(tmpStr);
                    break;

                case TSDB_DATA_TYPE_TINYINT:
8519
                case TSDB_DATA_TYPE_UTINYINT:
8520 8521 8522 8523 8524 8525
                    *((int8_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int8_t)*i)) =
                        (int8_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
8526
                case TSDB_DATA_TYPE_USMALLINT:
8527 8528 8529 8530 8531 8532
                    *((int16_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int16_t)*i)) =
                        (int16_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_BIGINT:
8533
                case TSDB_DATA_TYPE_UBIGINT:
8534 8535 8536 8537 8538 8539 8540 8541 8542 8543 8544 8545 8546 8547 8548 8549 8550 8551 8552 8553 8554 8555 8556 8557 8558 8559 8560 8561 8562 8563 8564 8565 8566 8567 8568 8569 8570 8571 8572 8573 8574 8575 8576 8577 8578 8579 8580 8581 8582 8583 8584 8585 8586 8587 8588 8589 8590 8591 8592 8593 8594 8595 8596 8597 8598 8599 8600 8601 8602 8603 8604 8605 8606 8607 8608 8609 8610 8611 8612
                    *((int64_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int64_t)*i)) =
                        (int64_t)atol(tmpStr);
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    *((int8_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int8_t)*i)) =
                        (int8_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    *((int64_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int64_t)*i)) =
                        (int64_t)atol(tmpStr);
                    break;

                case TSDB_DATA_TYPE_BINARY:
                case TSDB_DATA_TYPE_NCHAR:
                    tmpP = (char *)(*(uintptr_t*)(sampleBindBatchArray
                            +sizeof(char*)*c));
                    strcpy(tmpP + i*
                        (((stbInfo)?stbInfo->columns[c].dataLen:g_args.binwidth))
                    , tmpStr);
                    break;

                default:
                    break;
            }

            free(tmpStr);
        }
    }

    return 0;
}

static int parseSampleToStmtBatchForThread(
        threadInfo *pThreadInfo, SSuperTable *stbInfo,
        uint32_t timePrec,
        uint32_t batch)
{
    uint32_t columnCount = (stbInfo)?stbInfo->columnCount:g_args.columnCount;

    pThreadInfo->bind_ts_array = malloc(sizeof(int64_t) * batch);
    assert(pThreadInfo->bind_ts_array);

    pThreadInfo->bindParams = malloc(sizeof(TAOS_MULTI_BIND) * (columnCount + 1));
    assert(pThreadInfo->bindParams);

    pThreadInfo->is_null = malloc(batch);
    assert(pThreadInfo->is_null);

    return 0;
}

static int parseStbSampleToStmtBatchForThread(
        threadInfo *pThreadInfo,
        SSuperTable *stbInfo,
        uint32_t timePrec,
        uint32_t batch)
{
    return parseSampleToStmtBatchForThread(
        pThreadInfo, stbInfo, timePrec, batch);
}

static int parseNtbSampleToStmtBatchForThread(
        threadInfo *pThreadInfo, uint32_t timePrec, uint32_t batch)
{
    return parseSampleToStmtBatchForThread(
        pThreadInfo, NULL, timePrec, batch);
}

#else
static int parseSampleToStmt(
        threadInfo *pThreadInfo,
        SSuperTable *stbInfo, uint32_t timePrec)
{
    pThreadInfo->sampleBindArray =
8613
        (char *)calloc(1, sizeof(char *) * MAX_SAMPLES);
8614 8615 8616
    if (pThreadInfo->sampleBindArray == NULL) {
        errorPrint2("%s() LN%d, Failed to allocate %"PRIu64" bind array buffer\n",
                __func__, __LINE__,
8617
                (uint64_t)sizeof(char *) * MAX_SAMPLES);
8618 8619 8620 8621 8622 8623 8624
        return -1;
    }

    int32_t columnCount = (stbInfo)?stbInfo->columnCount:g_args.columnCount;
    char *sampleDataBuf = (stbInfo)?stbInfo->sampleDataBuf:g_sampleDataBuf;
    int64_t lenOfOneRow = (stbInfo)?stbInfo->lenOfOneRow:g_args.lenOfOneRow;

8625
    for (int i=0; i < MAX_SAMPLES; i++) {
8626 8627 8628 8629 8630 8631 8632 8633 8634 8635 8636 8637 8638 8639 8640 8641 8642 8643 8644 8645 8646 8647 8648 8649 8650 8651 8652 8653 8654 8655 8656 8657 8658 8659 8660 8661 8662 8663 8664 8665 8666 8667 8668 8669 8670 8671 8672 8673 8674 8675 8676 8677 8678 8679 8680
        char *bindArray =
            calloc(1, sizeof(TAOS_BIND) * (columnCount + 1));
        if (bindArray == NULL) {
            errorPrint2("%s() LN%d, Failed to allocate %d bind params\n",
                    __func__, __LINE__, (columnCount + 1));
            return -1;
        }

        TAOS_BIND *bind;
        int cursor = 0;

        for (int c = 0; c < columnCount + 1; c++) {
            bind = (TAOS_BIND *)((char *)bindArray + (sizeof(TAOS_BIND) * c));

            if (c == 0) {
                bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
                bind->buffer_length = sizeof(int64_t);
                bind->buffer = NULL; //bind_ts;
                bind->length = &bind->buffer_length;
                bind->is_null = NULL;
            } else {
                char data_type = (stbInfo)?
                    stbInfo->columns[c-1].data_type:
                    g_args.data_type[c-1];
                int32_t dataLen = (stbInfo)?
                    stbInfo->columns[c-1].dataLen:
                    g_args.binwidth;
                char *restStr = sampleDataBuf
                    + lenOfOneRow * i + cursor;
                int lengthOfRest = strlen(restStr);

                int index = 0;
                for (index = 0; index < lengthOfRest; index ++) {
                    if (restStr[index] == ',') {
                        break;
                    }
                }

                char *bindBuffer = calloc(1, index + 1);
                if (bindBuffer == NULL) {
                    errorPrint2("%s() LN%d, Failed to allocate %d bind buffer\n",
                            __func__, __LINE__, index + 1);
                    return -1;
                }

                strncpy(bindBuffer, restStr, index);
                cursor += index + 1; // skip ',' too

                if (-1 == prepareStmtBindArrayByType(
                            bind,
                            data_type,
                            dataLen,
                            timePrec,
                            bindBuffer)) {
                    free(bindBuffer);
8681
                    free(bindArray);
8682 8683 8684 8685 8686 8687 8688 8689 8690 8691 8692 8693 8694 8695 8696 8697 8698 8699 8700 8701 8702 8703 8704 8705 8706 8707 8708 8709 8710
                    return -1;
                }
                free(bindBuffer);
            }
        }
        *((uintptr_t *)(pThreadInfo->sampleBindArray + (sizeof(char *)) * i)) =
            (uintptr_t)bindArray;
    }

    return 0;
}

static int parseStbSampleToStmt(
        threadInfo *pThreadInfo,
        SSuperTable *stbInfo, uint32_t timePrec)
{
    return parseSampleToStmt(
        pThreadInfo,
        stbInfo, timePrec);
}

static int parseNtbSampleToStmt(
        threadInfo *pThreadInfo,
        uint32_t timePrec)
{
    return parseSampleToStmt(
        pThreadInfo,
        NULL,
        timePrec);
8711
}
8712

8713 8714
static int32_t prepareStbStmtBindStartTime(
        char *tableName,
8715 8716
        int64_t *ts,
        char *bindArray, SSuperTable *stbInfo,
8717
        int64_t startTime, int32_t recSeq)
8718 8719
{
    TAOS_BIND *bind;
8720

8721
    bind = (TAOS_BIND *)bindArray;
8722

8723
    int64_t *bind_ts = ts;
8724

8725 8726 8727 8728 8729 8730 8731 8732
    bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
    if (stbInfo->disorderRatio) {
        *bind_ts = startTime + getTSRandTail(
                stbInfo->timeStampStep, recSeq,
                stbInfo->disorderRatio,
                stbInfo->disorderRange);
    } else {
        *bind_ts = startTime + stbInfo->timeStampStep * recSeq;
8733
    }
8734 8735 8736 8737

    verbosePrint("%s() LN%d, tableName: %s, bind_ts=%"PRId64"\n",
            __func__, __LINE__, tableName, *bind_ts);

8738 8739 8740 8741
    bind->buffer_length = sizeof(int64_t);
    bind->buffer = bind_ts;
    bind->length = &bind->buffer_length;
    bind->is_null = NULL;
8742 8743 8744 8745

    return 0;
}

8746
static uint32_t execBindParam(
8747
        threadInfo *pThreadInfo,
8748 8749 8750
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
8751 8752
        uint64_t insertRows,
        uint64_t recordFrom,
8753 8754
        int64_t startTime,
        int64_t *pSamplePos)
8755
{
8756
    int ret;
8757 8758
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;
8759

8760 8761
    uint32_t k;
    for (k = 0; k < batch;) {
8762 8763
        char *bindArray = (char *)(*((uintptr_t *)
                    (pThreadInfo->sampleBindArray + (sizeof(char *)) * (*pSamplePos))));
8764
        /* columnCount + 1 (ts) */
8765 8766
        if (-1 == prepareStbStmtBindStartTime(
                    tableName,
8767 8768
                    pThreadInfo->bind_ts,
                    bindArray, stbInfo,
8769
                    startTime, k
8770
                    /* is column */)) {
8771
            return -1;
8772
        }
8773 8774
        ret = taos_stmt_bind_param(stmt, (TAOS_BIND *)bindArray);
        if (0 != ret) {
8775
            errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
8776 8777 8778
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
8779
        // if msg > 3MB, break
8780 8781
        ret = taos_stmt_add_batch(stmt);
        if (0 != ret) {
8782
            errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
8783 8784 8785
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
8786

8787 8788
        k++;
        recordFrom ++;
8789

8790
        (*pSamplePos) ++;
8791
        if ((*pSamplePos) == MAX_SAMPLES) {
8792 8793 8794
            *pSamplePos = 0;
        }

8795 8796 8797 8798
        if (recordFrom >= insertRows) {
            break;
        }
    }
8799 8800 8801

    return k;
}
8802
#endif
8803

8804
static int32_t prepareStbStmt(
8805
        threadInfo *pThreadInfo,
8806 8807 8808
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
8809 8810 8811 8812 8813
        uint64_t insertRows,
        uint64_t recordFrom,
        int64_t startTime,
        int64_t *pSamplePos)
{
8814 8815 8816
    int ret;
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;
8817

8818 8819 8820 8821 8822 8823 8824 8825 8826 8827 8828 8829
    if (AUTO_CREATE_SUBTBL == stbInfo->autoCreateTable) {
        char* tagsValBuf = NULL;

        if (0 == stbInfo->tagSource) {
            tagsValBuf = generateTagValuesForStb(stbInfo, tableSeq);
        } else {
            tagsValBuf = getTagValueFromTagSample(
                    stbInfo,
                    tableSeq % stbInfo->tagSampleCount);
        }

        if (NULL == tagsValBuf) {
8830
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
8831 8832 8833 8834 8835 8836 8837
                    __func__, __LINE__);
            return -1;
        }

        char *tagsArray = calloc(1, sizeof(TAOS_BIND) * stbInfo->tagCount);
        if (NULL == tagsArray) {
            tmfree(tagsValBuf);
8838
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
8839 8840 8841 8842 8843 8844 8845 8846 8847 8848 8849 8850 8851 8852 8853 8854 8855 8856
                    __func__, __LINE__);
            return -1;
        }

        if (-1 == prepareStbStmtBindTag(
                    tagsArray, stbInfo, tagsValBuf, pThreadInfo->time_precision
                    /* is tag */)) {
            tmfree(tagsValBuf);
            tmfree(tagsArray);
            return -1;
        }

        ret = taos_stmt_set_tbname_tags(stmt, tableName, (TAOS_BIND *)tagsArray);

        tmfree(tagsValBuf);
        tmfree(tagsArray);

        if (0 != ret) {
8857
            errorPrint2("%s() LN%d, stmt_set_tbname_tags() failed! reason: %s\n",
8858 8859 8860 8861 8862 8863
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    } else {
        ret = taos_stmt_set_tbname(stmt, tableName);
        if (0 != ret) {
8864
            errorPrint2("%s() LN%d, stmt_set_tbname() failed! reason: %s\n",
8865 8866 8867 8868 8869
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    }

8870
#if STMT_BIND_PARAM_BATCH == 1
8871
    return execStbBindParamBatch(
8872 8873 8874 8875 8876 8877 8878 8879 8880 8881 8882 8883 8884 8885 8886 8887 8888 8889 8890
        pThreadInfo,
        tableName,
        tableSeq,
        batch,
        insertRows,
        recordFrom,
        startTime,
        pSamplePos);
#else
    return execBindParam(
        pThreadInfo,
        tableName,
        tableSeq,
        batch,
        insertRows,
        recordFrom,
        startTime,
        pSamplePos);
#endif
8891
}
8892 8893

static int32_t generateStbProgressiveData(
8894
        SSuperTable *stbInfo,
8895
        char *tableName,
8896
        int64_t tableSeq,
8897
        char *dbName, char *buffer,
8898
        int64_t insertRows,
8899
        uint64_t recordFrom, int64_t startTime, int64_t *pSamplePos,
8900
        int64_t *pRemainderBufLen)
8901
{
8902 8903
    assert(buffer != NULL);
    char *pstr = buffer;
8904

8905
    memset(pstr, 0, *pRemainderBufLen);
8906

8907
    int64_t headLen = generateStbSQLHead(
8908
            stbInfo,
8909 8910
            tableName, tableSeq, dbName,
            buffer, *pRemainderBufLen);
8911

8912 8913 8914 8915 8916
    if (headLen <= 0) {
        return 0;
    }
    pstr += headLen;
    *pRemainderBufLen -= headLen;
8917

8918
    int64_t dataLen;
8919

8920
    return generateStbDataTail(stbInfo,
8921
            g_args.reqPerReq, pstr, *pRemainderBufLen,
8922 8923 8924
            insertRows, recordFrom,
            startTime,
            pSamplePos, &dataLen);
8925 8926
}

8927
static int32_t generateProgressiveDataWithoutStb(
8928
        char *tableName,
8929
        /* int64_t tableSeq, */
8930 8931
        threadInfo *pThreadInfo, char *buffer,
        int64_t insertRows,
8932
        uint64_t recordFrom, int64_t startTime, /*int64_t *pSamplePos, */
8933 8934
        int64_t *pRemainderBufLen)
{
8935 8936
    assert(buffer != NULL);
    char *pstr = buffer;
8937

8938
    memset(buffer, 0, *pRemainderBufLen);
8939

8940 8941 8942
    int64_t headLen = generateSQLHeadWithoutStb(
            tableName, pThreadInfo->db_name,
            buffer, *pRemainderBufLen);
8943

8944 8945 8946 8947 8948
    if (headLen <= 0) {
        return 0;
    }
    pstr += headLen;
    *pRemainderBufLen -= headLen;
8949

8950
    int64_t dataLen;
8951

8952
    return generateDataTailWithoutStb(
8953
            g_args.reqPerReq, pstr, *pRemainderBufLen, insertRows, recordFrom,
8954 8955
            startTime,
            /*pSamplePos, */&dataLen);
8956
}
8957

8958 8959
static void printStatPerThread(threadInfo *pThreadInfo)
{
8960 8961 8962
    if (0 == pThreadInfo->totalDelay)
        pThreadInfo->totalDelay = 1;

8963
    fprintf(stderr, "====thread[%d] completed total inserted rows: %"PRIu64 ", total affected rows: %"PRIu64". %.2f records/second====\n",
8964 8965 8966
            pThreadInfo->threadID,
            pThreadInfo->totalInsertRows,
            pThreadInfo->totalAffectedRows,
8967 8968
            (double)(pThreadInfo->totalAffectedRows/((double)pThreadInfo->totalDelay/1000000.0))
            );
8969 8970
}

8971 8972 8973 8974
#if STMT_BIND_PARAM_BATCH == 1
// stmt sync write interlace data
static void* syncWriteInterlaceStmtBatch(threadInfo *pThreadInfo, uint32_t interlaceRows) {
    debugPrint("[%d] %s() LN%d: ### stmt interlace write\n",
8975
            pThreadInfo->threadID, __func__, __LINE__);
8976

8977
    int64_t insertRows;
8978
    int64_t timeStampStep;
8979
    uint64_t insert_interval;
8980

8981
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
8982

8983 8984
    if (stbInfo) {
        insertRows = stbInfo->insertRows;
8985
        timeStampStep = stbInfo->timeStampStep;
8986 8987 8988
        insert_interval = stbInfo->insertInterval;
    } else {
        insertRows = g_args.insertRows;
8989
        timeStampStep = g_args.timestamp_step;
8990 8991
        insert_interval = g_args.insert_interval;
    }
8992

8993 8994 8995 8996 8997
    debugPrint("[%d] %s() LN%d: start_table_from=%"PRIu64" ntables=%"PRId64" insertRows=%"PRIu64"\n",
            pThreadInfo->threadID, __func__, __LINE__,
            pThreadInfo->start_table_from,
            pThreadInfo->ntables, insertRows);

8998 8999
    uint64_t timesInterlace = (insertRows / interlaceRows) + 1;
    uint32_t precalcBatch = interlaceRows;
9000

9001 9002
    if (precalcBatch > g_args.reqPerReq)
        precalcBatch = g_args.reqPerReq;
9003

9004 9005
    if (precalcBatch > MAX_SAMPLES)
        precalcBatch = MAX_SAMPLES;
9006 9007 9008 9009 9010 9011 9012 9013 9014 9015 9016 9017

    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;

    uint64_t st = 0;
    uint64_t et = UINT64_MAX;

    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;

    uint64_t tableSeq = pThreadInfo->start_table_from;
9018
    int64_t startTime;
9019 9020 9021 9022 9023 9024

    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;

    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;
9025
    pThreadInfo->samplePos = 0;
9026

9027 9028
    for (int64_t interlace = 0;
            interlace < timesInterlace; interlace ++) {
9029 9030 9031 9032 9033
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }

9034 9035
        int64_t generated = 0;
        int64_t samplePos;
9036

9037
        for (; tableSeq < pThreadInfo->start_table_from + pThreadInfo->ntables; tableSeq ++) {
9038 9039 9040 9041 9042 9043 9044 9045
            char tableName[TSDB_TABLE_NAME_LEN];
            getTableName(tableName, pThreadInfo, tableSeq);
            if (0 == strlen(tableName)) {
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
                        pThreadInfo->threadID, __func__, __LINE__);
                return NULL;
            }

9046 9047 9048 9049 9050 9051
            samplePos = pThreadInfo->samplePos;
            startTime = pThreadInfo->start_time
                + interlace * interlaceRows * timeStampStep;
            uint64_t remainRecPerTbl =
                    insertRows - interlaceRows * interlace;
            uint64_t recPerTbl = 0;
9052

9053 9054 9055 9056 9057
            uint64_t remainPerInterlace;
            if (remainRecPerTbl > interlaceRows) {
                remainPerInterlace = interlaceRows;
            } else {
                remainPerInterlace = remainRecPerTbl;
9058 9059
            }

9060
            while(remainPerInterlace > 0) {
9061

9062 9063 9064 9065 9066 9067 9068 9069 9070 9071
                uint32_t batch;
                if (remainPerInterlace > precalcBatch) {
                    batch = precalcBatch;
                } else {
                    batch = remainPerInterlace;
                }
                debugPrint("[%d] %s() LN%d, tableName:%s, batch:%d startTime:%"PRId64"\n",
                        pThreadInfo->threadID,
                        __func__, __LINE__,
                        tableName, batch, startTime);
9072

9073 9074 9075 9076 9077 9078 9079 9080 9081 9082 9083 9084 9085 9086 9087 9088 9089 9090
                if (stbInfo) {
                    generated = prepareStbStmt(
                            pThreadInfo,
                            tableName,
                            tableSeq,
                            batch,
                            insertRows, 0,
                            startTime,
                            &samplePos);
                } else {
                    generated = prepareStmtWithoutStb(
                            pThreadInfo,
                            tableName,
                            batch,
                            insertRows,
                            interlaceRows * interlace + recPerTbl,
                            startTime);
                }
9091

9092 9093 9094 9095 9096 9097 9098
                debugPrint("[%d] %s() LN%d, generated records is %"PRId64"\n",
                        pThreadInfo->threadID, __func__, __LINE__, generated);
                if (generated < 0) {
                    errorPrint2("[%d] %s() LN%d, generated records is %"PRId64"\n",
                            pThreadInfo->threadID, __func__, __LINE__, generated);
                    goto free_of_interlace_stmt;
                } else if (generated == 0) {
9099
                    break;
9100
                }
9101

9102 9103 9104
                recPerTbl += generated;
                remainPerInterlace -= generated;
                pThreadInfo->totalInsertRows += generated;
9105

9106 9107 9108
                verbosePrint("[%d] %s() LN%d totalInsertRows=%"PRIu64"\n",
                        pThreadInfo->threadID, __func__, __LINE__,
                        pThreadInfo->totalInsertRows);
9109

9110
                startTs = taosGetTimestampUs();
9111

9112
                int64_t affectedRows = execInsert(pThreadInfo, generated);
9113

9114 9115 9116 9117 9118 9119 9120
                endTs = taosGetTimestampUs();
                uint64_t delay = endTs - startTs;
                performancePrint("%s() LN%d, insert execution time is %10.2f ms\n",
                        __func__, __LINE__, delay / 1000.0);
                verbosePrint("[%d] %s() LN%d affectedRows=%"PRId64"\n",
                        pThreadInfo->threadID,
                        __func__, __LINE__, affectedRows);
9121

9122 9123 9124 9125
                if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
                if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
                pThreadInfo->cntDelay++;
                pThreadInfo->totalDelay += delay;
9126

9127 9128 9129 9130 9131 9132
                if (generated != affectedRows) {
                    errorPrint2("[%d] %s() LN%d execInsert() insert %"PRId64", affected rows: %"PRId64"\n\n",
                            pThreadInfo->threadID, __func__, __LINE__,
                            generated, affectedRows);
                    goto free_of_interlace_stmt;
                }
9133

9134
                pThreadInfo->totalAffectedRows += affectedRows;
9135

9136 9137 9138 9139 9140 9141 9142 9143 9144 9145 9146 9147 9148
                int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
                if (currentPercent > percentComplete ) {
                    printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
                    percentComplete = currentPercent;
                }
                int64_t  currentPrintTime = taosGetTimestampMs();
                if (currentPrintTime - lastPrintTime > 30*1000) {
                    printf("thread[%d] has currently inserted rows: %"PRIu64 ", affected rows: %"PRIu64 "\n",
                            pThreadInfo->threadID,
                            pThreadInfo->totalInsertRows,
                            pThreadInfo->totalAffectedRows);
                    lastPrintTime = currentPrintTime;
                }
9149

9150 9151
                startTime += (generated * timeStampStep);
            }
9152
        }
9153
        pThreadInfo->samplePos = samplePos;
9154

9155 9156 9157 9158
        if (tableSeq == pThreadInfo->start_table_from
                + pThreadInfo->ntables) {
            // turn to first table
            tableSeq = pThreadInfo->start_table_from;
9159

9160
            flagSleep = true;
9161 9162 9163 9164 9165 9166 9167 9168 9169 9170 9171 9172 9173 9174 9175 9176 9177 9178 9179 9180 9181 9182 9183 9184 9185 9186 9187 9188 9189
        }

        if ((insert_interval) && flagSleep) {
            et = taosGetTimestampMs();

            if (insert_interval > (et - st) ) {
                uint64_t sleepTime = insert_interval - (et -st);
                performancePrint("%s() LN%d sleep: %"PRId64" ms for insert interval\n",
                        __func__, __LINE__, sleepTime);
                taosMsleep(sleepTime); // ms
                sleepTimeTotal += insert_interval;
            }
        }
    }
    if (percentComplete < 100)
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);

free_of_interlace_stmt:
    printStatPerThread(pThreadInfo);
    return NULL;
}
#else
// stmt sync write interlace data
static void* syncWriteInterlaceStmt(threadInfo *pThreadInfo, uint32_t interlaceRows) {
    debugPrint("[%d] %s() LN%d: ### stmt interlace write\n",
            pThreadInfo->threadID, __func__, __LINE__);

    int64_t insertRows;
    uint64_t maxSqlLen;
9190
    int64_t timeStampStep;
9191 9192 9193 9194 9195 9196
    uint64_t insert_interval;

    SSuperTable* stbInfo = pThreadInfo->stbInfo;

    if (stbInfo) {
        insertRows = stbInfo->insertRows;
9197
        maxSqlLen = stbInfo->maxSqlLen;
9198
        timeStampStep = stbInfo->timeStampStep;
9199
        insert_interval = stbInfo->insertInterval;
9200
    } else {
9201
        insertRows = g_args.insertRows;
9202
        maxSqlLen = g_args.max_sql_len;
9203
        timeStampStep = g_args.timestamp_step;
9204
        insert_interval = g_args.insert_interval;
9205
    }
9206

9207 9208 9209 9210
    debugPrint("[%d] %s() LN%d: start_table_from=%"PRIu64" ntables=%"PRId64" insertRows=%"PRIu64"\n",
            pThreadInfo->threadID, __func__, __LINE__,
            pThreadInfo->start_table_from,
            pThreadInfo->ntables, insertRows);
9211

9212 9213 9214 9215 9216 9217 9218 9219 9220 9221 9222 9223 9224 9225 9226 9227 9228 9229 9230 9231 9232 9233 9234 9235 9236 9237 9238 9239 9240 9241 9242 9243 9244 9245 9246 9247 9248 9249 9250 9251 9252 9253 9254 9255 9256 9257 9258 9259 9260 9261 9262 9263
    uint32_t batchPerTbl = interlaceRows;
    uint32_t batchPerTblTimes;

    if (interlaceRows > g_args.reqPerReq)
        interlaceRows = g_args.reqPerReq;

    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
            g_args.reqPerReq / interlaceRows;
    } else {
        batchPerTblTimes = 1;
    }

    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;

    uint64_t st = 0;
    uint64_t et = UINT64_MAX;

    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;

    uint64_t tableSeq = pThreadInfo->start_table_from;
    int64_t startTime = pThreadInfo->start_time;

    uint64_t generatedRecPerTbl = 0;
    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;

    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

    while(pThreadInfo->totalInsertRows < pThreadInfo->ntables * insertRows) {
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }

        uint32_t recOfBatch = 0;

        int32_t generated;
        for (uint64_t i = 0; i < batchPerTblTimes; i ++) {
            char tableName[TSDB_TABLE_NAME_LEN];

            getTableName(tableName, pThreadInfo, tableSeq);
            if (0 == strlen(tableName)) {
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
                        pThreadInfo->threadID, __func__, __LINE__);
                return NULL;
            }

9264 9265 9266 9267
            debugPrint("[%d] %s() LN%d, tableName:%s, batch:%d startTime:%"PRId64"\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__,
                    tableName, batchPerTbl, startTime);
9268
            if (stbInfo) {
9269
                generated = prepareStbStmt(
9270 9271 9272 9273 9274 9275 9276 9277 9278 9279 9280 9281 9282 9283 9284 9285 9286 9287 9288 9289 9290 9291 9292 9293 9294 9295 9296 9297 9298 9299 9300 9301 9302 9303 9304 9305 9306 9307 9308 9309 9310
                        pThreadInfo,
                        tableName,
                        tableSeq,
                        batchPerTbl,
                        insertRows, 0,
                        startTime,
                        &(pThreadInfo->samplePos));
            } else {
                generated = prepareStmtWithoutStb(
                        pThreadInfo,
                        tableName,
                        batchPerTbl,
                        insertRows, i,
                        startTime);
            }

            debugPrint("[%d] %s() LN%d, generated records is %d\n",
                    pThreadInfo->threadID, __func__, __LINE__, generated);
            if (generated < 0) {
                errorPrint2("[%d] %s() LN%d, generated records is %d\n",
                        pThreadInfo->threadID, __func__, __LINE__, generated);
                goto free_of_interlace_stmt;
            } else if (generated == 0) {
                break;
            }

            tableSeq ++;
            recOfBatch += batchPerTbl;

            pThreadInfo->totalInsertRows += batchPerTbl;

            verbosePrint("[%d] %s() LN%d batchPerTbl=%d recOfBatch=%d\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    batchPerTbl, recOfBatch);

            if (tableSeq == pThreadInfo->start_table_from + pThreadInfo->ntables) {
                // turn to first table
                tableSeq = pThreadInfo->start_table_from;
                generatedRecPerTbl += batchPerTbl;

                startTime = pThreadInfo->start_time
9311
                    + generatedRecPerTbl * timeStampStep;
9312 9313 9314 9315 9316 9317 9318 9319 9320 9321 9322 9323 9324 9325 9326 9327 9328 9329 9330 9331 9332 9333 9334 9335 9336 9337 9338 9339 9340 9341 9342 9343 9344 9345 9346 9347 9348 9349 9350 9351 9352 9353 9354 9355 9356 9357 9358 9359 9360 9361 9362 9363 9364 9365 9366 9367 9368 9369 9370 9371 9372 9373 9374 9375 9376 9377 9378 9379 9380 9381 9382 9383 9384 9385 9386 9387 9388 9389 9390 9391 9392 9393 9394 9395 9396 9397 9398 9399 9400 9401 9402 9403 9404 9405 9406 9407

                flagSleep = true;
                if (generatedRecPerTbl >= insertRows)
                    break;

                int64_t remainRows = insertRows - generatedRecPerTbl;
                if ((remainRows > 0) && (batchPerTbl > remainRows))
                    batchPerTbl = remainRows;

                if (pThreadInfo->ntables * batchPerTbl < g_args.reqPerReq)
                    break;
            }

            verbosePrint("[%d] %s() LN%d generatedRecPerTbl=%"PRId64" insertRows=%"PRId64"\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    generatedRecPerTbl, insertRows);

            if ((g_args.reqPerReq - recOfBatch) < batchPerTbl)
                break;
        }

        verbosePrint("[%d] %s() LN%d recOfBatch=%d totalInsertRows=%"PRIu64"\n",
                pThreadInfo->threadID, __func__, __LINE__, recOfBatch,
                pThreadInfo->totalInsertRows);

        startTs = taosGetTimestampUs();

        if (recOfBatch == 0) {
            errorPrint2("[%d] %s() LN%d Failed to insert records of batch %d\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    batchPerTbl);
            if (batchPerTbl > 0) {
                errorPrint("\tIf the batch is %d, the length of the SQL to insert a row must be less then %"PRId64"\n",
                        batchPerTbl, maxSqlLen / batchPerTbl);
            }
            goto free_of_interlace_stmt;
        }
        int64_t affectedRows = execInsert(pThreadInfo, recOfBatch);

        endTs = taosGetTimestampUs();
        uint64_t delay = endTs - startTs;
        performancePrint("%s() LN%d, insert execution time is %10.2f ms\n",
                __func__, __LINE__, delay / 1000.0);
        verbosePrint("[%d] %s() LN%d affectedRows=%"PRId64"\n",
                pThreadInfo->threadID,
                __func__, __LINE__, affectedRows);

        if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
        if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
        pThreadInfo->cntDelay++;
        pThreadInfo->totalDelay += delay;

        if (recOfBatch != affectedRows) {
            errorPrint2("[%d] %s() LN%d execInsert insert %d, affected rows: %"PRId64"\n\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    recOfBatch, affectedRows);
            goto free_of_interlace_stmt;
        }

        pThreadInfo->totalAffectedRows += affectedRows;

        int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
        if (currentPercent > percentComplete ) {
            printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
            percentComplete = currentPercent;
        }
        int64_t  currentPrintTime = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
            printf("thread[%d] has currently inserted rows: %"PRIu64 ", affected rows: %"PRIu64 "\n",
                    pThreadInfo->threadID,
                    pThreadInfo->totalInsertRows,
                    pThreadInfo->totalAffectedRows);
            lastPrintTime = currentPrintTime;
        }

        if ((insert_interval) && flagSleep) {
            et = taosGetTimestampMs();

            if (insert_interval > (et - st) ) {
                uint64_t sleepTime = insert_interval - (et -st);
                performancePrint("%s() LN%d sleep: %"PRId64" ms for insert interval\n",
                        __func__, __LINE__, sleepTime);
                taosMsleep(sleepTime); // ms
                sleepTimeTotal += insert_interval;
            }
        }
    }
    if (percentComplete < 100)
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);

free_of_interlace_stmt:
    printStatPerThread(pThreadInfo);
    return NULL;
}

#endif
9408

9409 9410 9411 9412 9413 9414 9415
// sync write interlace data
static void* syncWriteInterlace(threadInfo *pThreadInfo, uint32_t interlaceRows) {
    debugPrint("[%d] %s() LN%d: ### interlace write\n",
            pThreadInfo->threadID, __func__, __LINE__);

    int64_t insertRows;
    uint64_t maxSqlLen;
9416
    int64_t timeStampStep;
9417 9418 9419 9420 9421 9422 9423
    uint64_t insert_interval;

    SSuperTable* stbInfo = pThreadInfo->stbInfo;

    if (stbInfo) {
        insertRows = stbInfo->insertRows;
        maxSqlLen = stbInfo->maxSqlLen;
9424
        timeStampStep = stbInfo->timeStampStep;
9425 9426 9427 9428
        insert_interval = stbInfo->insertInterval;
    } else {
        insertRows = g_args.insertRows;
        maxSqlLen = g_args.max_sql_len;
9429
        timeStampStep = g_args.timestamp_step;
9430 9431 9432 9433 9434 9435 9436 9437
        insert_interval = g_args.insert_interval;
    }

    debugPrint("[%d] %s() LN%d: start_table_from=%"PRIu64" ntables=%"PRId64" insertRows=%"PRIu64"\n",
            pThreadInfo->threadID, __func__, __LINE__,
            pThreadInfo->start_table_from,
            pThreadInfo->ntables, insertRows);
#if 1
9438 9439
    if (interlaceRows > g_args.reqPerReq)
        interlaceRows = g_args.reqPerReq;
9440

9441 9442
    uint32_t batchPerTbl = interlaceRows;
    uint32_t batchPerTblTimes;
9443

9444 9445
    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
9446
            g_args.reqPerReq / interlaceRows;
9447 9448 9449
    } else {
        batchPerTblTimes = 1;
    }
9450 9451 9452 9453 9454 9455 9456 9457
#else
    uint32_t batchPerTbl;
    if (interlaceRows > g_args.reqPerReq)
        batchPerTbl = g_args.reqPerReq;
    else
        batchPerTbl = interlaceRows;

    uint32_t batchPerTblTimes;
9458

9459 9460 9461 9462 9463 9464 9465
    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
            interlaceRows / batchPerTbl;
    } else {
        batchPerTblTimes = 1;
    }
#endif
9466 9467
    pThreadInfo->buffer = calloc(maxSqlLen, 1);
    if (NULL == pThreadInfo->buffer) {
9468
        errorPrint2( "%s() LN%d, Failed to alloc %"PRIu64" Bytes, reason:%s\n",
9469 9470 9471
                __func__, __LINE__, maxSqlLen, strerror(errno));
        return NULL;
    }
9472

9473 9474
    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;
9475

9476 9477
    uint64_t st = 0;
    uint64_t et = UINT64_MAX;
9478

9479 9480 9481
    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;
9482

9483 9484
    uint64_t tableSeq = pThreadInfo->start_table_from;
    int64_t startTime = pThreadInfo->start_time;
9485

9486 9487 9488
    uint64_t generatedRecPerTbl = 0;
    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;
9489

9490 9491 9492
    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

9493 9494 9495 9496 9497
    while(pThreadInfo->totalInsertRows < pThreadInfo->ntables * insertRows) {
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }
9498

9499 9500 9501
        // generate data
        memset(pThreadInfo->buffer, 0, maxSqlLen);
        uint64_t remainderBufLen = maxSqlLen;
9502

9503
        char *pstr = pThreadInfo->buffer;
9504

9505 9506 9507 9508
        int len = snprintf(pstr,
                strlen(STR_INSERT_INTO) + 1, "%s", STR_INSERT_INTO);
        pstr += len;
        remainderBufLen -= len;
9509

9510
        uint32_t recOfBatch = 0;
9511

9512
        int32_t generated;
9513 9514
        for (uint64_t i = 0; i < batchPerTblTimes; i ++) {
            char tableName[TSDB_TABLE_NAME_LEN];
9515

9516 9517
            getTableName(tableName, pThreadInfo, tableSeq);
            if (0 == strlen(tableName)) {
9518
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
9519 9520 9521 9522
                        pThreadInfo->threadID, __func__, __LINE__);
                free(pThreadInfo->buffer);
                return NULL;
            }
9523

9524
            uint64_t oldRemainderLen = remainderBufLen;
9525

9526
            if (stbInfo) {
9527 9528 9529 9530 9531 9532 9533 9534 9535
                generated = generateStbInterlaceData(
                        pThreadInfo,
                        tableName, batchPerTbl, i,
                        batchPerTblTimes,
                        tableSeq,
                        pstr,
                        insertRows,
                        startTime,
                        &remainderBufLen);
9536
            } else {
9537 9538 9539 9540 9541 9542 9543
                generated = generateInterlaceDataWithoutStb(
                        tableName, batchPerTbl,
                        tableSeq,
                        pThreadInfo->db_name, pstr,
                        insertRows,
                        startTime,
                        &remainderBufLen);
9544
            }
9545

9546 9547 9548
            debugPrint("[%d] %s() LN%d, generated records is %d\n",
                    pThreadInfo->threadID, __func__, __LINE__, generated);
            if (generated < 0) {
9549
                errorPrint2("[%d] %s() LN%d, generated records is %d\n",
9550 9551 9552 9553 9554
                        pThreadInfo->threadID, __func__, __LINE__, generated);
                goto free_of_interlace;
            } else if (generated == 0) {
                break;
            }
9555

9556 9557
            tableSeq ++;
            recOfBatch += batchPerTbl;
9558

9559 9560
            pstr += (oldRemainderLen - remainderBufLen);
            pThreadInfo->totalInsertRows += batchPerTbl;
9561

9562 9563 9564
            verbosePrint("[%d] %s() LN%d batchPerTbl=%d recOfBatch=%d\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    batchPerTbl, recOfBatch);
9565

9566 9567 9568 9569
            if (tableSeq == pThreadInfo->start_table_from + pThreadInfo->ntables) {
                // turn to first table
                tableSeq = pThreadInfo->start_table_from;
                generatedRecPerTbl += batchPerTbl;
9570

9571
                startTime = pThreadInfo->start_time
9572
                    + generatedRecPerTbl * timeStampStep;
9573

9574 9575 9576
                flagSleep = true;
                if (generatedRecPerTbl >= insertRows)
                    break;
9577

9578 9579 9580
                int64_t remainRows = insertRows - generatedRecPerTbl;
                if ((remainRows > 0) && (batchPerTbl > remainRows))
                    batchPerTbl = remainRows;
9581

9582
                if (pThreadInfo->ntables * batchPerTbl < g_args.reqPerReq)
9583 9584
                    break;
            }
9585

9586 9587 9588
            verbosePrint("[%d] %s() LN%d generatedRecPerTbl=%"PRId64" insertRows=%"PRId64"\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    generatedRecPerTbl, insertRows);
9589

9590
            if ((g_args.reqPerReq - recOfBatch) < batchPerTbl)
9591
                break;
9592
        }
9593

9594 9595 9596 9597 9598 9599
        verbosePrint("[%d] %s() LN%d recOfBatch=%d totalInsertRows=%"PRIu64"\n",
                pThreadInfo->threadID, __func__, __LINE__, recOfBatch,
                pThreadInfo->totalInsertRows);
        verbosePrint("[%d] %s() LN%d, buffer=%s\n",
                pThreadInfo->threadID, __func__, __LINE__, pThreadInfo->buffer);

9600
        startTs = taosGetTimestampUs();
9601 9602

        if (recOfBatch == 0) {
9603
            errorPrint2("[%d] %s() LN%d Failed to insert records of batch %d\n",
9604 9605 9606 9607 9608 9609 9610 9611 9612 9613 9614 9615
                    pThreadInfo->threadID, __func__, __LINE__,
                    batchPerTbl);
            if (batchPerTbl > 0) {
                errorPrint("\tIf the batch is %d, the length of the SQL to insert a row must be less then %"PRId64"\n",
                        batchPerTbl, maxSqlLen / batchPerTbl);
            }
            errorPrint("\tPlease check if the buffer length(%"PRId64") or batch(%d) is set with proper value!\n",
                    maxSqlLen, batchPerTbl);
            goto free_of_interlace;
        }
        int64_t affectedRows = execInsert(pThreadInfo, recOfBatch);

9616
        endTs = taosGetTimestampUs();
9617
        uint64_t delay = endTs - startTs;
9618 9619
        performancePrint("%s() LN%d, insert execution time is %10.2f ms\n",
                __func__, __LINE__, delay / 1000.0);
9620 9621 9622
        verbosePrint("[%d] %s() LN%d affectedRows=%"PRId64"\n",
                pThreadInfo->threadID,
                __func__, __LINE__, affectedRows);
9623

9624 9625 9626 9627
        if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
        if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
        pThreadInfo->cntDelay++;
        pThreadInfo->totalDelay += delay;
9628

9629
        if (recOfBatch != affectedRows) {
9630
            errorPrint2("[%d] %s() LN%d execInsert insert %d, affected rows: %"PRId64"\n%s\n",
9631 9632 9633 9634
                    pThreadInfo->threadID, __func__, __LINE__,
                    recOfBatch, affectedRows, pThreadInfo->buffer);
            goto free_of_interlace;
        }
9635

9636
        pThreadInfo->totalAffectedRows += affectedRows;
9637

9638 9639 9640 9641 9642
        int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
        if (currentPercent > percentComplete ) {
            printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
            percentComplete = currentPercent;
        }
9643 9644 9645
        int64_t  currentPrintTime = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
            printf("thread[%d] has currently inserted rows: %"PRIu64 ", affected rows: %"PRIu64 "\n",
9646 9647 9648
                    pThreadInfo->threadID,
                    pThreadInfo->totalInsertRows,
                    pThreadInfo->totalAffectedRows);
9649 9650
            lastPrintTime = currentPrintTime;
        }
9651

9652 9653
        if ((insert_interval) && flagSleep) {
            et = taosGetTimestampMs();
9654

9655 9656 9657 9658 9659 9660 9661 9662
            if (insert_interval > (et - st) ) {
                uint64_t sleepTime = insert_interval - (et -st);
                performancePrint("%s() LN%d sleep: %"PRId64" ms for insert interval\n",
                        __func__, __LINE__, sleepTime);
                taosMsleep(sleepTime); // ms
                sleepTimeTotal += insert_interval;
            }
        }
9663
    }
9664 9665
    if (percentComplete < 100)
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
9666

9667
free_of_interlace:
9668 9669 9670
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
9671 9672
}

9673 9674 9675 9676 9677 9678 9679 9680 9681 9682 9683 9684 9685 9686 9687 9688 9689 9690 9691 9692 9693 9694 9695 9696 9697 9698 9699 9700 9701 9702 9703 9704 9705 9706 9707 9708 9709 9710 9711 9712 9713 9714 9715 9716 9717 9718 9719 9720 9721 9722 9723 9724 9725 9726 9727 9728 9729 9730 9731 9732 9733 9734 9735 9736 9737 9738 9739 9740 9741 9742 9743 9744 9745 9746 9747 9748 9749 9750 9751 9752 9753 9754 9755 9756 9757 9758 9759 9760 9761 9762 9763 9764 9765 9766 9767 9768 9769 9770 9771 9772 9773 9774 9775 9776 9777 9778 9779 9780 9781 9782 9783 9784 9785 9786 9787 9788 9789 9790 9791 9792 9793 9794 9795 9796 9797 9798 9799 9800 9801 9802 9803 9804 9805 9806 9807 9808 9809 9810
static void* syncWriteProgressiveStmt(threadInfo *pThreadInfo) {
    debugPrint("%s() LN%d: ### stmt progressive write\n", __func__, __LINE__);

    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    int64_t timeStampStep =
        stbInfo?stbInfo->timeStampStep:g_args.timestamp_step;
    int64_t insertRows =
        (stbInfo)?stbInfo->insertRows:g_args.insertRows;
    verbosePrint("%s() LN%d insertRows=%"PRId64"\n",
            __func__, __LINE__, insertRows);

    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;

    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;

    pThreadInfo->samplePos = 0;

    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

    for (uint64_t tableSeq = pThreadInfo->start_table_from;
            tableSeq <= pThreadInfo->end_table_to;
            tableSeq ++) {
        int64_t start_time = pThreadInfo->start_time;

        for (uint64_t i = 0; i < insertRows;) {
            char tableName[TSDB_TABLE_NAME_LEN];
            getTableName(tableName, pThreadInfo, tableSeq);
            verbosePrint("%s() LN%d: tid=%d seq=%"PRId64" tableName=%s\n",
                    __func__, __LINE__,
                    pThreadInfo->threadID, tableSeq, tableName);
            if (0 == strlen(tableName)) {
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
                        pThreadInfo->threadID, __func__, __LINE__);
                return NULL;
            }

            // measure prepare + insert
            startTs = taosGetTimestampUs();

            int32_t generated;
            if (stbInfo) {
                generated = prepareStbStmt(
                        pThreadInfo,
                        tableName,
                        tableSeq,
                        (g_args.reqPerReq>stbInfo->insertRows)?
                        stbInfo->insertRows:
                        g_args.reqPerReq,
                        insertRows, i, start_time,
                        &(pThreadInfo->samplePos));
            } else {
                generated = prepareStmtWithoutStb(
                        pThreadInfo,
                        tableName,
                        g_args.reqPerReq,
                        insertRows, i,
                        start_time);
            }

            verbosePrint("[%d] %s() LN%d generated=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, generated);

            if (generated > 0)
                i += generated;
            else
                goto free_of_stmt_progressive;

            start_time +=  generated * timeStampStep;
            pThreadInfo->totalInsertRows += generated;

            // only measure insert
            // startTs = taosGetTimestampUs();

            int32_t affectedRows = execInsert(pThreadInfo, generated);

            endTs = taosGetTimestampUs();
            uint64_t delay = endTs - startTs;
            performancePrint("%s() LN%d, insert execution time is %10.f ms\n",
                    __func__, __LINE__, delay/1000.0);
            verbosePrint("[%d] %s() LN%d affectedRows=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, affectedRows);

            if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
            if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
            pThreadInfo->cntDelay++;
            pThreadInfo->totalDelay += delay;

            if (affectedRows < 0) {
                errorPrint2("%s() LN%d, affected rows: %d\n",
                        __func__, __LINE__, affectedRows);
                goto free_of_stmt_progressive;
            }

            pThreadInfo->totalAffectedRows += affectedRows;

            int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
            if (currentPercent > percentComplete ) {
                printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
                percentComplete = currentPercent;
            }
            int64_t  currentPrintTime = taosGetTimestampMs();
            if (currentPrintTime - lastPrintTime > 30*1000) {
                printf("thread[%d] has currently inserted rows: %"PRId64 ", affected rows: %"PRId64 "\n",
                        pThreadInfo->threadID,
                        pThreadInfo->totalInsertRows,
                        pThreadInfo->totalAffectedRows);
                lastPrintTime = currentPrintTime;
            }

            if (i >= insertRows)
                break;
        }   // insertRows

        if ((g_args.verbose_print) &&
                (tableSeq == pThreadInfo->ntables - 1) && (stbInfo)
                && (0 == strncasecmp(
                        stbInfo->dataSource,
                        "sample", strlen("sample")))) {
            verbosePrint("%s() LN%d samplePos=%"PRId64"\n",
                    __func__, __LINE__, pThreadInfo->samplePos);
        }
    } // tableSeq

    if (percentComplete < 100) {
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
    }

free_of_stmt_progressive:
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
}
9811
// sync insertion progressive data
9812
static void* syncWriteProgressive(threadInfo *pThreadInfo) {
9813
    debugPrint("%s() LN%d: ### progressive write\n", __func__, __LINE__);
9814

9815 9816
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    uint64_t maxSqlLen = stbInfo?stbInfo->maxSqlLen:g_args.max_sql_len;
9817
    int64_t timeStampStep =
9818
        stbInfo?stbInfo->timeStampStep:g_args.timestamp_step;
9819
    int64_t insertRows =
9820
        (stbInfo)?stbInfo->insertRows:g_args.insertRows;
9821
    verbosePrint("%s() LN%d insertRows=%"PRId64"\n",
9822
            __func__, __LINE__, insertRows);
9823

9824 9825
    pThreadInfo->buffer = calloc(maxSqlLen, 1);
    if (NULL == pThreadInfo->buffer) {
9826
        errorPrint2("Failed to alloc %"PRIu64" bytes, reason:%s\n",
9827 9828
                maxSqlLen,
                strerror(errno));
9829
        return NULL;
9830 9831 9832 9833 9834 9835 9836 9837 9838 9839 9840
    }

    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;

    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;

    pThreadInfo->samplePos = 0;

9841 9842 9843
    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

9844 9845 9846 9847 9848 9849 9850 9851 9852 9853 9854 9855
    for (uint64_t tableSeq = pThreadInfo->start_table_from;
            tableSeq <= pThreadInfo->end_table_to;
            tableSeq ++) {
        int64_t start_time = pThreadInfo->start_time;

        for (uint64_t i = 0; i < insertRows;) {
            char tableName[TSDB_TABLE_NAME_LEN];
            getTableName(tableName, pThreadInfo, tableSeq);
            verbosePrint("%s() LN%d: tid=%d seq=%"PRId64" tableName=%s\n",
                    __func__, __LINE__,
                    pThreadInfo->threadID, tableSeq, tableName);
            if (0 == strlen(tableName)) {
9856
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
9857 9858 9859 9860
                        pThreadInfo->threadID, __func__, __LINE__);
                free(pThreadInfo->buffer);
                return NULL;
            }
9861

9862
            int64_t remainderBufLen = maxSqlLen - 2000;
9863
            char *pstr = pThreadInfo->buffer;
9864

9865 9866
            int len = snprintf(pstr,
                    strlen(STR_INSERT_INTO) + 1, "%s", STR_INSERT_INTO);
9867

9868 9869
            pstr += len;
            remainderBufLen -= len;
9870

9871 9872 9873
            // measure prepare + insert
            startTs = taosGetTimestampUs();

9874
            int32_t generated;
9875 9876
            if (stbInfo) {
                if (stbInfo->iface == STMT_IFACE) {
9877
                    generated = prepareStbStmt(
9878 9879 9880
                            pThreadInfo,
                            tableName,
                            tableSeq,
9881 9882 9883
                            (g_args.reqPerReq>stbInfo->insertRows)?
                                stbInfo->insertRows:
                                g_args.reqPerReq,
9884 9885
                            insertRows, i, start_time,
                            &(pThreadInfo->samplePos));
9886 9887
                } else {
                    generated = generateStbProgressiveData(
9888 9889 9890
                            stbInfo,
                            tableName, tableSeq,
                            pThreadInfo->db_name, pstr,
9891 9892 9893 9894 9895 9896 9897
                            insertRows, i, start_time,
                            &(pThreadInfo->samplePos),
                            &remainderBufLen);
                }
            } else {
                if (g_args.iface == STMT_IFACE) {
                    generated = prepareStmtWithoutStb(
9898
                            pThreadInfo,
9899
                            tableName,
9900
                            g_args.reqPerReq,
9901 9902 9903 9904 9905 9906 9907 9908 9909 9910 9911 9912
                            insertRows, i,
                            start_time);
                } else {
                    generated = generateProgressiveDataWithoutStb(
                            tableName,
                            /*  tableSeq, */
                            pThreadInfo, pstr, insertRows,
                            i, start_time,
                            /* &(pThreadInfo->samplePos), */
                            &remainderBufLen);
                }
            }
9913 9914 9915 9916 9917

            verbosePrint("[%d] %s() LN%d generated=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, generated);

9918 9919 9920 9921
            if (generated > 0)
                i += generated;
            else
                goto free_of_progressive;
9922

9923 9924
            start_time +=  generated * timeStampStep;
            pThreadInfo->totalInsertRows += generated;
9925

9926 9927
            // only measure insert
            // startTs = taosGetTimestampUs();
9928

9929
            int32_t affectedRows = execInsert(pThreadInfo, generated);
9930

9931
            endTs = taosGetTimestampUs();
9932
            uint64_t delay = endTs - startTs;
9933 9934
            performancePrint("%s() LN%d, insert execution time is %10.f ms\n",
                    __func__, __LINE__, delay/1000.0);
9935 9936 9937
            verbosePrint("[%d] %s() LN%d affectedRows=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, affectedRows);
9938

9939 9940 9941 9942
            if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
            if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
            pThreadInfo->cntDelay++;
            pThreadInfo->totalDelay += delay;
9943

9944
            if (affectedRows < 0) {
9945
                errorPrint2("%s() LN%d, affected rows: %d\n",
9946 9947 9948
                        __func__, __LINE__, affectedRows);
                goto free_of_progressive;
            }
9949

9950
            pThreadInfo->totalAffectedRows += affectedRows;
9951

9952 9953 9954 9955 9956
            int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
            if (currentPercent > percentComplete ) {
                printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
                percentComplete = currentPercent;
            }
9957 9958 9959 9960 9961 9962 9963 9964
            int64_t  currentPrintTime = taosGetTimestampMs();
            if (currentPrintTime - lastPrintTime > 30*1000) {
                printf("thread[%d] has currently inserted rows: %"PRId64 ", affected rows: %"PRId64 "\n",
                        pThreadInfo->threadID,
                        pThreadInfo->totalInsertRows,
                        pThreadInfo->totalAffectedRows);
                lastPrintTime = currentPrintTime;
            }
9965

9966 9967
            if (i >= insertRows)
                break;
9968
        }   // insertRows
9969

9970
        if ((g_args.verbose_print) &&
9971
                (tableSeq == pThreadInfo->ntables - 1) && (stbInfo)
9972
                && (0 == strncasecmp(
9973
                        stbInfo->dataSource,
9974
                        "sample", strlen("sample")))) {
9975 9976 9977 9978
            verbosePrint("%s() LN%d samplePos=%"PRId64"\n",
                    __func__, __LINE__, pThreadInfo->samplePos);
        }
    } // tableSeq
9979 9980

    if (percentComplete < 100) {
9981
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
9982
    }
9983

9984
free_of_progressive:
9985 9986 9987
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
H
Hui Li 已提交
9988 9989
}

9990 9991
static void* syncWrite(void *sarg) {

9992
    threadInfo *pThreadInfo = (threadInfo *)sarg;
9993
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
9994

9995 9996
    setThreadName("syncWrite");

9997
    uint32_t interlaceRows = 0;
9998

9999
    if (stbInfo) {
10000
        if (stbInfo->interlaceRows < stbInfo->insertRows)
10001
            interlaceRows = stbInfo->interlaceRows;
10002
    } else {
10003 10004
        if (g_args.interlaceRows < g_args.insertRows)
            interlaceRows = g_args.interlaceRows;
10005
    }
10006

10007 10008
    if (interlaceRows > 0) {
        // interlace mode
10009 10010
        if (stbInfo) {
            if (STMT_IFACE == stbInfo->iface) {
10011
#if STMT_BIND_PARAM_BATCH == 1
10012
                return syncWriteInterlaceStmtBatch(pThreadInfo, interlaceRows);
10013
#else
10014
                return syncWriteInterlaceStmt(pThreadInfo, interlaceRows);
10015
#endif
10016 10017 10018
            } else {
                return syncWriteInterlace(pThreadInfo, interlaceRows);
            }
10019
        }
10020 10021 10022 10023 10024 10025 10026 10027
    } else {
      // progressive mode
      if (((stbInfo) && (STMT_IFACE == stbInfo->iface))
              || (STMT_IFACE == g_args.iface)) {
          return syncWriteProgressiveStmt(pThreadInfo);
      } else {
          return syncWriteProgressive(pThreadInfo);
      }
10028
    }
10029 10030

    return NULL;
10031 10032
}

10033
static void callBack(void *param, TAOS_RES *res, int code) {
10034
    threadInfo* pThreadInfo = (threadInfo*)param;
10035
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10036 10037

    int insert_interval =
10038
        stbInfo?stbInfo->insertInterval:g_args.insert_interval;
10039 10040 10041 10042 10043
    if (insert_interval) {
        pThreadInfo->et = taosGetTimestampMs();
        if ((pThreadInfo->et - pThreadInfo->st) < insert_interval) {
            taosMsleep(insert_interval - (pThreadInfo->et - pThreadInfo->st)); // ms
        }
H
Hui Li 已提交
10044 10045
    }

10046
    char *buffer = calloc(1, pThreadInfo->stbInfo->maxSqlLen);
10047 10048
    char data[MAX_DATA_SIZE];
    char *pstr = buffer;
10049
    pstr += sprintf(pstr, "INSERT INTO %s.%s%"PRId64" VALUES",
10050 10051
            pThreadInfo->db_name, pThreadInfo->tb_prefix,
            pThreadInfo->start_table_from);
10052
    //  if (pThreadInfo->counter >= pThreadInfo->stbInfo->insertRows) {
10053
    if (pThreadInfo->counter >= g_args.reqPerReq) {
10054 10055 10056 10057 10058 10059 10060 10061
        pThreadInfo->start_table_from++;
        pThreadInfo->counter = 0;
    }
    if (pThreadInfo->start_table_from > pThreadInfo->end_table_to) {
        tsem_post(&pThreadInfo->lock_sem);
        free(buffer);
        taos_free_result(res);
        return;
H
Hui Li 已提交
10062
    }
10063

10064
    for (int i = 0; i < g_args.reqPerReq; i++) {
10065
        int rand_num = taosRandom() % 100;
10066 10067
        if (0 != pThreadInfo->stbInfo->disorderRatio
                && rand_num < pThreadInfo->stbInfo->disorderRatio) {
10068
            int64_t d = pThreadInfo->lastTs
10069 10070
                - (taosRandom() % pThreadInfo->stbInfo->disorderRange + 1);
            generateStbRowData(pThreadInfo->stbInfo, data,
10071 10072
                    MAX_DATA_SIZE,
                    d);
10073
        } else {
10074
            generateStbRowData(pThreadInfo->stbInfo,
10075 10076 10077
                    data,
                    MAX_DATA_SIZE,
                    pThreadInfo->lastTs += 1000);
10078 10079 10080
        }
        pstr += sprintf(pstr, "%s", data);
        pThreadInfo->counter++;
H
Hui Li 已提交
10081

10082
        if (pThreadInfo->counter >= pThreadInfo->stbInfo->insertRows) {
10083 10084 10085 10086 10087 10088 10089 10090 10091 10092 10093
            break;
        }
    }

    if (insert_interval) {
        pThreadInfo->st = taosGetTimestampMs();
    }
    taos_query_a(pThreadInfo->taos, buffer, callBack, pThreadInfo);
    free(buffer);

    taos_free_result(res);
H
Hui Li 已提交
10094 10095
}

10096
static void *asyncWrite(void *sarg) {
10097
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10098
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10099

10100
    setThreadName("asyncWrite");
10101

10102 10103 10104
    pThreadInfo->st = 0;
    pThreadInfo->et = 0;
    pThreadInfo->lastTs = pThreadInfo->start_time;
10105

10106
    int insert_interval =
10107
        stbInfo?stbInfo->insertInterval:g_args.insert_interval;
10108 10109 10110 10111
    if (insert_interval) {
        pThreadInfo->st = taosGetTimestampMs();
    }
    taos_query_a(pThreadInfo->taos, "show databases", callBack, pThreadInfo);
10112

10113
    tsem_wait(&(pThreadInfo->lock_sem));
H
Hui Li 已提交
10114

10115
    return NULL;
H
Hui Li 已提交
10116 10117
}

10118 10119
static int convertHostToServAddr(char *host, uint16_t port, struct sockaddr_in *serv_addr)
{
10120 10121 10122
    uint16_t rest_port = port + TSDB_PORT_HTTP;
    struct hostent *server = gethostbyname(host);
    if ((server == NULL) || (server->h_addr == NULL)) {
10123
        errorPrint2("%s", "no such host");
10124 10125
        return -1;
    }
10126

10127
    debugPrint("h_name: %s\nh_addr=%p\nh_addretype: %s\nh_length: %d\n",
10128 10129 10130 10131 10132
            server->h_name,
            server->h_addr,
            (server->h_addrtype == AF_INET)?"ipv4":"ipv6",
            server->h_length);

10133 10134 10135
    memset(serv_addr, 0, sizeof(struct sockaddr_in));
    serv_addr->sin_family = AF_INET;
    serv_addr->sin_port = htons(rest_port);
10136
#ifdef WINDOWS
10137
    serv_addr->sin_addr.s_addr = inet_addr(host);
10138
#else
10139
    memcpy(&(serv_addr->sin_addr.s_addr), server->h_addr, server->h_length);
10140
#endif
10141
    return 0;
10142 10143
}

10144
static void startMultiThreadInsertData(int threads, char* db_name,
10145
        char* precision, SSuperTable* stbInfo) {
10146

10147 10148 10149 10150 10151 10152 10153 10154 10155
    int32_t timePrec = TSDB_TIME_PRECISION_MILLI;
    if (0 != precision[0]) {
        if (0 == strncasecmp(precision, "ms", 2)) {
            timePrec = TSDB_TIME_PRECISION_MILLI;
        } else if (0 == strncasecmp(precision, "us", 2)) {
            timePrec = TSDB_TIME_PRECISION_MICRO;
        } else if (0 == strncasecmp(precision, "ns", 2)) {
            timePrec = TSDB_TIME_PRECISION_NANO;
        } else {
10156
            errorPrint2("Not support precision: %s\n", precision);
10157
            exit(EXIT_FAILURE);
10158
        }
10159 10160
    }

10161
    int64_t startTime;
10162 10163
    if (stbInfo) {
        if (0 == strncasecmp(stbInfo->startTimestamp, "now", 3)) {
10164
            startTime = taosGetTimestamp(timePrec);
10165 10166
        } else {
            if (TSDB_CODE_SUCCESS != taosParseTime(
10167
                        stbInfo->startTimestamp,
10168
                        &startTime,
10169
                        strlen(stbInfo->startTimestamp),
10170 10171 10172 10173
                        timePrec, 0)) {
                ERROR_EXIT("failed to parse time!\n");
            }
        }
10174
    } else {
10175
        startTime = DEFAULT_START_TIME;
10176
    }
10177 10178
    debugPrint("%s() LN%d, startTime= %"PRId64"\n",
            __func__, __LINE__, startTime);
10179

10180
    // read sample data from file first
10181
    int ret;
10182
    if (stbInfo) {
10183 10184 10185 10186 10187 10188 10189 10190 10191
        ret = prepareSampleForStb(stbInfo);
    } else {
        ret = prepareSampleForNtb();
    }

    if (0 != ret) {
        errorPrint2("%s() LN%d, prepare sample data for stable failed!\n",
                __func__, __LINE__);
        exit(EXIT_FAILURE);
10192 10193
    }

10194 10195 10196 10197
    TAOS* taos0 = taos_connect(
            g_Dbs.host, g_Dbs.user,
            g_Dbs.password, db_name, g_Dbs.port);
    if (NULL == taos0) {
10198
        errorPrint2("%s() LN%d, connect to server fail , reason: %s\n",
10199
                __func__, __LINE__, taos_errstr(NULL));
10200
        exit(EXIT_FAILURE);
10201
    }
10202

10203 10204
    int64_t ntables = 0;
    uint64_t tableFrom;
10205

10206
    if (stbInfo) {
10207 10208
        int64_t limit;
        uint64_t offset;
10209

10210
        if ((NULL != g_args.sqlFile)
10211 10212 10213
                && (stbInfo->childTblExists == TBL_NO_EXISTS)
                && ((stbInfo->childTblOffset != 0)
                    || (stbInfo->childTblLimit >= 0))) {
10214 10215
            printf("WARNING: offset and limit will not be used since the child tables not exists!\n");
        }
10216

10217 10218 10219 10220 10221
        if (stbInfo->childTblExists == TBL_ALREADY_EXISTS) {
            if ((stbInfo->childTblLimit < 0)
                    || ((stbInfo->childTblOffset
                            + stbInfo->childTblLimit)
                        > (stbInfo->childTblCount))) {
10222 10223 10224 10225 10226 10227

                if (stbInfo->childTblCount < stbInfo->childTblOffset) {
                    printf("WARNING: offset will not be used since the child tables count is less then offset!\n");

                    stbInfo->childTblOffset = 0;
                }
10228 10229
                stbInfo->childTblLimit =
                    stbInfo->childTblCount - stbInfo->childTblOffset;
10230
            }
10231

10232 10233
            offset = stbInfo->childTblOffset;
            limit = stbInfo->childTblLimit;
10234
        } else {
10235
            limit = stbInfo->childTblCount;
10236 10237
            offset = 0;
        }
10238

10239 10240
        ntables = limit;
        tableFrom = offset;
10241

10242 10243 10244
        if ((stbInfo->childTblExists != TBL_NO_EXISTS)
                && ((stbInfo->childTblOffset + stbInfo->childTblLimit)
                    > stbInfo->childTblCount)) {
10245 10246 10247
            printf("WARNING: specified offset + limit > child table count!\n");
            prompt();
        }
10248

10249 10250
        if ((stbInfo->childTblExists != TBL_NO_EXISTS)
                && (0 == stbInfo->childTblLimit)) {
10251 10252 10253
            printf("WARNING: specified limit = 0, which cannot find table name to insert or query! \n");
            prompt();
        }
10254

10255
        stbInfo->childTblName = (char*)calloc(1,
10256
                limit * TSDB_TABLE_NAME_LEN);
10257
        if (stbInfo->childTblName == NULL) {
10258
            taos_close(taos0);
10259
            errorPrint2("%s() LN%d, alloc memory failed!\n", __func__, __LINE__);
10260
            exit(EXIT_FAILURE);
10261
        }
10262

10263 10264 10265
        int64_t childTblCount;
        getChildNameOfSuperTableWithLimitAndOffset(
                taos0,
10266
                db_name, stbInfo->stbName,
10267
                &stbInfo->childTblName, &childTblCount,
10268 10269
                limit,
                offset);
10270
        ntables = childTblCount; // CBD
10271
    } else {
10272
        ntables = g_args.ntables;
10273 10274
        tableFrom = 0;
    }
10275

10276
    taos_close(taos0);
10277

10278 10279 10280 10281 10282
    int64_t a = ntables / threads;
    if (a < 1) {
        threads = ntables;
        a = 1;
    }
10283

10284 10285 10286 10287
    int64_t b = 0;
    if (threads != 0) {
        b = ntables % threads;
    }
10288

10289 10290
    if ((stbInfo)
            && (stbInfo->iface == REST_IFACE)) {
10291 10292
        if (convertHostToServAddr(
                    g_Dbs.host, g_Dbs.port, &(g_Dbs.serv_addr)) != 0) {
10293
            ERROR_EXIT("convert host to server address");
10294 10295
        }
    }
10296

10297 10298
    pthread_t *pids = calloc(1, threads * sizeof(pthread_t));
    threadInfo *infos = calloc(1, threads * sizeof(threadInfo));
10299
    assert(pids != NULL);
10300
    assert(infos != NULL);
10301

10302 10303
    char *stmtBuffer = calloc(1, BUFFER_SIZE);
    assert(stmtBuffer);
10304 10305

#if STMT_BIND_PARAM_BATCH == 1
10306
    uint32_t interlaceRows = 0;
10307 10308 10309
    uint32_t batch;

    if (stbInfo) {
10310
        if (stbInfo->interlaceRows < stbInfo->insertRows)
10311 10312
            interlaceRows = stbInfo->interlaceRows;
    } else {
10313 10314
        if (g_args.interlaceRows < g_args.insertRows)
            interlaceRows = g_args.interlaceRows;
10315 10316 10317 10318 10319 10320 10321 10322 10323 10324 10325
    }

    if (interlaceRows > 0) {
        batch = interlaceRows;
    } else {
        batch = (g_args.reqPerReq>g_args.insertRows)?
            g_args.insertRows:g_args.reqPerReq;
    }

#endif

10326 10327 10328 10329 10330 10331 10332 10333 10334
    if ((g_args.iface == STMT_IFACE)
            || ((stbInfo)
                && (stbInfo->iface == STMT_IFACE))) {
        char *pstr = stmtBuffer;

        if ((stbInfo)
                && (AUTO_CREATE_SUBTBL
                    == stbInfo->autoCreateTable)) {
            pstr += sprintf(pstr, "INSERT INTO ? USING %s TAGS(?",
10335
                    stbInfo->stbName);
10336 10337 10338 10339 10340 10341 10342 10343 10344
            for (int tag = 0; tag < (stbInfo->tagCount - 1);
                    tag ++ ) {
                pstr += sprintf(pstr, ",?");
            }
            pstr += sprintf(pstr, ") VALUES(?");
        } else {
            pstr += sprintf(pstr, "INSERT INTO ? VALUES(?");
        }

10345 10346 10347
        int columnCount = (stbInfo)?
            stbInfo->columnCount:
            g_args.columnCount;
10348 10349 10350 10351 10352 10353 10354

        for (int col = 0; col < columnCount; col ++) {
            pstr += sprintf(pstr, ",?");
        }
        pstr += sprintf(pstr, ")");

        debugPrint("%s() LN%d, stmtBuffer: %s", __func__, __LINE__, stmtBuffer);
10355 10356 10357
#if STMT_BIND_PARAM_BATCH == 1
        parseSamplefileToStmtBatch(stbInfo);
#endif
10358 10359
    }

10360 10361 10362
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
        pThreadInfo->threadID = i;
10363

10364 10365
        tstrncpy(pThreadInfo->db_name, db_name, TSDB_DB_NAME_LEN);
        pThreadInfo->time_precision = timePrec;
10366
        pThreadInfo->stbInfo = stbInfo;
10367

10368
        pThreadInfo->start_time = startTime;
10369 10370
        pThreadInfo->minDelay = UINT64_MAX;

10371 10372
        if ((NULL == stbInfo) ||
                (stbInfo->iface != REST_IFACE)) {
10373 10374 10375 10376 10377
            //t_info->taos = taos;
            pThreadInfo->taos = taos_connect(
                    g_Dbs.host, g_Dbs.user,
                    g_Dbs.password, db_name, g_Dbs.port);
            if (NULL == pThreadInfo->taos) {
10378
                free(infos);
10379
                errorPrint2(
10380 10381 10382
                        "%s() LN%d, connect to server fail from insert sub thread, reason: %s\n",
                        __func__, __LINE__,
                        taos_errstr(NULL));
10383
                exit(EXIT_FAILURE);
10384
            }
10385

10386
            if ((g_args.iface == STMT_IFACE)
10387 10388
                    || ((stbInfo)
                        && (stbInfo->iface == STMT_IFACE))) {
10389

10390 10391
                pThreadInfo->stmt = taos_stmt_init(pThreadInfo->taos);
                if (NULL == pThreadInfo->stmt) {
10392 10393
                    free(pids);
                    free(infos);
10394
                    errorPrint2(
10395 10396 10397
                            "%s() LN%d, failed init stmt, reason: %s\n",
                            __func__, __LINE__,
                            taos_errstr(NULL));
10398
                    exit(EXIT_FAILURE);
10399
                }
10400

10401
                if (0 != taos_stmt_prepare(pThreadInfo->stmt, stmtBuffer, 0)) {
10402 10403
                    free(pids);
                    free(infos);
10404
                    free(stmtBuffer);
10405
                    errorPrint2("failed to execute taos_stmt_prepare. return 0x%x. reason: %s\n",
10406 10407
                            ret, taos_stmt_errstr(pThreadInfo->stmt));
                    exit(EXIT_FAILURE);
10408
                }
10409
                pThreadInfo->bind_ts = malloc(sizeof(int64_t));
10410 10411

                if (stbInfo) {
10412 10413 10414 10415 10416 10417 10418 10419 10420 10421 10422 10423 10424
#if STMT_BIND_PARAM_BATCH == 1
                    parseStbSampleToStmtBatchForThread(
                            pThreadInfo, stbInfo, timePrec, batch);
#else
                    parseStbSampleToStmt(pThreadInfo, stbInfo, timePrec);
#endif
                } else {
#if STMT_BIND_PARAM_BATCH == 1
                    parseNtbSampleToStmtBatchForThread(
                            pThreadInfo, timePrec, batch);
#else
                    parseNtbSampleToStmt(pThreadInfo, timePrec);
#endif
10425
                }
10426 10427
            }
        } else {
10428
            pThreadInfo->taos = NULL;
10429
        }
10430

10431 10432
        /*    if ((NULL == stbInfo)
              || (0 == stbInfo->multiThreadWriteOneTbl)) {
10433 10434 10435 10436 10437 10438 10439
              */
        pThreadInfo->start_table_from = tableFrom;
        pThreadInfo->ntables = i<b?a+1:a;
        pThreadInfo->end_table_to = i < b ? tableFrom + a : tableFrom + a - 1;
        tableFrom = pThreadInfo->end_table_to + 1;
        /*    } else {
              pThreadInfo->start_table_from = 0;
10440
              pThreadInfo->ntables = stbInfo->childTblCount;
10441 10442 10443 10444 10445 10446 10447 10448
              pThreadInfo->start_time = pThreadInfo->start_time + rand_int() % 10000 - rand_tinyint();
              }
              */
        tsem_init(&(pThreadInfo->lock_sem), 0, 0);
        if (ASYNC_MODE == g_Dbs.asyncMode) {
            pthread_create(pids + i, NULL, asyncWrite, pThreadInfo);
        } else {
            pthread_create(pids + i, NULL, syncWrite, pThreadInfo);
10449
        }
10450 10451
    }

10452 10453
    free(stmtBuffer);

10454
    int64_t start = taosGetTimestampUs();
10455

10456 10457
    for (int i = 0; i < threads; i++) {
        pthread_join(pids[i], NULL);
10458
    }
H
Hui Li 已提交
10459

10460 10461 10462 10463 10464
    uint64_t totalDelay = 0;
    uint64_t maxDelay = 0;
    uint64_t minDelay = UINT64_MAX;
    uint64_t cntDelay = 1;
    double  avgDelay = 0;
10465

10466 10467
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
10468

10469 10470 10471
        tsem_destroy(&(pThreadInfo->lock_sem));
        taos_close(pThreadInfo->taos);

10472 10473 10474
        if (pThreadInfo->stmt) {
            taos_stmt_close(pThreadInfo->stmt);
        }
10475

10476
        tmfree((char *)pThreadInfo->bind_ts);
10477
#if STMT_BIND_PARAM_BATCH == 1
10478 10479 10480 10481
        tmfree((char *)pThreadInfo->bind_ts_array);
        tmfree(pThreadInfo->bindParams);
        tmfree(pThreadInfo->is_null);
#else
10482
        if (pThreadInfo->sampleBindArray) {
10483
            for (int k = 0; k < MAX_SAMPLES; k++) {
10484 10485 10486
                uintptr_t *tmp = (uintptr_t *)(*(uintptr_t *)(
                            pThreadInfo->sampleBindArray
                            + sizeof(uintptr_t *) * k));
10487 10488 10489 10490
                int columnCount = (pThreadInfo->stbInfo)?
                    pThreadInfo->stbInfo->columnCount:
                    g_args.columnCount;
                for (int c = 1; c < columnCount + 1; c++) {
10491 10492 10493 10494 10495 10496 10497 10498
                    TAOS_BIND *bind = (TAOS_BIND *)((char *)tmp + (sizeof(TAOS_BIND) * c));
                    if (bind)
                        tmfree(bind->buffer);
                }
                tmfree((char *)tmp);
            }
            tmfree(pThreadInfo->sampleBindArray);
        }
10499
#endif
10500

10501 10502 10503 10504
        debugPrint("%s() LN%d, [%d] totalInsert=%"PRIu64" totalAffected=%"PRIu64"\n",
                __func__, __LINE__,
                pThreadInfo->threadID, pThreadInfo->totalInsertRows,
                pThreadInfo->totalAffectedRows);
10505 10506 10507
        if (stbInfo) {
            stbInfo->totalAffectedRows += pThreadInfo->totalAffectedRows;
            stbInfo->totalInsertRows += pThreadInfo->totalInsertRows;
10508 10509 10510 10511
        } else {
            g_args.totalAffectedRows += pThreadInfo->totalAffectedRows;
            g_args.totalInsertRows += pThreadInfo->totalInsertRows;
        }
H
Hui Li 已提交
10512

10513 10514 10515 10516 10517
        totalDelay  += pThreadInfo->totalDelay;
        cntDelay   += pThreadInfo->cntDelay;
        if (pThreadInfo->maxDelay > maxDelay) maxDelay = pThreadInfo->maxDelay;
        if (pThreadInfo->minDelay < minDelay) minDelay = pThreadInfo->minDelay;
    }
10518

10519 10520
    if (cntDelay == 0)    cntDelay = 1;
    avgDelay = (double)totalDelay / cntDelay;
10521

10522
    int64_t end = taosGetTimestampUs();
10523
    int64_t t = end - start;
10524
    if (0 == t) t = 1;
10525

10526
    double tInMs = (double) t / 1000000.0;
10527

10528
    if (stbInfo) {
10529
        fprintf(stderr, "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s.%s. %.2f records/second\n\n",
10530 10531
                tInMs, stbInfo->totalInsertRows,
                stbInfo->totalAffectedRows,
10532
                threads, db_name, stbInfo->stbName,
10533
                (double)(stbInfo->totalInsertRows/tInMs));
10534 10535 10536

        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10537
                    "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s.%s. %.2f records/second\n\n",
10538 10539
                    tInMs, stbInfo->totalInsertRows,
                    stbInfo->totalAffectedRows,
10540
                    threads, db_name, stbInfo->stbName,
10541
                    (double)(stbInfo->totalInsertRows/tInMs));
10542 10543
        }
    } else {
10544
        fprintf(stderr, "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s %.2f records/second\n\n",
10545 10546 10547
                tInMs, g_args.totalInsertRows,
                g_args.totalAffectedRows,
                threads, db_name,
10548
                (double)(g_args.totalInsertRows/tInMs));
10549 10550
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10551
                    "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s %.2f records/second\n\n",
10552 10553 10554
                    tInMs, g_args.totalInsertRows,
                    g_args.totalAffectedRows,
                    threads, db_name,
10555
                    (double)(g_args.totalInsertRows/tInMs));
10556
        }
10557
    }
10558

10559 10560 10561 10562 10563 10564 10565 10566 10567 10568 10569 10570
    if (minDelay != UINT64_MAX) {
        fprintf(stderr, "insert delay, avg: %10.2fms, max: %10.2fms, min: %10.2fms\n\n",
                (double)avgDelay/1000.0,
                (double)maxDelay/1000.0,
                (double)minDelay/1000.0);

        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult, "insert delay, avg:%10.2fms, max: %10.2fms, min: %10.2fms\n\n",
                    (double)avgDelay/1000.0,
                    (double)maxDelay/1000.0,
                    (double)minDelay/1000.0);
        }
10571
    }
10572

10573
    //taos_close(taos);
10574

10575 10576
    free(pids);
    free(infos);
H
Hui Li 已提交
10577 10578
}

10579
static void *queryNtableAggrFunc(void *sarg) {
10580 10581
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    TAOS *taos = pThreadInfo->taos;
10582
    setThreadName("queryNtableAggrFunc");
10583 10584 10585
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);

10586
    uint64_t startTime = pThreadInfo->start_time;
10587 10588 10589
    char *tb_prefix = pThreadInfo->tb_prefix;
    FILE *fp = fopen(pThreadInfo->filePath, "a");
    if (NULL == fp) {
10590
        errorPrint2("fopen %s fail, reason:%s.\n", pThreadInfo->filePath, strerror(errno));
10591
        free(command);
10592
        return NULL;
10593
    }
H
hzcheng 已提交
10594

10595
    int64_t insertRows;
10596
    /*  if (pThreadInfo->stbInfo) {
10597
        insertRows = pThreadInfo->stbInfo->insertRows; //  nrecords_per_table;
10598 10599
        } else {
        */
10600
    insertRows = g_args.insertRows;
10601 10602
    //  }

10603
    int64_t ntables = pThreadInfo->ntables; // pThreadInfo->end_table_to - pThreadInfo->start_table_from + 1;
10604
    int64_t totalData = insertRows * ntables;
10605 10606 10607 10608
    bool aggr_func = g_Dbs.aggr_func;

    char **aggreFunc;
    int n;
10609

10610 10611 10612 10613 10614 10615 10616 10617 10618
    if (g_args.demo_mode) {
        aggreFunc = g_aggreFuncDemo;
        n = aggr_func?(sizeof(g_aggreFuncDemo) / sizeof(g_aggreFuncDemo[0])) : 2;
    } else {
        aggreFunc = g_aggreFunc;
        n = aggr_func?(sizeof(g_aggreFunc) / sizeof(g_aggreFunc[0])) : 2;
    }

    if (!aggr_func) {
10619 10620 10621 10622 10623 10624 10625 10626
        printf("\nThe first field is either Binary or Bool. Aggregation functions are not supported.\n");
    }
    printf("%"PRId64" records:\n", totalData);
    fprintf(fp, "| QFunctions |    QRecords    |   QSpeed(R/s)   |  QLatency(ms) |\n");

    for (int j = 0; j < n; j++) {
        double totalT = 0;
        uint64_t count = 0;
10627 10628
        for (int64_t i = 0; i < ntables; i++) {
            sprintf(command, "SELECT %s FROM %s%"PRId64" WHERE ts>= %" PRIu64,
10629
                    aggreFunc[j], tb_prefix, i, startTime);
10630

10631 10632 10633
            double t = taosGetTimestampUs();
            debugPrint("%s() LN%d, sql command: %s\n",
                    __func__, __LINE__, command);
10634 10635 10636 10637
            TAOS_RES *pSql = taos_query(taos, command);
            int32_t code = taos_errno(pSql);

            if (code != 0) {
10638
                errorPrint2("Failed to query:%s\n", taos_errstr(pSql));
10639 10640 10641
                taos_free_result(pSql);
                taos_close(taos);
                fclose(fp);
10642
                free(command);
10643 10644
                return NULL;
            }
H
hzcheng 已提交
10645

10646 10647 10648
            while(taos_fetch_row(pSql) != NULL) {
                count++;
            }
H
hzcheng 已提交
10649

10650
            t = taosGetTimestampUs() - t;
10651 10652 10653 10654
            totalT += t;

            taos_free_result(pSql);
        }
H
hzcheng 已提交
10655

10656
        fprintf(fp, "|%10s  |   %"PRId64"   |  %12.2f   |   %10.2f  |\n",
10657 10658 10659
                aggreFunc[j][0] == '*' ? "   *   " : aggreFunc[j], totalData,
                (double)(ntables * insertRows) / totalT, totalT / 1000000);
        printf("select %10s took %.6f second(s)\n", aggreFunc[j], totalT / 1000000);
10660 10661 10662
    }
    fprintf(fp, "\n");
    fclose(fp);
10663
    free(command);
10664
    return NULL;
H
hzcheng 已提交
10665 10666
}

10667
static void *queryStableAggrFunc(void *sarg) {
10668 10669
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    TAOS *taos = pThreadInfo->taos;
10670
    setThreadName("queryStableAggrFunc");
10671 10672 10673
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);

10674 10675 10676
    FILE *fp = fopen(pThreadInfo->filePath, "a");
    if (NULL == fp) {
        printf("fopen %s fail, reason:%s.\n", pThreadInfo->filePath, strerror(errno));
10677
        free(command);
10678 10679
        return NULL;
    }
10680

10681
    int64_t insertRows = pThreadInfo->stbInfo->insertRows;
10682
    int64_t ntables = pThreadInfo->ntables; // pThreadInfo->end_table_to - pThreadInfo->start_table_from + 1;
10683
    int64_t totalData = insertRows * ntables;
10684 10685 10686 10687
    bool aggr_func = g_Dbs.aggr_func;

    char **aggreFunc;
    int n;
H
hzcheng 已提交
10688

10689 10690 10691 10692 10693 10694 10695 10696 10697
    if (g_args.demo_mode) {
        aggreFunc = g_aggreFuncDemo;
        n = aggr_func?(sizeof(g_aggreFuncDemo) / sizeof(g_aggreFuncDemo[0])) : 2;
    } else {
        aggreFunc = g_aggreFunc;
        n = aggr_func?(sizeof(g_aggreFunc) / sizeof(g_aggreFunc[0])) : 2;
    }

    if (!aggr_func) {
10698 10699
        printf("\nThe first field is either Binary or Bool. Aggregation functions are not supported.\n");
    }
10700

10701 10702
    printf("%"PRId64" records:\n", totalData);
    fprintf(fp, "Querying On %"PRId64" records:\n", totalData);
H
hzcheng 已提交
10703

10704 10705 10706
    for (int j = 0; j < n; j++) {
        char condition[COND_BUF_LEN] = "\0";
        char tempS[64] = "\0";
H
hzcheng 已提交
10707

10708
        int64_t m = 10 < ntables ? 10 : ntables;
H
hzcheng 已提交
10709

10710 10711
        for (int64_t i = 1; i <= m; i++) {
            if (i == 1) {
10712 10713 10714 10715 10716
                if (g_args.demo_mode) {
                    sprintf(tempS, "groupid = %"PRId64"", i);
                } else {
                    sprintf(tempS, "t0 = %"PRId64"", i);
                }
10717
            } else {
10718 10719 10720 10721 10722
                if (g_args.demo_mode) {
                    sprintf(tempS, " or groupid = %"PRId64" ", i);
                } else {
                    sprintf(tempS, " or t0 = %"PRId64" ", i);
                }
10723 10724
            }
            strncat(condition, tempS, COND_BUF_LEN - 1);
H
hzcheng 已提交
10725

10726
            sprintf(command, "SELECT %s FROM meters WHERE %s", aggreFunc[j], condition);
H
hzcheng 已提交
10727

10728
            printf("Where condition: %s\n", condition);
10729 10730 10731

            debugPrint("%s() LN%d, sql command: %s\n",
                    __func__, __LINE__, command);
10732
            fprintf(fp, "%s\n", command);
H
hzcheng 已提交
10733

10734
            double t = taosGetTimestampUs();
H
hzcheng 已提交
10735

10736 10737
            TAOS_RES *pSql = taos_query(taos, command);
            int32_t code = taos_errno(pSql);
S
Shuaiqiang Chang 已提交
10738

10739
            if (code != 0) {
10740
                errorPrint2("Failed to query:%s\n", taos_errstr(pSql));
10741 10742 10743
                taos_free_result(pSql);
                taos_close(taos);
                fclose(fp);
10744
                free(command);
10745 10746 10747 10748 10749 10750
                return NULL;
            }
            int count = 0;
            while(taos_fetch_row(pSql) != NULL) {
                count++;
            }
10751
            t = taosGetTimestampUs() - t;
H
hzcheng 已提交
10752

10753
            fprintf(fp, "| Speed: %12.2f(per s) | Latency: %.4f(ms) |\n",
10754 10755
                    ntables * insertRows / (t / 1000), t);
            printf("select %10s took %.6f second(s)\n\n", aggreFunc[j], t / 1000000);
H
hzcheng 已提交
10756

10757 10758 10759
            taos_free_result(pSql);
        }
        fprintf(fp, "\n");
H
hzcheng 已提交
10760
    }
10761
    fclose(fp);
10762
    free(command);
10763

10764
    return NULL;
H
hzcheng 已提交
10765 10766
}

10767 10768
static void prompt()
{
10769 10770 10771 10772
    if (!g_args.answer_yes) {
        printf("         Press enter key to continue or Ctrl-C to stop\n\n");
        (void)getchar();
    }
10773
}
H
Hui Li 已提交
10774

10775
static int insertTestProcess() {
10776

10777 10778 10779
    setupForAnsiEscape();
    int ret = printfInsertMeta();
    resetAfterAnsiEscape();
10780

10781 10782
    if (ret == -1)
        exit(EXIT_FAILURE);
10783

10784 10785 10786
    debugPrint("%d result file: %s\n", __LINE__, g_Dbs.resultFile);
    g_fpOfInsertResult = fopen(g_Dbs.resultFile, "a");
    if (NULL == g_fpOfInsertResult) {
10787
        errorPrint("Failed to open %s for save result\n", g_Dbs.resultFile);
10788 10789
        return -1;
    }
10790

10791 10792
    if (g_fpOfInsertResult)
        printfInsertMetaToFile(g_fpOfInsertResult);
10793

10794
    prompt();
10795

10796
    init_rand_data();
10797

10798
    // create database and super tables
10799 10800 10801 10802
    char *cmdBuffer = calloc(1, BUFFER_SIZE);
    assert(cmdBuffer);

    if(createDatabasesAndStables(cmdBuffer) != 0) {
10803 10804
        if (g_fpOfInsertResult)
            fclose(g_fpOfInsertResult);
10805
        free(cmdBuffer);
10806 10807
        return -1;
    }
10808
    free(cmdBuffer);
10809

10810
    // pretreatment
10811 10812 10813 10814 10815
    if (prepareSampleData() != 0) {
        if (g_fpOfInsertResult)
            fclose(g_fpOfInsertResult);
        return -1;
    }
10816

10817 10818
    double start;
    double end;
10819

10820
    if (g_totalChildTables > 0) {
10821 10822
        fprintf(stderr,
                "creating %"PRId64" table(s) with %d thread(s)\n\n",
10823
                g_totalChildTables, g_Dbs.threadCountForCreateTbl);
10824 10825 10826
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
                "creating %"PRId64" table(s) with %d thread(s)\n\n",
10827
                g_totalChildTables, g_Dbs.threadCountForCreateTbl);
10828 10829 10830 10831 10832 10833 10834 10835
        }

        // create child tables
        start = taosGetTimestampMs();
        createChildTables();
        end = taosGetTimestampMs();

        fprintf(stderr,
10836
                "\nSpent %.4f seconds to create %"PRId64" table(s) with %d thread(s), actual %"PRId64" table(s) created\n\n",
10837
                (end - start)/1000.0, g_totalChildTables,
10838
                g_Dbs.threadCountForCreateTbl, g_actualChildTables);
10839 10840
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10841
                "\nSpent %.4f seconds to create %"PRId64" table(s) with %d thread(s), actual %"PRId64" table(s) created\n\n",
10842
                (end - start)/1000.0, g_totalChildTables,
10843
                g_Dbs.threadCountForCreateTbl, g_actualChildTables);
10844
        }
10845
    }
10846

10847 10848 10849 10850 10851 10852
    // create sub threads for inserting data
    //start = taosGetTimestampMs();
    for (int i = 0; i < g_Dbs.dbCount; i++) {
        if (g_Dbs.use_metric) {
            if (g_Dbs.db[i].superTblCount > 0) {
                for (uint64_t j = 0; j < g_Dbs.db[i].superTblCount; j++) {
10853

10854
                    SSuperTable* stbInfo = &g_Dbs.db[i].superTbls[j];
10855

10856
                    if (stbInfo && (stbInfo->insertRows > 0)) {
10857 10858 10859 10860
                        startMultiThreadInsertData(
                                g_Dbs.threadCount,
                                g_Dbs.db[i].dbName,
                                g_Dbs.db[i].dbCfg.precision,
10861
                                stbInfo);
10862 10863 10864 10865
                    }
                }
            }
        } else {
10866
            startMultiThreadInsertData(
10867 10868 10869 10870
                    g_Dbs.threadCount,
                    g_Dbs.db[i].dbName,
                    g_Dbs.db[i].dbCfg.precision,
                    NULL);
10871
        }
H
Hui Li 已提交
10872
    }
10873
    //end = taosGetTimestampMs();
10874

10875 10876 10877 10878 10879 10880 10881 10882 10883 10884 10885
    //int64_t    totalInsertRows = 0;
    //int64_t    totalAffectedRows = 0;
    //for (int i = 0; i < g_Dbs.dbCount; i++) {
    //  for (int j = 0; j < g_Dbs.db[i].superTblCount; j++) {
    //  totalInsertRows+= g_Dbs.db[i].superTbls[j].totalInsertRows;
    //  totalAffectedRows += g_Dbs.db[i].superTbls[j].totalAffectedRows;
    //}
    //printf("Spent %.4f seconds to insert rows: %"PRId64", affected rows: %"PRId64" with %d thread(s)\n\n", end - start, totalInsertRows, totalAffectedRows, g_Dbs.threadCount);
    postFreeResource();

    return 0;
10886 10887
}

10888
static void *specifiedTableQuery(void *sarg) {
10889
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10890

10891
    setThreadName("specTableQuery");
10892

10893 10894 10895 10896 10897 10898 10899 10900
    if (pThreadInfo->taos == NULL) {
        TAOS * taos = NULL;
        taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                NULL,
                g_queryInfo.port);
        if (taos == NULL) {
10901
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
10902 10903 10904 10905 10906
                    pThreadInfo->threadID, taos_errstr(NULL));
            return NULL;
        } else {
            pThreadInfo->taos = taos;
        }
10907 10908
    }

10909 10910 10911 10912
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "use %s", g_queryInfo.dbName);
    if (0 != queryDbExec(pThreadInfo->taos, sqlStr, NO_INSERT_TYPE, false)) {
        taos_close(pThreadInfo->taos);
10913
        errorPrint("use database %s failed!\n\n",
10914
                g_queryInfo.dbName);
10915 10916
        return NULL;
    }
10917

10918 10919
    uint64_t st = 0;
    uint64_t et = 0;
10920

10921
    uint64_t queryTimes = g_queryInfo.specifiedQueryInfo.queryTimes;
10922

10923 10924 10925
    uint64_t totalQueried = 0;
    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
10926

10927 10928
    if (g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq][0] != '\0') {
        sprintf(pThreadInfo->filePath, "%s-%d",
10929 10930
                g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                pThreadInfo->threadID);
10931 10932
    }

10933 10934 10935 10936 10937 10938 10939
    while(queryTimes --) {
        if (g_queryInfo.specifiedQueryInfo.queryInterval && (et - st) <
                (int64_t)g_queryInfo.specifiedQueryInfo.queryInterval) {
            taosMsleep(g_queryInfo.specifiedQueryInfo.queryInterval - (et - st)); // ms
        }

        st = taosGetTimestampMs();
10940

10941 10942
        selectAndGetResult(pThreadInfo,
                g_queryInfo.specifiedQueryInfo.sql[pThreadInfo->querySeq]);
10943

10944 10945 10946
        et = taosGetTimestampMs();
        printf("=thread[%"PRId64"] use %s complete one sql, Spent %10.3f s\n",
                taosGetSelfPthreadId(), g_queryInfo.queryMode, (et - st)/1000.0);
10947

10948 10949
        totalQueried ++;
        g_queryInfo.specifiedQueryInfo.totalQueried ++;
10950

10951 10952 10953
        uint64_t  currentPrintTime = taosGetTimestampMs();
        uint64_t  endTs = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
10954
            debugPrint("%s() LN%d, endTs=%"PRIu64" ms, startTs=%"PRIu64" ms\n",
10955 10956
                    __func__, __LINE__, endTs, startTs);
            printf("thread[%d] has currently completed queries: %"PRIu64", QPS: %10.6f\n",
10957 10958
                    pThreadInfo->threadID,
                    totalQueried,
10959
                    (double)(totalQueried/((endTs-startTs)/1000.0)));
10960 10961
            lastPrintTime = currentPrintTime;
        }
10962
    }
10963
    return NULL;
10964
}
H
Hui Li 已提交
10965

10966
static void replaceChildTblName(char* inSql, char* outSql, int tblIndex) {
10967
    char sourceString[32] = "xxxx";
10968
    char subTblName[TSDB_TABLE_NAME_LEN];
10969 10970 10971
    sprintf(subTblName, "%s.%s",
            g_queryInfo.dbName,
            g_queryInfo.superQueryInfo.childTblName + tblIndex*TSDB_TABLE_NAME_LEN);
10972

10973
    //printf("inSql: %s\n", inSql);
10974

10975 10976 10977 10978 10979 10980 10981
    char* pos = strstr(inSql, sourceString);
    if (0 == pos) {
        return;
    }

    tstrncpy(outSql, inSql, pos - inSql + 1);
    //printf("1: %s\n", outSql);
10982
    strncat(outSql, subTblName, BUFFER_SIZE - 1);
10983
    //printf("2: %s\n", outSql);
10984
    strncat(outSql, pos+strlen(sourceString), BUFFER_SIZE - 1);
10985
    //printf("3: %s\n", outSql);
H
Hui Li 已提交
10986 10987
}

10988
static void *superTableQuery(void *sarg) {
10989 10990 10991
    char *sqlstr = calloc(1, BUFFER_SIZE);
    assert(sqlstr);

10992
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10993

10994
    setThreadName("superTableQuery");
10995

10996 10997 10998 10999 11000 11001 11002 11003 11004 11005
    if (pThreadInfo->taos == NULL) {
        TAOS * taos = NULL;
        taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                NULL,
                g_queryInfo.port);
        if (taos == NULL) {
            errorPrint("[%d] Failed to connect to TDengine, reason:%s\n",
                    pThreadInfo->threadID, taos_errstr(NULL));
11006
            free(sqlstr);
11007 11008 11009 11010
            return NULL;
        } else {
            pThreadInfo->taos = taos;
        }
11011
    }
11012

11013 11014
    uint64_t st = 0;
    uint64_t et = (int64_t)g_queryInfo.superQueryInfo.queryInterval;
11015

11016 11017 11018
    uint64_t queryTimes = g_queryInfo.superQueryInfo.queryTimes;
    uint64_t totalQueried = 0;
    uint64_t  startTs = taosGetTimestampMs();
H
Hui Li 已提交
11019

11020 11021 11022 11023 11024 11025 11026 11027 11028 11029 11030
    uint64_t  lastPrintTime = taosGetTimestampMs();
    while(queryTimes --) {
        if (g_queryInfo.superQueryInfo.queryInterval
                && (et - st) < (int64_t)g_queryInfo.superQueryInfo.queryInterval) {
            taosMsleep(g_queryInfo.superQueryInfo.queryInterval - (et - st)); // ms
            //printf("========sleep duration:%"PRId64 "========inserted rows:%d, table range:%d - %d\n", (1000 - (et - st)), i, pThreadInfo->start_table_from, pThreadInfo->end_table_to);
        }

        st = taosGetTimestampMs();
        for (int i = pThreadInfo->start_table_from; i <= pThreadInfo->end_table_to; i++) {
            for (int j = 0; j < g_queryInfo.superQueryInfo.sqlCount; j++) {
11031
                memset(sqlstr, 0, BUFFER_SIZE);
11032 11033 11034 11035 11036 11037 11038
                replaceChildTblName(g_queryInfo.superQueryInfo.sql[j], sqlstr, i);
                if (g_queryInfo.superQueryInfo.result[j][0] != '\0') {
                    sprintf(pThreadInfo->filePath, "%s-%d",
                            g_queryInfo.superQueryInfo.result[j],
                            pThreadInfo->threadID);
                }
                selectAndGetResult(pThreadInfo, sqlstr);
11039

11040 11041
                totalQueried++;
                g_queryInfo.superQueryInfo.totalQueried ++;
11042

11043 11044 11045 11046 11047 11048 11049 11050 11051 11052
                int64_t  currentPrintTime = taosGetTimestampMs();
                int64_t  endTs = taosGetTimestampMs();
                if (currentPrintTime - lastPrintTime > 30*1000) {
                    printf("thread[%d] has currently completed queries: %"PRIu64", QPS: %10.3f\n",
                            pThreadInfo->threadID,
                            totalQueried,
                            (double)(totalQueried/((endTs-startTs)/1000.0)));
                    lastPrintTime = currentPrintTime;
                }
            }
11053
        }
11054 11055 11056 11057 11058 11059
        et = taosGetTimestampMs();
        printf("####thread[%"PRId64"] complete all sqls to allocate all sub-tables[%"PRIu64" - %"PRIu64"] once queries duration:%.4fs\n\n",
                taosGetSelfPthreadId(),
                pThreadInfo->start_table_from,
                pThreadInfo->end_table_to,
                (double)(et - st)/1000.0);
H
Hui Li 已提交
11060
    }
11061

11062
    free(sqlstr);
11063
    return NULL;
11064 11065
}

11066
static int queryTestProcess() {
11067

11068 11069 11070
    setupForAnsiEscape();
    printfQueryMeta();
    resetAfterAnsiEscape();
11071

11072 11073 11074 11075 11076 11077 11078
    TAOS * taos = NULL;
    taos = taos_connect(g_queryInfo.host,
            g_queryInfo.user,
            g_queryInfo.password,
            NULL,
            g_queryInfo.port);
    if (taos == NULL) {
11079
        errorPrint("Failed to connect to TDengine, reason:%s\n",
11080
                taos_errstr(NULL));
11081
        exit(EXIT_FAILURE);
11082
    }
11083

11084 11085 11086
    if (0 != g_queryInfo.superQueryInfo.sqlCount) {
        getAllChildNameOfSuperTable(taos,
                g_queryInfo.dbName,
11087
                g_queryInfo.superQueryInfo.stbName,
11088 11089 11090
                &g_queryInfo.superQueryInfo.childTblName,
                &g_queryInfo.superQueryInfo.childTblCount);
    }
11091

11092
    prompt();
11093

11094 11095 11096
    if (g_args.debug_print || g_args.verbose_print) {
        printfQuerySystemInfo(taos);
    }
11097

11098 11099 11100
    if (0 == strncasecmp(g_queryInfo.queryMode, "rest", strlen("rest"))) {
        if (convertHostToServAddr(
                    g_queryInfo.host, g_queryInfo.port, &g_queryInfo.serv_addr) != 0)
11101
            ERROR_EXIT("convert host to server address");
11102
    }
11103

11104 11105 11106 11107 11108
    pthread_t  *pids  = NULL;
    threadInfo *infos = NULL;
    //==== create sub threads for query from specify table
    int nConcurrent = g_queryInfo.specifiedQueryInfo.concurrent;
    uint64_t nSqlCount = g_queryInfo.specifiedQueryInfo.sqlCount;
11109

11110
    uint64_t startTs = taosGetTimestampMs();
11111

11112
    if ((nSqlCount > 0) && (nConcurrent > 0)) {
11113

11114 11115
        pids  = calloc(1, nConcurrent * nSqlCount * sizeof(pthread_t));
        infos = calloc(1, nConcurrent * nSqlCount * sizeof(threadInfo));
11116

11117 11118 11119 11120
        if ((NULL == pids) || (NULL == infos)) {
            taos_close(taos);
            ERROR_EXIT("memory allocation failed for create threads\n");
        }
11121

11122 11123 11124 11125 11126 11127 11128 11129 11130
        for (uint64_t i = 0; i < nSqlCount; i++) {
            for (int j = 0; j < nConcurrent; j++) {
                uint64_t seq = i * nConcurrent + j;
                threadInfo *pThreadInfo = infos + seq;
                pThreadInfo->threadID = seq;
                pThreadInfo->querySeq = i;

                if (0 == strncasecmp(g_queryInfo.queryMode, "taosc", 5)) {

11131 11132
                    char sqlStr[TSDB_DB_NAME_LEN + 5];
                    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11133 11134 11135 11136
                    if (0 != queryDbExec(taos, sqlStr, NO_INSERT_TYPE, false)) {
                        taos_close(taos);
                        free(infos);
                        free(pids);
11137
                        errorPrint2("use database %s failed!\n\n",
11138 11139 11140
                                g_queryInfo.dbName);
                        return -1;
                    }
11141
                }
11142

11143
                pThreadInfo->taos = NULL;// workaround to use separate taos connection;
11144

11145 11146 11147
                pthread_create(pids + seq, NULL, specifiedTableQuery,
                        pThreadInfo);
            }
11148
        }
11149 11150
    } else {
        g_queryInfo.specifiedQueryInfo.concurrent = 0;
11151 11152
    }

11153
    taos_close(taos);
11154

11155 11156 11157 11158 11159 11160 11161
    pthread_t  *pidsOfSub  = NULL;
    threadInfo *infosOfSub = NULL;
    //==== create sub threads for query from all sub table of the super table
    if ((g_queryInfo.superQueryInfo.sqlCount > 0)
            && (g_queryInfo.superQueryInfo.threadCnt > 0)) {
        pidsOfSub  = calloc(1, g_queryInfo.superQueryInfo.threadCnt * sizeof(pthread_t));
        infosOfSub = calloc(1, g_queryInfo.superQueryInfo.threadCnt * sizeof(threadInfo));
11162

11163 11164 11165
        if ((NULL == pidsOfSub) || (NULL == infosOfSub)) {
            free(infos);
            free(pids);
11166

11167 11168
            ERROR_EXIT("memory allocation failed for create threads\n");
        }
11169

11170 11171
        int64_t ntables = g_queryInfo.superQueryInfo.childTblCount;
        int threads = g_queryInfo.superQueryInfo.threadCnt;
11172

11173 11174 11175 11176 11177
        int64_t a = ntables / threads;
        if (a < 1) {
            threads = ntables;
            a = 1;
        }
11178

11179 11180 11181 11182
        int64_t b = 0;
        if (threads != 0) {
            b = ntables % threads;
        }
11183

11184 11185 11186 11187
        uint64_t tableFrom = 0;
        for (int i = 0; i < threads; i++) {
            threadInfo *pThreadInfo = infosOfSub + i;
            pThreadInfo->threadID = i;
11188

11189 11190 11191 11192
            pThreadInfo->start_table_from = tableFrom;
            pThreadInfo->ntables = i<b?a+1:a;
            pThreadInfo->end_table_to = i < b ? tableFrom + a : tableFrom + a - 1;
            tableFrom = pThreadInfo->end_table_to + 1;
11193
            pThreadInfo->taos = NULL; // workaround to use separate taos connection;
11194 11195
            pthread_create(pidsOfSub + i, NULL, superTableQuery, pThreadInfo);
        }
11196

11197 11198 11199 11200
        g_queryInfo.superQueryInfo.threadCnt = threads;
    } else {
        g_queryInfo.superQueryInfo.threadCnt = 0;
    }
11201

11202 11203 11204 11205 11206 11207
    if ((nSqlCount > 0) && (nConcurrent > 0)) {
        for (int i = 0; i < nConcurrent; i++) {
            for (int j = 0; j < nSqlCount; j++) {
                pthread_join(pids[i * nSqlCount + j], NULL);
            }
        }
11208
    }
S
Shuaiqiang Chang 已提交
11209

11210 11211
    tmfree((char*)pids);
    tmfree((char*)infos);
11212

11213 11214 11215
    for (int i = 0; i < g_queryInfo.superQueryInfo.threadCnt; i++) {
        pthread_join(pidsOfSub[i], NULL);
    }
H
Hui Li 已提交
11216

11217 11218
    tmfree((char*)pidsOfSub);
    tmfree((char*)infosOfSub);
11219

11220
    //  taos_close(taos);// workaround to use separate taos connection;
11221
    uint64_t endTs = taosGetTimestampMs();
11222

11223 11224
    uint64_t totalQueried = g_queryInfo.specifiedQueryInfo.totalQueried +
        g_queryInfo.superQueryInfo.totalQueried;
11225

11226 11227 11228 11229
    fprintf(stderr, "==== completed total queries: %"PRIu64", the QPS of all threads: %10.3f====\n",
            totalQueried,
            (double)(totalQueried/((endTs-startTs)/1000.0)));
    return 0;
11230 11231
}

11232 11233
static void stable_sub_callback(
        TAOS_SUB* tsub, TAOS_RES *res, void* param, int code) {
11234
    if (res == NULL || taos_errno(res) != 0) {
11235
        errorPrint2("%s() LN%d, failed to subscribe result, code:%d, reason:%s\n",
11236 11237 11238
                __func__, __LINE__, code, taos_errstr(res));
        return;
    }
11239

11240 11241
    if (param)
        fetchResult(res, (threadInfo *)param);
11242
    // tao_unsubscribe() will free result.
11243 11244 11245 11246
}

static void specified_sub_callback(
        TAOS_SUB* tsub, TAOS_RES *res, void* param, int code) {
11247
    if (res == NULL || taos_errno(res) != 0) {
11248
        errorPrint2("%s() LN%d, failed to subscribe result, code:%d, reason:%s\n",
11249 11250 11251
                __func__, __LINE__, code, taos_errstr(res));
        return;
    }
11252

11253 11254
    if (param)
        fetchResult(res, (threadInfo *)param);
11255
    // tao_unsubscribe() will free result.
H
hzcheng 已提交
11256 11257
}

11258
static TAOS_SUB* subscribeImpl(
11259 11260 11261 11262
        QUERY_CLASS class,
        threadInfo *pThreadInfo,
        char *sql, char* topic, bool restart, uint64_t interval)
{
11263 11264 11265 11266 11267 11268 11269 11270 11271 11272 11273 11274 11275 11276 11277 11278 11279 11280 11281 11282 11283 11284 11285 11286
    TAOS_SUB* tsub = NULL;

    if ((SPECIFIED_CLASS == class)
            && (ASYNC_MODE == g_queryInfo.specifiedQueryInfo.asyncMode)) {
        tsub = taos_subscribe(
                pThreadInfo->taos,
                restart,
                topic, sql, specified_sub_callback, (void*)pThreadInfo,
                g_queryInfo.specifiedQueryInfo.subscribeInterval);
    } else if ((STABLE_CLASS == class)
            && (ASYNC_MODE == g_queryInfo.superQueryInfo.asyncMode)) {
        tsub = taos_subscribe(
                pThreadInfo->taos,
                restart,
                topic, sql, stable_sub_callback, (void*)pThreadInfo,
                g_queryInfo.superQueryInfo.subscribeInterval);
    } else {
        tsub = taos_subscribe(
                pThreadInfo->taos,
                restart,
                topic, sql, NULL, NULL, interval);
    }

    if (tsub == NULL) {
11287
        errorPrint2("failed to create subscription. topic:%s, sql:%s\n", topic, sql);
11288 11289 11290 11291
        return NULL;
    }

    return tsub;
11292
}
H
hzcheng 已提交
11293

11294
static void *superSubscribe(void *sarg) {
11295
    threadInfo *pThreadInfo = (threadInfo *)sarg;
11296 11297 11298
    char *subSqlStr = calloc(1, BUFFER_SIZE);
    assert(subSqlStr);

11299 11300 11301
    TAOS_SUB*    tsub[MAX_QUERY_SQL_COUNT] = {0};
    uint64_t tsubSeq;

11302 11303
    setThreadName("superSub");

11304
    if (pThreadInfo->ntables > MAX_QUERY_SQL_COUNT) {
11305
        free(subSqlStr);
11306 11307
        errorPrint("The table number(%"PRId64") of the thread is more than max query sql count: %d\n",
                pThreadInfo->ntables, MAX_QUERY_SQL_COUNT);
11308
        exit(EXIT_FAILURE);
11309
    }
11310

11311
    if (pThreadInfo->taos == NULL) {
11312 11313 11314 11315 11316 11317
        pThreadInfo->taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                g_queryInfo.dbName,
                g_queryInfo.port);
        if (pThreadInfo->taos == NULL) {
11318
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11319
                    pThreadInfo->threadID, taos_errstr(NULL));
11320
            free(subSqlStr);
11321 11322
            return NULL;
        }
11323 11324
    }

11325 11326
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11327 11328
    if (0 != queryDbExec(pThreadInfo->taos, sqlStr, NO_INSERT_TYPE, false)) {
        taos_close(pThreadInfo->taos);
11329
        errorPrint2("use database %s failed!\n\n",
11330
                g_queryInfo.dbName);
11331
        free(subSqlStr);
11332 11333
        return NULL;
    }
11334

11335 11336 11337 11338 11339 11340 11341 11342 11343 11344 11345
    char topic[32] = {0};
    for (uint64_t i = pThreadInfo->start_table_from;
            i <= pThreadInfo->end_table_to; i++) {
        tsubSeq = i - pThreadInfo->start_table_from;
        verbosePrint("%s() LN%d, [%d], start=%"PRId64" end=%"PRId64" i=%"PRIu64"\n",
                __func__, __LINE__,
                pThreadInfo->threadID,
                pThreadInfo->start_table_from,
                pThreadInfo->end_table_to, i);
        sprintf(topic, "taosdemo-subscribe-%"PRIu64"-%"PRIu64"",
                i, pThreadInfo->querySeq);
11346
        memset(subSqlStr, 0, BUFFER_SIZE);
11347 11348
        replaceChildTblName(
                g_queryInfo.superQueryInfo.sql[pThreadInfo->querySeq],
11349
                subSqlStr, i);
11350 11351 11352 11353 11354
        if (g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq][0] != 0) {
            sprintf(pThreadInfo->filePath, "%s-%d",
                    g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq],
                    pThreadInfo->threadID);
        }
11355

11356 11357
        verbosePrint("%s() LN%d, [%d] subSqlStr: %s\n",
                __func__, __LINE__, pThreadInfo->threadID, subSqlStr);
11358 11359
        tsub[tsubSeq] = subscribeImpl(
                STABLE_CLASS,
11360
                pThreadInfo, subSqlStr, topic,
11361 11362 11363 11364
                g_queryInfo.superQueryInfo.subscribeRestart,
                g_queryInfo.superQueryInfo.subscribeInterval);
        if (NULL == tsub[tsubSeq]) {
            taos_close(pThreadInfo->taos);
11365
            free(subSqlStr);
11366 11367 11368
            return NULL;
        }
    }
11369

11370 11371 11372 11373 11374 11375
    // start loop to consume result
    int consumed[MAX_QUERY_SQL_COUNT];
    for (int i = 0; i < MAX_QUERY_SQL_COUNT; i++) {
        consumed[i] = 0;
    }
    TAOS_RES* res = NULL;
11376

11377
    uint64_t st = 0, et = 0;
11378

11379 11380 11381 11382
    while ((g_queryInfo.superQueryInfo.endAfterConsume == -1)
            || (g_queryInfo.superQueryInfo.endAfterConsume >
                consumed[pThreadInfo->end_table_to
                - pThreadInfo->start_table_from])) {
11383

11384 11385 11386 11387 11388 11389 11390 11391 11392 11393 11394 11395 11396 11397 11398 11399 11400 11401 11402 11403 11404 11405 11406 11407 11408 11409 11410 11411 11412 11413 11414 11415 11416 11417 11418 11419 11420 11421
        verbosePrint("super endAfterConsume: %d, consumed: %d\n",
                g_queryInfo.superQueryInfo.endAfterConsume,
                consumed[pThreadInfo->end_table_to
                - pThreadInfo->start_table_from]);
        for (uint64_t i = pThreadInfo->start_table_from;
                i <= pThreadInfo->end_table_to; i++) {
            tsubSeq = i - pThreadInfo->start_table_from;
            if (ASYNC_MODE == g_queryInfo.superQueryInfo.asyncMode) {
                continue;
            }

            st = taosGetTimestampMs();
            performancePrint("st: %"PRIu64" et: %"PRIu64" st-et: %"PRIu64"\n", st, et, (st - et));
            res = taos_consume(tsub[tsubSeq]);
            et = taosGetTimestampMs();
            performancePrint("st: %"PRIu64" et: %"PRIu64" delta: %"PRIu64"\n", st, et, (et - st));

            if (res) {
                if (g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq][0] != 0) {
                    sprintf(pThreadInfo->filePath, "%s-%d",
                            g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq],
                            pThreadInfo->threadID);
                    fetchResult(res, pThreadInfo);
                }
                consumed[tsubSeq] ++;

                if ((g_queryInfo.superQueryInfo.resubAfterConsume != -1)
                        && (consumed[tsubSeq] >=
                            g_queryInfo.superQueryInfo.resubAfterConsume)) {
                    verbosePrint("%s() LN%d, keepProgress:%d, resub super table query: %"PRIu64"\n",
                            __func__, __LINE__,
                            g_queryInfo.superQueryInfo.subscribeKeepProgress,
                            pThreadInfo->querySeq);
                    taos_unsubscribe(tsub[tsubSeq],
                            g_queryInfo.superQueryInfo.subscribeKeepProgress);
                    consumed[tsubSeq]= 0;
                    tsub[tsubSeq] = subscribeImpl(
                            STABLE_CLASS,
11422
                            pThreadInfo, subSqlStr, topic,
11423 11424 11425 11426 11427
                            g_queryInfo.superQueryInfo.subscribeRestart,
                            g_queryInfo.superQueryInfo.subscribeInterval
                            );
                    if (NULL == tsub[tsubSeq]) {
                        taos_close(pThreadInfo->taos);
11428
                        free(subSqlStr);
11429 11430 11431 11432 11433 11434 11435 11436 11437 11438 11439
                        return NULL;
                    }
                }
            }
        }
    }
    verbosePrint("%s() LN%d, super endAfterConsume: %d, consumed: %d\n",
            __func__, __LINE__,
            g_queryInfo.superQueryInfo.endAfterConsume,
            consumed[pThreadInfo->end_table_to - pThreadInfo->start_table_from]);
    taos_free_result(res);
11440

11441 11442
    for (uint64_t i = pThreadInfo->start_table_from;
            i <= pThreadInfo->end_table_to; i++) {
11443 11444
        tsubSeq = i - pThreadInfo->start_table_from;
        taos_unsubscribe(tsub[tsubSeq], 0);
H
hzcheng 已提交
11445
    }
11446

11447
    taos_close(pThreadInfo->taos);
11448
    free(subSqlStr);
11449
    return NULL;
H
hzcheng 已提交
11450 11451
}

11452
static void *specifiedSubscribe(void *sarg) {
11453 11454
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    //  TAOS_SUB*  tsub = NULL;
11455

11456
    setThreadName("specSub");
11457

11458
    if (pThreadInfo->taos == NULL) {
11459 11460 11461 11462 11463 11464
        pThreadInfo->taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                g_queryInfo.dbName,
                g_queryInfo.port);
        if (pThreadInfo->taos == NULL) {
11465
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11466 11467 11468
                    pThreadInfo->threadID, taos_errstr(NULL));
            return NULL;
        }
11469 11470
    }

11471 11472
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11473 11474 11475 11476 11477 11478 11479 11480 11481 11482 11483
    if (0 != queryDbExec(pThreadInfo->taos, sqlStr, NO_INSERT_TYPE, false)) {
        taos_close(pThreadInfo->taos);
        return NULL;
    }

    sprintf(g_queryInfo.specifiedQueryInfo.topic[pThreadInfo->threadID],
            "taosdemo-subscribe-%"PRIu64"-%d",
            pThreadInfo->querySeq,
            pThreadInfo->threadID);
    if (g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq][0] != '\0') {
        sprintf(pThreadInfo->filePath, "%s-%d",
11484 11485
                g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                pThreadInfo->threadID);
11486 11487 11488 11489 11490 11491 11492 11493 11494 11495 11496 11497 11498
    }
    g_queryInfo.specifiedQueryInfo.tsub[pThreadInfo->threadID] = subscribeImpl(
            SPECIFIED_CLASS, pThreadInfo,
            g_queryInfo.specifiedQueryInfo.sql[pThreadInfo->querySeq],
            g_queryInfo.specifiedQueryInfo.topic[pThreadInfo->threadID],
            g_queryInfo.specifiedQueryInfo.subscribeRestart,
            g_queryInfo.specifiedQueryInfo.subscribeInterval);
    if (NULL == g_queryInfo.specifiedQueryInfo.tsub[pThreadInfo->threadID]) {
        taos_close(pThreadInfo->taos);
        return NULL;
    }

    // start loop to consume result
11499

11500 11501 11502 11503 11504 11505 11506 11507 11508 11509 11510 11511 11512 11513 11514 11515 11516 11517 11518 11519 11520 11521 11522 11523 11524 11525 11526 11527 11528 11529 11530 11531 11532 11533 11534 11535 11536 11537 11538 11539 11540 11541 11542 11543 11544 11545 11546 11547 11548 11549 11550 11551 11552 11553 11554 11555
    g_queryInfo.specifiedQueryInfo.consumed[pThreadInfo->threadID] = 0;
    while((g_queryInfo.specifiedQueryInfo.endAfterConsume[pThreadInfo->querySeq] == -1)
            || (g_queryInfo.specifiedQueryInfo.consumed[pThreadInfo->threadID] <
                g_queryInfo.specifiedQueryInfo.endAfterConsume[pThreadInfo->querySeq])) {

        printf("consumed[%d]: %d, endAfterConsum[%"PRId64"]: %d\n",
                pThreadInfo->threadID,
                g_queryInfo.specifiedQueryInfo.consumed[pThreadInfo->threadID],
                pThreadInfo->querySeq,
                g_queryInfo.specifiedQueryInfo.endAfterConsume[pThreadInfo->querySeq]);
        if (ASYNC_MODE == g_queryInfo.specifiedQueryInfo.asyncMode) {
            continue;
        }

        g_queryInfo.specifiedQueryInfo.res[pThreadInfo->threadID] = taos_consume(
                g_queryInfo.specifiedQueryInfo.tsub[pThreadInfo->threadID]);
        if (g_queryInfo.specifiedQueryInfo.res[pThreadInfo->threadID]) {
            if (g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq][0]
                    != 0) {
                sprintf(pThreadInfo->filePath, "%s-%d",
                        g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                        pThreadInfo->threadID);
            }
            fetchResult(
                    g_queryInfo.specifiedQueryInfo.res[pThreadInfo->threadID],
                    pThreadInfo);

            g_queryInfo.specifiedQueryInfo.consumed[pThreadInfo->threadID] ++;
            if ((g_queryInfo.specifiedQueryInfo.resubAfterConsume[pThreadInfo->querySeq] != -1)
                    && (g_queryInfo.specifiedQueryInfo.consumed[pThreadInfo->threadID] >=
                        g_queryInfo.specifiedQueryInfo.resubAfterConsume[pThreadInfo->querySeq])) {
                printf("keepProgress:%d, resub specified query: %"PRIu64"\n",
                        g_queryInfo.specifiedQueryInfo.subscribeKeepProgress,
                        pThreadInfo->querySeq);
                g_queryInfo.specifiedQueryInfo.consumed[pThreadInfo->threadID] = 0;
                taos_unsubscribe(g_queryInfo.specifiedQueryInfo.tsub[pThreadInfo->threadID],
                        g_queryInfo.specifiedQueryInfo.subscribeKeepProgress);
                g_queryInfo.specifiedQueryInfo.tsub[pThreadInfo->threadID] =
                    subscribeImpl(
                            SPECIFIED_CLASS,
                            pThreadInfo,
                            g_queryInfo.specifiedQueryInfo.sql[pThreadInfo->querySeq],
                            g_queryInfo.specifiedQueryInfo.topic[pThreadInfo->threadID],
                            g_queryInfo.specifiedQueryInfo.subscribeRestart,
                            g_queryInfo.specifiedQueryInfo.subscribeInterval);
                if (NULL == g_queryInfo.specifiedQueryInfo.tsub[pThreadInfo->threadID]) {
                    taos_close(pThreadInfo->taos);
                    return NULL;
                }
            }
        }
    }
    taos_free_result(g_queryInfo.specifiedQueryInfo.res[pThreadInfo->threadID]);
    taos_close(pThreadInfo->taos);

    return NULL;
H
hzcheng 已提交
11556 11557
}

11558
static int subscribeTestProcess() {
11559 11560 11561 11562 11563 11564 11565 11566 11567 11568
    setupForAnsiEscape();
    printfQueryMeta();
    resetAfterAnsiEscape();

    prompt();

    TAOS * taos = NULL;
    taos = taos_connect(g_queryInfo.host,
            g_queryInfo.user,
            g_queryInfo.password,
11569
            g_queryInfo.dbName,
11570 11571
            g_queryInfo.port);
    if (taos == NULL) {
11572
        errorPrint2("Failed to connect to TDengine, reason:%s\n",
11573
                taos_errstr(NULL));
11574
        exit(EXIT_FAILURE);
S
Shengliang Guan 已提交
11575
    }
11576

11577 11578 11579
    if (0 != g_queryInfo.superQueryInfo.sqlCount) {
        getAllChildNameOfSuperTable(taos,
                g_queryInfo.dbName,
11580
                g_queryInfo.superQueryInfo.stbName,
11581 11582
                &g_queryInfo.superQueryInfo.childTblName,
                &g_queryInfo.superQueryInfo.childTblCount);
S
Shengliang Guan 已提交
11583
    }
11584

11585
    taos_close(taos); // workaround to use separate taos connection;
11586 11587 11588 11589 11590 11591 11592 11593 11594

    pthread_t  *pids = NULL;
    threadInfo *infos = NULL;

    pthread_t  *pidsOfStable  = NULL;
    threadInfo *infosOfStable = NULL;

    //==== create threads for query for specified table
    if (g_queryInfo.specifiedQueryInfo.sqlCount <= 0) {
11595
        debugPrint("%s() LN%d, specified query sqlCount %d.\n",
11596 11597 11598 11599
                __func__, __LINE__,
                g_queryInfo.specifiedQueryInfo.sqlCount);
    } else {
        if (g_queryInfo.specifiedQueryInfo.concurrent <= 0) {
11600
            errorPrint2("%s() LN%d, specified query sqlCount %d.\n",
11601 11602
                    __func__, __LINE__,
                    g_queryInfo.specifiedQueryInfo.sqlCount);
11603
            exit(EXIT_FAILURE);
11604
        }
11605

11606
        pids  = calloc(
11607
                1,
11608 11609 11610 11611
                g_queryInfo.specifiedQueryInfo.sqlCount *
                g_queryInfo.specifiedQueryInfo.concurrent *
                sizeof(pthread_t));
        infos = calloc(
11612
                1,
11613 11614 11615 11616
                g_queryInfo.specifiedQueryInfo.sqlCount *
                g_queryInfo.specifiedQueryInfo.concurrent *
                sizeof(threadInfo));
        if ((NULL == pids) || (NULL == infos)) {
11617
            errorPrint2("%s() LN%d, malloc failed for create threads\n", __func__, __LINE__);
11618
            exit(EXIT_FAILURE);
11619
        }
11620

11621 11622 11623 11624 11625 11626
        for (int i = 0; i < g_queryInfo.specifiedQueryInfo.sqlCount; i++) {
            for (int j = 0; j < g_queryInfo.specifiedQueryInfo.concurrent; j++) {
                uint64_t seq = i * g_queryInfo.specifiedQueryInfo.concurrent + j;
                threadInfo *pThreadInfo = infos + seq;
                pThreadInfo->threadID = seq;
                pThreadInfo->querySeq = i;
11627
                pThreadInfo->taos = NULL;  // workaround to use separate taos connection;
11628 11629
                pthread_create(pids + seq, NULL, specifiedSubscribe, pThreadInfo);
            }
11630
        }
11631
    }
11632

11633 11634 11635 11636 11637 11638 11639 11640 11641 11642 11643 11644 11645 11646 11647 11648 11649 11650 11651
    //==== create threads for super table query
    if (g_queryInfo.superQueryInfo.sqlCount <= 0) {
        debugPrint("%s() LN%d, super table query sqlCount %d.\n",
                __func__, __LINE__,
                g_queryInfo.superQueryInfo.sqlCount);
    } else {
        if ((g_queryInfo.superQueryInfo.sqlCount > 0)
                && (g_queryInfo.superQueryInfo.threadCnt > 0)) {
            pidsOfStable  = calloc(
                    1,
                    g_queryInfo.superQueryInfo.sqlCount *
                    g_queryInfo.superQueryInfo.threadCnt *
                    sizeof(pthread_t));
            infosOfStable = calloc(
                    1,
                    g_queryInfo.superQueryInfo.sqlCount *
                    g_queryInfo.superQueryInfo.threadCnt *
                    sizeof(threadInfo));
            if ((NULL == pidsOfStable) || (NULL == infosOfStable)) {
11652
                errorPrint2("%s() LN%d, malloc failed for create threads\n",
11653 11654
                        __func__, __LINE__);
                // taos_close(taos);
11655
                exit(EXIT_FAILURE);
11656
            }
11657

11658 11659
            int64_t ntables = g_queryInfo.superQueryInfo.childTblCount;
            int threads = g_queryInfo.superQueryInfo.threadCnt;
11660

11661 11662 11663 11664
            int64_t a = ntables / threads;
            if (a < 1) {
                threads = ntables;
                a = 1;
11665
            }
11666

11667 11668 11669 11670 11671 11672 11673 11674 11675 11676 11677 11678 11679 11680 11681 11682 11683
            int64_t b = 0;
            if (threads != 0) {
                b = ntables % threads;
            }

            for (uint64_t i = 0; i < g_queryInfo.superQueryInfo.sqlCount; i++) {
                uint64_t tableFrom = 0;
                for (int j = 0; j < threads; j++) {
                    uint64_t seq = i * threads + j;
                    threadInfo *pThreadInfo = infosOfStable + seq;
                    pThreadInfo->threadID = seq;
                    pThreadInfo->querySeq = i;

                    pThreadInfo->start_table_from = tableFrom;
                    pThreadInfo->ntables = j<b?a+1:a;
                    pThreadInfo->end_table_to = j<b?tableFrom+a:tableFrom+a-1;
                    tableFrom = pThreadInfo->end_table_to + 1;
11684
                    pThreadInfo->taos = NULL; // workaround to use separate taos connection;
11685 11686 11687 11688
                    pthread_create(pidsOfStable + seq,
                            NULL, superSubscribe, pThreadInfo);
                }
            }
11689

11690 11691 11692 11693 11694 11695 11696
            g_queryInfo.superQueryInfo.threadCnt = threads;

            for (int i = 0; i < g_queryInfo.superQueryInfo.sqlCount; i++) {
                for (int j = 0; j < threads; j++) {
                    uint64_t seq = i * threads + j;
                    pthread_join(pidsOfStable[seq], NULL);
                }
11697 11698
            }
        }
11699
    }
11700

11701 11702 11703 11704 11705
    for (int i = 0; i < g_queryInfo.specifiedQueryInfo.sqlCount; i++) {
        for (int j = 0; j < g_queryInfo.specifiedQueryInfo.concurrent; j++) {
            uint64_t seq = i * g_queryInfo.specifiedQueryInfo.concurrent + j;
            pthread_join(pids[seq], NULL);
        }
11706
    }
H
hzcheng 已提交
11707

11708 11709
    tmfree((char*)pids);
    tmfree((char*)infos);
H
hzcheng 已提交
11710

11711 11712 11713 11714
    tmfree((char*)pidsOfStable);
    tmfree((char*)infosOfStable);
    //   taos_close(taos);
    return 0;
H
hzcheng 已提交
11715 11716
}

11717
static void initOfInsertMeta() {
11718
    memset(&g_Dbs, 0, sizeof(SDbs));
11719

11720 11721 11722 11723
    // set default values
    tstrncpy(g_Dbs.host, "127.0.0.1", MAX_HOSTNAME_SIZE);
    g_Dbs.port = 6030;
    tstrncpy(g_Dbs.user, TSDB_DEFAULT_USER, MAX_USERNAME_SIZE);
11724
    tstrncpy(g_Dbs.password, TSDB_DEFAULT_PASS, SHELL_MAX_PASSWORD_LEN);
11725
    g_Dbs.threadCount = 2;
11726

11727
    g_Dbs.use_metric = g_args.use_metric;
L
Liu Tao 已提交
11728 11729
}

11730
static void initOfQueryMeta() {
11731
    memset(&g_queryInfo, 0, sizeof(SQueryMetaInfo));
11732

11733 11734 11735 11736
    // set default values
    tstrncpy(g_queryInfo.host, "127.0.0.1", MAX_HOSTNAME_SIZE);
    g_queryInfo.port = 6030;
    tstrncpy(g_queryInfo.user, TSDB_DEFAULT_USER, MAX_USERNAME_SIZE);
11737
    tstrncpy(g_queryInfo.password, TSDB_DEFAULT_PASS, SHELL_MAX_PASSWORD_LEN);
L
Liu Tao 已提交
11738 11739
}

11740
static void setParaFromArg() {
11741 11742 11743 11744 11745
    if (g_args.host) {
        tstrncpy(g_Dbs.host, g_args.host, MAX_HOSTNAME_SIZE);
    } else {
        tstrncpy(g_Dbs.host, "127.0.0.1", MAX_HOSTNAME_SIZE);
    }
11746

11747 11748 11749
    if (g_args.user) {
        tstrncpy(g_Dbs.user, g_args.user, MAX_USERNAME_SIZE);
    }
11750

11751
    tstrncpy(g_Dbs.password, g_args.password, SHELL_MAX_PASSWORD_LEN);
L
Liu Tao 已提交
11752

11753 11754 11755
    if (g_args.port) {
        g_Dbs.port = g_args.port;
    }
11756

11757 11758
    g_Dbs.threadCount = g_args.nthreads;
    g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11759

11760 11761
    g_Dbs.dbCount = 1;
    g_Dbs.db[0].drop = true;
L
Liu Tao 已提交
11762

11763 11764
    tstrncpy(g_Dbs.db[0].dbName, g_args.database, TSDB_DB_NAME_LEN);
    g_Dbs.db[0].dbCfg.replica = g_args.replica;
11765
    tstrncpy(g_Dbs.db[0].dbCfg.precision, "ms", SMALL_BUFF_LEN);
11766

11767
    tstrncpy(g_Dbs.resultFile, g_args.output_file, MAX_FILE_NAME_LEN);
11768

11769
    g_Dbs.use_metric = g_args.use_metric;
L
Liu Tao 已提交
11770

11771
    g_Dbs.aggr_func = g_args.aggr_func;
11772

11773
    char dataString[TSDB_MAX_BYTES_PER_ROW];
11774 11775
    char *data_type = g_args.data_type;
    char **dataType = g_args.dataType;
L
Liu Tao 已提交
11776

11777
    memset(dataString, 0, TSDB_MAX_BYTES_PER_ROW);
H
hzcheng 已提交
11778

11779 11780 11781
    if ((data_type[0] == TSDB_DATA_TYPE_BINARY)
            || (data_type[0] == TSDB_DATA_TYPE_BOOL)
            || (data_type[0] == TSDB_DATA_TYPE_NCHAR)) {
11782
        g_Dbs.aggr_func = false;
11783
    }
11784

11785 11786
    if (g_args.use_metric) {
        g_Dbs.db[0].superTblCount = 1;
11787 11788 11789 11790
        tstrncpy(g_Dbs.db[0].superTbls[0].stbName, "meters", TSDB_TABLE_NAME_LEN);
        g_Dbs.db[0].superTbls[0].childTblCount = g_args.ntables;
        g_Dbs.threadCount = g_args.nthreads;
        g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11791
        g_Dbs.asyncMode = g_args.async_mode;
11792

11793 11794 11795 11796 11797
        g_Dbs.db[0].superTbls[0].autoCreateTable = PRE_CREATE_SUBTBL;
        g_Dbs.db[0].superTbls[0].childTblExists = TBL_NO_EXISTS;
        g_Dbs.db[0].superTbls[0].disorderRange = g_args.disorderRange;
        g_Dbs.db[0].superTbls[0].disorderRatio = g_args.disorderRatio;
        tstrncpy(g_Dbs.db[0].superTbls[0].childTblPrefix,
11798 11799
                g_args.tb_prefix, TBNAME_PREFIX_LEN);
        tstrncpy(g_Dbs.db[0].superTbls[0].dataSource, "rand", SMALL_BUFF_LEN);
11800

11801 11802 11803 11804 11805 11806 11807
        if (g_args.iface == INTERFACE_BUT) {
            g_Dbs.db[0].superTbls[0].iface = TAOSC_IFACE;
        } else {
            g_Dbs.db[0].superTbls[0].iface = g_args.iface;
        }
        tstrncpy(g_Dbs.db[0].superTbls[0].startTimestamp,
                "2017-07-14 10:40:00.000", MAX_TB_NAME_SIZE);
11808
        g_Dbs.db[0].superTbls[0].timeStampStep = g_args.timestamp_step;
11809

11810
        g_Dbs.db[0].superTbls[0].insertRows = g_args.insertRows;
11811
        g_Dbs.db[0].superTbls[0].maxSqlLen = g_args.max_sql_len;
11812

11813 11814
        g_Dbs.db[0].superTbls[0].columnCount = 0;
        for (int i = 0; i < MAX_NUM_COLUMNS; i++) {
11815
            if (data_type[i] == TSDB_DATA_TYPE_NULL) {
11816 11817
                break;
            }
11818

11819
            g_Dbs.db[0].superTbls[0].columns[i].data_type = data_type[i];
11820
            tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
11821
                    dataType[i], min(DATATYPE_BUFF_LEN, strlen(dataType[i]) + 1));
11822
            g_Dbs.db[0].superTbls[0].columns[i].dataLen = g_args.binwidth;
11823 11824 11825
            g_Dbs.db[0].superTbls[0].columnCount++;
        }

11826 11827
        if (g_Dbs.db[0].superTbls[0].columnCount > g_args.columnCount) {
            g_Dbs.db[0].superTbls[0].columnCount = g_args.columnCount;
11828 11829
        } else {
            for (int i = g_Dbs.db[0].superTbls[0].columnCount;
11830 11831
                    i < g_args.columnCount; i++) {
                g_Dbs.db[0].superTbls[0].columns[i].data_type = TSDB_DATA_TYPE_INT;
11832
                tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
11833
                        "INT", min(DATATYPE_BUFF_LEN, strlen("INT") + 1));
11834 11835 11836 11837 11838 11839
                g_Dbs.db[0].superTbls[0].columns[i].dataLen = 0;
                g_Dbs.db[0].superTbls[0].columnCount++;
            }
        }

        tstrncpy(g_Dbs.db[0].superTbls[0].tags[0].dataType,
11840
                "INT", min(DATATYPE_BUFF_LEN, strlen("INT") + 1));
11841 11842 11843
        g_Dbs.db[0].superTbls[0].tags[0].dataLen = 0;

        tstrncpy(g_Dbs.db[0].superTbls[0].tags[1].dataType,
11844
                "BINARY", min(DATATYPE_BUFF_LEN, strlen("BINARY") + 1));
11845
        g_Dbs.db[0].superTbls[0].tags[1].dataLen = g_args.binwidth;
11846 11847
        g_Dbs.db[0].superTbls[0].tagCount = 2;
    } else {
11848
        g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11849 11850
        g_Dbs.db[0].superTbls[0].tagCount = 0;
    }
11851 11852 11853 11854
}

/* Function to do regular expression check */
static int regexMatch(const char *s, const char *reg, int cflags) {
11855 11856 11857 11858 11859
    regex_t regex;
    char    msgbuf[100] = {0};

    /* Compile regular expression */
    if (regcomp(&regex, reg, cflags) != 0) {
11860
        ERROR_EXIT("Fail to compile regex\n");
11861 11862 11863 11864 11865 11866 11867 11868 11869 11870 11871 11872 11873
    }

    /* Execute regular expression */
    int reti = regexec(&regex, s, 0, NULL, 0);
    if (!reti) {
        regfree(&regex);
        return 1;
    } else if (reti == REG_NOMATCH) {
        regfree(&regex);
        return 0;
    } else {
        regerror(reti, &regex, msgbuf, sizeof(msgbuf));
        regfree(&regex);
11874 11875
        printf("Regex match failed: %s\n", msgbuf);
        exit(EXIT_FAILURE);
11876 11877
    }

11878 11879 11880 11881
    return 0;
}

static int isCommentLine(char *line) {
11882
    if (line == NULL) return 1;
11883

11884
    return regexMatch(line, "^\\s*#.*", REG_EXTENDED);
11885 11886
}

11887
static void querySqlFile(TAOS* taos, char* sqlFile)
11888
{
11889 11890 11891 11892 11893
    FILE *fp = fopen(sqlFile, "r");
    if (fp == NULL) {
        printf("failed to open file %s, reason:%s\n", sqlFile, strerror(errno));
        return;
    }
11894

11895 11896 11897 11898 11899
    int       read_len = 0;
    char *    cmd = calloc(1, TSDB_MAX_BYTES_PER_ROW);
    size_t    cmd_len = 0;
    char *    line = NULL;
    size_t    line_len = 0;
11900

11901
    double t = taosGetTimestampMs();
11902

11903 11904 11905
    while((read_len = tgetline(&line, &line_len, fp)) != -1) {
        if (read_len >= TSDB_MAX_BYTES_PER_ROW) continue;
        line[--read_len] = '\0';
11906

11907 11908 11909
        if (read_len == 0 || isCommentLine(line)) {  // line starts with #
            continue;
        }
11910

11911 11912 11913 11914 11915 11916
        if (line[read_len - 1] == '\\') {
            line[read_len - 1] = ' ';
            memcpy(cmd + cmd_len, line, read_len);
            cmd_len += read_len;
            continue;
        }
11917

11918 11919
        memcpy(cmd + cmd_len, line, read_len);
        if (0 != queryDbExec(taos, cmd, NO_INSERT_TYPE, false)) {
11920
            errorPrint2("%s() LN%d, queryDbExec %s failed!\n",
11921 11922 11923 11924 11925 11926 11927 11928
                    __func__, __LINE__, cmd);
            tmfree(cmd);
            tmfree(line);
            tmfclose(fp);
            return;
        }
        memset(cmd, 0, TSDB_MAX_BYTES_PER_ROW);
        cmd_len = 0;
11929
    }
H
hzcheng 已提交
11930

11931 11932
    t = taosGetTimestampMs() - t;
    printf("run %s took %.6f second(s)\n\n", sqlFile, t);
11933

11934 11935 11936 11937
    tmfree(cmd);
    tmfree(line);
    tmfclose(fp);
    return;
H
hzcheng 已提交
11938 11939
}

11940
static void testMetaFile() {
11941
    if (INSERT_TEST == g_args.test_mode) {
11942 11943
        if (g_Dbs.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_Dbs.cfgDir);
11944

11945
        insertTestProcess();
11946

11947
    } else if (QUERY_TEST == g_args.test_mode) {
11948 11949
        if (g_queryInfo.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_queryInfo.cfgDir);
11950

11951
        queryTestProcess();
11952

11953
    } else if (SUBSCRIBE_TEST == g_args.test_mode) {
11954 11955
        if (g_queryInfo.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_queryInfo.cfgDir);
11956

11957
        subscribeTestProcess();
11958

11959
    }  else {
11960
        ;
11961
    }
11962
}
11963

11964
static void queryAggrFunc() {
11965 11966 11967 11968 11969
    // query data

    pthread_t read_id;
    threadInfo *pThreadInfo = calloc(1, sizeof(threadInfo));
    assert(pThreadInfo);
11970
    pThreadInfo->start_time = DEFAULT_START_TIME;  // 2017-07-14 10:40:00.000
11971 11972 11973 11974 11975
    pThreadInfo->start_table_from = 0;

    if (g_args.use_metric) {
        pThreadInfo->ntables = g_Dbs.db[0].superTbls[0].childTblCount;
        pThreadInfo->end_table_to = g_Dbs.db[0].superTbls[0].childTblCount - 1;
11976
        pThreadInfo->stbInfo = &g_Dbs.db[0].superTbls[0];
11977
        tstrncpy(pThreadInfo->tb_prefix,
11978
                g_Dbs.db[0].superTbls[0].childTblPrefix, TBNAME_PREFIX_LEN);
11979
    } else {
11980 11981
        pThreadInfo->ntables = g_args.ntables;
        pThreadInfo->end_table_to = g_args.ntables -1;
11982 11983 11984 11985 11986 11987 11988 11989 11990 11991
        tstrncpy(pThreadInfo->tb_prefix, g_args.tb_prefix, TSDB_TABLE_NAME_LEN);
    }

    pThreadInfo->taos = taos_connect(
            g_Dbs.host,
            g_Dbs.user,
            g_Dbs.password,
            g_Dbs.db[0].dbName,
            g_Dbs.port);
    if (pThreadInfo->taos == NULL) {
11992
        free(pThreadInfo);
11993
        errorPrint2("Failed to connect to TDengine, reason:%s\n",
11994
                taos_errstr(NULL));
11995
        exit(EXIT_FAILURE);
11996 11997 11998 11999 12000
    }

    tstrncpy(pThreadInfo->filePath, g_Dbs.resultFile, MAX_FILE_NAME_LEN);

    if (!g_Dbs.use_metric) {
12001
        pthread_create(&read_id, NULL, queryNtableAggrFunc, pThreadInfo);
12002
    } else {
12003
        pthread_create(&read_id, NULL, queryStableAggrFunc, pThreadInfo);
12004 12005 12006
    }
    pthread_join(read_id, NULL);
    taos_close(pThreadInfo->taos);
12007
    free(pThreadInfo);
12008 12009
}

12010 12011
static void testCmdLine() {

12012 12013 12014
    if (strlen(configDir)) {
        wordexp_t full_path;
        if (wordexp(configDir, &full_path, 0) != 0) {
12015
            errorPrint("Invalid path %s\n", configDir);
12016 12017 12018 12019
            return;
        }
        taos_options(TSDB_OPTION_CONFIGDIR, full_path.we_wordv[0]);
        wordfree(&full_path);
12020 12021
    }

12022 12023
    g_args.test_mode = INSERT_TEST;
    insertTestProcess();
12024

12025 12026 12027
    if (g_Dbs.aggr_func) {
        queryAggrFunc();
    }
12028 12029
}

12030
int main(int argc, char *argv[]) {
12031
    parse_args(argc, argv, &g_args);
12032

12033
    debugPrint("meta file: %s\n", g_args.metaFile);
12034

12035
    if (g_args.metaFile) {
12036
        g_totalChildTables = 0;
12037 12038
        initOfInsertMeta();
        initOfQueryMeta();
12039

12040 12041 12042 12043
        if (false == getInfoFromJsonFile(g_args.metaFile)) {
            printf("Failed to read %s\n", g_args.metaFile);
            return 1;
        }
12044

12045
        testMetaFile();
12046
    } else {
12047 12048 12049 12050 12051 12052 12053 12054 12055 12056 12057 12058
        memset(&g_Dbs, 0, sizeof(SDbs));
        setParaFromArg();

        if (NULL != g_args.sqlFile) {
            TAOS* qtaos = taos_connect(
                    g_Dbs.host,
                    g_Dbs.user,
                    g_Dbs.password,
                    g_Dbs.db[0].dbName,
                    g_Dbs.port);
            querySqlFile(qtaos, g_args.sqlFile);
            taos_close(qtaos);
12059

12060 12061 12062
        } else {
            testCmdLine();
        }
12063

12064 12065 12066 12067 12068
        if (g_dupstr)
            free(g_dupstr);
    }

    return 0;
H
hzcheng 已提交
12069
}