taosdemo.c 444.7 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
        } else {
            taos_close(taos);
4276
            free(command);
4277
            errorPrint2("%s() LN%d, config error tag type : %s\n",
4278
                    __func__, __LINE__, dataType);
4279
            exit(EXIT_FAILURE);
4280
        }
4281
    }
4282

4283
    len -= 1;
4284
    len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len, ")");
4285

4286
    superTbl->lenOfTagOfOneRow = lenOfTagOfOneRow;
4287

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

4298
    debugPrint("create supertable %s success!\n\n", superTbl->stbName);
4299
    free(command);
4300
    return 0;
4301 4302
}

4303
int createDatabasesAndStables(char *command) {
4304 4305 4306 4307
    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) {
4308
        errorPrint2("Failed to connect to TDengine, reason:%s\n", taos_errstr(NULL));
4309
        return -1;
4310
    }
4311

4312 4313 4314 4315 4316 4317 4318
    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;
            }
4319

4320 4321
            int dataLen = 0;
            dataLen += snprintf(command + dataLen,
4322
                    BUFFER_SIZE - dataLen, "CREATE DATABASE IF NOT EXISTS %s",
4323
                    g_Dbs.db[i].dbName);
4324

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

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

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

4414
        int validStbCount = 0;
4415

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

4421 4422 4423
            if ((ret != 0) || (g_Dbs.db[i].drop)) {
                ret = createSuperTable(taos, g_Dbs.db[i].dbName,
                        &g_Dbs.db[i].superTbls[j]);
4424

4425 4426 4427 4428
                if (0 != ret) {
                    errorPrint("create super table %"PRIu64" failed!\n\n", j);
                    continue;
                }
4429 4430
            } else {
                ret = getSuperTableFromServer(taos, g_Dbs.db[i].dbName,
4431
                    &g_Dbs.db[i].superTbls[j]);
4432 4433 4434 4435 4436
                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;
                }
4437 4438 4439 4440 4441 4442 4443 4444
            }
            validStbCount ++;
        }
        g_Dbs.db[i].superTblCount = validStbCount;
    }

    taos_close(taos);
    return 0;
4445 4446
}

4447 4448
static void* createTable(void *sarg)
{
4449
    threadInfo *pThreadInfo = (threadInfo *)sarg;
4450
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
4451

4452 4453
    setThreadName("createTable");

4454
    uint64_t  lastPrintTime = taosGetTimestampMs();
4455

4456
    int buff_len = BUFFER_SIZE;
4457

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

    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,
4475
                    "CREATE TABLE IF NOT EXISTS %s.%s%"PRIu64" %s;",
4476 4477 4478
                    pThreadInfo->db_name,
                    g_args.tb_prefix, i,
                    pThreadInfo->cols);
4479
            batchNum ++;
4480
        } else {
4481 4482
            if (stbInfo == NULL) {
                free(pThreadInfo->buffer);
4483
                errorPrint2("%s() LN%d, use metric, but super table info is NULL\n",
4484
                        __func__, __LINE__);
4485
                exit(EXIT_FAILURE);
4486 4487 4488 4489 4490
            } else {
                if (0 == len) {
                    batchNum = 0;
                    memset(pThreadInfo->buffer, 0, buff_len);
                    len += snprintf(pThreadInfo->buffer + len,
4491
                            buff_len - len, "CREATE TABLE ");
4492
                }
4493

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

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

4527
        len = 0;
S
Shengliang Guan 已提交
4528

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

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

4545 4546 4547
    if (0 != len) {
        if (0 != queryDbExec(pThreadInfo->taos, pThreadInfo->buffer,
                    NO_INSERT_TYPE, false)) {
4548
            errorPrint2("queryDbExec() failed. buffer:\n%s\n", pThreadInfo->buffer);
4549
        }
4550
    }
4551

4552 4553
    free(pThreadInfo->buffer);
    return NULL;
4554 4555
}

4556
static int startMultiThreadCreateChildTable(
4557
        char* cols, int threads, uint64_t tableFrom, int64_t ntables,
4558
        char* db_name, SSuperTable* stbInfo) {
4559

4560 4561
    pthread_t *pids = calloc(1, threads * sizeof(pthread_t));
    threadInfo *infos = calloc(1, threads * sizeof(threadInfo));
4562

4563
    if ((NULL == pids) || (NULL == infos)) {
4564
        ERROR_EXIT("createChildTable malloc failed\n");
4565
    }
4566

4567 4568 4569
    if (threads < 1) {
        threads = 1;
    }
4570

4571 4572 4573 4574
    int64_t a = ntables / threads;
    if (a < 1) {
        threads = ntables;
        a = 1;
4575
    }
4576

4577 4578
    int64_t b = 0;
    b = ntables % threads;
4579

4580 4581 4582 4583
    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);
4584
        pThreadInfo->stbInfo = stbInfo;
4585 4586 4587 4588 4589 4590 4591 4592
        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) {
4593
            errorPrint2("%s() LN%d, Failed to connect to TDengine, reason:%s\n",
4594 4595 4596 4597 4598
                    __func__, __LINE__, taos_errstr(NULL));
            free(pids);
            free(infos);
            return -1;
        }
4599

4600 4601 4602 4603 4604 4605 4606
        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;
4607
        pThreadInfo->tables_created = 0;
4608 4609
        pthread_create(pids + i, NULL, createTable, pThreadInfo);
    }
4610

4611 4612 4613
    for (int i = 0; i < threads; i++) {
        pthread_join(pids[i], NULL);
    }
4614

4615 4616 4617
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
        taos_close(pThreadInfo->taos);
4618 4619

        g_actualChildTables += pThreadInfo->tables_created;
4620 4621 4622 4623 4624 4625
    }

    free(pids);
    free(infos);

    return 0;
4626 4627
}

4628
static void createChildTables() {
4629
    char tblColsBuf[TSDB_MAX_BYTES_PER_ROW];
4630
    int len;
4631

4632 4633 4634 4635 4636
    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++) {
4637 4638 4639 4640
                    if ((AUTO_CREATE_SUBTBL
                                == g_Dbs.db[i].superTbls[j].autoCreateTable)
                            || (TBL_ALREADY_EXISTS
                                == g_Dbs.db[i].superTbls[j].childTblExists)) {
4641 4642 4643
                        continue;
                    }
                    verbosePrint("%s() LN%d: %s\n", __func__, __LINE__,
4644
                            g_Dbs.db[i].superTbls[j].colsOfCreateChildTable);
4645 4646 4647 4648 4649 4650 4651
                    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,
4652
                            g_Dbs.threadCountForCreateTbl,
4653 4654 4655 4656 4657 4658 4659
                            startFrom,
                            g_Dbs.db[i].superTbls[j].childTblCount,
                            g_Dbs.db[i].dbName, &(g_Dbs.db[i].superTbls[j]));
                }
            }
        } else {
            // normal table
4660
            len = snprintf(tblColsBuf, TSDB_MAX_BYTES_PER_ROW, "(TS TIMESTAMP");
4661 4662 4663
            for (int j = 0; j < g_args.columnCount; j++) {
                if ((strncasecmp(g_args.dataType[j], "BINARY", strlen("BINARY")) == 0)
                        || (strncasecmp(g_args.dataType[j],
4664
                                "NCHAR", strlen("NCHAR")) == 0)) {
4665
                    snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len,
4666
                            ",C%d %s(%d)", j, g_args.dataType[j], g_args.binwidth);
4667
                } else {
4668
                    snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len,
4669
                            ",C%d %s", j, g_args.dataType[j]);
4670 4671 4672
                }
                len = strlen(tblColsBuf);
            }
4673

4674
            snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len, ")");
4675 4676 4677

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

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

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

4705 4706 4707
    if (stbInfo->tagDataBuf) {
        free(stbInfo->tagDataBuf);
        stbInfo->tagDataBuf = NULL;
4708 4709
    }

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

4719 4720 4721 4722 4723 4724 4725 4726 4727
    while((readLen = tgetline(&line, &n, fp)) != -1) {
        if (('\r' == line[readLen - 1]) || ('\n' == line[readLen - 1])) {
            line[--readLen] = 0;
        }

        if (readLen == 0) {
            continue;
        }

4728
        memcpy(tagDataBuf + count * stbInfo->lenOfTagOfOneRow, line, readLen);
4729 4730 4731 4732
        count++;

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

4750 4751
    stbInfo->tagDataBuf = tagDataBuf;
    stbInfo->tagSampleCount = count;
4752 4753 4754 4755

    free(line);
    fclose(fp);
    return 0;
4756 4757 4758
}

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

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

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

4790 4791 4792
        if (('\r' == line[readLen - 1]) || ('\n' == line[readLen - 1])) {
            line[--readLen] = 0;
        }
4793

4794 4795 4796
        if (readLen == 0) {
            continue;
        }
4797

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

4804
        memcpy(stbInfo->sampleDataBuf + getRows * stbInfo->lenOfOneRow,
4805 4806 4807
                line, readLen);
        getRows++;

4808
        if (getRows == MAX_SAMPLES) {
4809 4810
            break;
        }
4811 4812
    }

4813 4814 4815
    fclose(fp);
    tmfree(line);
    return 0;
4816 4817
}

4818
static bool getColumnAndTagTypeFromInsertJsonFile(
4819
        cJSON* stbInfo, SSuperTable* superTbls) {
4820 4821 4822 4823 4824
    bool  ret = false;

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

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

4840 4841 4842
    int count = 1;
    int index = 0;
    StrColumn    columnCase;
4843

4844 4845 4846 4847
    //superTbls->columnCount = columnSize;
    for (int k = 0; k < columnSize; ++k) {
        cJSON* column = cJSON_GetArrayItem(columns, k);
        if (column == NULL) continue;
4848

4849 4850 4851 4852 4853
        count = 1;
        cJSON* countObj = cJSON_GetObjectItem(column, "count");
        if (countObj && countObj->type == cJSON_Number) {
            count = countObj->valueint;
        } else if (countObj && countObj->type != cJSON_Number) {
4854
            errorPrint("%s", "failed to read json, column count not found\n");
4855 4856 4857 4858 4859 4860 4861 4862 4863 4864
            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) {
4865
            errorPrint("%s", "failed to read json, column type not found\n");
4866 4867
            goto PARSE_OVER;
        }
4868 4869 4870
        //tstrncpy(superTbls->columns[k].dataType, dataType->valuestring, DATATYPE_BUFF_LEN);
        tstrncpy(columnCase.dataType, dataType->valuestring,
                min(DATATYPE_BUFF_LEN, strlen(dataType->valuestring) + 1));
4871 4872 4873 4874 4875 4876 4877 4878 4879

        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 {
4880
            columnCase.dataLen = SMALL_BUFF_LEN;
4881 4882 4883 4884
        }

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

4888 4889 4890 4891 4892 4893
            superTbls->columns[index].dataLen = columnCase.dataLen;
            index++;
        }
    }

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

    superTbls->columnCount = index;
4900

4901 4902 4903 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
    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;
4932 4933 4934 4935 4936 4937 4938 4939 4940 4941 4942 4943
        } 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;
4944 4945 4946 4947 4948
        } else {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_NULL;
        }
    }

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

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

4965 4966 4967 4968 4969 4970 4971 4972 4973 4974
    //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) {
4975
            errorPrint("%s", "failed to read json, column count not found\n");
4976 4977 4978 4979 4980 4981 4982 4983 4984 4985
            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) {
4986
            errorPrint("%s", "failed to read json, tag type not found\n");
4987 4988
            goto PARSE_OVER;
        }
4989 4990
        tstrncpy(columnCase.dataType, dataType->valuestring,
                min(DATATYPE_BUFF_LEN, strlen(dataType->valuestring) + 1));
4991 4992 4993 4994 4995

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

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

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

5016
    superTbls->tagCount = index;
5017

5018 5019 5020 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
    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;
5049 5050 5051 5052 5053 5054 5055 5056 5057 5058 5059 5060
        } 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;
5061 5062 5063 5064 5065
        } else {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_NULL;
        }
    }

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

5073
PARSE_OVER:
5074
    return ret;
5075 5076 5077
}

static bool getMetaFromInsertJsonFile(cJSON* root) {
5078
    bool  ret = false;
5079

5080 5081 5082
    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);
5083
    }
5084

5085 5086 5087 5088 5089
    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);
5090
    } else {
5091
        errorPrint("%s", "failed to read json, host not found\n");
5092
        goto PARSE_OVER;
5093 5094
    }

5095 5096 5097 5098 5099
    cJSON* port = cJSON_GetObjectItem(root, "port");
    if (port && port->type == cJSON_Number) {
        g_Dbs.port = port->valueint;
    } else if (!port) {
        g_Dbs.port = 6030;
5100
    }
5101

5102 5103 5104 5105 5106
    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);
5107 5108
    }

5109 5110
    cJSON* password = cJSON_GetObjectItem(root, "password");
    if (password && password->type == cJSON_String && password->valuestring != NULL) {
5111
        tstrncpy(g_Dbs.password, password->valuestring, SHELL_MAX_PASSWORD_LEN);
5112
    } else if (!password) {
5113
        tstrncpy(g_Dbs.password, "taosdata", SHELL_MAX_PASSWORD_LEN);
5114 5115
    }

5116 5117 5118 5119 5120
    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);
5121 5122
    }

5123 5124 5125 5126 5127
    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;
5128
    } else {
5129
        errorPrint("%s", "failed to read json, threads not found\n");
5130
        goto PARSE_OVER;
5131
    }
5132

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

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

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

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

5172 5173 5174 5175 5176 5177 5178 5179 5180 5181
    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);
5182
    } else {
5183 5184 5185
        errorPrint("%s() LN%d, failed to read json, max_sql_len input mistake\n",
                __func__, __LINE__);
        goto PARSE_OVER;
5186 5187
    }

5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198 5199 5200 5201
    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;
        }
5202
        g_args.reqPerReq = numRecPerReq->valueint;
5203
    } else if (!numRecPerReq) {
5204
        g_args.reqPerReq = MAX_RECORDS_PER_REQ;
5205
    } else {
5206 5207 5208
        errorPrint("%s() LN%d, failed to read json, num_of_records_per_req not found\n",
                __func__, __LINE__);
        goto PARSE_OVER;
5209 5210
    }

5211 5212 5213 5214 5215 5216 5217 5218 5219 5220 5221 5222 5223
    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.
5224
    } else {
5225 5226
        errorPrint("%s", "failed to read json, confirm_parameter_prompt input mistake\n");
        goto PARSE_OVER;
5227 5228
    }

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

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

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

5253 5254 5255 5256
    g_Dbs.dbCount = dbSize;
    for (int i = 0; i < dbSize; ++i) {
        cJSON* dbinfos = cJSON_GetArrayItem(dbs, i);
        if (dbinfos == NULL) continue;
5257

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

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

5272 5273 5274 5275 5276 5277 5278 5279 5280 5281
        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 {
5282
            errorPrint("%s", "failed to read json, drop input mistake\n");
5283 5284
            goto PARSE_OVER;
        }
5285

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

5298 5299 5300 5301 5302
        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;
5303
        } else {
5304
            errorPrint("%s", "failed to read json, update not found\n");
5305
            goto PARSE_OVER;
5306
        }
5307

5308 5309 5310 5311 5312 5313
        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 {
5314
            errorPrint("%s", "failed to read json, replica not found\n");
5315 5316
            goto PARSE_OVER;
        }
5317

5318 5319 5320 5321 5322 5323
        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 {
5324
            errorPrint("%s", "failed to read json, keep not found\n");
5325 5326
            goto PARSE_OVER;
        }
5327

5328 5329 5330 5331 5332
        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;
5333
        } else {
5334
            errorPrint("%s", "failed to read json, days not found\n");
5335 5336
            goto PARSE_OVER;
        }
5337

5338 5339 5340 5341 5342 5343
        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 {
5344
            errorPrint("%s", "failed to read json, cache not found\n");
5345
            goto PARSE_OVER;
5346 5347
        }

5348 5349 5350 5351 5352 5353
        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 {
5354
            errorPrint("%s", "failed to read json, block not found\n");
5355
            goto PARSE_OVER;
5356 5357
        }

5358 5359 5360 5361 5362 5363 5364 5365 5366 5367 5368 5369 5370 5371 5372 5373
        //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 {
5374
            errorPrint("%s", "failed to read json, minRows not found\n");
5375 5376
            goto PARSE_OVER;
        }
5377

5378 5379 5380 5381 5382 5383
        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 {
5384
            errorPrint("%s", "failed to read json, maxRows not found\n");
5385 5386
            goto PARSE_OVER;
        }
5387

5388 5389 5390 5391 5392 5393
        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 {
5394
            errorPrint("%s", "failed to read json, comp not found\n");
5395 5396
            goto PARSE_OVER;
        }
5397

5398 5399 5400 5401 5402 5403
        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 {
5404
            errorPrint("%s", "failed to read json, walLevel not found\n");
5405 5406
            goto PARSE_OVER;
        }
5407

5408 5409 5410 5411 5412
        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;
5413
        } else {
5414
            errorPrint("%s", "failed to read json, cacheLast not found\n");
5415
            goto PARSE_OVER;
5416
        }
5417

5418 5419 5420 5421 5422
        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;
5423
        } else {
5424 5425
            printf("failed to read json, quorum input mistake");
            goto PARSE_OVER;
5426
        }
5427 5428 5429 5430 5431 5432 5433

        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 {
5434
            errorPrint("%s", "failed to read json, fsync input mistake\n");
5435
            goto PARSE_OVER;
5436 5437
        }

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

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

5453 5454 5455 5456
        g_Dbs.db[i].superTblCount = stbSize;
        for (int j = 0; j < stbSize; ++j) {
            cJSON* stbInfo = cJSON_GetArrayItem(stables, j);
            if (stbInfo == NULL) continue;
5457

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

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

            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 {
5491
                errorPrint("%s", "failed to read json, auto_create_table not found\n");
5492 5493
                goto PARSE_OVER;
            }
5494

5495 5496 5497 5498
            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) {
5499
                g_Dbs.db[i].superTbls[j].batchCreateTableNum = 10;
5500
            } else {
5501
                errorPrint("%s", "failed to read json, batch_create_tbl_num not found\n");
5502 5503 5504 5505 5506 5507 5508 5509 5510 5511 5512 5513 5514 5515 5516 5517 5518 5519 5520
                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 {
5521 5522
                errorPrint("%s",
                        "failed to read json, child_table_exists not found\n");
5523 5524 5525 5526 5527 5528 5529 5530 5531
                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) {
5532 5533
                errorPrint("%s",
                        "failed to read json, childtable_count input mistake\n");
5534 5535 5536
                goto PARSE_OVER;
            }
            g_Dbs.db[i].superTbls[j].childTblCount = count->valueint;
5537
            g_totalChildTables += g_Dbs.db[i].superTbls[j].childTblCount;
5538 5539 5540 5541 5542

            cJSON *dataSource = cJSON_GetObjectItem(stbInfo, "data_source");
            if (dataSource && dataSource->type == cJSON_String
                    && dataSource->valuestring != NULL) {
                tstrncpy(g_Dbs.db[i].superTbls[j].dataSource,
5543 5544
                        dataSource->valuestring,
                        min(SMALL_BUFF_LEN, strlen(dataSource->valuestring) + 1));
5545
            } else if (!dataSource) {
5546 5547
                tstrncpy(g_Dbs.db[i].superTbls[j].dataSource, "rand",
                        min(SMALL_BUFF_LEN, strlen("rand") + 1));
5548
            } else {
5549
                errorPrint("%s", "failed to read json, data_source not found\n");
5550 5551 5552 5553 5554 5555 5556 5557 5558 5559 5560 5561 5562
                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 {
5563 5564
                    errorPrint("failed to read json, insert_mode %s not recognized\n",
                            stbIface->valuestring);
5565 5566 5567 5568 5569 5570 5571 5572 5573 5574 5575 5576 5577
                    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) {
5578
                    errorPrint("%s", "failed to read json, childtable_limit\n");
5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590
                    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)) {
5591
                    errorPrint("%s", "failed to read json, childtable_offset\n");
5592 5593 5594 5595 5596 5597 5598 5599 5600 5601 5602 5603 5604 5605 5606
                    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 {
5607
                errorPrint("%s", "failed to read json, start_timestamp not found\n");
5608 5609 5610 5611 5612 5613 5614
                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) {
5615
                g_Dbs.db[i].superTbls[j].timeStampStep = g_args.timestamp_step;
5616
            } else {
5617
                errorPrint("%s", "failed to read json, timestamp_step not found\n");
5618 5619 5620 5621 5622 5623 5624
                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,
5625 5626 5627
                        sampleFormat->valuestring,
                        min(SMALL_BUFF_LEN,
                            strlen(sampleFormat->valuestring) + 1));
5628
            } else if (!sampleFormat) {
5629 5630
                tstrncpy(g_Dbs.db[i].superTbls[j].sampleFormat, "csv",
                        SMALL_BUFF_LEN);
5631
            } else {
5632
                errorPrint("%s", "failed to read json, sample_format not found\n");
5633 5634 5635 5636 5637 5638 5639
                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,
5640 5641 5642
                        sampleFile->valuestring,
                        min(MAX_FILE_NAME_LEN,
                            strlen(sampleFile->valuestring) + 1));
5643
            } else if (!sampleFile) {
5644 5645
                memset(g_Dbs.db[i].superTbls[j].sampleFile, 0,
                        MAX_FILE_NAME_LEN);
5646
            } else {
5647
                errorPrint("%s", "failed to read json, sample_file not found\n");
5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663 5664
                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 {
5665
                errorPrint("%s", "failed to read json, tags_file not found\n");
5666 5667 5668 5669 5670 5671 5672 5673 5674 5675 5676 5677 5678 5679 5680
                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 {
5681
                errorPrint("%s", "failed to read json, stbMaxSqlLen input mistake\n");
5682 5683 5684 5685 5686 5687 5688 5689 5690 5691 5692 5693 5694 5695 5696 5697
                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 {
5698
               errorPrint("%s", "failed to read json, multiThreadWriteOneTbl not found\n");
5699 5700 5701 5702 5703 5704
               goto PARSE_OVER;
               }
               */
            cJSON* insertRows = cJSON_GetObjectItem(stbInfo, "insert_rows");
            if (insertRows && insertRows->type == cJSON_Number) {
                if (insertRows->valueint < 0) {
5705
                    errorPrint("%s", "failed to read json, insert_rows input mistake\n");
5706 5707 5708 5709 5710 5711
                    goto PARSE_OVER;
                }
                g_Dbs.db[i].superTbls[j].insertRows = insertRows->valueint;
            } else if (!insertRows) {
                g_Dbs.db[i].superTbls[j].insertRows = 0x7FFFFFFFFFFFFFFF;
            } else {
5712
                errorPrint("%s", "failed to read json, insert_rows input mistake\n");
5713 5714 5715 5716 5717 5718
                goto PARSE_OVER;
            }

            cJSON* stbInterlaceRows = cJSON_GetObjectItem(stbInfo, "interlace_rows");
            if (stbInterlaceRows && stbInterlaceRows->type == cJSON_Number) {
                if (stbInterlaceRows->valueint < 0) {
5719
                    errorPrint("%s", "failed to read json, interlace rows input mistake\n");
5720 5721 5722 5723 5724 5725 5726 5727 5728 5729 5730 5731 5732 5733
                    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) {
5734
                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
5735 5736
            } else {
                errorPrint(
5737
                        "%s", "failed to read json, interlace rows input mistake\n");
5738 5739 5740 5741 5742 5743 5744 5745 5746 5747 5748 5749 5750 5751 5752
                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 {
5753
                errorPrint("%s", "failed to read json, disorderRatio not found\n");
5754 5755 5756 5757 5758 5759 5760 5761 5762
                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 {
5763
                errorPrint("%s", "failed to read json, disorderRange not found\n");
5764 5765 5766 5767 5768 5769 5770
                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) {
5771
                    errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5772 5773 5774
                    goto PARSE_OVER;
                }
            } else if (!insertInterval) {
5775
                verbosePrint("%s() LN%d: stable insert interval be overrode by global %"PRIu64".\n",
5776 5777 5778
                        __func__, __LINE__, g_args.insert_interval);
                g_Dbs.db[i].superTbls[j].insertInterval = g_args.insert_interval;
            } else {
5779
                errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5780 5781
                goto PARSE_OVER;
            }
5782

5783 5784 5785 5786 5787
            int retVal = getColumnAndTagTypeFromInsertJsonFile(
                    stbInfo, &g_Dbs.db[i].superTbls[j]);
            if (false == retVal) {
                goto PARSE_OVER;
            }
5788
        }
5789
    }
5790

5791
    ret = true;
5792

5793
PARSE_OVER:
5794
    return ret;
5795 5796 5797
}

static bool getMetaFromQueryJsonFile(cJSON* root) {
5798
    bool  ret = false;
5799

5800 5801 5802 5803
    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 已提交
5804

5805 5806 5807 5808 5809
    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 已提交
5810
    } else {
5811
        errorPrint("%s", "failed to read json, host not found\n");
5812
        goto PARSE_OVER;
H
hzcheng 已提交
5813
    }
5814

5815 5816 5817 5818 5819
    cJSON* port = cJSON_GetObjectItem(root, "port");
    if (port && port->type == cJSON_Number) {
        g_queryInfo.port = port->valueint;
    } else if (!port) {
        g_queryInfo.port = 6030;
5820
    }
5821

5822 5823 5824 5825 5826
    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); ;
5827 5828
    }

5829 5830
    cJSON* password = cJSON_GetObjectItem(root, "password");
    if (password && password->type == cJSON_String && password->valuestring != NULL) {
5831
        tstrncpy(g_queryInfo.password, password->valuestring, SHELL_MAX_PASSWORD_LEN);
5832
    } else if (!password) {
5833
        tstrncpy(g_queryInfo.password, "taosdata", SHELL_MAX_PASSWORD_LEN);;
5834
    }
5835

5836 5837 5838 5839 5840 5841 5842 5843 5844 5845 5846 5847
    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;
5848
    } else {
5849
        errorPrint("%s", "failed to read json, confirm_parameter_prompt not found\n");
5850
        goto PARSE_OVER;
5851
    }
5852

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

5867 5868 5869 5870
    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) {
5871
        errorPrint("%s", "failed to read json, databases not found\n");
5872
        goto PARSE_OVER;
5873
    }
5874

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

5889 5890 5891 5892 5893 5894
    // 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) {
5895
        errorPrint("%s", "failed to read json, super_table_query not found\n");
5896 5897
        goto PARSE_OVER;
    } else {
5898 5899 5900 5901 5902 5903
        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;
        }
5904

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

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

5924 5925 5926 5927
        cJSON* concurrent = cJSON_GetObjectItem(specifiedQuery, "concurrent");
        if (concurrent && concurrent->type == cJSON_Number) {
            if (concurrent->valueint <= 0) {
                errorPrint(
5928
                        "query sqlCount %d or concurrent %d is not correct.\n",
5929 5930 5931 5932 5933 5934 5935 5936
                        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;
        }
5937

5938 5939 5940 5941 5942 5943 5944 5945
        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 {
5946
                errorPrint("%s", "failed to read json, async mode input error\n");
5947 5948 5949 5950 5951
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.asyncMode = SYNC_MODE;
        }
5952

5953 5954 5955 5956 5957 5958 5959 5960
        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;
        }
5961

5962 5963 5964 5965 5966 5967 5968
        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 {
5969
                errorPrint("%s", "failed to read json, subscribe restart error\n");
5970 5971 5972 5973 5974
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.subscribeRestart = true;
        }
5975

5976 5977 5978 5979 5980 5981 5982 5983 5984
        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 {
5985
                errorPrint("%s", "failed to read json, subscribe keepProgress error\n");
5986 5987 5988 5989 5990
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.subscribeKeepProgress = 0;
        }
5991

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

6010 6011 6012 6013 6014 6015 6016
            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) {
6017
                    errorPrint("%s", "failed to read json, sql not found\n");
6018 6019 6020
                    goto PARSE_OVER;
                }
                tstrncpy(g_queryInfo.specifiedQueryInfo.sql[j],
6021
                        sqlStr->valuestring, BUFFER_SIZE);
6022 6023 6024 6025 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

                // 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 {
6057 6058
                    errorPrint("%s",
                            "failed to read json, super query result file not found\n");
6059 6060 6061 6062
                    goto PARSE_OVER;
                }
            }
        }
6063
    }
6064

6065 6066 6067 6068 6069 6070
    // 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) {
6071
        errorPrint("%s", "failed to read json, sub_table_query not found\n");
6072
        ret = true;
6073 6074
        goto PARSE_OVER;
    } else {
6075 6076 6077 6078 6079 6080 6081 6082 6083 6084
        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) {
6085 6086
                errorPrint("failed to read json, query_times: %"PRId64", need be a valid (>0) number\n",
                        superQueryTimes->valueint);
6087 6088 6089 6090 6091 6092
                goto PARSE_OVER;
            }
            g_queryInfo.superQueryInfo.queryTimes = superQueryTimes->valueint;
        } else if (!superQueryTimes) {
            g_queryInfo.superQueryInfo.queryTimes = g_args.query_times;
        } else {
6093
            errorPrint("%s", "failed to read json, query_times input mistake\n");
6094 6095
            goto PARSE_OVER;
        }
6096

6097 6098 6099
        cJSON* threads = cJSON_GetObjectItem(superQuery, "threads");
        if (threads && threads->type == cJSON_Number) {
            if (threads->valueint <= 0) {
6100
                errorPrint("%s", "failed to read json, threads input mistake\n");
6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117 6118
                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) {
6119
            tstrncpy(g_queryInfo.superQueryInfo.stbName, stblname->valuestring,
6120 6121
                    TSDB_TABLE_NAME_LEN);
        } else {
6122
            errorPrint("%s", "failed to read json, super table name input error\n");
6123 6124 6125 6126 6127 6128 6129 6130 6131 6132 6133
            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 {
6134
                errorPrint("%s", "failed to read json, async mode input error\n");
6135 6136 6137 6138 6139 6140 6141 6142 6143
                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) {
6144
                errorPrint("%s", "failed to read json, interval input mistake\n");
6145 6146 6147 6148 6149 6150 6151 6152 6153 6154 6155 6156 6157 6158 6159 6160 6161
                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 {
6162
                errorPrint("%s", "failed to read json, subscribe restart error\n");
6163 6164 6165 6166 6167 6168 6169 6170 6171 6172 6173 6174 6175 6176 6177
                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 {
6178 6179
                errorPrint("%s",
                        "failed to read json, subscribe super table keepProgress error\n");
6180 6181 6182 6183 6184 6185 6186
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.superQueryInfo.subscribeKeepProgress = 0;
        }

        // default value is -1, which mean do not resub
6187
        g_queryInfo.superQueryInfo.endAfterConsume = -1;
6188 6189 6190 6191 6192 6193 6194 6195 6196
        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;
6197

6198
        // default value is -1, which mean do not resub
6199
        g_queryInfo.superQueryInfo.resubAfterConsume = -1;
6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215
        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) {
6216
            errorPrint("%s", "failed to read json, super sqls not found\n");
6217 6218 6219 6220
            goto PARSE_OVER;
        } else {
            int superSqlSize = cJSON_GetArraySize(superSqls);
            if (superSqlSize > MAX_QUERY_SQL_COUNT) {
6221 6222
                errorPrint("failed to read json, query sql size overflow, max is %d\n",
                        MAX_QUERY_SQL_COUNT);
6223 6224
                goto PARSE_OVER;
            }
6225

6226 6227 6228 6229
            g_queryInfo.superQueryInfo.sqlCount = superSqlSize;
            for (int j = 0; j < superSqlSize; ++j) {
                cJSON* sql = cJSON_GetArrayItem(superSqls, j);
                if (sql == NULL) continue;
6230

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

                cJSON *result = cJSON_GetObjectItem(sql, "result");
                if (result != NULL && result->type == cJSON_String
6242
                        && result->valuestring != NULL) {
6243 6244 6245 6246 6247
                    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 {
6248
                    errorPrint("%s", "failed to read json, sub query result file not found\n");
6249 6250 6251
                    goto PARSE_OVER;
                }
            }
6252
        }
6253
    }
H
hzcheng 已提交
6254

6255
    ret = true;
H
hzcheng 已提交
6256

6257
PARSE_OVER:
6258
    return ret;
6259
}
H
hzcheng 已提交
6260

6261
static bool getInfoFromJsonFile(char* file) {
6262
    debugPrint("%s %d %s\n", __func__, __LINE__, file);
6263

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

6270 6271 6272 6273 6274 6275 6276
    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);
6277
        errorPrint("failed to read %s, content is null", file);
6278 6279
        return false;
    }
H
Hui Li 已提交
6280

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

6288 6289 6290 6291 6292 6293 6294 6295 6296
    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 {
6297
            errorPrint("%s", "failed to read json, filetype not support\n");
6298 6299 6300 6301
            goto PARSE_OVER;
        }
    } else if (!filetype) {
        g_args.test_mode = INSERT_TEST;
6302
    } else {
6303
        errorPrint("%s", "failed to read json, filetype not found\n");
6304
        goto PARSE_OVER;
6305
    }
H
hzcheng 已提交
6306

6307 6308 6309 6310 6311 6312
    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 {
6313 6314
        errorPrint("%s",
                "input json file type error! please input correct file type: insert or query or subscribe\n");
6315 6316
        goto PARSE_OVER;
    }
6317

6318
PARSE_OVER:
6319 6320 6321 6322
    free(content);
    cJSON_Delete(root);
    fclose(fp);
    return ret;
6323
}
H
hzcheng 已提交
6324

6325
static int prepareSampleData() {
6326 6327 6328 6329 6330 6331 6332
    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;
                }
            }
6333
        }
6334
    }
6335

6336
    return 0;
6337 6338
}

6339
static void postFreeResource() {
6340
    tmfclose(g_fpOfInsertResult);
6341

6342 6343 6344
    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) {
6345
                tmfree(g_Dbs.db[i].superTbls[j].colsOfCreateChildTable);
6346 6347 6348
                g_Dbs.db[i].superTbls[j].colsOfCreateChildTable = NULL;
            }
            if (0 != g_Dbs.db[i].superTbls[j].sampleDataBuf) {
6349
                tmfree(g_Dbs.db[i].superTbls[j].sampleDataBuf);
6350 6351
                g_Dbs.db[i].superTbls[j].sampleDataBuf = NULL;
            }
6352

6353 6354 6355 6356 6357 6358 6359 6360 6361 6362 6363 6364 6365
#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
6366
            if (0 != g_Dbs.db[i].superTbls[j].tagDataBuf) {
6367
                tmfree(g_Dbs.db[i].superTbls[j].tagDataBuf);
6368 6369 6370
                g_Dbs.db[i].superTbls[j].tagDataBuf = NULL;
            }
            if (0 != g_Dbs.db[i].superTbls[j].childTblName) {
6371
                tmfree(g_Dbs.db[i].superTbls[j].childTblName);
6372 6373 6374
                g_Dbs.db[i].superTbls[j].childTblName = NULL;
            }
        }
S
Shuaiqiang Chang 已提交
6375
    }
6376 6377 6378 6379 6380 6381 6382 6383 6384 6385

    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);
6386

6387 6388 6389 6390 6391 6392 6393 6394 6395 6396 6397 6398 6399
    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
6400
}
S
Shuaiqiang Chang 已提交
6401

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

    int    dataLen = 0;

    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
            "(%" PRId64 ", ", timestamp);
    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
            "%s",
6416 6417
            stbInfo->sampleDataBuf
            + stbInfo->lenOfOneRow * (*sampleUsePos));
6418 6419 6420 6421 6422
    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen, ")");

    (*sampleUsePos)++;

    return dataLen;
6423
}
S
Shuaiqiang Chang 已提交
6424

6425 6426
static int64_t generateStbRowData(
        SSuperTable* stbInfo,
6427 6428 6429
        char* recBuf,
        int64_t remainderBufLen,
        int64_t timestamp)
6430
{
6431 6432 6433
    int64_t   dataLen = 0;
    char  *pstr = recBuf;
    int64_t maxLen = MAX_DATA_SIZE;
6434
    int tmpLen;
6435 6436

    dataLen += snprintf(pstr + dataLen, maxLen - dataLen,
6437
            "(%" PRId64 "", timestamp);
6438 6439

    for (int i = 0; i < stbInfo->columnCount; i++) {
6440 6441 6442 6443 6444
        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)) {
6445
            if (stbInfo->columns[i].dataLen > TSDB_MAX_BINARY_LEN) {
6446
                errorPrint2("binary or nchar length overflow, max size:%u\n",
6447 6448 6449
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
6450

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

6465
        } else {
6466 6467 6468 6469 6470 6471 6472 6473 6474 6475 6476
            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;
6477

6478 6479 6480 6481 6482 6483
                case TSDB_DATA_TYPE_UINT:
                    tmp = rand_uint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, INT_BUFF_LEN));
                    break;

6484 6485
                case TSDB_DATA_TYPE_BIGINT:
                    tmp = rand_bigint_str();
6486 6487 6488 6489 6490 6491 6492 6493
                    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));
6494 6495 6496 6497 6498 6499 6500 6501 6502
                    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();
                        }
6503
                    } else {
6504
                        tmp = rand_float_str();
6505
                    }
6506
                    tmpLen = strlen(tmp);
6507
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, FLOAT_BUFF_LEN));
6508 6509 6510 6511 6512
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    tmp = rand_double_str();
                    tmpLen = strlen(tmp);
6513
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, DOUBLE_BUFF_LEN));
6514 6515 6516 6517 6518
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
                    tmp = rand_smallint_str();
                    tmpLen = strlen(tmp);
6519 6520 6521 6522 6523 6524 6525
                    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));
6526 6527 6528 6529 6530
                    break;

                case TSDB_DATA_TYPE_TINYINT:
                    tmp = rand_tinyint_str();
                    tmpLen = strlen(tmp);
6531 6532 6533 6534 6535 6536 6537
                    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));
6538 6539 6540 6541 6542
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    tmp = rand_bool_str();
                    tmpLen = strlen(tmp);
6543
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BOOL_BUFF_LEN));
6544 6545 6546 6547 6548
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    tmp = rand_bigint_str();
                    tmpLen = strlen(tmp);
6549
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
6550 6551 6552 6553 6554 6555 6556 6557 6558
                    break;

                case TSDB_DATA_TYPE_NULL:
                    break;

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

6565
        if (dataLen > (remainderBufLen - (128)))
6566
            return 0;
6567
    }
6568

6569
    dataLen += snprintf(pstr + dataLen, 2, ")");
6570

6571 6572
    verbosePrint("%s() LN%d, dataLen:%"PRId64"\n", __func__, __LINE__, dataLen);
    verbosePrint("%s() LN%d, recBuf:\n\t%s\n", __func__, __LINE__, recBuf);
6573

6574
    return strlen(recBuf);
6575
}
S
Shuaiqiang Chang 已提交
6576

6577
static int64_t generateData(char *recBuf, char *data_type,
6578
        int64_t timestamp, int lenOfBinary) {
6579 6580
    memset(recBuf, 0, MAX_DATA_SIZE);
    char *pstr = recBuf;
6581
    pstr += sprintf(pstr, "(%"PRId64"", timestamp);
6582

6583
    int columnCount = g_args.columnCount;
6584

6585 6586
    bool b;
    char *s;
6587
    for (int i = 0; i < columnCount; i++) {
6588 6589 6590 6591 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
        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;

6646 6647 6648 6649 6650 6651 6652 6653 6654 6655 6656 6657 6658 6659 6660 6661
            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;

6662 6663 6664 6665 6666 6667 6668
            case TSDB_DATA_TYPE_NULL:
                break;

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

6672
        if (strlen(recBuf) > MAX_DATA_SIZE) {
6673
            ERROR_EXIT("column length too long, abort");
6674
        }
6675 6676
    }

6677
    pstr += sprintf(pstr, ")");
6678

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

6681
    return (int32_t)strlen(recBuf);
6682 6683
}

6684 6685 6686 6687 6688 6689
static int generateSampleFromRand(
        char *sampleDataBuf,
        uint64_t lenOfOneRow,
        int columnCount,
        StrColumn *columns
        )
6690 6691 6692 6693
{
    char data[MAX_DATA_SIZE];
    memset(data, 0, MAX_DATA_SIZE);

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

6701
    for (int i=0; i < MAX_SAMPLES; i++) {
6702
        uint64_t pos = 0;
6703
        memset(buff, 0, lenOfOneRow);
6704

6705 6706 6707 6708 6709 6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722 6723 6724 6725 6726
        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;
                    rand_string(data, dataLen);
                    pos += sprintf(buff + pos, "%s,", data);
                    break;

                case TSDB_DATA_TYPE_INT:
                    if ((g_args.demo_mode) && (c == 1)) {
                        tmp = demo_voltage_int_str();
6727
                    } else {
6728
                        tmp = rand_int_str();
6729
                    }
6730 6731 6732
                    pos += sprintf(buff + pos, "%s,", tmp);
                    break;

6733 6734 6735 6736
                case TSDB_DATA_TYPE_UINT:
                    pos += sprintf(buff + pos, "%s,", rand_uint_str());
                    break;

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

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

6745 6746 6747 6748 6749 6750 6751 6752 6753 6754 6755 6756 6757 6758 6759 6760 6761 6762 6763 6764 6765
                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;

6766 6767 6768 6769
                case TSDB_DATA_TYPE_USMALLINT:
                    pos += sprintf(buff + pos, "%s,", rand_usmallint_str());
                    break;

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

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

6778 6779 6780 6781 6782 6783 6784 6785 6786 6787 6788 6789 6790 6791 6792 6793
                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);
6794 6795
            }
        }
6796

6797
        *(buff + pos - 1) = 0;
6798
        memcpy(sampleDataBuf + i * lenOfOneRow, buff, pos);
6799 6800 6801 6802 6803 6804
    }

    free(buff);
    return 0;
}

6805 6806 6807 6808 6809 6810 6811 6812 6813 6814 6815 6816 6817 6818 6819 6820 6821 6822 6823
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() {
6824
    g_sampleDataBuf = calloc(g_args.lenOfOneRow * MAX_SAMPLES, 1);
6825 6826 6827
    if (NULL == g_sampleDataBuf) {
        errorPrint2("%s() LN%d, Failed to calloc %"PRIu64" Bytes, reason:%s\n",
                __func__, __LINE__,
6828
                g_args.lenOfOneRow * MAX_SAMPLES,
6829 6830 6831 6832 6833 6834 6835 6836
                strerror(errno));
        return -1;
    }

    return generateSampleFromRandForNtb();
}

static int prepareSampleForStb(SSuperTable *stbInfo) {
6837

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

6848
    int ret;
6849 6850 6851 6852 6853
    if (0 == strncasecmp(stbInfo->dataSource, "sample", strlen("sample"))) {
        ret = generateSampleFromCsvForStb(stbInfo);
    } else {
        ret = generateSampleFromRandForStb(stbInfo);
    }
6854

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

6863
    return 0;
6864 6865
}

6866
static int32_t execInsert(threadInfo *pThreadInfo, uint32_t k)
6867
{
6868
    int32_t affectedRows;
6869
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
6870

6871
    uint16_t iface;
6872 6873
    if (stbInfo)
        iface = stbInfo->iface;
6874 6875 6876 6877 6878 6879
    else {
        if (g_args.iface == INTERFACE_BUT)
            iface = TAOSC_IFACE;
        else
            iface = g_args.iface;
    }
6880 6881 6882

    debugPrint("[%d] %s() LN%d %s\n", pThreadInfo->threadID,
            __func__, __LINE__,
6883 6884
            (iface==TAOSC_IFACE)?
            "taosc":(iface==REST_IFACE)?"rest":"stmt");
6885 6886 6887

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

6891 6892 6893 6894 6895 6896
            affectedRows = queryDbExec(
                    pThreadInfo->taos,
                    pThreadInfo->buffer, INSERT_TYPE, false);
            break;

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

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

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

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

        default:
6924
            errorPrint2("%s() LN%d: unknown insert mode: %d\n",
6925
                    __func__, __LINE__, stbInfo->iface);
6926
            affectedRows = 0;
6927
    }
6928

6929
    return affectedRows;
6930 6931
}

6932 6933
static void getTableName(char *pTblName,
        threadInfo* pThreadInfo, uint64_t tableSeq)
6934
{
6935 6936 6937 6938
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    if (stbInfo) {
        if (AUTO_CREATE_SUBTBL != stbInfo->autoCreateTable) {
            if (stbInfo->childTblLimit > 0) {
6939
                snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s",
6940 6941
                        stbInfo->childTblName +
                        (tableSeq - stbInfo->childTblOffset) * TSDB_TABLE_NAME_LEN);
6942 6943 6944 6945 6946 6947
            } 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",
6948
                        stbInfo->childTblName + tableSeq * TSDB_TABLE_NAME_LEN);
6949 6950 6951
            }
        } else {
            snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s%"PRIu64"",
6952
                    stbInfo->childTblPrefix, tableSeq);
6953
        }
6954
    } else {
6955 6956
        snprintf(pTblName, TSDB_TABLE_NAME_LEN, "%s%"PRIu64"",
                g_args.tb_prefix, tableSeq);
6957 6958 6959
    }
}

6960
static int32_t generateDataTailWithoutStb(
6961
        uint32_t batch, char* buffer,
6962
        int64_t remainderBufLen, int64_t insertRows,
6963
        uint64_t recordFrom, int64_t startTime,
6964
        /* int64_t *pSamplePos, */int64_t *dataLen) {
6965

6966 6967
    uint64_t len = 0;
    char *pstr = buffer;
6968

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

6971 6972
    int32_t k = 0;
    for (k = 0; k < batch;) {
6973
        char *data = pstr;
6974
        memset(data, 0, MAX_DATA_SIZE);
6975

6976
        int64_t retLen = 0;
6977

6978
        char *data_type = g_args.data_type;
6979
        int lenOfBinary = g_args.binwidth;
6980

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

6994 6995
        if (len > remainderBufLen)
            break;
6996

6997
        pstr += retLen;
6998 6999 7000
        k++;
        len += retLen;
        remainderBufLen -= retLen;
7001

7002 7003
        verbosePrint("%s() LN%d len=%"PRIu64" k=%d \nbuffer=%s\n",
                __func__, __LINE__, len, k, buffer);
7004

7005
        recordFrom ++;
7006

7007 7008 7009
        if (recordFrom >= insertRows) {
            break;
        }
7010
    }
7011

7012 7013
    *dataLen = len;
    return k;
7014 7015
}

7016 7017 7018 7019 7020 7021 7022 7023 7024 7025 7026 7027 7028 7029 7030 7031 7032
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(
7033
        SSuperTable* stbInfo,
7034
        uint32_t batch, char* buffer,
7035
        int64_t remainderBufLen, int64_t insertRows,
7036
        uint64_t recordFrom, int64_t startTime,
7037
        int64_t *pSamplePos, int64_t *dataLen) {
7038
    uint64_t len = 0;
7039

7040
    char *pstr = buffer;
7041

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

7051 7052
    int32_t k;
    for (k = 0; k < batch;) {
7053
        char *data = pstr;
7054

7055
        int64_t lenOfRow = 0;
7056

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

7081 7082 7083 7084
        if (lenOfRow == 0) {
            data[0] = '\0';
            break;
        }
7085 7086 7087
        if ((lenOfRow + 1) > remainderBufLen) {
            break;
        }
7088

7089
        pstr += lenOfRow;
7090 7091 7092
        k++;
        len += lenOfRow;
        remainderBufLen -= lenOfRow;
7093

7094 7095
        verbosePrint("%s() LN%d len=%"PRIu64" k=%u \nbuffer=%s\n",
                __func__, __LINE__, len, k, buffer);
7096

7097
        recordFrom ++;
7098

7099 7100 7101
        if (recordFrom >= insertRows) {
            break;
        }
7102
    }
7103

7104 7105
    *dataLen = len;
    return k;
7106
}
7107

7108 7109 7110

static int generateSQLHeadWithoutStb(char *tableName,
        char *dbName,
7111
        char *buffer, int remainderBufLen)
7112
{
7113
    int len;
7114

7115
    char headBuf[HEAD_BUFF_LEN];
7116

7117 7118 7119 7120 7121 7122
    len = snprintf(
            headBuf,
            HEAD_BUFF_LEN,
            "%s.%s values",
            dbName,
            tableName);
7123

7124 7125
    if (len > remainderBufLen)
        return -1;
7126

7127
    tstrncpy(buffer, headBuf, len + 1);
7128

7129
    return len;
7130 7131 7132
}

static int generateStbSQLHead(
7133
        SSuperTable* stbInfo,
7134
        char *tableName, int64_t tableSeq,
7135 7136 7137
        char *dbName,
        char *buffer, int remainderBufLen)
{
7138
    int len;
7139

7140
    char headBuf[HEAD_BUFF_LEN];
7141

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

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

7183 7184
    if (len > remainderBufLen)
        return -1;
7185

7186
    tstrncpy(buffer, headBuf, len + 1);
7187

7188
    return len;
7189 7190
}

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

7205
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
7206
    int headLen = generateStbSQLHead(
7207
            stbInfo,
7208 7209
            tableName, tableSeq, pThreadInfo->db_name,
            pstr, *pRemainderBufLen);
7210

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

7218 7219
    pstr += headLen;
    *pRemainderBufLen -= headLen;
7220

7221
    int64_t dataLen = 0;
7222

7223
    verbosePrint("[%d] %s() LN%d i=%"PRIu64" batchPerTblTimes=%u batchPerTbl = %u\n",
7224 7225
            pThreadInfo->threadID, __func__, __LINE__,
            i, batchPerTblTimes, batchPerTbl);
7226

7227
    if (0 == strncasecmp(stbInfo->startTimestamp, "now", 3)) {
7228 7229
        startTime = taosGetTimestamp(pThreadInfo->time_precision);
    }
7230

7231
    int32_t k = generateStbDataTail(
7232
            stbInfo,
7233 7234 7235
            batchPerTbl, pstr, *pRemainderBufLen, insertRows, 0,
            startTime,
            &(pThreadInfo->samplePos), &dataLen);
7236

7237 7238 7239 7240 7241 7242 7243 7244 7245 7246
    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;
    }
7247

7248
    return k;
7249 7250 7251
}

static int64_t generateInterlaceDataWithoutStb(
7252
        char *tableName, uint32_t batch,
7253 7254 7255
        uint64_t tableSeq,
        char *dbName, char *buffer,
        int64_t insertRows,
7256
        int64_t startTime,
7257 7258
        uint64_t *pRemainderBufLen)
{
7259 7260
    assert(buffer);
    char *pstr = buffer;
7261

7262 7263
    int headLen = generateSQLHeadWithoutStb(
            tableName, dbName,
7264 7265
            pstr, *pRemainderBufLen);

7266 7267 7268
    if (headLen <= 0) {
        return 0;
    }
7269

7270 7271
    pstr += headLen;
    *pRemainderBufLen -= headLen;
7272

7273
    int64_t dataLen = 0;
7274

7275
    int32_t k = generateDataTailWithoutStb(
7276
            batch, pstr, *pRemainderBufLen, insertRows, 0,
7277
            startTime,
7278
            &dataLen);
7279

7280 7281 7282 7283 7284 7285 7286 7287 7288 7289 7290 7291
    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;
7292 7293
}

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

    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;
7321

7322 7323 7324 7325 7326 7327 7328 7329 7330 7331
            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;
            }
7332

7333 7334 7335 7336
            bind->length = &bind->buffer_length;
            bind->buffer = bind_binary;
            bind->is_null = NULL;
            break;
7337

7338 7339 7340 7341 7342 7343 7344
        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;
7345

7346 7347 7348 7349 7350 7351 7352 7353
            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);
            }
7354

7355 7356 7357 7358 7359
            bind->buffer_length = strlen(bind_nchar);
            bind->buffer = bind_nchar;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7360

7361 7362 7363
        case TSDB_DATA_TYPE_INT:
            bind_int = malloc(sizeof(int32_t));
            assert(bind_int);
7364

7365 7366 7367 7368 7369 7370 7371 7372 7373 7374 7375
            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;
7376 7377 7378 7379 7380 7381 7382 7383 7384 7385 7386 7387 7388 7389 7390 7391
        
        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;
7392

7393 7394 7395
        case TSDB_DATA_TYPE_BIGINT:
            bind_bigint = malloc(sizeof(int64_t));
            assert(bind_bigint);
7396

7397 7398 7399 7400 7401 7402 7403 7404 7405 7406 7407
            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;
7408

7409 7410 7411 7412 7413 7414 7415 7416 7417 7418 7419 7420 7421 7422 7423 7424
        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;

7425 7426 7427 7428 7429 7430
        case TSDB_DATA_TYPE_FLOAT:
            bind_float = malloc(sizeof(float));
            assert(bind_float);

            if (value) {
                *bind_float = (float)atof(value);
7431
            } else {
7432
                *bind_float = rand_float();
7433
            }
7434 7435 7436 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
            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;

7473 7474 7475 7476 7477 7478 7479 7480 7481 7482 7483 7484 7485 7486 7487 7488
        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;

7489 7490 7491 7492 7493 7494 7495 7496 7497 7498 7499 7500 7501 7502 7503 7504
        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;

7505 7506 7507 7508 7509 7510 7511 7512 7513 7514 7515 7516 7517 7518 7519 7520
        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;

7521 7522 7523 7524 7525 7526 7527 7528 7529 7530 7531 7532 7533 7534 7535 7536 7537 7538 7539
        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;
7540

7541 7542 7543 7544 7545 7546 7547 7548 7549 7550 7551 7552
        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++;
7553
                    }
7554 7555 7556 7557 7558 7559 7560 7561 7562 7563 7564
                    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);
7565 7566
                }
            } else {
7567
                *bind_ts2 = rand_bigint();
7568
            }
7569 7570 7571 7572 7573 7574 7575 7576 7577 7578 7579 7580 7581
            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);
7582 7583 7584 7585 7586 7587 7588
    }

    return 0;
}

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

    switch(data_type) {
        case TSDB_DATA_TYPE_BINARY:
7609

7610 7611 7612 7613 7614 7615
            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;
7616

7617 7618 7619 7620 7621 7622 7623 7624
            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;
            }
7625

7626 7627 7628
            bind->length = &bind->buffer_length;
            bind->buffer = bind_binary;
            bind->is_null = NULL;
7629

7630 7631
            *ptr += bind->buffer_length;
            break;
7632

7633 7634 7635 7636 7637 7638 7639
        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;
7640

7641 7642 7643 7644 7645 7646
            bind->buffer_type = TSDB_DATA_TYPE_NCHAR;
            if (value) {
                strncpy(bind_nchar, value, strlen(value));
            } else {
                rand_string(bind_nchar, dataLen);
            }
7647

7648 7649 7650 7651
            bind->buffer_length = strlen(bind_nchar);
            bind->buffer = bind_nchar;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7652

7653 7654
            *ptr += bind->buffer_length;
            break;
7655

7656 7657
        case TSDB_DATA_TYPE_INT:
            bind_int = (int32_t *)*ptr;
7658

7659 7660 7661 7662 7663 7664 7665 7666 7667 7668
            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;
7669

7670 7671
            *ptr += bind->buffer_length;
            break;
7672

7673 7674 7675 7676 7677 7678 7679 7680 7681 7682 7683 7684 7685 7686 7687 7688 7689
        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;

7690 7691
        case TSDB_DATA_TYPE_BIGINT:
            bind_bigint = (int64_t *)*ptr;
7692

7693 7694 7695 7696 7697 7698 7699 7700 7701 7702
            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;
7703

7704 7705
            *ptr += bind->buffer_length;
            break;
7706

7707 7708 7709 7710 7711 7712 7713 7714 7715 7716 7717 7718 7719 7720 7721 7722 7723
        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;

7724 7725
        case TSDB_DATA_TYPE_FLOAT:
            bind_float = (float *)*ptr;
7726

7727 7728 7729 7730 7731 7732 7733 7734 7735 7736
            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;
7737

7738 7739
            *ptr += bind->buffer_length;
            break;
7740

7741 7742
        case TSDB_DATA_TYPE_DOUBLE:
            bind_double = (double *)*ptr;
7743

7744 7745
            if (value) {
                *bind_double = atof(value);
7746
            } else {
7747
                *bind_double = rand_double();
7748
            }
7749 7750 7751 7752 7753 7754 7755 7756 7757 7758 7759 7760 7761 7762 7763 7764 7765 7766 7767 7768 7769 7770 7771 7772 7773 7774
            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;

7775 7776 7777 7778 7779 7780 7781 7782 7783 7784 7785 7786 7787 7788 7789 7790 7791
        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;

7792 7793 7794 7795 7796 7797 7798 7799 7800 7801 7802 7803 7804 7805 7806 7807 7808
        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;

7809 7810 7811 7812 7813 7814 7815 7816 7817 7818 7819 7820 7821 7822 7823 7824 7825
        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;

7826 7827 7828 7829 7830 7831 7832 7833
        case TSDB_DATA_TYPE_BOOL:
            bind_bool = (int8_t *)*ptr;

            if (value) {
                if (strncasecmp(value, "true", 4)) {
                    *bind_bool = true;
                } else {
                    *bind_bool = false;
7834
                }
7835 7836 7837 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
            } 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);
7869 7870
                }
            } else {
7871
                *bind_ts2 = rand_bigint();
7872
            }
7873 7874 7875 7876 7877
            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;
7878

7879 7880 7881 7882 7883 7884
            *ptr += bind->buffer_length;
            break;

        default:
            errorPrint2("No support data type: %d\n", data_type);
            return -1;
7885 7886 7887 7888 7889 7890
    }

    return 0;
}

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

7906
    char *data_type = g_args.data_type;
7907

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

7915 7916 7917 7918 7919 7920
    int32_t k = 0;
    for (k = 0; k < batch;) {
        /* columnCount + 1 (ts) */

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

7921
        int64_t *bind_ts = pThreadInfo->bind_ts;
7922 7923

        bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
7924 7925 7926

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

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

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

7970
    free(bindArray);
7971 7972 7973
    return k;
}

7974 7975 7976 7977 7978 7979 7980 7981 7982 7983 7984
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,
7985
                    stbInfo->tags[t].data_type,
7986 7987 7988 7989 7990 7991 7992 7993 7994 7995 7996 7997 7998
                    stbInfo->tags[t].dataLen,
                    timePrec,
                    NULL)) {
            return -1;
        }
    }

    return 0;
}

static int32_t prepareStbStmtBindRand(
        int64_t *ts,
        char *bindArray, SSuperTable *stbInfo,
7999
        int64_t startTime, int32_t recSeq,
8000
        int32_t timePrec)
8001 8002 8003 8004 8005 8006 8007
{
    char data[MAX_DATA_SIZE];
    memset(data, 0, MAX_DATA_SIZE);
    char *ptr = data;

    TAOS_BIND *bind;

8008 8009
    for (int i = 0; i < stbInfo->columnCount + 1; i ++) {
        bind = (TAOS_BIND *)((char *)bindArray + (sizeof(TAOS_BIND) * i));
8010

8011 8012
        if (i == 0) {
            int64_t *bind_ts = ts;
8013

8014 8015 8016 8017 8018 8019 8020 8021 8022 8023 8024 8025 8026 8027 8028 8029 8030
            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,
8031
                    stbInfo->columns[i-1].data_type,
8032 8033 8034 8035 8036 8037 8038
                    stbInfo->columns[i-1].dataLen,
                    timePrec,
                    &ptr,
                    NULL)) {
            return -1;
        }
    }
8039

8040 8041 8042 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
    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
8156
static int execStbBindParamBatch(
8157 8158 8159 8160 8161 8162 8163 8164 8165 8166 8167 8168 8169
        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;
8170 8171 8172
    assert(stbInfo);

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

8174
    uint32_t thisBatch = MAX_SAMPLES - (*pSamplePos);
8175 8176 8177 8178 8179 8180 8181 8182 8183 8184 8185 8186 8187 8188 8189 8190 8191 8192 8193 8194 8195 8196

    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 {
8197
            data_type = stbInfo->columns[c-1].data_type;
8198 8199 8200 8201 8202

            char *tmpP;

            switch(data_type) {
                case TSDB_DATA_TYPE_BINARY:
8203 8204 8205 8206 8207 8208 8209 8210 8211 8212 8213 8214 8215 8216
                    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;

8217 8218
                case TSDB_DATA_TYPE_NCHAR:
                    param->buffer_length =
8219
                        stbInfo->columns[c-1].dataLen;
8220 8221 8222 8223 8224

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

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

8229
                    param->buffer = (void *)(tmpP + *pSamplePos * param->buffer_length);
8230 8231 8232
                    break;

                case TSDB_DATA_TYPE_INT:
8233
                case TSDB_DATA_TYPE_UINT:
8234
                    param->buffer_length = sizeof(int32_t);
8235
                    param->buffer =
8236
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8237
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8238 8239 8240
                    break;

                case TSDB_DATA_TYPE_TINYINT:
8241
                case TSDB_DATA_TYPE_UTINYINT:
8242
                    param->buffer_length = sizeof(int8_t);
8243
                    param->buffer =
8244 8245 8246
                        (void *)((uintptr_t)*(uintptr_t*)(
                                    stbInfo->sampleBindBatchArray
                                    +sizeof(char*)*(c-1))
8247
                                + stbInfo->columns[c-1].dataLen*(*pSamplePos));
8248 8249 8250
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
8251
                case TSDB_DATA_TYPE_USMALLINT:
8252
                    param->buffer_length = sizeof(int16_t);
8253
                    param->buffer =
8254
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8255
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8256 8257 8258
                    break;

                case TSDB_DATA_TYPE_BIGINT:
8259
                case TSDB_DATA_TYPE_UBIGINT:
8260
                    param->buffer_length = sizeof(int64_t);
8261
                    param->buffer =
8262
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8263
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8264 8265 8266 8267
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    param->buffer_length = sizeof(int8_t);
8268
                    param->buffer =
8269
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8270
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8271 8272 8273 8274
                    break;

                case TSDB_DATA_TYPE_FLOAT:
                    param->buffer_length = sizeof(float);
8275
                    param->buffer =
8276
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8277
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8278 8279 8280 8281
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    param->buffer_length = sizeof(double);
8282
                    param->buffer =
8283
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8284
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8285 8286 8287 8288
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    param->buffer_length = sizeof(int64_t);
8289
                    param->buffer =
8290
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8291
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8292 8293 8294 8295 8296 8297 8298 8299 8300 8301 8302 8303 8304 8305 8306 8307 8308 8309 8310 8311
                    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 *
8312
                         stbInfo->columns[c].dataLen
8313 8314 8315 8316 8317 8318 8319 8320 8321 8322 8323 8324 8325 8326 8327 8328 8329 8330 8331 8332 8333 8334 8335 8336 8337 8338 8339 8340
                        );
            } 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) ++;
8341
        if ((*pSamplePos) == MAX_SAMPLES) {
8342 8343 8344 8345 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
            *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:
8396
            case TSDB_DATA_TYPE_UINT:
8397
                tmpP = calloc(1, sizeof(int) * MAX_SAMPLES);
8398 8399 8400 8401 8402
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_TINYINT:
8403
            case TSDB_DATA_TYPE_UTINYINT:
8404
                tmpP = calloc(1, sizeof(int8_t) * MAX_SAMPLES);
8405 8406 8407 8408 8409
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
8410
            case TSDB_DATA_TYPE_USMALLINT:
8411
                tmpP = calloc(1, sizeof(int16_t) * MAX_SAMPLES);
8412 8413 8414 8415 8416
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BIGINT:
8417
            case TSDB_DATA_TYPE_UBIGINT:
8418
                tmpP = calloc(1, sizeof(int64_t) * MAX_SAMPLES);
8419 8420 8421 8422 8423
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BOOL:
8424
                tmpP = calloc(1, sizeof(int8_t) * MAX_SAMPLES);
8425 8426 8427 8428 8429
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_FLOAT:
8430
                tmpP = calloc(1, sizeof(float) * MAX_SAMPLES);
8431 8432 8433 8434 8435
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_DOUBLE:
8436
                tmpP = calloc(1, sizeof(double) * MAX_SAMPLES);
8437 8438 8439 8440 8441 8442
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BINARY:
            case TSDB_DATA_TYPE_NCHAR:
8443
                tmpP = calloc(1, MAX_SAMPLES *
8444 8445 8446 8447 8448 8449
                        (((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:
8450
                tmpP = calloc(1, sizeof(int64_t) * MAX_SAMPLES);
8451 8452 8453 8454 8455 8456 8457 8458 8459 8460 8461 8462 8463 8464
                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;

8465
    for (int i=0; i < MAX_SAMPLES; i++) {
8466 8467 8468 8469 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
        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:
8496
                case TSDB_DATA_TYPE_UINT:
8497 8498 8499 8500 8501 8502 8503 8504 8505 8506 8507 8508 8509 8510 8511 8512 8513 8514
                    *((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:
8515
                case TSDB_DATA_TYPE_UTINYINT:
8516 8517 8518 8519 8520 8521
                    *((int8_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int8_t)*i)) =
                        (int8_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
8522
                case TSDB_DATA_TYPE_USMALLINT:
8523 8524 8525 8526 8527 8528
                    *((int16_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int16_t)*i)) =
                        (int16_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_BIGINT:
8529
                case TSDB_DATA_TYPE_UBIGINT:
8530 8531 8532 8533 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
                    *((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 =
8609
        (char *)calloc(1, sizeof(char *) * MAX_SAMPLES);
8610 8611 8612
    if (pThreadInfo->sampleBindArray == NULL) {
        errorPrint2("%s() LN%d, Failed to allocate %"PRIu64" bind array buffer\n",
                __func__, __LINE__,
8613
                (uint64_t)sizeof(char *) * MAX_SAMPLES);
8614 8615 8616 8617 8618 8619 8620
        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;

8621
    for (int i=0; i < MAX_SAMPLES; i++) {
8622 8623 8624 8625 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
        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);
8677
                    free(bindArray);
8678 8679 8680 8681 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
                    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);
8707
}
8708

8709 8710
static int32_t prepareStbStmtBindStartTime(
        char *tableName,
8711 8712
        int64_t *ts,
        char *bindArray, SSuperTable *stbInfo,
8713
        int64_t startTime, int32_t recSeq)
8714 8715
{
    TAOS_BIND *bind;
8716

8717
    bind = (TAOS_BIND *)bindArray;
8718

8719
    int64_t *bind_ts = ts;
8720

8721 8722 8723 8724 8725 8726 8727 8728
    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;
8729
    }
8730 8731 8732 8733

    verbosePrint("%s() LN%d, tableName: %s, bind_ts=%"PRId64"\n",
            __func__, __LINE__, tableName, *bind_ts);

8734 8735 8736 8737
    bind->buffer_length = sizeof(int64_t);
    bind->buffer = bind_ts;
    bind->length = &bind->buffer_length;
    bind->is_null = NULL;
8738 8739 8740 8741

    return 0;
}

8742
static uint32_t execBindParam(
8743
        threadInfo *pThreadInfo,
8744 8745 8746
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
8747 8748
        uint64_t insertRows,
        uint64_t recordFrom,
8749 8750
        int64_t startTime,
        int64_t *pSamplePos)
8751
{
8752
    int ret;
8753 8754
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;
8755

8756 8757
    uint32_t k;
    for (k = 0; k < batch;) {
8758 8759
        char *bindArray = (char *)(*((uintptr_t *)
                    (pThreadInfo->sampleBindArray + (sizeof(char *)) * (*pSamplePos))));
8760
        /* columnCount + 1 (ts) */
8761 8762
        if (-1 == prepareStbStmtBindStartTime(
                    tableName,
8763 8764
                    pThreadInfo->bind_ts,
                    bindArray, stbInfo,
8765
                    startTime, k
8766
                    /* is column */)) {
8767
            return -1;
8768
        }
8769 8770
        ret = taos_stmt_bind_param(stmt, (TAOS_BIND *)bindArray);
        if (0 != ret) {
8771
            errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
8772 8773 8774
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
8775
        // if msg > 3MB, break
8776 8777
        ret = taos_stmt_add_batch(stmt);
        if (0 != ret) {
8778
            errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
8779 8780 8781
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
8782

8783 8784
        k++;
        recordFrom ++;
8785

8786
        (*pSamplePos) ++;
8787
        if ((*pSamplePos) == MAX_SAMPLES) {
8788 8789 8790
            *pSamplePos = 0;
        }

8791 8792 8793 8794
        if (recordFrom >= insertRows) {
            break;
        }
    }
8795 8796 8797

    return k;
}
8798
#endif
8799

8800
static int32_t prepareStbStmt(
8801
        threadInfo *pThreadInfo,
8802 8803 8804
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
8805 8806 8807 8808 8809
        uint64_t insertRows,
        uint64_t recordFrom,
        int64_t startTime,
        int64_t *pSamplePos)
{
8810 8811 8812
    int ret;
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;
8813

8814 8815 8816 8817 8818 8819 8820 8821 8822 8823 8824 8825
    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) {
8826
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
8827 8828 8829 8830 8831 8832 8833
                    __func__, __LINE__);
            return -1;
        }

        char *tagsArray = calloc(1, sizeof(TAOS_BIND) * stbInfo->tagCount);
        if (NULL == tagsArray) {
            tmfree(tagsValBuf);
8834
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
8835 8836 8837 8838 8839 8840 8841 8842 8843 8844 8845 8846 8847 8848 8849 8850 8851 8852
                    __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) {
8853
            errorPrint2("%s() LN%d, stmt_set_tbname_tags() failed! reason: %s\n",
8854 8855 8856 8857 8858 8859
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    } else {
        ret = taos_stmt_set_tbname(stmt, tableName);
        if (0 != ret) {
8860
            errorPrint2("%s() LN%d, stmt_set_tbname() failed! reason: %s\n",
8861 8862 8863 8864 8865
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    }

8866
#if STMT_BIND_PARAM_BATCH == 1
8867
    return execStbBindParamBatch(
8868 8869 8870 8871 8872 8873 8874 8875 8876 8877 8878 8879 8880 8881 8882 8883 8884 8885 8886
        pThreadInfo,
        tableName,
        tableSeq,
        batch,
        insertRows,
        recordFrom,
        startTime,
        pSamplePos);
#else
    return execBindParam(
        pThreadInfo,
        tableName,
        tableSeq,
        batch,
        insertRows,
        recordFrom,
        startTime,
        pSamplePos);
#endif
8887
}
8888 8889

static int32_t generateStbProgressiveData(
8890
        SSuperTable *stbInfo,
8891
        char *tableName,
8892
        int64_t tableSeq,
8893
        char *dbName, char *buffer,
8894
        int64_t insertRows,
8895
        uint64_t recordFrom, int64_t startTime, int64_t *pSamplePos,
8896
        int64_t *pRemainderBufLen)
8897
{
8898 8899
    assert(buffer != NULL);
    char *pstr = buffer;
8900

8901
    memset(pstr, 0, *pRemainderBufLen);
8902

8903
    int64_t headLen = generateStbSQLHead(
8904
            stbInfo,
8905 8906
            tableName, tableSeq, dbName,
            buffer, *pRemainderBufLen);
8907

8908 8909 8910 8911 8912
    if (headLen <= 0) {
        return 0;
    }
    pstr += headLen;
    *pRemainderBufLen -= headLen;
8913

8914
    int64_t dataLen;
8915

8916
    return generateStbDataTail(stbInfo,
8917
            g_args.reqPerReq, pstr, *pRemainderBufLen,
8918 8919 8920
            insertRows, recordFrom,
            startTime,
            pSamplePos, &dataLen);
8921 8922
}

8923
static int32_t generateProgressiveDataWithoutStb(
8924
        char *tableName,
8925
        /* int64_t tableSeq, */
8926 8927
        threadInfo *pThreadInfo, char *buffer,
        int64_t insertRows,
8928
        uint64_t recordFrom, int64_t startTime, /*int64_t *pSamplePos, */
8929 8930
        int64_t *pRemainderBufLen)
{
8931 8932
    assert(buffer != NULL);
    char *pstr = buffer;
8933

8934
    memset(buffer, 0, *pRemainderBufLen);
8935

8936 8937 8938
    int64_t headLen = generateSQLHeadWithoutStb(
            tableName, pThreadInfo->db_name,
            buffer, *pRemainderBufLen);
8939

8940 8941 8942 8943 8944
    if (headLen <= 0) {
        return 0;
    }
    pstr += headLen;
    *pRemainderBufLen -= headLen;
8945

8946
    int64_t dataLen;
8947

8948
    return generateDataTailWithoutStb(
8949
            g_args.reqPerReq, pstr, *pRemainderBufLen, insertRows, recordFrom,
8950 8951
            startTime,
            /*pSamplePos, */&dataLen);
8952
}
8953

8954 8955
static void printStatPerThread(threadInfo *pThreadInfo)
{
8956 8957 8958
    if (0 == pThreadInfo->totalDelay)
        pThreadInfo->totalDelay = 1;

8959
    fprintf(stderr, "====thread[%d] completed total inserted rows: %"PRIu64 ", total affected rows: %"PRIu64". %.2f records/second====\n",
8960 8961 8962
            pThreadInfo->threadID,
            pThreadInfo->totalInsertRows,
            pThreadInfo->totalAffectedRows,
8963 8964
            (double)(pThreadInfo->totalAffectedRows/((double)pThreadInfo->totalDelay/1000000.0))
            );
8965 8966
}

8967 8968 8969 8970
#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",
8971
            pThreadInfo->threadID, __func__, __LINE__);
8972

8973
    int64_t insertRows;
8974
    int64_t timeStampStep;
8975
    uint64_t insert_interval;
8976

8977
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
8978

8979 8980
    if (stbInfo) {
        insertRows = stbInfo->insertRows;
8981
        timeStampStep = stbInfo->timeStampStep;
8982 8983 8984
        insert_interval = stbInfo->insertInterval;
    } else {
        insertRows = g_args.insertRows;
8985
        timeStampStep = g_args.timestamp_step;
8986 8987
        insert_interval = g_args.insert_interval;
    }
8988

8989 8990 8991 8992 8993
    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);

8994 8995
    uint64_t timesInterlace = (insertRows / interlaceRows) + 1;
    uint32_t precalcBatch = interlaceRows;
8996

8997 8998
    if (precalcBatch > g_args.reqPerReq)
        precalcBatch = g_args.reqPerReq;
8999

9000 9001
    if (precalcBatch > MAX_SAMPLES)
        precalcBatch = MAX_SAMPLES;
9002 9003 9004 9005 9006 9007 9008 9009 9010 9011 9012 9013

    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;
9014
    int64_t startTime;
9015 9016 9017 9018 9019 9020

    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;

    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;
9021
    pThreadInfo->samplePos = 0;
9022

9023 9024
    for (int64_t interlace = 0;
            interlace < timesInterlace; interlace ++) {
9025 9026 9027 9028 9029
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }

9030 9031
        int64_t generated = 0;
        int64_t samplePos;
9032

9033
        for (; tableSeq < pThreadInfo->start_table_from + pThreadInfo->ntables; tableSeq ++) {
9034 9035 9036 9037 9038 9039 9040 9041
            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;
            }

9042 9043 9044 9045 9046 9047
            samplePos = pThreadInfo->samplePos;
            startTime = pThreadInfo->start_time
                + interlace * interlaceRows * timeStampStep;
            uint64_t remainRecPerTbl =
                    insertRows - interlaceRows * interlace;
            uint64_t recPerTbl = 0;
9048

9049 9050 9051 9052 9053
            uint64_t remainPerInterlace;
            if (remainRecPerTbl > interlaceRows) {
                remainPerInterlace = interlaceRows;
            } else {
                remainPerInterlace = remainRecPerTbl;
9054 9055
            }

9056
            while(remainPerInterlace > 0) {
9057

9058 9059 9060 9061 9062 9063 9064 9065 9066 9067
                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);
9068

9069 9070 9071 9072 9073 9074 9075 9076 9077 9078 9079 9080 9081 9082 9083 9084 9085 9086
                if (stbInfo) {
                    generated = prepareStbStmt(
                            pThreadInfo,
                            tableName,
                            tableSeq,
                            batch,
                            insertRows, 0,
                            startTime,
                            &samplePos);
                } else {
                    generated = prepareStmtWithoutStb(
                            pThreadInfo,
                            tableName,
                            batch,
                            insertRows,
                            interlaceRows * interlace + recPerTbl,
                            startTime);
                }
9087

9088 9089 9090 9091 9092 9093 9094
                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) {
9095
                    break;
9096
                }
9097

9098 9099 9100
                recPerTbl += generated;
                remainPerInterlace -= generated;
                pThreadInfo->totalInsertRows += generated;
9101

9102 9103 9104
                verbosePrint("[%d] %s() LN%d totalInsertRows=%"PRIu64"\n",
                        pThreadInfo->threadID, __func__, __LINE__,
                        pThreadInfo->totalInsertRows);
9105

9106
                startTs = taosGetTimestampUs();
9107

9108
                int64_t affectedRows = execInsert(pThreadInfo, generated);
9109

9110 9111 9112 9113 9114 9115 9116
                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);
9117

9118 9119 9120 9121
                if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
                if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
                pThreadInfo->cntDelay++;
                pThreadInfo->totalDelay += delay;
9122

9123 9124 9125 9126 9127 9128
                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;
                }
9129

9130
                pThreadInfo->totalAffectedRows += affectedRows;
9131

9132 9133 9134 9135 9136 9137 9138 9139 9140 9141 9142 9143 9144
                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;
                }
9145

9146 9147
                startTime += (generated * timeStampStep);
            }
9148
        }
9149
        pThreadInfo->samplePos = samplePos;
9150

9151 9152 9153 9154
        if (tableSeq == pThreadInfo->start_table_from
                + pThreadInfo->ntables) {
            // turn to first table
            tableSeq = pThreadInfo->start_table_from;
9155

9156
            flagSleep = true;
9157 9158 9159 9160 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
        }

        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;
9186
    int64_t timeStampStep;
9187 9188 9189 9190 9191 9192
    uint64_t insert_interval;

    SSuperTable* stbInfo = pThreadInfo->stbInfo;

    if (stbInfo) {
        insertRows = stbInfo->insertRows;
9193
        maxSqlLen = stbInfo->maxSqlLen;
9194
        timeStampStep = stbInfo->timeStampStep;
9195
        insert_interval = stbInfo->insertInterval;
9196
    } else {
9197
        insertRows = g_args.insertRows;
9198
        maxSqlLen = g_args.max_sql_len;
9199
        timeStampStep = g_args.timestamp_step;
9200
        insert_interval = g_args.insert_interval;
9201
    }
9202

9203 9204 9205 9206
    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);
9207

9208 9209 9210 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
    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;
            }

9260 9261 9262 9263
            debugPrint("[%d] %s() LN%d, tableName:%s, batch:%d startTime:%"PRId64"\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__,
                    tableName, batchPerTbl, startTime);
9264
            if (stbInfo) {
9265
                generated = prepareStbStmt(
9266 9267 9268 9269 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
                        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
9307
                    + generatedRecPerTbl * timeStampStep;
9308 9309 9310 9311 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

                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
9404

9405 9406 9407 9408 9409 9410 9411
// 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;
9412
    int64_t timeStampStep;
9413 9414 9415 9416 9417 9418 9419
    uint64_t insert_interval;

    SSuperTable* stbInfo = pThreadInfo->stbInfo;

    if (stbInfo) {
        insertRows = stbInfo->insertRows;
        maxSqlLen = stbInfo->maxSqlLen;
9420
        timeStampStep = stbInfo->timeStampStep;
9421 9422 9423 9424
        insert_interval = stbInfo->insertInterval;
    } else {
        insertRows = g_args.insertRows;
        maxSqlLen = g_args.max_sql_len;
9425
        timeStampStep = g_args.timestamp_step;
9426 9427 9428 9429 9430 9431 9432 9433
        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
9434 9435
    if (interlaceRows > g_args.reqPerReq)
        interlaceRows = g_args.reqPerReq;
9436

9437 9438
    uint32_t batchPerTbl = interlaceRows;
    uint32_t batchPerTblTimes;
9439

9440 9441
    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
9442
            g_args.reqPerReq / interlaceRows;
9443 9444 9445
    } else {
        batchPerTblTimes = 1;
    }
9446 9447 9448 9449 9450 9451 9452 9453
#else
    uint32_t batchPerTbl;
    if (interlaceRows > g_args.reqPerReq)
        batchPerTbl = g_args.reqPerReq;
    else
        batchPerTbl = interlaceRows;

    uint32_t batchPerTblTimes;
9454

9455 9456 9457 9458 9459 9460 9461
    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
            interlaceRows / batchPerTbl;
    } else {
        batchPerTblTimes = 1;
    }
#endif
9462 9463
    pThreadInfo->buffer = calloc(maxSqlLen, 1);
    if (NULL == pThreadInfo->buffer) {
9464
        errorPrint2( "%s() LN%d, Failed to alloc %"PRIu64" Bytes, reason:%s\n",
9465 9466 9467
                __func__, __LINE__, maxSqlLen, strerror(errno));
        return NULL;
    }
9468

9469 9470
    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;
9471

9472 9473
    uint64_t st = 0;
    uint64_t et = UINT64_MAX;
9474

9475 9476 9477
    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;
9478

9479 9480
    uint64_t tableSeq = pThreadInfo->start_table_from;
    int64_t startTime = pThreadInfo->start_time;
9481

9482 9483 9484
    uint64_t generatedRecPerTbl = 0;
    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;
9485

9486 9487 9488
    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

9489 9490 9491 9492 9493
    while(pThreadInfo->totalInsertRows < pThreadInfo->ntables * insertRows) {
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }
9494

9495 9496 9497
        // generate data
        memset(pThreadInfo->buffer, 0, maxSqlLen);
        uint64_t remainderBufLen = maxSqlLen;
9498

9499
        char *pstr = pThreadInfo->buffer;
9500

9501 9502 9503 9504
        int len = snprintf(pstr,
                strlen(STR_INSERT_INTO) + 1, "%s", STR_INSERT_INTO);
        pstr += len;
        remainderBufLen -= len;
9505

9506
        uint32_t recOfBatch = 0;
9507

9508
        int32_t generated;
9509 9510
        for (uint64_t i = 0; i < batchPerTblTimes; i ++) {
            char tableName[TSDB_TABLE_NAME_LEN];
9511

9512 9513
            getTableName(tableName, pThreadInfo, tableSeq);
            if (0 == strlen(tableName)) {
9514
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
9515 9516 9517 9518
                        pThreadInfo->threadID, __func__, __LINE__);
                free(pThreadInfo->buffer);
                return NULL;
            }
9519

9520
            uint64_t oldRemainderLen = remainderBufLen;
9521

9522
            if (stbInfo) {
9523 9524 9525 9526 9527 9528 9529 9530 9531
                generated = generateStbInterlaceData(
                        pThreadInfo,
                        tableName, batchPerTbl, i,
                        batchPerTblTimes,
                        tableSeq,
                        pstr,
                        insertRows,
                        startTime,
                        &remainderBufLen);
9532
            } else {
9533 9534 9535 9536 9537 9538 9539
                generated = generateInterlaceDataWithoutStb(
                        tableName, batchPerTbl,
                        tableSeq,
                        pThreadInfo->db_name, pstr,
                        insertRows,
                        startTime,
                        &remainderBufLen);
9540
            }
9541

9542 9543 9544
            debugPrint("[%d] %s() LN%d, generated records is %d\n",
                    pThreadInfo->threadID, __func__, __LINE__, generated);
            if (generated < 0) {
9545
                errorPrint2("[%d] %s() LN%d, generated records is %d\n",
9546 9547 9548 9549 9550
                        pThreadInfo->threadID, __func__, __LINE__, generated);
                goto free_of_interlace;
            } else if (generated == 0) {
                break;
            }
9551

9552 9553
            tableSeq ++;
            recOfBatch += batchPerTbl;
9554

9555 9556
            pstr += (oldRemainderLen - remainderBufLen);
            pThreadInfo->totalInsertRows += batchPerTbl;
9557

9558 9559 9560
            verbosePrint("[%d] %s() LN%d batchPerTbl=%d recOfBatch=%d\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    batchPerTbl, recOfBatch);
9561

9562 9563 9564 9565
            if (tableSeq == pThreadInfo->start_table_from + pThreadInfo->ntables) {
                // turn to first table
                tableSeq = pThreadInfo->start_table_from;
                generatedRecPerTbl += batchPerTbl;
9566

9567
                startTime = pThreadInfo->start_time
9568
                    + generatedRecPerTbl * timeStampStep;
9569

9570 9571 9572
                flagSleep = true;
                if (generatedRecPerTbl >= insertRows)
                    break;
9573

9574 9575 9576
                int64_t remainRows = insertRows - generatedRecPerTbl;
                if ((remainRows > 0) && (batchPerTbl > remainRows))
                    batchPerTbl = remainRows;
9577

9578
                if (pThreadInfo->ntables * batchPerTbl < g_args.reqPerReq)
9579 9580
                    break;
            }
9581

9582 9583 9584
            verbosePrint("[%d] %s() LN%d generatedRecPerTbl=%"PRId64" insertRows=%"PRId64"\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    generatedRecPerTbl, insertRows);
9585

9586
            if ((g_args.reqPerReq - recOfBatch) < batchPerTbl)
9587
                break;
9588
        }
9589

9590 9591 9592 9593 9594 9595
        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);

9596
        startTs = taosGetTimestampUs();
9597 9598

        if (recOfBatch == 0) {
9599
            errorPrint2("[%d] %s() LN%d Failed to insert records of batch %d\n",
9600 9601 9602 9603 9604 9605 9606 9607 9608 9609 9610 9611
                    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);

9612
        endTs = taosGetTimestampUs();
9613
        uint64_t delay = endTs - startTs;
9614 9615
        performancePrint("%s() LN%d, insert execution time is %10.2f ms\n",
                __func__, __LINE__, delay / 1000.0);
9616 9617 9618
        verbosePrint("[%d] %s() LN%d affectedRows=%"PRId64"\n",
                pThreadInfo->threadID,
                __func__, __LINE__, affectedRows);
9619

9620 9621 9622 9623
        if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
        if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
        pThreadInfo->cntDelay++;
        pThreadInfo->totalDelay += delay;
9624

9625
        if (recOfBatch != affectedRows) {
9626
            errorPrint2("[%d] %s() LN%d execInsert insert %d, affected rows: %"PRId64"\n%s\n",
9627 9628 9629 9630
                    pThreadInfo->threadID, __func__, __LINE__,
                    recOfBatch, affectedRows, pThreadInfo->buffer);
            goto free_of_interlace;
        }
9631

9632
        pThreadInfo->totalAffectedRows += affectedRows;
9633

9634 9635 9636 9637 9638
        int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
        if (currentPercent > percentComplete ) {
            printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
            percentComplete = currentPercent;
        }
9639 9640 9641
        int64_t  currentPrintTime = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
            printf("thread[%d] has currently inserted rows: %"PRIu64 ", affected rows: %"PRIu64 "\n",
9642 9643 9644
                    pThreadInfo->threadID,
                    pThreadInfo->totalInsertRows,
                    pThreadInfo->totalAffectedRows);
9645 9646
            lastPrintTime = currentPrintTime;
        }
9647

9648 9649
        if ((insert_interval) && flagSleep) {
            et = taosGetTimestampMs();
9650

9651 9652 9653 9654 9655 9656 9657 9658
            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;
            }
        }
9659
    }
9660 9661
    if (percentComplete < 100)
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
9662

9663
free_of_interlace:
9664 9665 9666
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
9667 9668
}

9669 9670 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
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;
}
9807
// sync insertion progressive data
9808
static void* syncWriteProgressive(threadInfo *pThreadInfo) {
9809
    debugPrint("%s() LN%d: ### progressive write\n", __func__, __LINE__);
9810

9811 9812
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    uint64_t maxSqlLen = stbInfo?stbInfo->maxSqlLen:g_args.max_sql_len;
9813
    int64_t timeStampStep =
9814
        stbInfo?stbInfo->timeStampStep:g_args.timestamp_step;
9815
    int64_t insertRows =
9816
        (stbInfo)?stbInfo->insertRows:g_args.insertRows;
9817
    verbosePrint("%s() LN%d insertRows=%"PRId64"\n",
9818
            __func__, __LINE__, insertRows);
9819

9820 9821
    pThreadInfo->buffer = calloc(maxSqlLen, 1);
    if (NULL == pThreadInfo->buffer) {
9822
        errorPrint2("Failed to alloc %"PRIu64" bytes, reason:%s\n",
9823 9824
                maxSqlLen,
                strerror(errno));
9825
        return NULL;
9826 9827 9828 9829 9830 9831 9832 9833 9834 9835 9836
    }

    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;

    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;

    pThreadInfo->samplePos = 0;

9837 9838 9839
    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

9840 9841 9842 9843 9844 9845 9846 9847 9848 9849 9850 9851
    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)) {
9852
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
9853 9854 9855 9856
                        pThreadInfo->threadID, __func__, __LINE__);
                free(pThreadInfo->buffer);
                return NULL;
            }
9857

9858
            int64_t remainderBufLen = maxSqlLen - 2000;
9859
            char *pstr = pThreadInfo->buffer;
9860

9861 9862
            int len = snprintf(pstr,
                    strlen(STR_INSERT_INTO) + 1, "%s", STR_INSERT_INTO);
9863

9864 9865
            pstr += len;
            remainderBufLen -= len;
9866

9867 9868 9869
            // measure prepare + insert
            startTs = taosGetTimestampUs();

9870
            int32_t generated;
9871 9872
            if (stbInfo) {
                if (stbInfo->iface == STMT_IFACE) {
9873
                    generated = prepareStbStmt(
9874 9875 9876
                            pThreadInfo,
                            tableName,
                            tableSeq,
9877 9878 9879
                            (g_args.reqPerReq>stbInfo->insertRows)?
                                stbInfo->insertRows:
                                g_args.reqPerReq,
9880 9881
                            insertRows, i, start_time,
                            &(pThreadInfo->samplePos));
9882 9883
                } else {
                    generated = generateStbProgressiveData(
9884 9885 9886
                            stbInfo,
                            tableName, tableSeq,
                            pThreadInfo->db_name, pstr,
9887 9888 9889 9890 9891 9892 9893
                            insertRows, i, start_time,
                            &(pThreadInfo->samplePos),
                            &remainderBufLen);
                }
            } else {
                if (g_args.iface == STMT_IFACE) {
                    generated = prepareStmtWithoutStb(
9894
                            pThreadInfo,
9895
                            tableName,
9896
                            g_args.reqPerReq,
9897 9898 9899 9900 9901 9902 9903 9904 9905 9906 9907 9908
                            insertRows, i,
                            start_time);
                } else {
                    generated = generateProgressiveDataWithoutStb(
                            tableName,
                            /*  tableSeq, */
                            pThreadInfo, pstr, insertRows,
                            i, start_time,
                            /* &(pThreadInfo->samplePos), */
                            &remainderBufLen);
                }
            }
9909 9910 9911 9912 9913

            verbosePrint("[%d] %s() LN%d generated=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, generated);

9914 9915 9916 9917
            if (generated > 0)
                i += generated;
            else
                goto free_of_progressive;
9918

9919 9920
            start_time +=  generated * timeStampStep;
            pThreadInfo->totalInsertRows += generated;
9921

9922 9923
            // only measure insert
            // startTs = taosGetTimestampUs();
9924

9925
            int32_t affectedRows = execInsert(pThreadInfo, generated);
9926

9927
            endTs = taosGetTimestampUs();
9928
            uint64_t delay = endTs - startTs;
9929 9930
            performancePrint("%s() LN%d, insert execution time is %10.f ms\n",
                    __func__, __LINE__, delay/1000.0);
9931 9932 9933
            verbosePrint("[%d] %s() LN%d affectedRows=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, affectedRows);
9934

9935 9936 9937 9938
            if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
            if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
            pThreadInfo->cntDelay++;
            pThreadInfo->totalDelay += delay;
9939

9940
            if (affectedRows < 0) {
9941
                errorPrint2("%s() LN%d, affected rows: %d\n",
9942 9943 9944
                        __func__, __LINE__, affectedRows);
                goto free_of_progressive;
            }
9945

9946
            pThreadInfo->totalAffectedRows += affectedRows;
9947

9948 9949 9950 9951 9952
            int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
            if (currentPercent > percentComplete ) {
                printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
                percentComplete = currentPercent;
            }
9953 9954 9955 9956 9957 9958 9959 9960
            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;
            }
9961

9962 9963
            if (i >= insertRows)
                break;
9964
        }   // insertRows
9965

9966
        if ((g_args.verbose_print) &&
9967
                (tableSeq == pThreadInfo->ntables - 1) && (stbInfo)
9968
                && (0 == strncasecmp(
9969
                        stbInfo->dataSource,
9970
                        "sample", strlen("sample")))) {
9971 9972 9973 9974
            verbosePrint("%s() LN%d samplePos=%"PRId64"\n",
                    __func__, __LINE__, pThreadInfo->samplePos);
        }
    } // tableSeq
9975 9976

    if (percentComplete < 100) {
9977
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
9978
    }
9979

9980
free_of_progressive:
9981 9982 9983
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
H
Hui Li 已提交
9984 9985
}

9986 9987
static void* syncWrite(void *sarg) {

9988
    threadInfo *pThreadInfo = (threadInfo *)sarg;
9989
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
9990

9991 9992
    setThreadName("syncWrite");

9993
    uint32_t interlaceRows = 0;
9994

9995
    if (stbInfo) {
9996
        if (stbInfo->interlaceRows < stbInfo->insertRows)
9997
            interlaceRows = stbInfo->interlaceRows;
9998
    } else {
9999 10000
        if (g_args.interlaceRows < g_args.insertRows)
            interlaceRows = g_args.interlaceRows;
10001
    }
10002

10003 10004
    if (interlaceRows > 0) {
        // interlace mode
10005 10006
        if (stbInfo) {
            if (STMT_IFACE == stbInfo->iface) {
10007
#if STMT_BIND_PARAM_BATCH == 1
10008
                return syncWriteInterlaceStmtBatch(pThreadInfo, interlaceRows);
10009
#else
10010
                return syncWriteInterlaceStmt(pThreadInfo, interlaceRows);
10011
#endif
10012 10013 10014
            } else {
                return syncWriteInterlace(pThreadInfo, interlaceRows);
            }
10015
        }
10016 10017 10018 10019 10020 10021 10022 10023
    } else {
      // progressive mode
      if (((stbInfo) && (STMT_IFACE == stbInfo->iface))
              || (STMT_IFACE == g_args.iface)) {
          return syncWriteProgressiveStmt(pThreadInfo);
      } else {
          return syncWriteProgressive(pThreadInfo);
      }
10024
    }
10025 10026

    return NULL;
10027 10028
}

10029
static void callBack(void *param, TAOS_RES *res, int code) {
10030
    threadInfo* pThreadInfo = (threadInfo*)param;
10031
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10032 10033

    int insert_interval =
10034
        stbInfo?stbInfo->insertInterval:g_args.insert_interval;
10035 10036 10037 10038 10039
    if (insert_interval) {
        pThreadInfo->et = taosGetTimestampMs();
        if ((pThreadInfo->et - pThreadInfo->st) < insert_interval) {
            taosMsleep(insert_interval - (pThreadInfo->et - pThreadInfo->st)); // ms
        }
H
Hui Li 已提交
10040 10041
    }

10042
    char *buffer = calloc(1, pThreadInfo->stbInfo->maxSqlLen);
10043 10044
    char data[MAX_DATA_SIZE];
    char *pstr = buffer;
10045
    pstr += sprintf(pstr, "INSERT INTO %s.%s%"PRId64" VALUES",
10046 10047
            pThreadInfo->db_name, pThreadInfo->tb_prefix,
            pThreadInfo->start_table_from);
10048
    //  if (pThreadInfo->counter >= pThreadInfo->stbInfo->insertRows) {
10049
    if (pThreadInfo->counter >= g_args.reqPerReq) {
10050 10051 10052 10053 10054 10055 10056 10057
        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 已提交
10058
    }
10059

10060
    for (int i = 0; i < g_args.reqPerReq; i++) {
10061
        int rand_num = taosRandom() % 100;
10062 10063
        if (0 != pThreadInfo->stbInfo->disorderRatio
                && rand_num < pThreadInfo->stbInfo->disorderRatio) {
10064
            int64_t d = pThreadInfo->lastTs
10065 10066
                - (taosRandom() % pThreadInfo->stbInfo->disorderRange + 1);
            generateStbRowData(pThreadInfo->stbInfo, data,
10067 10068
                    MAX_DATA_SIZE,
                    d);
10069
        } else {
10070
            generateStbRowData(pThreadInfo->stbInfo,
10071 10072 10073
                    data,
                    MAX_DATA_SIZE,
                    pThreadInfo->lastTs += 1000);
10074 10075 10076
        }
        pstr += sprintf(pstr, "%s", data);
        pThreadInfo->counter++;
H
Hui Li 已提交
10077

10078
        if (pThreadInfo->counter >= pThreadInfo->stbInfo->insertRows) {
10079 10080 10081 10082 10083 10084 10085 10086 10087 10088 10089
            break;
        }
    }

    if (insert_interval) {
        pThreadInfo->st = taosGetTimestampMs();
    }
    taos_query_a(pThreadInfo->taos, buffer, callBack, pThreadInfo);
    free(buffer);

    taos_free_result(res);
H
Hui Li 已提交
10090 10091
}

10092
static void *asyncWrite(void *sarg) {
10093
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10094
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10095

10096
    setThreadName("asyncWrite");
10097

10098 10099 10100
    pThreadInfo->st = 0;
    pThreadInfo->et = 0;
    pThreadInfo->lastTs = pThreadInfo->start_time;
10101

10102
    int insert_interval =
10103
        stbInfo?stbInfo->insertInterval:g_args.insert_interval;
10104 10105 10106 10107
    if (insert_interval) {
        pThreadInfo->st = taosGetTimestampMs();
    }
    taos_query_a(pThreadInfo->taos, "show databases", callBack, pThreadInfo);
10108

10109
    tsem_wait(&(pThreadInfo->lock_sem));
H
Hui Li 已提交
10110

10111
    return NULL;
H
Hui Li 已提交
10112 10113
}

10114 10115
static int convertHostToServAddr(char *host, uint16_t port, struct sockaddr_in *serv_addr)
{
10116 10117 10118
    uint16_t rest_port = port + TSDB_PORT_HTTP;
    struct hostent *server = gethostbyname(host);
    if ((server == NULL) || (server->h_addr == NULL)) {
10119
        errorPrint2("%s", "no such host");
10120 10121
        return -1;
    }
10122

10123
    debugPrint("h_name: %s\nh_addr=%p\nh_addretype: %s\nh_length: %d\n",
10124 10125 10126 10127 10128
            server->h_name,
            server->h_addr,
            (server->h_addrtype == AF_INET)?"ipv4":"ipv6",
            server->h_length);

10129 10130 10131
    memset(serv_addr, 0, sizeof(struct sockaddr_in));
    serv_addr->sin_family = AF_INET;
    serv_addr->sin_port = htons(rest_port);
10132
#ifdef WINDOWS
10133
    serv_addr->sin_addr.s_addr = inet_addr(host);
10134
#else
10135
    memcpy(&(serv_addr->sin_addr.s_addr), server->h_addr, server->h_length);
10136
#endif
10137
    return 0;
10138 10139
}

10140
static void startMultiThreadInsertData(int threads, char* db_name,
10141
        char* precision, SSuperTable* stbInfo) {
10142

10143 10144 10145 10146 10147 10148 10149 10150 10151
    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 {
10152
            errorPrint2("Not support precision: %s\n", precision);
10153
            exit(EXIT_FAILURE);
10154
        }
10155 10156
    }

10157
    int64_t startTime;
10158 10159
    if (stbInfo) {
        if (0 == strncasecmp(stbInfo->startTimestamp, "now", 3)) {
10160
            startTime = taosGetTimestamp(timePrec);
10161 10162
        } else {
            if (TSDB_CODE_SUCCESS != taosParseTime(
10163
                        stbInfo->startTimestamp,
10164
                        &startTime,
10165
                        strlen(stbInfo->startTimestamp),
10166 10167 10168 10169
                        timePrec, 0)) {
                ERROR_EXIT("failed to parse time!\n");
            }
        }
10170
    } else {
10171
        startTime = DEFAULT_START_TIME;
10172
    }
10173 10174
    debugPrint("%s() LN%d, startTime= %"PRId64"\n",
            __func__, __LINE__, startTime);
10175

10176
    // read sample data from file first
10177
    int ret;
10178
    if (stbInfo) {
10179 10180 10181 10182 10183 10184 10185 10186 10187
        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);
10188 10189
    }

10190 10191 10192 10193
    TAOS* taos0 = taos_connect(
            g_Dbs.host, g_Dbs.user,
            g_Dbs.password, db_name, g_Dbs.port);
    if (NULL == taos0) {
10194
        errorPrint2("%s() LN%d, connect to server fail , reason: %s\n",
10195
                __func__, __LINE__, taos_errstr(NULL));
10196
        exit(EXIT_FAILURE);
10197
    }
10198

10199 10200
    int64_t ntables = 0;
    uint64_t tableFrom;
10201

10202
    if (stbInfo) {
10203 10204
        int64_t limit;
        uint64_t offset;
10205

10206
        if ((NULL != g_args.sqlFile)
10207 10208 10209
                && (stbInfo->childTblExists == TBL_NO_EXISTS)
                && ((stbInfo->childTblOffset != 0)
                    || (stbInfo->childTblLimit >= 0))) {
10210 10211
            printf("WARNING: offset and limit will not be used since the child tables not exists!\n");
        }
10212

10213 10214 10215 10216 10217
        if (stbInfo->childTblExists == TBL_ALREADY_EXISTS) {
            if ((stbInfo->childTblLimit < 0)
                    || ((stbInfo->childTblOffset
                            + stbInfo->childTblLimit)
                        > (stbInfo->childTblCount))) {
10218 10219 10220 10221 10222 10223

                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;
                }
10224 10225
                stbInfo->childTblLimit =
                    stbInfo->childTblCount - stbInfo->childTblOffset;
10226
            }
10227

10228 10229
            offset = stbInfo->childTblOffset;
            limit = stbInfo->childTblLimit;
10230
        } else {
10231
            limit = stbInfo->childTblCount;
10232 10233
            offset = 0;
        }
10234

10235 10236
        ntables = limit;
        tableFrom = offset;
10237

10238 10239 10240
        if ((stbInfo->childTblExists != TBL_NO_EXISTS)
                && ((stbInfo->childTblOffset + stbInfo->childTblLimit)
                    > stbInfo->childTblCount)) {
10241 10242 10243
            printf("WARNING: specified offset + limit > child table count!\n");
            prompt();
        }
10244

10245 10246
        if ((stbInfo->childTblExists != TBL_NO_EXISTS)
                && (0 == stbInfo->childTblLimit)) {
10247 10248 10249
            printf("WARNING: specified limit = 0, which cannot find table name to insert or query! \n");
            prompt();
        }
10250

10251
        stbInfo->childTblName = (char*)calloc(1,
10252
                limit * TSDB_TABLE_NAME_LEN);
10253
        if (stbInfo->childTblName == NULL) {
10254
            taos_close(taos0);
10255
            errorPrint2("%s() LN%d, alloc memory failed!\n", __func__, __LINE__);
10256
            exit(EXIT_FAILURE);
10257
        }
10258

10259 10260 10261
        int64_t childTblCount;
        getChildNameOfSuperTableWithLimitAndOffset(
                taos0,
10262
                db_name, stbInfo->stbName,
10263
                &stbInfo->childTblName, &childTblCount,
10264 10265
                limit,
                offset);
10266
        ntables = childTblCount; // CBD
10267
    } else {
10268
        ntables = g_args.ntables;
10269 10270
        tableFrom = 0;
    }
10271

10272
    taos_close(taos0);
10273

10274 10275 10276 10277 10278
    int64_t a = ntables / threads;
    if (a < 1) {
        threads = ntables;
        a = 1;
    }
10279

10280 10281 10282 10283
    int64_t b = 0;
    if (threads != 0) {
        b = ntables % threads;
    }
10284

10285 10286
    if ((stbInfo)
            && (stbInfo->iface == REST_IFACE)) {
10287 10288
        if (convertHostToServAddr(
                    g_Dbs.host, g_Dbs.port, &(g_Dbs.serv_addr)) != 0) {
10289
            ERROR_EXIT("convert host to server address");
10290 10291
        }
    }
10292

10293 10294
    pthread_t *pids = calloc(1, threads * sizeof(pthread_t));
    threadInfo *infos = calloc(1, threads * sizeof(threadInfo));
10295
    assert(pids != NULL);
10296
    assert(infos != NULL);
10297

10298 10299
    char *stmtBuffer = calloc(1, BUFFER_SIZE);
    assert(stmtBuffer);
10300 10301

#if STMT_BIND_PARAM_BATCH == 1
10302
    uint32_t interlaceRows = 0;
10303 10304 10305
    uint32_t batch;

    if (stbInfo) {
10306
        if (stbInfo->interlaceRows < stbInfo->insertRows)
10307 10308
            interlaceRows = stbInfo->interlaceRows;
    } else {
10309 10310
        if (g_args.interlaceRows < g_args.insertRows)
            interlaceRows = g_args.interlaceRows;
10311 10312 10313 10314 10315 10316 10317 10318 10319 10320 10321
    }

    if (interlaceRows > 0) {
        batch = interlaceRows;
    } else {
        batch = (g_args.reqPerReq>g_args.insertRows)?
            g_args.insertRows:g_args.reqPerReq;
    }

#endif

10322 10323 10324 10325 10326 10327 10328 10329 10330
    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(?",
10331
                    stbInfo->stbName);
10332 10333 10334 10335 10336 10337 10338 10339 10340
            for (int tag = 0; tag < (stbInfo->tagCount - 1);
                    tag ++ ) {
                pstr += sprintf(pstr, ",?");
            }
            pstr += sprintf(pstr, ") VALUES(?");
        } else {
            pstr += sprintf(pstr, "INSERT INTO ? VALUES(?");
        }

10341 10342 10343
        int columnCount = (stbInfo)?
            stbInfo->columnCount:
            g_args.columnCount;
10344 10345 10346 10347 10348 10349 10350

        for (int col = 0; col < columnCount; col ++) {
            pstr += sprintf(pstr, ",?");
        }
        pstr += sprintf(pstr, ")");

        debugPrint("%s() LN%d, stmtBuffer: %s", __func__, __LINE__, stmtBuffer);
10351 10352 10353
#if STMT_BIND_PARAM_BATCH == 1
        parseSamplefileToStmtBatch(stbInfo);
#endif
10354 10355
    }

10356 10357 10358
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
        pThreadInfo->threadID = i;
10359

10360 10361
        tstrncpy(pThreadInfo->db_name, db_name, TSDB_DB_NAME_LEN);
        pThreadInfo->time_precision = timePrec;
10362
        pThreadInfo->stbInfo = stbInfo;
10363

10364
        pThreadInfo->start_time = startTime;
10365 10366
        pThreadInfo->minDelay = UINT64_MAX;

10367 10368
        if ((NULL == stbInfo) ||
                (stbInfo->iface != REST_IFACE)) {
10369 10370 10371 10372 10373
            //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) {
10374
                free(infos);
10375
                errorPrint2(
10376 10377 10378
                        "%s() LN%d, connect to server fail from insert sub thread, reason: %s\n",
                        __func__, __LINE__,
                        taos_errstr(NULL));
10379
                exit(EXIT_FAILURE);
10380
            }
10381

10382
            if ((g_args.iface == STMT_IFACE)
10383 10384
                    || ((stbInfo)
                        && (stbInfo->iface == STMT_IFACE))) {
10385

10386 10387
                pThreadInfo->stmt = taos_stmt_init(pThreadInfo->taos);
                if (NULL == pThreadInfo->stmt) {
10388 10389
                    free(pids);
                    free(infos);
10390
                    errorPrint2(
10391 10392 10393
                            "%s() LN%d, failed init stmt, reason: %s\n",
                            __func__, __LINE__,
                            taos_errstr(NULL));
10394
                    exit(EXIT_FAILURE);
10395
                }
10396

10397
                if (0 != taos_stmt_prepare(pThreadInfo->stmt, stmtBuffer, 0)) {
10398 10399
                    free(pids);
                    free(infos);
10400
                    free(stmtBuffer);
10401
                    errorPrint2("failed to execute taos_stmt_prepare. return 0x%x. reason: %s\n",
10402 10403
                            ret, taos_stmt_errstr(pThreadInfo->stmt));
                    exit(EXIT_FAILURE);
10404
                }
10405
                pThreadInfo->bind_ts = malloc(sizeof(int64_t));
10406 10407

                if (stbInfo) {
10408 10409 10410 10411 10412 10413 10414 10415 10416 10417 10418 10419 10420
#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
10421
                }
10422 10423
            }
        } else {
10424
            pThreadInfo->taos = NULL;
10425
        }
10426

10427 10428
        /*    if ((NULL == stbInfo)
              || (0 == stbInfo->multiThreadWriteOneTbl)) {
10429 10430 10431 10432 10433 10434 10435
              */
        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;
10436
              pThreadInfo->ntables = stbInfo->childTblCount;
10437 10438 10439 10440 10441 10442 10443 10444
              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);
10445
        }
10446 10447
    }

10448 10449
    free(stmtBuffer);

10450
    int64_t start = taosGetTimestampUs();
10451

10452 10453
    for (int i = 0; i < threads; i++) {
        pthread_join(pids[i], NULL);
10454
    }
H
Hui Li 已提交
10455

10456 10457 10458 10459 10460
    uint64_t totalDelay = 0;
    uint64_t maxDelay = 0;
    uint64_t minDelay = UINT64_MAX;
    uint64_t cntDelay = 1;
    double  avgDelay = 0;
10461

10462 10463
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
10464

10465 10466 10467
        tsem_destroy(&(pThreadInfo->lock_sem));
        taos_close(pThreadInfo->taos);

10468 10469 10470
        if (pThreadInfo->stmt) {
            taos_stmt_close(pThreadInfo->stmt);
        }
10471

10472
        tmfree((char *)pThreadInfo->bind_ts);
10473
#if STMT_BIND_PARAM_BATCH == 1
10474 10475 10476 10477
        tmfree((char *)pThreadInfo->bind_ts_array);
        tmfree(pThreadInfo->bindParams);
        tmfree(pThreadInfo->is_null);
#else
10478
        if (pThreadInfo->sampleBindArray) {
10479
            for (int k = 0; k < MAX_SAMPLES; k++) {
10480 10481 10482
                uintptr_t *tmp = (uintptr_t *)(*(uintptr_t *)(
                            pThreadInfo->sampleBindArray
                            + sizeof(uintptr_t *) * k));
10483 10484 10485 10486
                int columnCount = (pThreadInfo->stbInfo)?
                    pThreadInfo->stbInfo->columnCount:
                    g_args.columnCount;
                for (int c = 1; c < columnCount + 1; c++) {
10487 10488 10489 10490 10491 10492 10493 10494
                    TAOS_BIND *bind = (TAOS_BIND *)((char *)tmp + (sizeof(TAOS_BIND) * c));
                    if (bind)
                        tmfree(bind->buffer);
                }
                tmfree((char *)tmp);
            }
            tmfree(pThreadInfo->sampleBindArray);
        }
10495
#endif
10496

10497 10498 10499 10500
        debugPrint("%s() LN%d, [%d] totalInsert=%"PRIu64" totalAffected=%"PRIu64"\n",
                __func__, __LINE__,
                pThreadInfo->threadID, pThreadInfo->totalInsertRows,
                pThreadInfo->totalAffectedRows);
10501 10502 10503
        if (stbInfo) {
            stbInfo->totalAffectedRows += pThreadInfo->totalAffectedRows;
            stbInfo->totalInsertRows += pThreadInfo->totalInsertRows;
10504 10505 10506 10507
        } else {
            g_args.totalAffectedRows += pThreadInfo->totalAffectedRows;
            g_args.totalInsertRows += pThreadInfo->totalInsertRows;
        }
H
Hui Li 已提交
10508

10509 10510 10511 10512 10513
        totalDelay  += pThreadInfo->totalDelay;
        cntDelay   += pThreadInfo->cntDelay;
        if (pThreadInfo->maxDelay > maxDelay) maxDelay = pThreadInfo->maxDelay;
        if (pThreadInfo->minDelay < minDelay) minDelay = pThreadInfo->minDelay;
    }
10514

10515 10516
    if (cntDelay == 0)    cntDelay = 1;
    avgDelay = (double)totalDelay / cntDelay;
10517

10518
    int64_t end = taosGetTimestampUs();
10519
    int64_t t = end - start;
10520
    if (0 == t) t = 1;
10521

10522
    double tInMs = (double) t / 1000000.0;
10523

10524
    if (stbInfo) {
10525
        fprintf(stderr, "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s.%s. %.2f records/second\n\n",
10526 10527
                tInMs, stbInfo->totalInsertRows,
                stbInfo->totalAffectedRows,
10528
                threads, db_name, stbInfo->stbName,
10529
                (double)(stbInfo->totalInsertRows/tInMs));
10530 10531 10532

        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10533
                    "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s.%s. %.2f records/second\n\n",
10534 10535
                    tInMs, stbInfo->totalInsertRows,
                    stbInfo->totalAffectedRows,
10536
                    threads, db_name, stbInfo->stbName,
10537
                    (double)(stbInfo->totalInsertRows/tInMs));
10538 10539
        }
    } else {
10540
        fprintf(stderr, "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s %.2f records/second\n\n",
10541 10542 10543
                tInMs, g_args.totalInsertRows,
                g_args.totalAffectedRows,
                threads, db_name,
10544
                (double)(g_args.totalInsertRows/tInMs));
10545 10546
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10547
                    "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s %.2f records/second\n\n",
10548 10549 10550
                    tInMs, g_args.totalInsertRows,
                    g_args.totalAffectedRows,
                    threads, db_name,
10551
                    (double)(g_args.totalInsertRows/tInMs));
10552
        }
10553
    }
10554

10555 10556 10557 10558 10559 10560 10561 10562 10563 10564 10565 10566
    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);
        }
10567
    }
10568

10569
    //taos_close(taos);
10570

10571 10572
    free(pids);
    free(infos);
H
Hui Li 已提交
10573 10574
}

10575
static void *queryNtableAggrFunc(void *sarg) {
10576 10577
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    TAOS *taos = pThreadInfo->taos;
10578
    setThreadName("queryNtableAggrFunc");
10579 10580 10581
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);

10582
    uint64_t startTime = pThreadInfo->start_time;
10583 10584 10585
    char *tb_prefix = pThreadInfo->tb_prefix;
    FILE *fp = fopen(pThreadInfo->filePath, "a");
    if (NULL == fp) {
10586
        errorPrint2("fopen %s fail, reason:%s.\n", pThreadInfo->filePath, strerror(errno));
10587
        free(command);
10588
        return NULL;
10589
    }
H
hzcheng 已提交
10590

10591
    int64_t insertRows;
10592
    /*  if (pThreadInfo->stbInfo) {
10593
        insertRows = pThreadInfo->stbInfo->insertRows; //  nrecords_per_table;
10594 10595
        } else {
        */
10596
    insertRows = g_args.insertRows;
10597 10598
    //  }

10599
    int64_t ntables = pThreadInfo->ntables; // pThreadInfo->end_table_to - pThreadInfo->start_table_from + 1;
10600
    int64_t totalData = insertRows * ntables;
10601 10602 10603 10604
    bool aggr_func = g_Dbs.aggr_func;

    char **aggreFunc;
    int n;
10605

10606 10607 10608 10609 10610 10611 10612 10613 10614
    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) {
10615 10616 10617 10618 10619 10620 10621 10622
        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;
10623 10624
        for (int64_t i = 0; i < ntables; i++) {
            sprintf(command, "SELECT %s FROM %s%"PRId64" WHERE ts>= %" PRIu64,
10625
                    aggreFunc[j], tb_prefix, i, startTime);
10626

10627 10628 10629
            double t = taosGetTimestampUs();
            debugPrint("%s() LN%d, sql command: %s\n",
                    __func__, __LINE__, command);
10630 10631 10632 10633
            TAOS_RES *pSql = taos_query(taos, command);
            int32_t code = taos_errno(pSql);

            if (code != 0) {
10634
                errorPrint2("Failed to query:%s\n", taos_errstr(pSql));
10635 10636 10637
                taos_free_result(pSql);
                taos_close(taos);
                fclose(fp);
10638
                free(command);
10639 10640
                return NULL;
            }
H
hzcheng 已提交
10641

10642 10643 10644
            while(taos_fetch_row(pSql) != NULL) {
                count++;
            }
H
hzcheng 已提交
10645

10646
            t = taosGetTimestampUs() - t;
10647 10648 10649 10650
            totalT += t;

            taos_free_result(pSql);
        }
H
hzcheng 已提交
10651

10652
        fprintf(fp, "|%10s  |   %"PRId64"   |  %12.2f   |   %10.2f  |\n",
10653 10654 10655
                aggreFunc[j][0] == '*' ? "   *   " : aggreFunc[j], totalData,
                (double)(ntables * insertRows) / totalT, totalT / 1000000);
        printf("select %10s took %.6f second(s)\n", aggreFunc[j], totalT / 1000000);
10656 10657 10658
    }
    fprintf(fp, "\n");
    fclose(fp);
10659
    free(command);
10660
    return NULL;
H
hzcheng 已提交
10661 10662
}

10663
static void *queryStableAggrFunc(void *sarg) {
10664 10665
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    TAOS *taos = pThreadInfo->taos;
10666
    setThreadName("queryStableAggrFunc");
10667 10668 10669
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);

10670 10671 10672
    FILE *fp = fopen(pThreadInfo->filePath, "a");
    if (NULL == fp) {
        printf("fopen %s fail, reason:%s.\n", pThreadInfo->filePath, strerror(errno));
10673
        free(command);
10674 10675
        return NULL;
    }
10676

10677
    int64_t insertRows = pThreadInfo->stbInfo->insertRows;
10678
    int64_t ntables = pThreadInfo->ntables; // pThreadInfo->end_table_to - pThreadInfo->start_table_from + 1;
10679
    int64_t totalData = insertRows * ntables;
10680 10681 10682 10683
    bool aggr_func = g_Dbs.aggr_func;

    char **aggreFunc;
    int n;
H
hzcheng 已提交
10684

10685 10686 10687 10688 10689 10690 10691 10692 10693
    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) {
10694 10695
        printf("\nThe first field is either Binary or Bool. Aggregation functions are not supported.\n");
    }
10696

10697 10698
    printf("%"PRId64" records:\n", totalData);
    fprintf(fp, "Querying On %"PRId64" records:\n", totalData);
H
hzcheng 已提交
10699

10700 10701 10702
    for (int j = 0; j < n; j++) {
        char condition[COND_BUF_LEN] = "\0";
        char tempS[64] = "\0";
H
hzcheng 已提交
10703

10704
        int64_t m = 10 < ntables ? 10 : ntables;
H
hzcheng 已提交
10705

10706 10707
        for (int64_t i = 1; i <= m; i++) {
            if (i == 1) {
10708 10709 10710 10711 10712
                if (g_args.demo_mode) {
                    sprintf(tempS, "groupid = %"PRId64"", i);
                } else {
                    sprintf(tempS, "t0 = %"PRId64"", i);
                }
10713
            } else {
10714 10715 10716 10717 10718
                if (g_args.demo_mode) {
                    sprintf(tempS, " or groupid = %"PRId64" ", i);
                } else {
                    sprintf(tempS, " or t0 = %"PRId64" ", i);
                }
10719 10720
            }
            strncat(condition, tempS, COND_BUF_LEN - 1);
H
hzcheng 已提交
10721

10722
            sprintf(command, "SELECT %s FROM meters WHERE %s", aggreFunc[j], condition);
H
hzcheng 已提交
10723

10724
            printf("Where condition: %s\n", condition);
10725 10726 10727

            debugPrint("%s() LN%d, sql command: %s\n",
                    __func__, __LINE__, command);
10728
            fprintf(fp, "%s\n", command);
H
hzcheng 已提交
10729

10730
            double t = taosGetTimestampUs();
H
hzcheng 已提交
10731

10732 10733
            TAOS_RES *pSql = taos_query(taos, command);
            int32_t code = taos_errno(pSql);
S
Shuaiqiang Chang 已提交
10734

10735
            if (code != 0) {
10736
                errorPrint2("Failed to query:%s\n", taos_errstr(pSql));
10737 10738 10739
                taos_free_result(pSql);
                taos_close(taos);
                fclose(fp);
10740
                free(command);
10741 10742 10743 10744 10745 10746
                return NULL;
            }
            int count = 0;
            while(taos_fetch_row(pSql) != NULL) {
                count++;
            }
10747
            t = taosGetTimestampUs() - t;
H
hzcheng 已提交
10748

10749
            fprintf(fp, "| Speed: %12.2f(per s) | Latency: %.4f(ms) |\n",
10750 10751
                    ntables * insertRows / (t / 1000), t);
            printf("select %10s took %.6f second(s)\n\n", aggreFunc[j], t / 1000000);
H
hzcheng 已提交
10752

10753 10754 10755
            taos_free_result(pSql);
        }
        fprintf(fp, "\n");
H
hzcheng 已提交
10756
    }
10757
    fclose(fp);
10758
    free(command);
10759

10760
    return NULL;
H
hzcheng 已提交
10761 10762
}

10763 10764
static void prompt()
{
10765 10766 10767 10768
    if (!g_args.answer_yes) {
        printf("         Press enter key to continue or Ctrl-C to stop\n\n");
        (void)getchar();
    }
10769
}
H
Hui Li 已提交
10770

10771
static int insertTestProcess() {
10772

10773 10774 10775
    setupForAnsiEscape();
    int ret = printfInsertMeta();
    resetAfterAnsiEscape();
10776

10777 10778
    if (ret == -1)
        exit(EXIT_FAILURE);
10779

10780 10781 10782
    debugPrint("%d result file: %s\n", __LINE__, g_Dbs.resultFile);
    g_fpOfInsertResult = fopen(g_Dbs.resultFile, "a");
    if (NULL == g_fpOfInsertResult) {
10783
        errorPrint("Failed to open %s for save result\n", g_Dbs.resultFile);
10784 10785
        return -1;
    }
10786

10787 10788
    if (g_fpOfInsertResult)
        printfInsertMetaToFile(g_fpOfInsertResult);
10789

10790
    prompt();
10791

10792
    init_rand_data();
10793

10794
    // create database and super tables
10795 10796 10797 10798
    char *cmdBuffer = calloc(1, BUFFER_SIZE);
    assert(cmdBuffer);

    if(createDatabasesAndStables(cmdBuffer) != 0) {
10799 10800
        if (g_fpOfInsertResult)
            fclose(g_fpOfInsertResult);
10801
        free(cmdBuffer);
10802 10803
        return -1;
    }
10804
    free(cmdBuffer);
10805

10806
    // pretreatment
10807 10808 10809 10810 10811
    if (prepareSampleData() != 0) {
        if (g_fpOfInsertResult)
            fclose(g_fpOfInsertResult);
        return -1;
    }
10812

10813 10814
    double start;
    double end;
10815

10816
    if (g_totalChildTables > 0) {
10817 10818
        fprintf(stderr,
                "creating %"PRId64" table(s) with %d thread(s)\n\n",
10819
                g_totalChildTables, g_Dbs.threadCountForCreateTbl);
10820 10821 10822
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
                "creating %"PRId64" table(s) with %d thread(s)\n\n",
10823
                g_totalChildTables, g_Dbs.threadCountForCreateTbl);
10824 10825 10826 10827 10828 10829 10830 10831
        }

        // create child tables
        start = taosGetTimestampMs();
        createChildTables();
        end = taosGetTimestampMs();

        fprintf(stderr,
10832
                "\nSpent %.4f seconds to create %"PRId64" table(s) with %d thread(s), actual %"PRId64" table(s) created\n\n",
10833
                (end - start)/1000.0, g_totalChildTables,
10834
                g_Dbs.threadCountForCreateTbl, g_actualChildTables);
10835 10836
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10837
                "\nSpent %.4f seconds to create %"PRId64" table(s) with %d thread(s), actual %"PRId64" table(s) created\n\n",
10838
                (end - start)/1000.0, g_totalChildTables,
10839
                g_Dbs.threadCountForCreateTbl, g_actualChildTables);
10840
        }
10841
    }
10842

10843 10844 10845 10846 10847 10848
    // 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++) {
10849

10850
                    SSuperTable* stbInfo = &g_Dbs.db[i].superTbls[j];
10851

10852
                    if (stbInfo && (stbInfo->insertRows > 0)) {
10853 10854 10855 10856
                        startMultiThreadInsertData(
                                g_Dbs.threadCount,
                                g_Dbs.db[i].dbName,
                                g_Dbs.db[i].dbCfg.precision,
10857
                                stbInfo);
10858 10859 10860 10861
                    }
                }
            }
        } else {
10862
            startMultiThreadInsertData(
10863 10864 10865 10866
                    g_Dbs.threadCount,
                    g_Dbs.db[i].dbName,
                    g_Dbs.db[i].dbCfg.precision,
                    NULL);
10867
        }
H
Hui Li 已提交
10868
    }
10869
    //end = taosGetTimestampMs();
10870

10871 10872 10873 10874 10875 10876 10877 10878 10879 10880 10881
    //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;
10882 10883
}

10884
static void *specifiedTableQuery(void *sarg) {
10885
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10886

10887
    setThreadName("specTableQuery");
10888

10889 10890 10891 10892 10893 10894 10895 10896
    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) {
10897
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
10898 10899 10900 10901 10902
                    pThreadInfo->threadID, taos_errstr(NULL));
            return NULL;
        } else {
            pThreadInfo->taos = taos;
        }
10903 10904
    }

10905 10906 10907 10908
    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);
10909
        errorPrint("use database %s failed!\n\n",
10910
                g_queryInfo.dbName);
10911 10912
        return NULL;
    }
10913

10914 10915
    uint64_t st = 0;
    uint64_t et = 0;
10916

10917
    uint64_t queryTimes = g_queryInfo.specifiedQueryInfo.queryTimes;
10918

10919 10920 10921
    uint64_t totalQueried = 0;
    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
10922

10923 10924
    if (g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq][0] != '\0') {
        sprintf(pThreadInfo->filePath, "%s-%d",
10925 10926
                g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                pThreadInfo->threadID);
10927 10928
    }

10929 10930 10931 10932 10933 10934 10935
    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();
10936

10937 10938
        selectAndGetResult(pThreadInfo,
                g_queryInfo.specifiedQueryInfo.sql[pThreadInfo->querySeq]);
10939

10940 10941 10942
        et = taosGetTimestampMs();
        printf("=thread[%"PRId64"] use %s complete one sql, Spent %10.3f s\n",
                taosGetSelfPthreadId(), g_queryInfo.queryMode, (et - st)/1000.0);
10943

10944 10945
        totalQueried ++;
        g_queryInfo.specifiedQueryInfo.totalQueried ++;
10946

10947 10948 10949
        uint64_t  currentPrintTime = taosGetTimestampMs();
        uint64_t  endTs = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
10950
            debugPrint("%s() LN%d, endTs=%"PRIu64" ms, startTs=%"PRIu64" ms\n",
10951 10952
                    __func__, __LINE__, endTs, startTs);
            printf("thread[%d] has currently completed queries: %"PRIu64", QPS: %10.6f\n",
10953 10954
                    pThreadInfo->threadID,
                    totalQueried,
10955
                    (double)(totalQueried/((endTs-startTs)/1000.0)));
10956 10957
            lastPrintTime = currentPrintTime;
        }
10958
    }
10959
    return NULL;
10960
}
H
Hui Li 已提交
10961

10962
static void replaceChildTblName(char* inSql, char* outSql, int tblIndex) {
10963
    char sourceString[32] = "xxxx";
10964
    char subTblName[TSDB_TABLE_NAME_LEN];
10965 10966 10967
    sprintf(subTblName, "%s.%s",
            g_queryInfo.dbName,
            g_queryInfo.superQueryInfo.childTblName + tblIndex*TSDB_TABLE_NAME_LEN);
10968

10969
    //printf("inSql: %s\n", inSql);
10970

10971 10972 10973 10974 10975 10976 10977
    char* pos = strstr(inSql, sourceString);
    if (0 == pos) {
        return;
    }

    tstrncpy(outSql, inSql, pos - inSql + 1);
    //printf("1: %s\n", outSql);
10978
    strncat(outSql, subTblName, BUFFER_SIZE - 1);
10979
    //printf("2: %s\n", outSql);
10980
    strncat(outSql, pos+strlen(sourceString), BUFFER_SIZE - 1);
10981
    //printf("3: %s\n", outSql);
H
Hui Li 已提交
10982 10983
}

10984
static void *superTableQuery(void *sarg) {
10985 10986 10987
    char *sqlstr = calloc(1, BUFFER_SIZE);
    assert(sqlstr);

10988
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10989

10990
    setThreadName("superTableQuery");
10991

10992 10993 10994 10995 10996 10997 10998 10999 11000 11001
    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));
11002
            free(sqlstr);
11003 11004 11005 11006
            return NULL;
        } else {
            pThreadInfo->taos = taos;
        }
11007
    }
11008

11009 11010
    uint64_t st = 0;
    uint64_t et = (int64_t)g_queryInfo.superQueryInfo.queryInterval;
11011

11012 11013 11014
    uint64_t queryTimes = g_queryInfo.superQueryInfo.queryTimes;
    uint64_t totalQueried = 0;
    uint64_t  startTs = taosGetTimestampMs();
H
Hui Li 已提交
11015

11016 11017 11018 11019 11020 11021 11022 11023 11024 11025 11026
    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++) {
11027
                memset(sqlstr, 0, BUFFER_SIZE);
11028 11029 11030 11031 11032 11033 11034
                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);
11035

11036 11037
                totalQueried++;
                g_queryInfo.superQueryInfo.totalQueried ++;
11038

11039 11040 11041 11042 11043 11044 11045 11046 11047 11048
                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;
                }
            }
11049
        }
11050 11051 11052 11053 11054 11055
        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 已提交
11056
    }
11057

11058
    free(sqlstr);
11059
    return NULL;
11060 11061
}

11062
static int queryTestProcess() {
11063

11064 11065 11066
    setupForAnsiEscape();
    printfQueryMeta();
    resetAfterAnsiEscape();
11067

11068 11069 11070 11071 11072 11073 11074
    TAOS * taos = NULL;
    taos = taos_connect(g_queryInfo.host,
            g_queryInfo.user,
            g_queryInfo.password,
            NULL,
            g_queryInfo.port);
    if (taos == NULL) {
11075
        errorPrint("Failed to connect to TDengine, reason:%s\n",
11076
                taos_errstr(NULL));
11077
        exit(EXIT_FAILURE);
11078
    }
11079

11080 11081 11082
    if (0 != g_queryInfo.superQueryInfo.sqlCount) {
        getAllChildNameOfSuperTable(taos,
                g_queryInfo.dbName,
11083
                g_queryInfo.superQueryInfo.stbName,
11084 11085 11086
                &g_queryInfo.superQueryInfo.childTblName,
                &g_queryInfo.superQueryInfo.childTblCount);
    }
11087

11088
    prompt();
11089

11090 11091 11092
    if (g_args.debug_print || g_args.verbose_print) {
        printfQuerySystemInfo(taos);
    }
11093

11094 11095 11096
    if (0 == strncasecmp(g_queryInfo.queryMode, "rest", strlen("rest"))) {
        if (convertHostToServAddr(
                    g_queryInfo.host, g_queryInfo.port, &g_queryInfo.serv_addr) != 0)
11097
            ERROR_EXIT("convert host to server address");
11098
    }
11099

11100 11101 11102 11103 11104
    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;
11105

11106
    uint64_t startTs = taosGetTimestampMs();
11107

11108
    if ((nSqlCount > 0) && (nConcurrent > 0)) {
11109

11110 11111
        pids  = calloc(1, nConcurrent * nSqlCount * sizeof(pthread_t));
        infos = calloc(1, nConcurrent * nSqlCount * sizeof(threadInfo));
11112

11113 11114 11115 11116
        if ((NULL == pids) || (NULL == infos)) {
            taos_close(taos);
            ERROR_EXIT("memory allocation failed for create threads\n");
        }
11117

11118 11119 11120 11121 11122 11123 11124 11125 11126
        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)) {

11127 11128
                    char sqlStr[TSDB_DB_NAME_LEN + 5];
                    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11129 11130 11131 11132
                    if (0 != queryDbExec(taos, sqlStr, NO_INSERT_TYPE, false)) {
                        taos_close(taos);
                        free(infos);
                        free(pids);
11133
                        errorPrint2("use database %s failed!\n\n",
11134 11135 11136
                                g_queryInfo.dbName);
                        return -1;
                    }
11137
                }
11138

11139
                pThreadInfo->taos = NULL;// workaround to use separate taos connection;
11140

11141 11142 11143
                pthread_create(pids + seq, NULL, specifiedTableQuery,
                        pThreadInfo);
            }
11144
        }
11145 11146
    } else {
        g_queryInfo.specifiedQueryInfo.concurrent = 0;
11147 11148
    }

11149
    taos_close(taos);
11150

11151 11152 11153 11154 11155 11156 11157
    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));
11158

11159 11160 11161
        if ((NULL == pidsOfSub) || (NULL == infosOfSub)) {
            free(infos);
            free(pids);
11162

11163 11164
            ERROR_EXIT("memory allocation failed for create threads\n");
        }
11165

11166 11167
        int64_t ntables = g_queryInfo.superQueryInfo.childTblCount;
        int threads = g_queryInfo.superQueryInfo.threadCnt;
11168

11169 11170 11171 11172 11173
        int64_t a = ntables / threads;
        if (a < 1) {
            threads = ntables;
            a = 1;
        }
11174

11175 11176 11177 11178
        int64_t b = 0;
        if (threads != 0) {
            b = ntables % threads;
        }
11179

11180 11181 11182 11183
        uint64_t tableFrom = 0;
        for (int i = 0; i < threads; i++) {
            threadInfo *pThreadInfo = infosOfSub + i;
            pThreadInfo->threadID = i;
11184

11185 11186 11187 11188
            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;
11189
            pThreadInfo->taos = NULL; // workaround to use separate taos connection;
11190 11191
            pthread_create(pidsOfSub + i, NULL, superTableQuery, pThreadInfo);
        }
11192

11193 11194 11195 11196
        g_queryInfo.superQueryInfo.threadCnt = threads;
    } else {
        g_queryInfo.superQueryInfo.threadCnt = 0;
    }
11197

11198 11199 11200 11201 11202 11203
    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);
            }
        }
11204
    }
S
Shuaiqiang Chang 已提交
11205

11206 11207
    tmfree((char*)pids);
    tmfree((char*)infos);
11208

11209 11210 11211
    for (int i = 0; i < g_queryInfo.superQueryInfo.threadCnt; i++) {
        pthread_join(pidsOfSub[i], NULL);
    }
H
Hui Li 已提交
11212

11213 11214
    tmfree((char*)pidsOfSub);
    tmfree((char*)infosOfSub);
11215

11216
    //  taos_close(taos);// workaround to use separate taos connection;
11217
    uint64_t endTs = taosGetTimestampMs();
11218

11219 11220
    uint64_t totalQueried = g_queryInfo.specifiedQueryInfo.totalQueried +
        g_queryInfo.superQueryInfo.totalQueried;
11221

11222 11223 11224 11225
    fprintf(stderr, "==== completed total queries: %"PRIu64", the QPS of all threads: %10.3f====\n",
            totalQueried,
            (double)(totalQueried/((endTs-startTs)/1000.0)));
    return 0;
11226 11227
}

11228 11229
static void stable_sub_callback(
        TAOS_SUB* tsub, TAOS_RES *res, void* param, int code) {
11230
    if (res == NULL || taos_errno(res) != 0) {
11231
        errorPrint2("%s() LN%d, failed to subscribe result, code:%d, reason:%s\n",
11232 11233 11234
                __func__, __LINE__, code, taos_errstr(res));
        return;
    }
11235

11236 11237
    if (param)
        fetchResult(res, (threadInfo *)param);
11238
    // tao_unsubscribe() will free result.
11239 11240 11241 11242
}

static void specified_sub_callback(
        TAOS_SUB* tsub, TAOS_RES *res, void* param, int code) {
11243
    if (res == NULL || taos_errno(res) != 0) {
11244
        errorPrint2("%s() LN%d, failed to subscribe result, code:%d, reason:%s\n",
11245 11246 11247
                __func__, __LINE__, code, taos_errstr(res));
        return;
    }
11248

11249 11250
    if (param)
        fetchResult(res, (threadInfo *)param);
11251
    // tao_unsubscribe() will free result.
H
hzcheng 已提交
11252 11253
}

11254
static TAOS_SUB* subscribeImpl(
11255 11256 11257 11258
        QUERY_CLASS class,
        threadInfo *pThreadInfo,
        char *sql, char* topic, bool restart, uint64_t interval)
{
11259 11260 11261 11262 11263 11264 11265 11266 11267 11268 11269 11270 11271 11272 11273 11274 11275 11276 11277 11278 11279 11280 11281 11282
    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) {
11283
        errorPrint2("failed to create subscription. topic:%s, sql:%s\n", topic, sql);
11284 11285 11286 11287
        return NULL;
    }

    return tsub;
11288
}
H
hzcheng 已提交
11289

11290
static void *superSubscribe(void *sarg) {
11291
    threadInfo *pThreadInfo = (threadInfo *)sarg;
11292 11293 11294
    char *subSqlStr = calloc(1, BUFFER_SIZE);
    assert(subSqlStr);

11295 11296 11297
    TAOS_SUB*    tsub[MAX_QUERY_SQL_COUNT] = {0};
    uint64_t tsubSeq;

11298 11299
    setThreadName("superSub");

11300
    if (pThreadInfo->ntables > MAX_QUERY_SQL_COUNT) {
11301
        free(subSqlStr);
11302 11303
        errorPrint("The table number(%"PRId64") of the thread is more than max query sql count: %d\n",
                pThreadInfo->ntables, MAX_QUERY_SQL_COUNT);
11304
        exit(EXIT_FAILURE);
11305
    }
11306

11307
    if (pThreadInfo->taos == NULL) {
11308 11309 11310 11311 11312 11313
        pThreadInfo->taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                g_queryInfo.dbName,
                g_queryInfo.port);
        if (pThreadInfo->taos == NULL) {
11314
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11315
                    pThreadInfo->threadID, taos_errstr(NULL));
11316
            free(subSqlStr);
11317 11318
            return NULL;
        }
11319 11320
    }

11321 11322
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11323 11324
    if (0 != queryDbExec(pThreadInfo->taos, sqlStr, NO_INSERT_TYPE, false)) {
        taos_close(pThreadInfo->taos);
11325
        errorPrint2("use database %s failed!\n\n",
11326
                g_queryInfo.dbName);
11327
        free(subSqlStr);
11328 11329
        return NULL;
    }
11330

11331 11332 11333 11334 11335 11336 11337 11338 11339 11340 11341
    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);
11342
        memset(subSqlStr, 0, BUFFER_SIZE);
11343 11344
        replaceChildTblName(
                g_queryInfo.superQueryInfo.sql[pThreadInfo->querySeq],
11345
                subSqlStr, i);
11346 11347 11348 11349 11350
        if (g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq][0] != 0) {
            sprintf(pThreadInfo->filePath, "%s-%d",
                    g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq],
                    pThreadInfo->threadID);
        }
11351

11352 11353
        verbosePrint("%s() LN%d, [%d] subSqlStr: %s\n",
                __func__, __LINE__, pThreadInfo->threadID, subSqlStr);
11354 11355
        tsub[tsubSeq] = subscribeImpl(
                STABLE_CLASS,
11356
                pThreadInfo, subSqlStr, topic,
11357 11358 11359 11360
                g_queryInfo.superQueryInfo.subscribeRestart,
                g_queryInfo.superQueryInfo.subscribeInterval);
        if (NULL == tsub[tsubSeq]) {
            taos_close(pThreadInfo->taos);
11361
            free(subSqlStr);
11362 11363 11364
            return NULL;
        }
    }
11365

11366 11367 11368 11369 11370 11371
    // 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;
11372

11373
    uint64_t st = 0, et = 0;
11374

11375 11376 11377 11378
    while ((g_queryInfo.superQueryInfo.endAfterConsume == -1)
            || (g_queryInfo.superQueryInfo.endAfterConsume >
                consumed[pThreadInfo->end_table_to
                - pThreadInfo->start_table_from])) {
11379

11380 11381 11382 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
        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,
11418
                            pThreadInfo, subSqlStr, topic,
11419 11420 11421 11422 11423
                            g_queryInfo.superQueryInfo.subscribeRestart,
                            g_queryInfo.superQueryInfo.subscribeInterval
                            );
                    if (NULL == tsub[tsubSeq]) {
                        taos_close(pThreadInfo->taos);
11424
                        free(subSqlStr);
11425 11426 11427 11428 11429 11430 11431 11432 11433 11434 11435
                        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);
11436

11437 11438
    for (uint64_t i = pThreadInfo->start_table_from;
            i <= pThreadInfo->end_table_to; i++) {
11439 11440
        tsubSeq = i - pThreadInfo->start_table_from;
        taos_unsubscribe(tsub[tsubSeq], 0);
H
hzcheng 已提交
11441
    }
11442

11443
    taos_close(pThreadInfo->taos);
11444
    free(subSqlStr);
11445
    return NULL;
H
hzcheng 已提交
11446 11447
}

11448
static void *specifiedSubscribe(void *sarg) {
11449 11450
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    //  TAOS_SUB*  tsub = NULL;
11451

11452
    setThreadName("specSub");
11453

11454
    if (pThreadInfo->taos == NULL) {
11455 11456 11457 11458 11459 11460
        pThreadInfo->taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                g_queryInfo.dbName,
                g_queryInfo.port);
        if (pThreadInfo->taos == NULL) {
11461
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11462 11463 11464
                    pThreadInfo->threadID, taos_errstr(NULL));
            return NULL;
        }
11465 11466
    }

11467 11468
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11469 11470 11471 11472 11473 11474 11475 11476 11477 11478 11479
    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",
11480 11481
                g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                pThreadInfo->threadID);
11482 11483 11484 11485 11486 11487 11488 11489 11490 11491 11492 11493 11494
    }
    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
11495

11496 11497 11498 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
    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 已提交
11552 11553
}

11554
static int subscribeTestProcess() {
11555 11556 11557 11558 11559 11560 11561 11562 11563 11564
    setupForAnsiEscape();
    printfQueryMeta();
    resetAfterAnsiEscape();

    prompt();

    TAOS * taos = NULL;
    taos = taos_connect(g_queryInfo.host,
            g_queryInfo.user,
            g_queryInfo.password,
11565
            g_queryInfo.dbName,
11566 11567
            g_queryInfo.port);
    if (taos == NULL) {
11568
        errorPrint2("Failed to connect to TDengine, reason:%s\n",
11569
                taos_errstr(NULL));
11570
        exit(EXIT_FAILURE);
S
Shengliang Guan 已提交
11571
    }
11572

11573 11574 11575
    if (0 != g_queryInfo.superQueryInfo.sqlCount) {
        getAllChildNameOfSuperTable(taos,
                g_queryInfo.dbName,
11576
                g_queryInfo.superQueryInfo.stbName,
11577 11578
                &g_queryInfo.superQueryInfo.childTblName,
                &g_queryInfo.superQueryInfo.childTblCount);
S
Shengliang Guan 已提交
11579
    }
11580

11581
    taos_close(taos); // workaround to use separate taos connection;
11582 11583 11584 11585 11586 11587 11588 11589 11590

    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) {
11591
        debugPrint("%s() LN%d, specified query sqlCount %d.\n",
11592 11593 11594 11595
                __func__, __LINE__,
                g_queryInfo.specifiedQueryInfo.sqlCount);
    } else {
        if (g_queryInfo.specifiedQueryInfo.concurrent <= 0) {
11596
            errorPrint2("%s() LN%d, specified query sqlCount %d.\n",
11597 11598
                    __func__, __LINE__,
                    g_queryInfo.specifiedQueryInfo.sqlCount);
11599
            exit(EXIT_FAILURE);
11600
        }
11601

11602
        pids  = calloc(
11603
                1,
11604 11605 11606 11607
                g_queryInfo.specifiedQueryInfo.sqlCount *
                g_queryInfo.specifiedQueryInfo.concurrent *
                sizeof(pthread_t));
        infos = calloc(
11608
                1,
11609 11610 11611 11612
                g_queryInfo.specifiedQueryInfo.sqlCount *
                g_queryInfo.specifiedQueryInfo.concurrent *
                sizeof(threadInfo));
        if ((NULL == pids) || (NULL == infos)) {
11613
            errorPrint2("%s() LN%d, malloc failed for create threads\n", __func__, __LINE__);
11614
            exit(EXIT_FAILURE);
11615
        }
11616

11617 11618 11619 11620 11621 11622
        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;
11623
                pThreadInfo->taos = NULL;  // workaround to use separate taos connection;
11624 11625
                pthread_create(pids + seq, NULL, specifiedSubscribe, pThreadInfo);
            }
11626
        }
11627
    }
11628

11629 11630 11631 11632 11633 11634 11635 11636 11637 11638 11639 11640 11641 11642 11643 11644 11645 11646 11647
    //==== 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)) {
11648
                errorPrint2("%s() LN%d, malloc failed for create threads\n",
11649 11650
                        __func__, __LINE__);
                // taos_close(taos);
11651
                exit(EXIT_FAILURE);
11652
            }
11653

11654 11655
            int64_t ntables = g_queryInfo.superQueryInfo.childTblCount;
            int threads = g_queryInfo.superQueryInfo.threadCnt;
11656

11657 11658 11659 11660
            int64_t a = ntables / threads;
            if (a < 1) {
                threads = ntables;
                a = 1;
11661
            }
11662

11663 11664 11665 11666 11667 11668 11669 11670 11671 11672 11673 11674 11675 11676 11677 11678 11679
            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;
11680
                    pThreadInfo->taos = NULL; // workaround to use separate taos connection;
11681 11682 11683 11684
                    pthread_create(pidsOfStable + seq,
                            NULL, superSubscribe, pThreadInfo);
                }
            }
11685

11686 11687 11688 11689 11690 11691 11692
            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);
                }
11693 11694
            }
        }
11695
    }
11696

11697 11698 11699 11700 11701
    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);
        }
11702
    }
H
hzcheng 已提交
11703

11704 11705
    tmfree((char*)pids);
    tmfree((char*)infos);
H
hzcheng 已提交
11706

11707 11708 11709 11710
    tmfree((char*)pidsOfStable);
    tmfree((char*)infosOfStable);
    //   taos_close(taos);
    return 0;
H
hzcheng 已提交
11711 11712
}

11713
static void initOfInsertMeta() {
11714
    memset(&g_Dbs, 0, sizeof(SDbs));
11715

11716 11717 11718 11719
    // 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);
11720
    tstrncpy(g_Dbs.password, TSDB_DEFAULT_PASS, SHELL_MAX_PASSWORD_LEN);
11721
    g_Dbs.threadCount = 2;
11722

11723
    g_Dbs.use_metric = g_args.use_metric;
L
Liu Tao 已提交
11724 11725
}

11726
static void initOfQueryMeta() {
11727
    memset(&g_queryInfo, 0, sizeof(SQueryMetaInfo));
11728

11729 11730 11731 11732
    // 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);
11733
    tstrncpy(g_queryInfo.password, TSDB_DEFAULT_PASS, SHELL_MAX_PASSWORD_LEN);
L
Liu Tao 已提交
11734 11735
}

11736
static void setParaFromArg() {
11737 11738 11739 11740 11741
    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);
    }
11742

11743 11744 11745
    if (g_args.user) {
        tstrncpy(g_Dbs.user, g_args.user, MAX_USERNAME_SIZE);
    }
11746

11747
    tstrncpy(g_Dbs.password, g_args.password, SHELL_MAX_PASSWORD_LEN);
L
Liu Tao 已提交
11748

11749 11750 11751
    if (g_args.port) {
        g_Dbs.port = g_args.port;
    }
11752

11753 11754
    g_Dbs.threadCount = g_args.nthreads;
    g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11755

11756 11757
    g_Dbs.dbCount = 1;
    g_Dbs.db[0].drop = true;
L
Liu Tao 已提交
11758

11759 11760
    tstrncpy(g_Dbs.db[0].dbName, g_args.database, TSDB_DB_NAME_LEN);
    g_Dbs.db[0].dbCfg.replica = g_args.replica;
11761
    tstrncpy(g_Dbs.db[0].dbCfg.precision, "ms", SMALL_BUFF_LEN);
11762

11763
    tstrncpy(g_Dbs.resultFile, g_args.output_file, MAX_FILE_NAME_LEN);
11764

11765
    g_Dbs.use_metric = g_args.use_metric;
L
Liu Tao 已提交
11766

11767
    g_Dbs.aggr_func = g_args.aggr_func;
11768

11769
    char dataString[TSDB_MAX_BYTES_PER_ROW];
11770 11771
    char *data_type = g_args.data_type;
    char **dataType = g_args.dataType;
L
Liu Tao 已提交
11772

11773
    memset(dataString, 0, TSDB_MAX_BYTES_PER_ROW);
H
hzcheng 已提交
11774

11775 11776 11777
    if ((data_type[0] == TSDB_DATA_TYPE_BINARY)
            || (data_type[0] == TSDB_DATA_TYPE_BOOL)
            || (data_type[0] == TSDB_DATA_TYPE_NCHAR)) {
11778
        g_Dbs.aggr_func = false;
11779
    }
11780

11781 11782
    if (g_args.use_metric) {
        g_Dbs.db[0].superTblCount = 1;
11783 11784 11785 11786
        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;
11787
        g_Dbs.asyncMode = g_args.async_mode;
11788

11789 11790 11791 11792 11793
        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,
11794 11795
                g_args.tb_prefix, TBNAME_PREFIX_LEN);
        tstrncpy(g_Dbs.db[0].superTbls[0].dataSource, "rand", SMALL_BUFF_LEN);
11796

11797 11798 11799 11800 11801 11802 11803
        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);
11804
        g_Dbs.db[0].superTbls[0].timeStampStep = g_args.timestamp_step;
11805

11806
        g_Dbs.db[0].superTbls[0].insertRows = g_args.insertRows;
11807
        g_Dbs.db[0].superTbls[0].maxSqlLen = g_args.max_sql_len;
11808

11809 11810
        g_Dbs.db[0].superTbls[0].columnCount = 0;
        for (int i = 0; i < MAX_NUM_COLUMNS; i++) {
11811
            if (data_type[i] == TSDB_DATA_TYPE_NULL) {
11812 11813
                break;
            }
11814

11815
            g_Dbs.db[0].superTbls[0].columns[i].data_type = data_type[i];
11816
            tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
11817
                    dataType[i], min(DATATYPE_BUFF_LEN, strlen(dataType[i]) + 1));
11818
            g_Dbs.db[0].superTbls[0].columns[i].dataLen = g_args.binwidth;
11819 11820 11821
            g_Dbs.db[0].superTbls[0].columnCount++;
        }

11822 11823
        if (g_Dbs.db[0].superTbls[0].columnCount > g_args.columnCount) {
            g_Dbs.db[0].superTbls[0].columnCount = g_args.columnCount;
11824 11825
        } else {
            for (int i = g_Dbs.db[0].superTbls[0].columnCount;
11826 11827
                    i < g_args.columnCount; i++) {
                g_Dbs.db[0].superTbls[0].columns[i].data_type = TSDB_DATA_TYPE_INT;
11828
                tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
11829
                        "INT", min(DATATYPE_BUFF_LEN, strlen("INT") + 1));
11830 11831 11832 11833 11834 11835
                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,
11836
                "INT", min(DATATYPE_BUFF_LEN, strlen("INT") + 1));
11837 11838 11839
        g_Dbs.db[0].superTbls[0].tags[0].dataLen = 0;

        tstrncpy(g_Dbs.db[0].superTbls[0].tags[1].dataType,
11840
                "BINARY", min(DATATYPE_BUFF_LEN, strlen("BINARY") + 1));
11841
        g_Dbs.db[0].superTbls[0].tags[1].dataLen = g_args.binwidth;
11842 11843
        g_Dbs.db[0].superTbls[0].tagCount = 2;
    } else {
11844
        g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11845 11846
        g_Dbs.db[0].superTbls[0].tagCount = 0;
    }
11847 11848 11849 11850
}

/* Function to do regular expression check */
static int regexMatch(const char *s, const char *reg, int cflags) {
11851 11852 11853 11854 11855
    regex_t regex;
    char    msgbuf[100] = {0};

    /* Compile regular expression */
    if (regcomp(&regex, reg, cflags) != 0) {
11856
        ERROR_EXIT("Fail to compile regex\n");
11857 11858 11859 11860 11861 11862 11863 11864 11865 11866 11867 11868 11869
    }

    /* 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);
11870 11871
        printf("Regex match failed: %s\n", msgbuf);
        exit(EXIT_FAILURE);
11872 11873
    }

11874 11875 11876 11877
    return 0;
}

static int isCommentLine(char *line) {
11878
    if (line == NULL) return 1;
11879

11880
    return regexMatch(line, "^\\s*#.*", REG_EXTENDED);
11881 11882
}

11883
static void querySqlFile(TAOS* taos, char* sqlFile)
11884
{
11885 11886 11887 11888 11889
    FILE *fp = fopen(sqlFile, "r");
    if (fp == NULL) {
        printf("failed to open file %s, reason:%s\n", sqlFile, strerror(errno));
        return;
    }
11890

11891 11892 11893 11894 11895
    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;
11896

11897
    double t = taosGetTimestampMs();
11898

11899 11900 11901
    while((read_len = tgetline(&line, &line_len, fp)) != -1) {
        if (read_len >= TSDB_MAX_BYTES_PER_ROW) continue;
        line[--read_len] = '\0';
11902

11903 11904 11905
        if (read_len == 0 || isCommentLine(line)) {  // line starts with #
            continue;
        }
11906

11907 11908 11909 11910 11911 11912
        if (line[read_len - 1] == '\\') {
            line[read_len - 1] = ' ';
            memcpy(cmd + cmd_len, line, read_len);
            cmd_len += read_len;
            continue;
        }
11913

11914 11915
        memcpy(cmd + cmd_len, line, read_len);
        if (0 != queryDbExec(taos, cmd, NO_INSERT_TYPE, false)) {
11916
            errorPrint2("%s() LN%d, queryDbExec %s failed!\n",
11917 11918 11919 11920 11921 11922 11923 11924
                    __func__, __LINE__, cmd);
            tmfree(cmd);
            tmfree(line);
            tmfclose(fp);
            return;
        }
        memset(cmd, 0, TSDB_MAX_BYTES_PER_ROW);
        cmd_len = 0;
11925
    }
H
hzcheng 已提交
11926

11927 11928
    t = taosGetTimestampMs() - t;
    printf("run %s took %.6f second(s)\n\n", sqlFile, t);
11929

11930 11931 11932 11933
    tmfree(cmd);
    tmfree(line);
    tmfclose(fp);
    return;
H
hzcheng 已提交
11934 11935
}

11936
static void testMetaFile() {
11937
    if (INSERT_TEST == g_args.test_mode) {
11938 11939
        if (g_Dbs.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_Dbs.cfgDir);
11940

11941
        insertTestProcess();
11942

11943
    } else if (QUERY_TEST == g_args.test_mode) {
11944 11945
        if (g_queryInfo.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_queryInfo.cfgDir);
11946

11947
        queryTestProcess();
11948

11949
    } else if (SUBSCRIBE_TEST == g_args.test_mode) {
11950 11951
        if (g_queryInfo.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_queryInfo.cfgDir);
11952

11953
        subscribeTestProcess();
11954

11955
    }  else {
11956
        ;
11957
    }
11958
}
11959

11960
static void queryAggrFunc() {
11961 11962 11963 11964 11965
    // query data

    pthread_t read_id;
    threadInfo *pThreadInfo = calloc(1, sizeof(threadInfo));
    assert(pThreadInfo);
11966
    pThreadInfo->start_time = DEFAULT_START_TIME;  // 2017-07-14 10:40:00.000
11967 11968 11969 11970 11971
    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;
11972
        pThreadInfo->stbInfo = &g_Dbs.db[0].superTbls[0];
11973
        tstrncpy(pThreadInfo->tb_prefix,
11974
                g_Dbs.db[0].superTbls[0].childTblPrefix, TBNAME_PREFIX_LEN);
11975
    } else {
11976 11977
        pThreadInfo->ntables = g_args.ntables;
        pThreadInfo->end_table_to = g_args.ntables -1;
11978 11979 11980 11981 11982 11983 11984 11985 11986 11987
        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) {
11988
        free(pThreadInfo);
11989
        errorPrint2("Failed to connect to TDengine, reason:%s\n",
11990
                taos_errstr(NULL));
11991
        exit(EXIT_FAILURE);
11992 11993 11994 11995 11996
    }

    tstrncpy(pThreadInfo->filePath, g_Dbs.resultFile, MAX_FILE_NAME_LEN);

    if (!g_Dbs.use_metric) {
11997
        pthread_create(&read_id, NULL, queryNtableAggrFunc, pThreadInfo);
11998
    } else {
11999
        pthread_create(&read_id, NULL, queryStableAggrFunc, pThreadInfo);
12000 12001 12002
    }
    pthread_join(read_id, NULL);
    taos_close(pThreadInfo->taos);
12003
    free(pThreadInfo);
12004 12005
}

12006 12007
static void testCmdLine() {

12008 12009 12010
    if (strlen(configDir)) {
        wordexp_t full_path;
        if (wordexp(configDir, &full_path, 0) != 0) {
12011
            errorPrint("Invalid path %s\n", configDir);
12012 12013 12014 12015
            return;
        }
        taos_options(TSDB_OPTION_CONFIGDIR, full_path.we_wordv[0]);
        wordfree(&full_path);
12016 12017
    }

12018 12019
    g_args.test_mode = INSERT_TEST;
    insertTestProcess();
12020

12021 12022 12023
    if (g_Dbs.aggr_func) {
        queryAggrFunc();
    }
12024 12025
}

12026
int main(int argc, char *argv[]) {
12027
    parse_args(argc, argv, &g_args);
12028

12029
    debugPrint("meta file: %s\n", g_args.metaFile);
12030

12031
    if (g_args.metaFile) {
12032
        g_totalChildTables = 0;
12033 12034
        initOfInsertMeta();
        initOfQueryMeta();
12035

12036 12037 12038 12039
        if (false == getInfoFromJsonFile(g_args.metaFile)) {
            printf("Failed to read %s\n", g_args.metaFile);
            return 1;
        }
12040

12041
        testMetaFile();
12042
    } else {
12043 12044 12045 12046 12047 12048 12049 12050 12051 12052 12053 12054
        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);
12055

12056 12057 12058
        } else {
            testCmdLine();
        }
12059

12060 12061 12062 12063 12064
        if (g_dupstr)
            free(g_dupstr);
    }

    return 0;
H
hzcheng 已提交
12065
}
12066