taosdemo.c 454.9 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
    char *   database;
    int      replica;
    char *   tb_prefix;
228
    bool     escapeChar;
229 230 231
    char *   sqlFile;
    bool     use_metric;
    bool     drop_database;
232
    bool     aggr_func;
233 234 235 236 237 238
    bool     answer_yes;
    bool     debug_print;
    bool     verbose_print;
    bool     performance_print;
    char *   output_file;
    bool     async_mode;
239 240
    char     data_type[MAX_NUM_COLUMNS+1];
    char     *dataType[MAX_NUM_COLUMNS+1];
241
    uint32_t binwidth;
242 243 244
    uint32_t columnCount;
    uint64_t lenOfOneRow;
    uint32_t nthreads;
245
    uint64_t insert_interval;
246
    uint64_t timestamp_step;
247
    int64_t  query_times;
248
    uint32_t interlaceRows;
249
    uint32_t reqPerReq;                  // num_of_records_per_req
250
    uint64_t max_sql_len;
251 252
    int64_t  ntables;
    int64_t  insertRows;
253 254
    int      abort;
    uint32_t disorderRatio;               // 0: no disorder, >0: x%
255
    int      disorderRange;               // ms, us or ns. according to database precision
256 257 258 259
    uint32_t method_of_delete;
    uint64_t totalInsertRows;
    uint64_t totalAffectedRows;
    bool     demo_mode;                  // use default column name and semi-random data
260 261 262
} SArguments;

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

typedef struct SSuperTable_S {
271
    char         stbName[TSDB_TABLE_NAME_LEN];
272 273
    char         dataSource[SMALL_BUFF_LEN];  // rand_gen or sample
    char         childTblPrefix[TBNAME_PREFIX_LEN];
274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291
    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];
292
    char         sampleFormat[SMALL_BUFF_LEN];  // csv, json
293 294 295 296 297 298 299 300 301
    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;
302
    bool         escapeChar;
303 304 305 306 307
    char*        colsOfCreateChildTable;
    uint64_t     lenOfOneRow;
    uint64_t     lenOfTagOfOneRow;

    char*        sampleDataBuf;
308
    bool         useSampleTs;
309 310 311 312 313 314

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

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

typedef struct {
325 326 327 328 329 330 331
    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;
332
    char     keeplist[64];
333 334 335 336 337 338 339 340
    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;
341
    char     precision[SMALL_BUFF_LEN];   // time resolution
342 343
    int8_t   update;
    char     status[16];
344 345
} SDbInfo;

346
typedef struct SDbCfg_S {
347 348 349 350 351 352 353 354 355 356 357 358 359 360
    //  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;
361
    char      precision[SMALL_BUFF_LEN];
362 363 364
} SDbCfg;

typedef struct SDataBase_S {
365 366 367 368 369
    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];
370 371 372
} SDataBase;

typedef struct SDbs_S {
373 374 375 376 377 378
    char        cfgDir[MAX_FILE_NAME_LEN];
    char        host[MAX_HOSTNAME_SIZE];
    struct      sockaddr_in serv_addr;

    uint16_t    port;
    char        user[MAX_USERNAME_SIZE];
379
    char        password[SHELL_MAX_PASSWORD_LEN];
380 381
    char        resultFile[MAX_FILE_NAME_LEN];
    bool        use_metric;
382
    bool        aggr_func;
383 384 385
    bool        asyncMode;

    uint32_t    threadCount;
386
    uint32_t    threadCountForCreateTbl;
387 388 389 390 391 392
    uint32_t    dbCount;
    SDataBase   db[MAX_DB_COUNT];

    // statistics
    uint64_t    totalInsertRows;
    uint64_t    totalAffectedRows;
393

394 395
} SDbs;

396
typedef struct SpecifiedQueryInfo_S {
397
    uint64_t     queryInterval;  // 0: unlimited  > 0   loop/s
398 399 400 401 402 403 404
    uint32_t     concurrent;
    int          sqlCount;
    uint32_t     asyncMode; // 0: sync, 1: async
    uint64_t     subscribeInterval; // ms
    uint64_t     queryTimes;
    bool         subscribeRestart;
    int          subscribeKeepProgress;
405
    char         sql[MAX_QUERY_SQL_COUNT][BUFFER_SIZE+1];
406 407 408 409 410 411 412 413
    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;
414
} SpecifiedQueryInfo;
415

416
typedef struct SuperQueryInfo_S {
417
    char         stbName[TSDB_TABLE_NAME_LEN];
418
    uint64_t     queryInterval;  // 0: unlimited  > 0   loop/s
419 420 421 422 423 424 425
    uint32_t     threadCnt;
    uint32_t     asyncMode; // 0: sync, 1: async
    uint64_t     subscribeInterval; // ms
    bool         subscribeRestart;
    int          subscribeKeepProgress;
    uint64_t     queryTimes;
    int64_t      childTblCount;
426
    char         childTblPrefix[TBNAME_PREFIX_LEN];    // 20 characters reserved for seq
427
    int          sqlCount;
428
    char         sql[MAX_QUERY_SQL_COUNT][BUFFER_SIZE+1];
429 430 431 432 433 434 435
    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;
436
} SuperQueryInfo;
437 438

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

    SpecifiedQueryInfo  specifiedQueryInfo;
    SuperQueryInfo      superQueryInfo;
    uint64_t     totalQueried;
451 452 453
} SQueryMetaInfo;

typedef struct SThreadInfo_S {
454 455
    TAOS *    taos;
    TAOS_STMT *stmt;
456
    int64_t     *bind_ts;
457 458 459 460 461 462 463 464 465

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

466 467 468 469 470 471 472 473 474
    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;
475
    int64_t   tables_created;
476 477 478 479
    uint64_t  data_of_rate;
    int64_t   start_time;
    char*     cols;
    bool      use_metric;
480
    SSuperTable* stbInfo;
481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
    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;
506

507
    int       sockfd;
508 509
} threadInfo;

510
#ifdef WINDOWS
511 512
#define _CRT_RAND_S

513
#include <windows.h>
514 515 516 517 518
#include <winsock2.h>

typedef unsigned __int32 uint32_t;

#pragma comment ( lib, "ws2_32.lib" )
519 520
// Some old MinGW/CYGWIN distributions don't define this:
#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
521
#define ENABLE_VIRTUAL_TERMINAL_PROCESSING  0x0004
522
#endif // ENABLE_VIRTUAL_TERMINAL_PROCESSING
523 524 525 526

static HANDLE g_stdoutHandle;
static DWORD g_consoleMode;

527
static void setupForAnsiEscape(void) {
528 529
    DWORD mode = 0;
    g_stdoutHandle = GetStdHandle(STD_OUTPUT_HANDLE);
530

531 532 533
    if(g_stdoutHandle == INVALID_HANDLE_VALUE) {
        exit(GetLastError());
    }
534

535 536 537
    if(!GetConsoleMode(g_stdoutHandle, &mode)) {
        exit(GetLastError());
    }
538

539
    g_consoleMode = mode;
540

541 542
    // Enable ANSI escape codes
    mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
543

544 545 546
    if(!SetConsoleMode(g_stdoutHandle, mode)) {
        exit(GetLastError());
    }
547 548
}

549
static void resetAfterAnsiEscape(void) {
550 551
    // Reset colors
    printf("\x1b[0m");
552

553 554 555 556
    // Reset console mode
    if(!SetConsoleMode(g_stdoutHandle, g_consoleMode)) {
        exit(GetLastError());
    }
557
}
558

559
static int taosRandom()
560 561 562 563 564 565
{
    int number;
    rand_s(&number);

    return number;
}
566
#else   // Not windows
567
static void setupForAnsiEscape(void) {}
568

569
static void resetAfterAnsiEscape(void) {
570 571
    // Reset colors
    printf("\x1b[0m");
572
}
573

574 575
#include <time.h>

576
static int taosRandom()
577
{
578
    return rand();
579 580
}

581
#endif // ifdef Windows
582

583
static void prompt();
584
static int createDatabasesAndStables();
585
static void createChildTables();
586
static int queryDbExec(TAOS *taos, char *command, QUERY_TYPE type, bool quiet);
587
static int postProceSql(char *host, uint16_t port, char* sqlstr, threadInfo *pThreadInfo);
588 589
static int64_t getTSRandTail(int64_t timeStampStep, int32_t seq,
        int disorderRatio, int disorderRange);
590 591
static bool getInfoFromJsonFile(char* file);
static void init_rand_data();
592
static int regexMatch(const char *s, const char *reg, int cflags);
593 594 595

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

596
int32_t  g_randint[MAX_PREPARED_RAND];
597
uint32_t  g_randuint[MAX_PREPARED_RAND];
598
int64_t  g_randbigint[MAX_PREPARED_RAND];
599
uint64_t  g_randubigint[MAX_PREPARED_RAND];
600 601 602 603 604
float    g_randfloat[MAX_PREPARED_RAND];
double   g_randdouble[MAX_PREPARED_RAND];

char    *g_randbool_buff = NULL;
char    *g_randint_buff = NULL;
605
char    *g_randuint_buff = NULL;
606 607
char    *g_rand_voltage_buff = NULL;
char    *g_randbigint_buff = NULL;
608
char    *g_randubigint_buff = NULL;
609
char    *g_randsmallint_buff = NULL;
610
char    *g_randusmallint_buff = NULL;
611
char    *g_randtinyint_buff = NULL;
612
char    *g_randutinyint_buff = NULL;
613 614 615 616 617
char    *g_randfloat_buff = NULL;
char    *g_rand_current_buff = NULL;
char    *g_rand_phase_buff = NULL;
char    *g_randdouble_buff = NULL;

618 619 620
char    *g_aggreFuncDemo[] = {"*", "count(*)", "avg(current)", "sum(current)",
    "max(current)", "min(current)", "first(current)", "last(current)"};

621 622
char    *g_aggreFunc[] = {"*", "count(*)", "avg(C0)", "sum(C0)",
    "max(C0)", "min(C0)", "first(C0)", "last(C0)"};
623

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

static SDbs            g_Dbs;
684
static int64_t         g_totalChildTables = DEFAULT_CHILDTABLES;
685
static int64_t         g_actualChildTables = 0;
686 687 688
static SQueryMetaInfo  g_queryInfo;
static FILE *          g_fpOfInsertResult = NULL;

689 690 691 692
#if _MSC_VER <= 1900
#define __func__ __FUNCTION__
#endif

693
#define debugPrint(fmt, ...) \
694
    do { if (g_args.debug_print || g_args.verbose_print) \
695
        fprintf(stderr, "DEBG: "fmt, __VA_ARGS__); } while(0)
696

697
#define verbosePrint(fmt, ...) \
698 699
    do { if (g_args.verbose_print) \
        fprintf(stderr, "VERB: "fmt, __VA_ARGS__); } while(0)
700

701 702
#define performancePrint(fmt, ...) \
    do { if (g_args.performance_print) \
703
        fprintf(stderr, "PERF: "fmt, __VA_ARGS__); } while(0)
704

705
#define errorPrint(fmt, ...) \
706 707 708 709 710 711 712
    do {\
        fprintf(stderr, " \033[31m");\
        fprintf(stderr, "ERROR: "fmt, __VA_ARGS__);\
        fprintf(stderr, " \033[0m");\
    } while(0)

#define errorPrint2(fmt, ...) \
713 714 715 716 717 718 719 720 721 722 723 724 725
    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");\
726
        errorPrint(fmt, __VA_ARGS__);\
727
    } while(0)
728

729 730 731
// for strncpy buffer overflow
#define min(a, b) (((a) < (b)) ? (a) : (b))

732

733
///////////////////////////////////////////////////
H
hzcheng 已提交
734

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

737 738 739 740 741 742 743 744
#ifndef TAOSDEMO_COMMIT_SHA1
#define TAOSDEMO_COMMIT_SHA1 "unknown"
#endif

#ifndef TD_VERNUMBER
#define TD_VERNUMBER    "unknown"
#endif

745
#ifndef TAOSDEMO_STATUS
746 747 748 749 750 751 752 753 754
#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) {
755
        printf("taosdemo version %s-%s\n",
756 757
                tdengine_ver, taosdemo_ver);
    } else {
758
        printf("taosdemo version %s-%s, status:%s\n",
759 760 761 762
                tdengine_ver, taosdemo_ver, taosdemo_status);
    }
}

763
static void printHelp() {
764 765 766
    char indent[10] = "  ";
    printf("%s\n\n", "Usage: taosdemo [OPTION...]");
    printf("%s%s%s%s\n", indent, "-f, --file=FILE", "\t\t",
767
            "The meta file to the execution procedure.");
768 769
    printf("%s%s%s%s\n", indent, "-u, --user=USER", "\t\t",
            "The user name to use when connecting to the server.");
770
#ifdef _TD_POWER_
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 'powerdb'");
773
    printf("%s%s%s%s\n", indent, "-c, --config-dir=CONFIG_DIR", "\t",
774
            "Configuration directory. By default is '/etc/power/'.");
775
#elif (_TD_TQ_ == true)
776
    printf("%s%s%s%s\n", indent, "-p, --password", "\t\t",
777
            "The password to use when connecting to the server. By default is 'tqueue'");
778
    printf("%s%s%s%s\n", indent, "-c, --config-dir=CONFIG_DIR", "\t",
779
            "Configuration directory. By default is '/etc/tq/'.");
780 781 782 783 784
#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/'.");
785
#else
786 787 788 789
    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.");
790
#endif
791 792 793 794 795
    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",
796
            "The interface (taosc, rest, and stmt) taosdemo uses. By default use 'taosc'.");
797
    printf("%s%s%s%s\n", indent, "-d, --database=DATABASE", "\t",
798
            "Destination database. By default is 'test'.");
799
    printf("%s%s%s%s\n", indent, "-a, --replica=REPLICA", "\t\t",
800
            "Set the replica parameters of the database, By default use 1, min: 1, max: 3.");
801
    printf("%s%s%s%s\n", indent, "-m, --table-prefix=TABLEPREFIX", "\t",
802
            "Table prefix name. By default use 'd'.");
803 804
    printf("%s%s%s%s\n", indent, "-E, --escape-character", "\t",
            "Use escape character for Both Stable and normmal table name");
805 806
    printf("%s%s%s%s\n", indent, "-s, --sql-file=FILE", "\t\t",
            "The select sql file.");
807 808
    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",
809
            "Direct output to the named file. By default use './output.txt'.");
810
    printf("%s%s%s%s\n", indent, "-q, --query-mode=MODE", "\t\t",
811
            "Query mode -- 0: SYNC, 1: ASYNC. By default use SYNC.");
812
    printf("%s%s%s%s\n", indent, "-b, --data-type=DATATYPE", "\t",
813
            "The data_type of columns, By default use: FLOAT,INT,FLOAT. NCHAR and BINARY can also use custom length. Eg: NCHAR(16),BINARY(8)");
814
    printf("%s%s%s%s%d\n", indent, "-w, --binwidth=WIDTH", "\t\t",
815
            "The width of data_type 'BINARY' or 'NCHAR'. By default use ",
816 817
            g_args.binwidth);
    printf("%s%s%s%s%d%s%d\n", indent, "-l, --columns=COLUMNS", "\t\t",
818
            "The number of columns per record. Demo mode by default is ",
819
            DEFAULT_DATATYPE_NUM,
820
            " (float, int, float). Max values is ",
821 822
            MAX_NUM_COLUMNS);
    printf("%s%s%s%s\n", indent, indent, indent,
823
            "\t\t\t\tAll of the new column(s) type is INT. If use -b to specify column type, -l will be ignored.");
824 825
    printf("%s%s%s%s%d.\n", indent, "-T, --threads=NUMBER", "\t\t",
            "The number of threads. By default use ", DEFAULT_NTHREADS);
826
    printf("%s%s%s%s\n", indent, "-i, --insert-interval=NUMBER", "\t",
827
            "The sleep time (ms) between insertion. By default is 0.");
828
    printf("%s%s%s%s%d.\n", indent, "-S, --time-step=TIME_STEP", "\t",
829
            "The timestamp step between insertion. By default is ",
830
            DEFAULT_TIMESTAMP_STEP);
831
    printf("%s%s%s%s%d.\n", indent, "-B, --interlace-rows=NUMBER", "\t",
832
            "The interlace rows of insertion. By default is ",
833 834
            DEFAULT_INTERLACE_ROWS);
    printf("%s%s%s%s\n", indent, "-r, --rec-per-req=NUMBER", "\t",
835
            "The number of records per request. By default is 30000.");
836
    printf("%s%s%s%s\n", indent, "-t, --tables=NUMBER", "\t\t",
837
            "The number of tables. By default is 10000.");
838
    printf("%s%s%s%s\n", indent, "-n, --records=NUMBER", "\t\t",
839
            "The number of records per table. By default is 10000.");
840
    printf("%s%s%s%s\n", indent, "-M, --random", "\t\t\t",
841
            "The value of records generated are totally random.");
842 843 844 845
    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.");
846
    printf("%s%s%s%s\n", indent, "-O, --disorder=NUMBER", "\t\t",
847
            "Insert order mode--0: In order, 1 ~ 50: disorder ratio. By default is in order.");
848
    printf("%s%s%s%s\n", indent, "-R, --disorder-range=NUMBER", "\t",
849
            "Out of order data's range. Unit is ms. By default is 1000.");
850
    printf("%s%s%s%s\n", indent, "-g, --debug", "\t\t\t",
851
            "Print debug info.");
852 853 854 855
    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");
856
    printf("%s%s\n", indent, "-V, --version\t\t\tPrint program version.");
857 858
    /*    printf("%s%s%s%s\n", indent, "-D", indent,
          "Delete database if exists. 0: no, 1: yes, default is 1");
859
          */
860 861 862 863
    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");
864 865
}

866 867
static bool isStringNumber(char *input)
{
868 869 870 871
    int len = strlen(input);
    if (0 == len) {
        return false;
    }
872

873 874 875 876
    for (int i = 0; i < len; i++) {
        if (!isdigit(input[i]))
            return false;
    }
877

878
    return true;
879 880
}

881 882 883 884 885 886
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");
}

887
static void errorUnrecognized(char *program, char *wrong_arg)
888 889 890 891 892
{
    fprintf(stderr, "%s: unrecognized options '%s'\n", program, wrong_arg);
    fprintf(stderr, "Try `taosdemo --help' or `taosdemo --usage' for more information.\n");
}

893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
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");
}

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

922
    for (int i = 1; i < argc; i++) {
923
        if ((0 == strncmp(argv[i], "-f", strlen("-f")))
924
                || (0 == strncmp(argv[i], "--file", strlen("--file")))) {
925
            arguments->demo_mode = false;
926

927 928 929
            if (2 == strlen(argv[i])) {
                if (i+1 == argc) {
                    errorPrintReqArg(argv[0], "f");
930 931 932
                    exit(EXIT_FAILURE);
                }
                arguments->metaFile = argv[++i];
933 934
            } else if (0 == strncmp(argv[i], "-f", strlen("-f"))) {
                arguments->metaFile = (char *)(argv[i] + strlen("-f"));
935
            } else if (strlen("--file") == strlen(argv[i])) {
936 937 938 939 940
                if (i+1 == argc) {
                    errorPrintReqArg3(argv[0], "--file");
                    exit(EXIT_FAILURE);
                }
                arguments->metaFile = argv[++i];
941 942
            } else if (0 == strncmp(argv[i], "--file=", strlen("--file="))) {
                arguments->metaFile = (char *)(argv[i] + strlen("--file="));
943
            } else {
944
                errorUnrecognized(argv[0], argv[i]);
945
                exit(EXIT_FAILURE);
946
            }
947
        } else if ((0 == strncmp(argv[i], "-c", strlen("-c")))
948 949 950 951 952 953 954
                || (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);
955
            } else if (0 == strncmp(argv[i], "-c", strlen("-c"))) {
956
                tstrncpy(configDir, (char *)(argv[i] + strlen("-c")), TSDB_FILENAME_LEN);
957
            } else if (strlen("--config-dir") == strlen(argv[i])) {
958 959 960 961 962
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--config-dir");
                    exit(EXIT_FAILURE);
                }
                tstrncpy(configDir, argv[++i], TSDB_FILENAME_LEN);
963 964
            } else if (0 == strncmp(argv[i], "--config-dir=", strlen("--config-dir="))) {
                tstrncpy(configDir, (char *)(argv[i] + strlen("--config-dir=")), TSDB_FILENAME_LEN);
965
            } else {
966
                errorUnrecognized(argv[0], argv[i]);
967
                exit(EXIT_FAILURE);
968
            }
969
        } else if ((0 == strncmp(argv[i], "-h", strlen("-h")))
970 971 972 973 974 975 976
                || (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];
977 978
            } else if (0 == strncmp(argv[i], "-h", strlen("-h"))) {
                arguments->host = (char *)(argv[i] + strlen("-h"));
979
            } else if (strlen("--host") == strlen(argv[i])) {
980 981 982 983 984
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--host");
                    exit(EXIT_FAILURE);
                }
                arguments->host = argv[++i];
985 986
            } else if (0 == strncmp(argv[i], "--host=", strlen("--host="))) {
                arguments->host = (char *)(argv[i] + strlen("--host="));
987
            } else {
988
                errorUnrecognized(argv[0], argv[i]);
989
                exit(EXIT_FAILURE);
990
            }
991 992 993
        } else if (strcmp(argv[i], "-PP") == 0) {
            arguments->performance_print = true;
        } else if ((0 == strncmp(argv[i], "-P", strlen("-P")))
994
                || (0 == strncmp(argv[i], "--port", strlen("--port")))) {
995 996 997
            uint64_t port;
            char strPort[BIGINT_BUFF_LEN];

998 999 1000 1001
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "P");
                    exit(EXIT_FAILURE);
1002 1003 1004
                } else if (isStringNumber(argv[i+1])) {
                    tstrncpy(strPort, argv[++i], BIGINT_BUFF_LEN);
                } else {
1005 1006 1007 1008 1009
                    errorPrintReqArg2(argv[0], "P");
                    exit(EXIT_FAILURE);
                }
            } else if (0 == strncmp(argv[i], "--port=", strlen("--port="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--port=")))) {
1010 1011 1012 1013
                    tstrncpy(strPort, (char *)(argv[i]+strlen("--port=")), BIGINT_BUFF_LEN);
                } else {
                    errorPrintReqArg2(argv[0], "--port");
                    exit(EXIT_FAILURE);
1014
                }
1015 1016
            } else if (0 == strncmp(argv[i], "-P", strlen("-P"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-P")))) {
1017 1018 1019 1020
                    tstrncpy(strPort, (char *)(argv[i]+strlen("-P")), BIGINT_BUFF_LEN);
                } else {
                    errorPrintReqArg2(argv[0], "--port");
                    exit(EXIT_FAILURE);
1021 1022 1023 1024 1025
                }
            } else if (strlen("--port") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--port");
                    exit(EXIT_FAILURE);
1026 1027 1028
                } else if (isStringNumber(argv[i+1])) {
                    tstrncpy(strPort, argv[++i], BIGINT_BUFF_LEN);
                } else {
1029 1030 1031 1032
                    errorPrintReqArg2(argv[0], "--port");
                    exit(EXIT_FAILURE);
                }
            } else {
1033
                errorUnrecognized(argv[0], argv[i]);
1034
                exit(EXIT_FAILURE);
1035
            }
1036 1037 1038 1039 1040 1041 1042 1043

            port = atoi(strPort);
            if (port > 65535) {
                errorWrongValue("taosdump", "-P or --port", strPort);
                exit(EXIT_FAILURE);
            }
            arguments->port = (uint16_t)port;

1044
        } else if ((0 == strncmp(argv[i], "-I", strlen("-I")))
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
                || (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 {
1058
                    errorWrongValue(argv[0], "-I", argv[i+1]);
1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072
                    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);
                }
1073 1074 1075 1076 1077 1078 1079 1080
            } 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 {
1081 1082
                    errorWrongValue(argv[0], "-I",
                            (char *)(argv[i] + strlen("-I")));
1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096
                    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 {
1097
                    errorWrongValue(argv[0], "--interface", argv[i+1]);
1098 1099 1100 1101
                    exit(EXIT_FAILURE);
                }
                i++;
            } else {
1102
                errorUnrecognized(argv[0], argv[i]);
1103
                exit(EXIT_FAILURE);
1104
            }
1105
        } else if ((0 == strncmp(argv[i], "-u", strlen("-u")))
1106 1107 1108 1109 1110 1111 1112
                || (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];
1113 1114
            } else if (0 == strncmp(argv[i], "-u", strlen("-u"))) {
                arguments->user = (char *)(argv[i++] + strlen("-u"));
1115 1116
            } else if (0 == strncmp(argv[i], "--user=", strlen("--user="))) {
                arguments->user = (char *)(argv[i++] + strlen("--user="));
1117 1118 1119 1120 1121 1122 1123
            } else if (strlen("--user") == strlen(argv[i])) {
                if (argc == i+1) {
                    errorPrintReqArg3(argv[0], "--user");
                    exit(EXIT_FAILURE);
                }
                arguments->user = argv[++i];
            } else {
1124
                errorUnrecognized(argv[0], argv[i]);
1125
                exit(EXIT_FAILURE);
1126
            }
1127
        } else if ((0 == strncmp(argv[i], "-p", strlen("-p")))
1128 1129
                || (0 == strcmp(argv[i], "--password"))) {
            if ((strlen(argv[i]) == 2) || (0 == strcmp(argv[i], "--password"))) {
1130 1131
                printf("Enter password: ");
                taosSetConsoleEcho(false);
1132 1133 1134
                if (scanf("%s", arguments->password) > 1) {
                    fprintf(stderr, "password read error!\n");
                }
1135
                taosSetConsoleEcho(true);
1136
            } else {
1137
                tstrncpy(arguments->password, (char *)(argv[i] + 2), SHELL_MAX_PASSWORD_LEN);
1138
            }
1139
        } else if ((0 == strncmp(argv[i], "-o", strlen("-o")))
1140 1141 1142
                || (0 == strncmp(argv[i], "--output", strlen("--output")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
1143
                    errorPrintReqArg3(argv[0], "--output");
1144 1145 1146 1147 1148
                    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="));
1149 1150 1151 1152 1153 1154 1155 1156 1157
            } 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 {
1158
                errorUnrecognized(argv[0], argv[i]);
1159
                exit(EXIT_FAILURE);
1160
            }
1161
        } else if ((0 == strncmp(argv[i], "-s", strlen("-s")))
1162 1163 1164 1165 1166 1167 1168 1169
                || (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="))) {
1170
                arguments->sqlFile = (char *)(argv[i++] + strlen("--sql-file="));
1171
            } else if (0 == strncmp(argv[i], "-s", strlen("-s"))) {
1172
                arguments->sqlFile = (char *)(argv[i++] + strlen("-s"));
1173 1174 1175 1176 1177 1178 1179
            } 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 {
1180
                errorUnrecognized(argv[0], argv[i]);
1181
                exit(EXIT_FAILURE);
1182
            }
1183
        } else if ((0 == strncmp(argv[i], "-q", strlen("-q")))
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200
                || (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);
                }
1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217
            } 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 {
1218
                errorUnrecognized(argv[0], argv[i]);
1219
                exit(EXIT_FAILURE);
1220
            }
1221
        } else if ((0 == strncmp(argv[i], "-T", strlen("-T")))
1222 1223 1224 1225 1226 1227 1228 1229 1230
                || (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);
                }
1231
                arguments->nthreads = atoi(argv[++i]);
1232 1233
            } else if (0 == strncmp(argv[i], "--threads=", strlen("--threads="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--threads=")))) {
1234
                    arguments->nthreads = atoi((char *)(argv[i]+strlen("--threads=")));
1235 1236 1237 1238
                } else {
                    errorPrintReqArg2(argv[0], "--threads");
                    exit(EXIT_FAILURE);
                }
1239 1240
            } else if (0 == strncmp(argv[i], "-T", strlen("-T"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-T")))) {
1241
                    arguments->nthreads = atoi((char *)(argv[i]+strlen("-T")));
1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253
                } 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);
                }
1254
                arguments->nthreads = atoi(argv[++i]);
1255
            } else {
1256
                errorUnrecognized(argv[0], argv[i]);
1257
                exit(EXIT_FAILURE);
1258
            }
1259
        } else if ((0 == strncmp(argv[i], "-i", strlen("-i")))
1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
                || (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="))) {
1271
                if (isStringNumber((char *)(argv[i] + strlen("--insert-interval=")))) {
1272 1273 1274 1275 1276
                    arguments->insert_interval = atoi((char *)(argv[i]+strlen("--insert-interval=")));
                } else {
                    errorPrintReqArg3(argv[0], "--insert-innterval");
                    exit(EXIT_FAILURE);
                }
1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293
            } 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 {
1294
                errorUnrecognized(argv[0], argv[i]);
1295 1296
                exit(EXIT_FAILURE);
            }
1297
        } else if ((0 == strncmp(argv[i], "-S", strlen("-S")))
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314
                || (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);
                }
1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
            } 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 {
1332
                errorUnrecognized(argv[0], argv[i]);
1333
                exit(EXIT_FAILURE);
1334
            }
1335 1336 1337 1338 1339 1340 1341 1342
        } 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]);
1343
        } else if ((0 == strncmp(argv[i], "-B", strlen("-B")))
1344
                || (0 == strncmp(argv[i], "--interlace-rows", strlen("--interlace-rows")))) {
1345
            if (strlen("-B") == strlen(argv[i])) {
1346 1347 1348 1349 1350 1351 1352
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "B");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "B");
                    exit(EXIT_FAILURE);
                }
1353
                arguments->interlaceRows = atoi(argv[++i]);
1354 1355
            } else if (0 == strncmp(argv[i], "--interlace-rows=", strlen("--interlace-rows="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--interlace-rows=")))) {
1356
                    arguments->interlaceRows = atoi((char *)(argv[i]+strlen("--interlace-rows=")));
1357 1358 1359 1360
                } else {
                    errorPrintReqArg2(argv[0], "--interlace-rows");
                    exit(EXIT_FAILURE);
                }
1361 1362
            } else if (0 == strncmp(argv[i], "-B", strlen("-B"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-B")))) {
1363
                    arguments->interlaceRows = atoi((char *)(argv[i]+strlen("-B")));
1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
                } 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);
                }
1376
                arguments->interlaceRows = atoi(argv[++i]);
1377
            } else {
1378
                errorUnrecognized(argv[0], argv[i]);
1379
                exit(EXIT_FAILURE);
1380
            }
1381
        } else if ((0 == strncmp(argv[i], "-r", strlen("-r")))
1382
                || (0 == strncmp(argv[i], "--rec-per-req", 13))) {
1383
            if (strlen("-r") == strlen(argv[i])) {
1384 1385 1386 1387 1388 1389 1390
                if (argc == i+1) {
                    errorPrintReqArg(argv[0], "r");
                    exit(EXIT_FAILURE);
                } else if (!isStringNumber(argv[i+1])) {
                    errorPrintReqArg2(argv[0], "r");
                    exit(EXIT_FAILURE);
                }
1391
                arguments->reqPerReq = atoi(argv[++i]);
1392 1393
            } else if (0 == strncmp(argv[i], "--rec-per-req=", strlen("--rec-per-req="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--rec-per-req=")))) {
1394
                    arguments->reqPerReq = atoi((char *)(argv[i]+strlen("--rec-per-req=")));
1395 1396 1397 1398
                } else {
                    errorPrintReqArg2(argv[0], "--rec-per-req");
                    exit(EXIT_FAILURE);
                }
1399 1400
            } else if (0 == strncmp(argv[i], "-r", strlen("-r"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-r")))) {
1401
                    arguments->reqPerReq = atoi((char *)(argv[i]+strlen("-r")));
1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413
                } 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);
                }
1414
                arguments->reqPerReq = atoi(argv[++i]);
1415
            } else {
1416
                errorUnrecognized(argv[0], argv[i]);
1417
                exit(EXIT_FAILURE);
1418
            }
1419
        } else if ((0 == strncmp(argv[i], "-t", strlen("-t")))
1420 1421 1422 1423 1424 1425 1426 1427 1428
                || (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);
                }
1429
                arguments->ntables = atoi(argv[++i]);
1430 1431
            } else if (0 == strncmp(argv[i], "--tables=", strlen("--tables="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--tables=")))) {
1432
                    arguments->ntables = atoi((char *)(argv[i]+strlen("--tables=")));
1433 1434 1435 1436
                } else {
                    errorPrintReqArg2(argv[0], "--tables");
                    exit(EXIT_FAILURE);
                }
1437 1438
            } else if (0 == strncmp(argv[i], "-t", strlen("-t"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-t")))) {
1439
                    arguments->ntables = atoi((char *)(argv[i]+strlen("-t")));
1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
                } 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);
                }
1452
                arguments->ntables = atoi(argv[++i]);
1453
            } else {
1454
                errorUnrecognized(argv[0], argv[i]);
1455
                exit(EXIT_FAILURE);
1456
            }
1457

1458
            g_totalChildTables = arguments->ntables;
1459
        } else if ((0 == strncmp(argv[i], "-n", strlen("-n")))
1460 1461 1462 1463 1464 1465 1466 1467 1468
                || (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);
                }
1469
                arguments->insertRows = atoi(argv[++i]);
1470 1471
            } else if (0 == strncmp(argv[i], "--records=", strlen("--records="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--records=")))) {
1472
                    arguments->insertRows = atoi((char *)(argv[i]+strlen("--records=")));
1473 1474 1475 1476
                } else {
                    errorPrintReqArg2(argv[0], "--records");
                    exit(EXIT_FAILURE);
                }
1477 1478
            } else if (0 == strncmp(argv[i], "-n", strlen("-n"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-n")))) {
1479
                    arguments->insertRows = atoi((char *)(argv[i]+strlen("-n")));
1480
                } else {
1481
                    errorPrintReqArg2(argv[0], "-n");
1482 1483 1484 1485 1486 1487 1488 1489 1490 1491
                    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);
                }
1492
                arguments->insertRows = atoi(argv[++i]);
1493
            } else {
1494
                errorUnrecognized(argv[0], argv[i]);
1495
                exit(EXIT_FAILURE);
1496
            }
1497
        } else if ((0 == strncmp(argv[i], "-d", strlen("-d")))
1498 1499 1500
                || (0 == strncmp(argv[i], "--database", strlen("--database")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
1501
                    errorPrintReqArg(argv[0], "d");
1502 1503 1504 1505 1506
                    exit(EXIT_FAILURE);
                }
                arguments->database = argv[++i];
            } else if (0 == strncmp(argv[i], "--database=", strlen("--database="))) {
                arguments->output_file = (char *)(argv[i] + strlen("--database="));
1507 1508 1509 1510 1511 1512 1513 1514 1515
            } 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 {
1516
                errorUnrecognized(argv[0], argv[i]);
1517
                exit(EXIT_FAILURE);
1518
            }
1519
        } else if ((0 == strncmp(argv[i], "-l", strlen("-l")))
1520
                || (0 == strncmp(argv[i], "--columns", strlen("--columns")))) {
1521
            arguments->demo_mode = false;
1522 1523 1524 1525 1526 1527 1528 1529
            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);
                }
1530
                arguments->columnCount = atoi(argv[++i]);
1531 1532
            } else if (0 == strncmp(argv[i], "--columns=", strlen("--columns="))) {
                if (isStringNumber((char *)(argv[i] + strlen("--columns=")))) {
1533
                    arguments->columnCount = atoi((char *)(argv[i]+strlen("--columns=")));
1534 1535 1536 1537 1538 1539
                } else {
                    errorPrintReqArg2(argv[0], "--columns");
                    exit(EXIT_FAILURE);
                }
            } else if (0 == strncmp(argv[i], "-l", strlen("-l"))) {
                if (isStringNumber((char *)(argv[i] + strlen("-l")))) {
1540
                    arguments->columnCount = atoi((char *)(argv[i]+strlen("-l")));
1541
                } else {
1542 1543 1544 1545 1546 1547 1548 1549
                    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])) {
1550
                    errorPrintReqArg2(argv[0], "--columns");
1551 1552
                    exit(EXIT_FAILURE);
                }
1553
                arguments->columnCount = atoi(argv[++i]);
1554
            } else {
1555
                errorUnrecognized(argv[0], argv[i]);
1556
                exit(EXIT_FAILURE);
1557
            }
1558

1559
            if (arguments->columnCount > MAX_NUM_COLUMNS) {
1560
                printf("WARNING: max acceptable columns count is %d\n", MAX_NUM_COLUMNS);
1561
                prompt();
1562
                arguments->columnCount = MAX_NUM_COLUMNS;
1563
            }
1564

1565 1566 1567
            for (int col = DEFAULT_DATATYPE_NUM; col < arguments->columnCount; col ++) {
                arguments->dataType[col] = "INT";
                arguments->data_type[col] = TSDB_DATA_TYPE_INT;
1568
            }
1569 1570 1571
            for (int col = arguments->columnCount; col < MAX_NUM_COLUMNS; col++) {
                arguments->dataType[col] = NULL;
                arguments->data_type[col] = TSDB_DATA_TYPE_NULL;
1572
            }
1573
        } else if ((0 == strncmp(argv[i], "-b", strlen("-b")))
1574
                || (0 == strncmp(argv[i], "--data-type", strlen("--data-type")))) {
1575
            arguments->demo_mode = false;
1576 1577 1578 1579 1580 1581 1582 1583 1584 1585

            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="));
1586 1587 1588 1589 1590 1591 1592 1593 1594
            } 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 {
1595
                errorUnrecognized(argv[0], argv[i]);
1596
                exit(EXIT_FAILURE);
1597
            }
1598 1599

            if (strstr(dataType, ",") == NULL) {
1600
                // only one col
1601 1602 1603 1604 1605 1606 1607 1608
                if (strcasecmp(dataType, "INT")
                        && strcasecmp(dataType, "FLOAT")
                        && strcasecmp(dataType, "TINYINT")
                        && strcasecmp(dataType, "BOOL")
                        && strcasecmp(dataType, "SMALLINT")
                        && strcasecmp(dataType, "BIGINT")
                        && strcasecmp(dataType, "DOUBLE")
                        && strcasecmp(dataType, "TIMESTAMP")
1609 1610 1611
                        && !regexMatch(dataType,
                            "^(NCHAR|BINARY)(\\([1-9][0-9]*\\))?$",
                            REG_ICASE | REG_EXTENDED)
1612 1613 1614 1615
                        && strcasecmp(dataType, "UTINYINT")
                        && strcasecmp(dataType, "USMALLINT")
                        && strcasecmp(dataType, "UINT")
                        && strcasecmp(dataType, "UBIGINT")) {
1616 1617 1618 1619
                    printHelp();
                    errorPrint("%s", "-b: Invalid data_type!\n");
                    exit(EXIT_FAILURE);
                }
1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
                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;
1633 1634 1635
                } else if (1 == regexMatch(dataType,
                            "^BINARY(\\([1-9][0-9]*\\))?$",
                            REG_ICASE | REG_EXTENDED)) {
1636
                    arguments->data_type[0] = TSDB_DATA_TYPE_BINARY;
1637 1638 1639
                } else if (1 == regexMatch(dataType,
                            "^NCHAR(\\([1-9][0-9]*\\))?$",
                            REG_ICASE | REG_EXTENDED)) {
1640 1641 1642 1643 1644
                    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;
1645 1646 1647 1648 1649 1650 1651 1652
                } 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;
1653 1654 1655 1656 1657
                } else {
                    arguments->data_type[0] = TSDB_DATA_TYPE_NULL;
                }
                arguments->dataType[1] = NULL;
                arguments->data_type[1] = TSDB_DATA_TYPE_NULL;
1658 1659 1660
            } else {
                // more than one col
                int index = 0;
1661
                g_dupstr = strdup(dataType);
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672
                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, "TIMESTAMP")
1673
                            && !regexMatch(token, "^(NCHAR|BINARY)(\\([1-9][0-9]*\\))?$", REG_ICASE | REG_EXTENDED)
1674 1675 1676 1677
                            && strcasecmp(token, "UTINYINT")
                            && strcasecmp(token, "USMALLINT")
                            && strcasecmp(token, "UINT")
                            && strcasecmp(token, "UBIGINT")) {
1678 1679 1680 1681 1682
                        printHelp();
                        free(g_dupstr);
                        errorPrint("%s", "-b: Invalid data_type!\n");
                        exit(EXIT_FAILURE);
                    }
1683 1684 1685 1686 1687 1688 1689 1690 1691 1692

                    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")) {
1693
                        arguments->data_type[index] = TSDB_DATA_TYPE_DOUBLE;
1694 1695
                    } else if (0 == strcasecmp(token, "TINYINT")) {
                        arguments->data_type[index] = TSDB_DATA_TYPE_TINYINT;
1696
                    } else if (1 == regexMatch(token, "^BINARY(\\([1-9][0-9]*\\))?$", REG_ICASE |
1697
                    REG_EXTENDED)) {
1698
                        arguments->data_type[index] = TSDB_DATA_TYPE_BINARY;
1699
                    } else if (1 == regexMatch(token, "^NCHAR(\\([1-9][0-9]*\\))?$", REG_ICASE |
1700
                    REG_EXTENDED)) {
1701 1702 1703 1704 1705
                        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;
1706 1707 1708 1709 1710 1711 1712 1713
                    } 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;
1714 1715 1716 1717 1718
                    } else {
                        arguments->data_type[index] = TSDB_DATA_TYPE_NULL;
                    }
                    arguments->dataType[index] = token;
                    index ++;
1719
                    token = strsep(&running, ",");
1720
                    if (index >= MAX_NUM_COLUMNS) break;
1721
                }
1722 1723
                arguments->dataType[index] = NULL;
                arguments->data_type[index] = TSDB_DATA_TYPE_NULL;
1724
            }
1725
        } else if ((0 == strncmp(argv[i], "-w", strlen("-w")))
1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742
                || (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);
                }
1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759
            } 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 {
1760
                errorUnrecognized(argv[0], argv[i]);
1761
                exit(EXIT_FAILURE);
1762
            }
1763
        } else if ((0 == strncmp(argv[i], "-m", strlen("-m")))
1764 1765 1766
                || (0 == strncmp(argv[i], "--table-prefix", strlen("--table-prefix")))) {
            if (2 == strlen(argv[i])) {
                if (argc == i+1) {
1767
                    errorPrintReqArg(argv[0], "m");
1768 1769 1770 1771 1772
                    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="));
1773 1774 1775 1776 1777 1778 1779 1780 1781
            } 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 {
1782
                errorUnrecognized(argv[0], argv[i]);
1783
                exit(EXIT_FAILURE);
1784
            }
1785 1786 1787
        } else if ((0 == strncmp(argv[i], "-E", strlen("-E")))
                || (0 == strncmp(argv[i], "--escape-character", strlen("--escape-character")))) {
            arguments->escapeChar = true;
1788 1789
        } else if ((strcmp(argv[i], "-N") == 0)
                || (0 == strcmp(argv[i], "--normal-table"))) {
1790
            arguments->demo_mode = false;
1791
            arguments->use_metric = false;
1792 1793
        } else if ((strcmp(argv[i], "-M") == 0)
                || (0 == strcmp(argv[i], "--random"))) {
1794
            arguments->demo_mode = false;
1795
        } else if ((strcmp(argv[i], "-x") == 0)
1796 1797
                || (0 == strcmp(argv[i], "--aggr-func"))) {
            arguments->aggr_func = true;
1798 1799
        } else if ((strcmp(argv[i], "-y") == 0)
                || (0 == strcmp(argv[i], "--answer-yes"))) {
1800
            arguments->answer_yes = true;
1801 1802
        } else if ((strcmp(argv[i], "-g") == 0)
                || (0 == strcmp(argv[i], "--debug"))) {
1803 1804 1805
            arguments->debug_print = true;
        } else if (strcmp(argv[i], "-gg") == 0) {
            arguments->verbose_print = true;
1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850
        } 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 {
1851
                errorUnrecognized(argv[0], argv[i]);
1852 1853
                exit(EXIT_FAILURE);
            }
1854
        } else if ((0 == strncmp(argv[i], "-O", strlen("-O")))
1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871
                || (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);
                }
1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888
            } 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 {
1889
                errorUnrecognized(argv[0], argv[i]);
1890
                exit(EXIT_FAILURE);
1891
            }
1892

1893
            if (arguments->disorderRatio > 50) {
1894 1895
                errorPrint("Invalid disorder ratio %d, will be set to %d\n",
                        arguments->disorderRatio, 50);
1896 1897
                arguments->disorderRatio = 50;
            }
1898
        } else if ((0 == strncmp(argv[i], "-a", strlen("-a")))
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918
                || (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);
                }
1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936
            } 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 {
1937
                errorUnrecognized(argv[0], argv[i]);
1938
                exit(EXIT_FAILURE);
1939
            }
1940

1941
            if (arguments->replica > 3 || arguments->replica < 1) {
1942 1943
                errorPrint("Invalid replica value %d, will be set to %d\n",
                        arguments->replica, 1);
1944 1945 1946 1947 1948
                arguments->replica = 1;
            }
        } else if (strcmp(argv[i], "-D") == 0) {
            arguments->method_of_delete = atoi(argv[++i]);
            if (arguments->method_of_delete > 3) {
1949
                errorPrint("%s", "\n\t-D need a value (0~3) number following!\n");
1950 1951
                exit(EXIT_FAILURE);
            }
1952 1953
        } else if ((strcmp(argv[i], "--version") == 0)
                || (strcmp(argv[i], "-V") == 0)) {
1954 1955
            printVersion();
            exit(0);
1956 1957
        } else if ((strcmp(argv[i], "--help") == 0)
                || (strcmp(argv[i], "-?") == 0)) {
1958 1959
            printHelp();
            exit(0);
1960 1961 1962 1963
        } 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\
1964
                    [-b DATATYPES] [-w WIDTH_OF_BINARY] [-l COLUMNS] [-T THREADNUMBER]\n\
1965 1966 1967 1968
                    [-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);
1969
        } else {
1970 1971
            // to simulate argp_option output
            if (strlen(argv[i]) > 2) {
1972 1973 1974 1975 1976 1977 1978 1979 1980
                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]);
                }
1981 1982 1983 1984 1985
            } 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");
1986 1987
            exit(EXIT_FAILURE);
        }
S
TD-1057  
Shengliang Guan 已提交
1988
    }
1989

1990
    int columnCount;
1991
    for (columnCount = 0; columnCount < MAX_NUM_COLUMNS; columnCount ++) {
1992
        if (g_args.dataType[columnCount] == NULL) {
1993 1994
            break;
        }
1995 1996
    }

1997
    if (0 == columnCount) {
1998
        ERROR_EXIT("data type error!");
1999
    }
2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013
    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:
2014
            case TSDB_DATA_TYPE_UINT:
2015 2016 2017 2018
                g_args.lenOfOneRow += INT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BIGINT:
2019
            case TSDB_DATA_TYPE_UBIGINT:
2020 2021 2022 2023
                g_args.lenOfOneRow += BIGINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
2024
            case TSDB_DATA_TYPE_USMALLINT:
2025 2026 2027 2028
                g_args.lenOfOneRow += SMALLINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TINYINT:
2029
            case TSDB_DATA_TYPE_UTINYINT:
2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053
                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);
        }
    }
2054

2055
    if (((arguments->debug_print) && (NULL != arguments->metaFile))
2056 2057 2058 2059 2060 2061 2062 2063 2064 2065
            || 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");
2066
        if (*(arguments->dataType)) {
2067
            printf("# Specified data type:               ");
2068
            for (int c = 0; c < MAX_NUM_COLUMNS; c++)
2069 2070
                if (arguments->dataType[c])
                    printf("%s,", arguments->dataType[c]);
2071 2072 2073 2074 2075 2076 2077
                else
                    break;
            printf("\n");
        }
        printf("# Insertion interval:                %"PRIu64"\n",
                arguments->insert_interval);
        printf("# Number of records per req:         %u\n",
2078
                arguments->reqPerReq);
2079 2080
        printf("# Max SQL length:                    %"PRIu64"\n",
                arguments->max_sql_len);
2081
        printf("# Length of Binary:                  %d\n", arguments->binwidth);
2082
        printf("# Number of Threads:                 %d\n", arguments->nthreads);
2083
        printf("# Number of Tables:                  %"PRId64"\n",
2084
                arguments->ntables);
2085
        printf("# Number of Data per Table:          %"PRId64"\n",
2086
                arguments->insertRows);
2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099
        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();
2100
    }
2101
}
H
hzcheng 已提交
2102

2103
static void tmfclose(FILE *fp) {
2104 2105 2106
    if (NULL != fp) {
        fclose(fp);
    }
2107
}
2108

2109
static void tmfree(char *buf) {
2110 2111
    if (NULL != buf) {
        free(buf);
2112
        buf = NULL;
2113
    }
2114
}
H
Hui Li 已提交
2115

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

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

2120 2121
    TAOS_RES *res = taos_query(taos, command);
    int32_t code = taos_errno(res);
2122

2123 2124
    if (code != 0) {
        if (!quiet) {
2125
            errorPrint2("Failed to execute <%s>, reason: %s\n",
2126 2127 2128 2129 2130
                    command, taos_errstr(res));
        }
        taos_free_result(res);
        //taos_close(taos);
        return -1;
2131
    }
H
hzcheng 已提交
2132

2133 2134 2135 2136
    if (INSERT_TYPE == type) {
        int affectedRows = taos_affected_rows(res);
        taos_free_result(res);
        return affectedRows;
2137
    }
H
hzcheng 已提交
2138

2139
    taos_free_result(res);
2140
    return 0;
2141
}
H
hzcheng 已提交
2142

2143
static void appendResultBufToFile(char *resultBuf, threadInfo *pThreadInfo)
2144
{
2145 2146
    pThreadInfo->fp = fopen(pThreadInfo->filePath, "at");
    if (pThreadInfo->fp == NULL) {
2147
        errorPrint2(
2148 2149 2150 2151
                "%s() LN%d, failed to open result file: %s, result will not save to file\n",
                __func__, __LINE__, pThreadInfo->filePath);
        return;
    }
2152

2153 2154 2155
    fprintf(pThreadInfo->fp, "%s", resultBuf);
    tmfclose(pThreadInfo->fp);
    pThreadInfo->fp = NULL;
2156 2157
}

2158
static void fetchResult(TAOS_RES *res, threadInfo* pThreadInfo) {
2159 2160 2161 2162 2163 2164 2165
    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) {
2166
        errorPrint2("%s() LN%d, failed to malloc, warning: save result to file slowly!\n",
2167 2168 2169
                __func__, __LINE__);
        return ;
    }
H
hzcheng 已提交
2170

2171
    int64_t   totalLen = 0;
H
hzcheng 已提交
2172

2173 2174
    // fetch the records row by row
    while((row = taos_fetch_row(res))) {
2175
        if (totalLen >= (100*1024*1024 - HEAD_BUFF_LEN*2)) {
2176 2177 2178 2179 2180 2181
            if (strlen(pThreadInfo->filePath) > 0)
                appendResultBufToFile(databuf, pThreadInfo);
            totalLen = 0;
            memset(databuf, 0, 100*1024*1024);
        }
        num_rows++;
2182
        char  temp[HEAD_BUFF_LEN] = {0};
2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197
        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);
2198
}
H
hzcheng 已提交
2199

2200
static void selectAndGetResult(
2201 2202
        threadInfo *pThreadInfo, char *command)
{
2203 2204 2205
    if (0 == strncasecmp(g_queryInfo.queryMode, "taosc", strlen("taosc"))) {
        TAOS_RES *res = taos_query(pThreadInfo->taos, command);
        if (res == NULL || taos_errno(res) != 0) {
2206
            errorPrint2("%s() LN%d, failed to execute sql:%s, reason:%s\n",
2207 2208 2209 2210
                    __func__, __LINE__, command, taos_errstr(res));
            taos_free_result(res);
            return;
        }
2211

2212 2213
        fetchResult(res, pThreadInfo);
        taos_free_result(res);
2214

2215 2216
    } else if (0 == strncasecmp(g_queryInfo.queryMode, "rest", strlen("rest"))) {
        int retCode = postProceSql(
2217
                g_queryInfo.host, g_queryInfo.port,
2218 2219 2220 2221 2222
                command,
                pThreadInfo);
        if (0 != retCode) {
            printf("====restful return fail, threadID[%d]\n", pThreadInfo->threadID);
        }
2223

2224
    } else {
2225
        errorPrint2("%s() LN%d, unknown query mode: %s\n",
2226 2227
                __func__, __LINE__, g_queryInfo.queryMode);
    }
2228
}
H
hzcheng 已提交
2229

2230
static char *rand_bool_str() {
2231 2232 2233
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2234
    return g_randbool_buff + ((cursor % MAX_PREPARED_RAND) * BOOL_BUFF_LEN);
2235 2236
}

2237
static int32_t rand_bool() {
2238 2239
    static int cursor;
    cursor++;
2240 2241
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randint[cursor % MAX_PREPARED_RAND] % 2;
2242 2243
}

2244 2245 2246 2247 2248
static char *rand_tinyint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2249 2250
    return g_randtinyint_buff +
        ((cursor % MAX_PREPARED_RAND) * TINYINT_BUFF_LEN);
2251 2252 2253 2254
}

static int32_t rand_tinyint()
{
2255 2256
    static int cursor;
    cursor++;
2257 2258
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randint[cursor % MAX_PREPARED_RAND] % 128;
2259 2260
}

2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277
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;
}

2278 2279 2280 2281 2282
static char *rand_smallint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2283 2284
    return g_randsmallint_buff +
        ((cursor % MAX_PREPARED_RAND) * SMALLINT_BUFF_LEN);
2285 2286
}

2287 2288
static int32_t rand_smallint()
{
2289 2290
    static int cursor;
    cursor++;
2291
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309
    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;
2310 2311
}

2312 2313 2314 2315 2316
static char *rand_int_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2317
    return g_randint_buff + ((cursor % MAX_PREPARED_RAND) * INT_BUFF_LEN);
2318 2319 2320 2321
}

static int32_t rand_int()
{
2322 2323
    static int cursor;
    cursor++;
2324 2325
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randint[cursor % MAX_PREPARED_RAND];
2326 2327
}

2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343
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];
}

2344 2345 2346 2347 2348
static char *rand_bigint_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2349 2350
    return g_randbigint_buff +
        ((cursor % MAX_PREPARED_RAND) * BIGINT_BUFF_LEN);
2351 2352
}

2353 2354
static int64_t rand_bigint()
{
2355 2356
    static int cursor;
    cursor++;
2357 2358
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randbigint[cursor % MAX_PREPARED_RAND];
2359 2360
}

2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377
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];
}

2378 2379 2380 2381 2382
static char *rand_float_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2383
    return g_randfloat_buff + ((cursor % MAX_PREPARED_RAND) * FLOAT_BUFF_LEN);
2384 2385
}

S
Shengliang Guan 已提交
2386

2387 2388
static float rand_float()
{
2389 2390
    static int cursor;
    cursor++;
2391 2392
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return g_randfloat[cursor % MAX_PREPARED_RAND];
2393 2394
}

2395 2396 2397 2398 2399
static char *demo_current_float_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2400 2401
    return g_rand_current_buff +
        ((cursor % MAX_PREPARED_RAND) * FLOAT_BUFF_LEN);
2402 2403 2404 2405
}

static float UNUSED_FUNC demo_current_float()
{
2406 2407
    static int cursor;
    cursor++;
2408 2409 2410
    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);
2411 2412 2413 2414 2415 2416 2417
}

static char *demo_voltage_int_str()
{
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2418 2419
    return g_rand_voltage_buff +
        ((cursor % MAX_PREPARED_RAND) * INT_BUFF_LEN);
2420 2421
}

2422 2423
static int32_t UNUSED_FUNC demo_voltage_int()
{
2424 2425
    static int cursor;
    cursor++;
2426 2427
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
    return 215 + g_randint[cursor % MAX_PREPARED_RAND] % 10;
2428 2429 2430 2431 2432 2433
}

static char *demo_phase_float_str() {
    static int cursor;
    cursor++;
    if (cursor > (MAX_PREPARED_RAND - 1)) cursor = 0;
2434
    return g_rand_phase_buff + ((cursor % MAX_PREPARED_RAND) * FLOAT_BUFF_LEN);
2435 2436
}

2437
static float UNUSED_FUNC demo_phase_float() {
2438 2439
    static int cursor;
    cursor++;
2440 2441 2442
    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);
2443 2444
}

2445 2446 2447 2448 2449 2450
#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)
{
2451 2452 2453 2454 2455 2456 2457 2458
    str[0] = 0;
    if (size > 0) {
        int n;
        for (n = 0; n < size; n++) {
            str[n] = charNum[n % 10];
        }
        str[n] = 0;
    }
2459 2460 2461
}
#endif

2462
static const char charset[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890";
2463

2464
static void rand_string(char *str, int size) {
2465 2466 2467 2468 2469 2470 2471 2472 2473
    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;
2474 2475 2476
    }
}

2477 2478 2479 2480 2481 2482 2483 2484 2485 2486
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()
{
2487 2488 2489
    static int cursor;
    cursor++;
    cursor = cursor % MAX_PREPARED_RAND;
2490
    return g_randdouble[cursor];
2491 2492 2493
}

static void init_rand_data() {
2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514

    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);
2515 2516 2517 2518 2519 2520 2521 2522
    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);
2523

2524
    for (int i = 0; i < MAX_PREPARED_RAND; i++) {
2525 2526
        g_randint[i] = (int)(taosRandom() % RAND_MAX - (RAND_MAX >> 1));
        g_randuint[i] = (int)(taosRandom());
2527 2528 2529 2530 2531 2532 2533 2534
        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",
2535
                g_randint[i] % 32768);
2536 2537
        sprintf(g_randtinyint_buff + i * TINYINT_BUFF_LEN, "%d",
                g_randint[i] % 128);
2538 2539 2540 2541 2542 2543 2544 2545 2546
        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());
2547 2548
        sprintf(g_randbigint_buff + i * BIGINT_BUFF_LEN, "%"PRId64"",
                g_randbigint[i]);
2549 2550
        sprintf(g_randubigint_buff + i * BIGINT_BUFF_LEN, "%"PRId64"",
                g_randubigint[i]);
2551

2552
        g_randfloat[i] = (float)(taosRandom() / 1000.0) * (taosRandom() % 2 > 0.5 ? 1 : -1);
2553 2554 2555 2556 2557 2558 2559 2560 2561
        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));

2562
        g_randdouble[i] = (double)(taosRandom() / 1000000.0) * (taosRandom() % 2 > 0.5 ? 1 : -1);
2563 2564
        sprintf(g_randdouble_buff + i * DOUBLE_BUFF_LEN, "%f",
                g_randdouble[i]);
2565
    }
2566 2567
}

2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587
#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)

2588
static int printfInsertMeta() {
2589
    SHOW_PARSE_RESULT_START();
2590

2591 2592 2593
    if (g_args.demo_mode) {
        printf("\ntaosdemo is simulating data generated by power equipment monitoring...\n\n");
    } else {
2594
        printf("\ntaosdemo is simulating random data as you request..\n\n");
2595
    }
2596

2597 2598 2599
    if (g_args.iface != INTERFACE_BUT) {
        // first time if no iface specified
        printf("interface:                  \033[33m%s\033[0m\n",
2600 2601
                (g_args.iface==TAOSC_IFACE)?"taosc":
                (g_args.iface==REST_IFACE)?"rest":"stmt");
2602 2603
    }

2604 2605 2606 2607 2608 2609 2610 2611
    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",
2612
            g_Dbs.threadCountForCreateTbl);
2613 2614 2615
    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",
2616
            g_args.reqPerReq);
2617 2618 2619 2620
    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);
2621

2622 2623 2624 2625 2626
    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) {
2627
            printf("  drop:                  \033[33m no\033[0m\n");
2628
        } else {
2629
            printf("  drop:                  \033[33m yes\033[0m\n");
2630
        }
2631

2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680
        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))
2681 2682
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "us", 2))
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ns", 2))) {
2683 2684 2685 2686 2687 2688 2689 2690
                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;
            }
        }
2691

2692

2693 2694 2695 2696 2697
        if (g_args.use_metric) {
            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);
2698

2699 2700
                printf("      stbName:           \033[33m%s\033[0m\n",
                        g_Dbs.db[i].superTbls[j].stbName);
2701

2702 2703 2704 2705 2706
                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");
2707
                } else {
2708
                    printf("      autoCreateTable:   \033[33m%s\033[0m\n",  "error");
2709 2710
                }

2711 2712 2713 2714
                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");
2715
                } else {
2716 2717 2718 2719 2720 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
                    printf("      childTblExists:    \033[33m%s\033[0m\n",  "error");
                }

                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) {
                printf("      multiThreadWriteOneTbl:  \033[33m no\033[0m\n");
                }else {
                printf("      multiThreadWriteOneTbl:  \033[33m yes\033[0m\n");
                }
                */
                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);
                }

                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);
2767 2768
                printf("      useSampleTs:       \033[33m%s\033[0m\n",
                        g_Dbs.db[i].superTbls[j].useSampleTs ? "yes (warning: disorderRange/disorderRatio is disabled)" : "no");
2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785
                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[%d]:\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);
                    }
2786
                }
2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805
                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");
2806
            }
2807 2808 2809 2810 2811
        } else {
            printf("  childTblCount:     \033[33m%"PRId64"\033[0m\n",
                        g_args.ntables);
            printf("  insertRows:        \033[33m%"PRId64"\033[0m\n",
                        g_args.insertRows);
2812
        }
2813
        printf("\n");
2814
    }
2815

2816
    SHOW_PARSE_RESULT_END();
2817

2818
    return 0;
2819 2820 2821
}

static void printfInsertMetaToFile(FILE* fp) {
2822

2823
    SHOW_PARSE_RESULT_START_TO_FILE(fp);
2824

2825 2826 2827 2828 2829
    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);
2830
    fprintf(fp, "thread num of create table: %d\n", g_Dbs.threadCountForCreateTbl);
2831
    fprintf(fp, "number of records per req:  %u\n", g_args.reqPerReq);
2832 2833
    fprintf(fp, "max sql length:             %"PRIu64"\n", g_args.max_sql_len);
    fprintf(fp, "database count:          %d\n", g_Dbs.dbCount);
2834

2835 2836 2837 2838 2839 2840 2841 2842
    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");
        }
2843

2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881
        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))
2882
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ns", 2))
2883 2884 2885 2886 2887 2888 2889 2890
                    || (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);
            }
        }
2891

2892 2893 2894 2895 2896 2897
        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",
2898
                    g_Dbs.db[i].superTbls[j].stbName);
2899 2900 2901 2902 2903 2904 2905 2906 2907

            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");
            }
2908

2909 2910 2911 2912 2913 2914 2915 2916
            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");
            }
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 2959 2960 2961 2962 2963 2964 2965 2966 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
            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");
2998
        }
2999
        fprintf(fp, "\n");
3000
    }
3001

3002
    SHOW_PARSE_RESULT_END_TO_FILE(fp);
3003 3004 3005
}

static void printfQueryMeta() {
3006

3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056
    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",
3057
                    g_queryInfo.superQueryInfo.stbName);
3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075
            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");
        }
3076
    }
3077

3078
    SHOW_PARSE_RESULT_END();
3079 3080
}

3081
static char* formatTimestamp(char* buf, int64_t val, int precision) {
3082
    time_t tt;
3083
    if (precision == TSDB_TIME_PRECISION_MICRO) {
3084
        tt = (time_t)(val / 1000000);
3085 3086
    } if (precision == TSDB_TIME_PRECISION_NANO) {
        tt = (time_t)(val / 1000000000);
3087 3088 3089
    } else {
        tt = (time_t)(val / 1000);
    }
3090

3091 3092 3093 3094 3095 3096 3097
    /* 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;
       }
       */
3098 3099

#ifdef WINDOWS
3100
    if (tt < 0) tt = 0;
3101 3102
#endif

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

3106
    if (precision == TSDB_TIME_PRECISION_MICRO) {
3107
        sprintf(buf + pos, ".%06d", (int)(val % 1000000));
3108 3109
    } else if (precision == TSDB_TIME_PRECISION_NANO) {
        sprintf(buf + pos, ".%09d", (int)(val % 1000000000));
3110 3111 3112
    } else {
        sprintf(buf + pos, ".%03d", (int)(val % 1000));
    }
3113

3114
    return buf;
3115 3116
}

3117
static void xDumpFieldToFile(FILE* fp, const char* val,
3118
        TAOS_FIELD* field, int32_t length, int precision) {
3119

3120 3121 3122 3123
    if (val == NULL) {
        fprintf(fp, "%s", TSDB_DATA_NULL_STR);
        return;
    }
3124

3125 3126 3127
    char buf[TSDB_MAX_BYTES_PER_ROW];
    switch (field->type) {
        case TSDB_DATA_TYPE_BOOL:
3128
            fprintf(fp, "%d", ((((int32_t)(*((int8_t*)val))) == 1) ? 1 : 0));
3129
            break;
3130

3131 3132 3133
        case TSDB_DATA_TYPE_TINYINT:
            fprintf(fp, "%d", *((int8_t *)val));
            break;
3134

3135 3136 3137 3138
        case TSDB_DATA_TYPE_UTINYINT:
            fprintf(fp, "%d", *((uint8_t *)val));
            break;

3139 3140 3141
        case TSDB_DATA_TYPE_SMALLINT:
            fprintf(fp, "%d", *((int16_t *)val));
            break;
3142

3143 3144 3145 3146
        case TSDB_DATA_TYPE_USMALLINT:
            fprintf(fp, "%d", *((uint16_t *)val));
            break;

3147 3148 3149
        case TSDB_DATA_TYPE_INT:
            fprintf(fp, "%d", *((int32_t *)val));
            break;
3150

3151 3152 3153 3154
        case TSDB_DATA_TYPE_UINT:
            fprintf(fp, "%d", *((uint32_t *)val));
            break;

3155
        case TSDB_DATA_TYPE_BIGINT:
3156
            fprintf(fp, "%"PRId64"", *((int64_t *)val));
3157
            break;
3158

3159 3160 3161 3162
        case TSDB_DATA_TYPE_UBIGINT:
            fprintf(fp, "%"PRId64"", *((uint64_t *)val));
            break;

3163 3164 3165
        case TSDB_DATA_TYPE_FLOAT:
            fprintf(fp, "%.5f", GET_FLOAT_VAL(val));
            break;
3166

3167 3168 3169
        case TSDB_DATA_TYPE_DOUBLE:
            fprintf(fp, "%.9f", GET_DOUBLE_VAL(val));
            break;
3170

3171 3172 3173 3174 3175 3176
        case TSDB_DATA_TYPE_BINARY:
        case TSDB_DATA_TYPE_NCHAR:
            memcpy(buf, val, length);
            buf[length] = 0;
            fprintf(fp, "\'%s\'", buf);
            break;
3177

3178 3179 3180 3181
        case TSDB_DATA_TYPE_TIMESTAMP:
            formatTimestamp(buf, *(int64_t*)val, precision);
            fprintf(fp, "'%s'", buf);
            break;
3182

3183 3184 3185
        default:
            break;
    }
3186 3187 3188
}

static int xDumpResultToFile(const char* fname, TAOS_RES* tres) {
3189 3190 3191 3192
    TAOS_ROW row = taos_fetch_row(tres);
    if (row == NULL) {
        return 0;
    }
3193

3194 3195
    FILE* fp = fopen(fname, "at");
    if (fp == NULL) {
3196
        errorPrint2("%s() LN%d, failed to open file: %s\n",
3197 3198 3199
                __func__, __LINE__, fname);
        return -1;
    }
3200

3201 3202 3203
    int num_fields = taos_num_fields(tres);
    TAOS_FIELD *fields = taos_fetch_fields(tres);
    int precision = taos_result_precision(tres);
3204

3205 3206 3207 3208 3209
    for (int col = 0; col < num_fields; col++) {
        if (col > 0) {
            fprintf(fp, ",");
        }
        fprintf(fp, "%s", fields[col].name);
3210 3211 3212
    }
    fputc('\n', fp);

3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227
    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);
3228

3229
    fclose(fp);
3230

3231
    return numOfRows;
3232 3233
}

3234
static int getDbFromServer(TAOS * taos, SDbInfo** dbInfos) {
3235 3236 3237
    TAOS_RES * res;
    TAOS_ROW row = NULL;
    int count = 0;
3238

3239 3240
    res = taos_query(taos, "show databases;");
    int32_t code = taos_errno(res);
3241

3242
    if (code != 0) {
3243
        errorPrint2("failed to run <show databases>, reason: %s\n",
3244 3245
                taos_errstr(res));
        return -1;
3246
    }
3247

3248
    TAOS_FIELD *fields = taos_fetch_fields(res);
3249

3250 3251 3252 3253 3254 3255
    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;
        }
3256

3257 3258
        dbInfos[count] = (SDbInfo *)calloc(1, sizeof(SDbInfo));
        if (dbInfos[count] == NULL) {
3259
            errorPrint2("failed to allocate memory for some dbInfo[%d]\n", count);
3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298
            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;
        }
3299 3300
    }

3301
    return count;
3302 3303
}

3304
static void printfDbInfoForQueryToFile(
3305
        char* filename, SDbInfo* dbInfos, int index) {
3306

3307 3308
    if (filename[0] == 0)
        return;
3309

3310 3311 3312 3313 3314
    FILE *fp = fopen(filename, "at");
    if (fp == NULL) {
        errorPrint( "failed to open file: %s\n", filename);
        return;
    }
3315

3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336
    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");
3337

3338
    fclose(fp);
3339 3340 3341
}

static void printfQuerySystemInfo(TAOS * taos) {
3342 3343
    char filename[MAX_FILE_NAME_LEN] = {0};
    char buffer[1024] = {0};
3344 3345 3346 3347 3348 3349
    TAOS_RES* res;

    time_t t;
    struct tm* lt;
    time(&t);
    lt = localtime(&t);
3350
    snprintf(filename, MAX_FILE_NAME_LEN, "querySystemInfo-%d-%d-%d %d:%d:%d",
3351 3352 3353 3354 3355 3356
            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);
3357
    xDumpResultToFile(filename, res);
3358

3359 3360
    // show dnodes
    res = taos_query(taos, "show dnodes;");
3361
    xDumpResultToFile(filename, res);
3362
    //fetchResult(res, filename);
3363

3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381
    // 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
3382
        snprintf(buffer, 1024, "show %s.vgroups;", dbInfos[i]->name);
3383 3384 3385 3386
        res = taos_query(taos, buffer);
        xDumpResultToFile(filename, res);

        // show db.stables
3387
        snprintf(buffer, 1024, "show %s.stables;", dbInfos[i]->name);
3388 3389 3390 3391
        res = taos_query(taos, buffer);
        xDumpResultToFile(filename, res);
        free(dbInfos[i]);
    }
3392

3393
    free(dbInfos);
3394 3395
}

3396
static int postProceSql(char *host, uint16_t port,
3397
        char* sqlstr, threadInfo *pThreadInfo)
3398
{
3399
    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";
3400

3401
    char *url = "/rest/sql";
3402

3403 3404 3405 3406
    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;
3407

3408
    int req_buf_len = strlen(sqlstr) + REQ_EXTRA_BUF_LEN;
3409

3410 3411
    request_buf = malloc(req_buf_len);
    if (NULL == request_buf) {
3412
        errorPrint("%s", "cannot allocate memory.\n");
3413
        exit(EXIT_FAILURE);
3414
    }
3415

3416 3417
    char userpass_buf[INPUT_BUF_LEN];
    int mod_table[] = {0, 2, 1};
3418

3419
    static char base64[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
3420 3421 3422 3423 3424 3425 3426
        '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', '+', '/'};
3427

3428
    snprintf(userpass_buf, INPUT_BUF_LEN, "%s:%s",
3429
            g_Dbs.user, g_Dbs.password);
3430 3431
    size_t userpass_buf_len = strlen(userpass_buf);
    size_t encoded_len = 4 * ((userpass_buf_len +2) / 3);
3432

3433
    char base64_buf[INPUT_BUF_LEN];
3434

3435
    memset(base64_buf, 0, INPUT_BUF_LEN);
3436

3437
    for (int n = 0, m = 0; n < userpass_buf_len;) {
3438 3439 3440 3441 3442 3443 3444
        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;
3445

3446 3447 3448 3449
        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];
3450
    }
3451

3452
    for (int l = 0; l < mod_table[userpass_buf_len % 3]; l++)
3453
        base64_buf[encoded_len - 1 - l] = '=';
3454

3455 3456 3457
    debugPrint("%s() LN%d: auth string base64 encoded: %s\n",
            __func__, __LINE__, base64_buf);
    char *auth = base64_buf;
3458

3459 3460 3461 3462 3463 3464
    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);
3465
        ERROR_EXIT("too long request");
3466 3467
    }
    verbosePrint("%s() LN%d: Request:\n%s\n", __func__, __LINE__, request_buf);
3468

3469 3470 3471
    req_str_len = strlen(request_buf);
    sent = 0;
    do {
3472
#ifdef WINDOWS
3473
        bytes = send(pThreadInfo->sockfd, request_buf + sent, req_str_len - sent, 0);
3474
#else
3475
        bytes = write(pThreadInfo->sockfd, request_buf + sent, req_str_len - sent);
3476
#endif
3477
        if (bytes < 0)
3478
            ERROR_EXIT("writing message to socket");
3479 3480 3481 3482
        if (bytes == 0)
            break;
        sent+=bytes;
    } while(sent < req_str_len);
3483

3484 3485 3486
    memset(response_buf, 0, RESP_BUF_LEN);
    resp_len = sizeof(response_buf) - 1;
    received = 0;
3487 3488 3489 3490 3491

    char resEncodingChunk[] = "Encoding: chunked";
    char resHttp[] = "HTTP/1.1 ";
    char resHttpOk[] = "HTTP/1.1 200 OK";

3492
    do {
3493
#ifdef WINDOWS
3494
        bytes = recv(pThreadInfo->sockfds, response_buf + received, resp_len - received, 0);
3495
#else
3496
        bytes = read(pThreadInfo->sockfd, response_buf + received, resp_len - received);
3497
#endif
3498
        verbosePrint("%s() LN%d: bytes:%d\n", __func__, __LINE__, bytes);
3499 3500
        if (bytes < 0) {
            free(request_buf);
3501
            ERROR_EXIT("reading response from socket");
3502 3503 3504 3505
        }
        if (bytes == 0)
            break;
        received += bytes;
3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518

        verbosePrint("%s() LN%d: received:%d resp_len:%d, response_buf:\n%s\n",
                __func__, __LINE__, received, resp_len, response_buf);

        if (((NULL != strstr(response_buf, resEncodingChunk))
                    && (NULL != strstr(response_buf, resHttp)))
                || ((NULL != strstr(response_buf, resHttpOk))
                    && (NULL != strstr(response_buf, "\"status\":")))) {
            debugPrint(
                    "%s() LN%d: received:%d resp_len:%d, response_buf:\n%s\n",
                    __func__, __LINE__, received, resp_len, response_buf);
            break;
        } 
3519
    } while(received < resp_len);
3520

3521 3522
    if (received == resp_len) {
        free(request_buf);
3523
        ERROR_EXIT("storing complete response from socket");
3524
    }
H
Haojun Liao 已提交
3525

3526
    response_buf[RESP_BUF_LEN - 1] = '\0';
H
Haojun Liao 已提交
3527

3528
    if (strlen(pThreadInfo->filePath) > 0) {
3529
        appendResultBufToFile(response_buf, pThreadInfo);
3530 3531
    }

3532
    free(request_buf);
3533

3534 3535 3536 3537 3538
    if (NULL == strstr(response_buf, resHttpOk)) {
        errorPrint("%s() LN%d, Response:\n%s\n",
                __func__, __LINE__, response_buf);
        return -1;
    }
3539
    return 0;
3540 3541
}

3542
static char* getTagValueFromTagSample(SSuperTable* stbInfo, int tagUsePos) {
3543 3544
    char*  dataBuf = (char*)calloc(TSDB_MAX_SQL_LEN+1, 1);
    if (NULL == dataBuf) {
3545
        errorPrint2("%s() LN%d, calloc failed! size:%d\n",
3546 3547 3548
                __func__, __LINE__, TSDB_MAX_SQL_LEN+1);
        return NULL;
    }
3549

3550 3551 3552
    int    dataLen = 0;
    dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
            "(%s)", stbInfo->tagDataBuf + stbInfo->lenOfTagOfOneRow * tagUsePos);
3553

3554
    return dataBuf;
3555 3556
}

3557 3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575
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) {
3576 3577 3578
    char*  dataBuf = (char*)calloc(TSDB_MAX_SQL_LEN+1, 1);
    if (NULL == dataBuf) {
        printf("calloc failed! size:%d\n", TSDB_MAX_SQL_LEN+1);
3579
        return NULL;
3580
    }
3581

3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594
    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;
            }
3595

3596 3597
            int32_t tagBufLen = stbInfo->tags[i].dataLen + 1;
            char *buf = generateBinaryNCharTagValues(tableSeq, tagBufLen);
3598 3599 3600 3601 3602
            if (NULL == buf) {
                tmfree(dataBuf);
                return NULL;
            }
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3603
                    "\'%s\',", buf);
3604 3605 3606
            tmfree(buf);
        } else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "int", strlen("int"))) {
3607 3608 3609
            if ((g_args.demo_mode) && (i == 0)) {
                dataLen += snprintf(dataBuf + dataLen,
                        TSDB_MAX_SQL_LEN - dataLen,
3610
                        "%"PRId64",", (tableSeq % 10) + 1);
3611 3612 3613
            } else {
                dataLen += snprintf(dataBuf + dataLen,
                        TSDB_MAX_SQL_LEN - dataLen,
3614
                        "%"PRId64",", tableSeq);
3615
            }
3616 3617 3618
        } else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "bigint", strlen("bigint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3619
                    "%"PRId64",", rand_bigint());
3620 3621 3622
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "float", strlen("float"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3623
                    "%f,", rand_float());
3624 3625 3626
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "double", strlen("double"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3627
                    "%f,", rand_double());
3628 3629 3630
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "smallint", strlen("smallint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3631
                    "%d,", rand_smallint());
3632 3633 3634
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "tinyint", strlen("tinyint"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3635
                    "%d,", rand_tinyint());
3636 3637 3638
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "bool", strlen("bool"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3639
                    "%d,", rand_bool());
3640 3641 3642
        }  else if (0 == strncasecmp(stbInfo->tags[i].dataType,
                    "timestamp", strlen("timestamp"))) {
            dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen,
3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659
                    "%"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());
3660
        }  else {
3661
            errorPrint2("No support data type: %s\n", stbInfo->tags[i].dataType);
3662 3663 3664
            tmfree(dataBuf);
            return NULL;
        }
3665
    }
3666

3667
    dataLen -= 1;
3668 3669
    dataLen += snprintf(dataBuf + dataLen, TSDB_MAX_SQL_LEN - dataLen, ")");
    return dataBuf;
3670 3671
}

3672
static int calcRowLen(SSuperTable*  superTbls) {
3673 3674 3675 3676 3677 3678
    int colIndex;
    int  lenOfOneRow = 0;

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

3679 3680 3681 3682 3683 3684 3685 3686 3687 3688
        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:
3689
            case TSDB_DATA_TYPE_UINT:
3690 3691 3692 3693
                lenOfOneRow += INT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_BIGINT:
3694
            case TSDB_DATA_TYPE_UBIGINT:
3695 3696 3697 3698
                lenOfOneRow += BIGINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
3699
            case TSDB_DATA_TYPE_USMALLINT:
3700 3701 3702 3703
                lenOfOneRow += SMALLINT_BUFF_LEN;
                break;

            case TSDB_DATA_TYPE_TINYINT:
3704
            case TSDB_DATA_TYPE_UTINYINT:
3705 3706 3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725 3726
                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);
3727 3728 3729 3730 3731 3732 3733 3734
        }
    }

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

    int tagIndex;
    int lenOfTagOfOneRow = 0;
    for (tagIndex = 0; tagIndex < superTbls->tagCount; tagIndex++) {
3735 3736 3737 3738
        char * dataType = superTbls->tags[tagIndex].dataType;
        switch (superTbls->tags[tagIndex].data_type)
        {
        case TSDB_DATA_TYPE_BINARY:
3739
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + 3;
3740 3741
            break;
        case TSDB_DATA_TYPE_NCHAR:
3742
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + 3;
3743 3744 3745
            break;
        case TSDB_DATA_TYPE_INT:
        case TSDB_DATA_TYPE_UINT:
3746
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + INT_BUFF_LEN;
3747 3748 3749
            break;
        case TSDB_DATA_TYPE_BIGINT:
        case TSDB_DATA_TYPE_UBIGINT:
3750
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + BIGINT_BUFF_LEN;
3751 3752 3753
            break;
        case TSDB_DATA_TYPE_SMALLINT:
        case TSDB_DATA_TYPE_USMALLINT:
3754
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + SMALLINT_BUFF_LEN;
3755 3756 3757
            break;
        case TSDB_DATA_TYPE_TINYINT:
        case TSDB_DATA_TYPE_UTINYINT:
3758
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + TINYINT_BUFF_LEN;
3759 3760
            break;
        case TSDB_DATA_TYPE_BOOL:
3761
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + BOOL_BUFF_LEN;
3762 3763
            break;
        case TSDB_DATA_TYPE_FLOAT:
3764
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + FLOAT_BUFF_LEN;
3765 3766
            break;
        case TSDB_DATA_TYPE_DOUBLE:
3767
            lenOfTagOfOneRow += superTbls->tags[tagIndex].dataLen + DOUBLE_BUFF_LEN;
3768 3769
            break;
        default:
3770
            errorPrint2("get error tag type : %s\n", dataType);
3771
            exit(EXIT_FAILURE);
3772
        }
3773 3774
    }

3775
    superTbls->lenOfTagOfOneRow = lenOfTagOfOneRow;
3776

3777
    return 0;
3778 3779
}

3780
static int getChildNameOfSuperTableWithLimitAndOffset(TAOS * taos,
3781
        char* dbName, char* stbName, char** childTblNameOfSuperTbl,
3782
        int64_t* childTblCountOfSuperTbl, int64_t limit, uint64_t offset, bool escapChar) {
3783

3784
    char command[1024] = "\0";
3785
    char limitBuf[100] = "\0";
3786

3787 3788
    TAOS_RES * res;
    TAOS_ROW row = NULL;
3789

3790
    char* childTblName = *childTblNameOfSuperTbl;
3791

3792 3793
    snprintf(limitBuf, 100, " limit %"PRId64" offset %"PRIu64"",
        limit, offset);
3794

3795
    //get all child table name use cmd: select tbname from superTblName;
3796 3797
    snprintf(command, 1024, escapChar ? "select tbname from %s.`%s` %s" :
            "select tbname from %s.%s %s", dbName, stbName, limitBuf);
3798

3799 3800 3801 3802
    res = taos_query(taos, command);
    int32_t code = taos_errno(res);
    if (code != 0) {
        taos_free_result(res);
3803
        taos_close(taos);
3804
        errorPrint2("%s() LN%d, failed to run command %s\n",
3805
                __func__, __LINE__, command);
3806
        exit(EXIT_FAILURE);
3807
    }
3808

3809 3810 3811 3812 3813 3814 3815
    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);
3816
            errorPrint2("%s() LN%d, failed to allocate memory!\n", __func__, __LINE__);
3817
            exit(EXIT_FAILURE);
3818
        }
3819 3820
    }

3821 3822 3823 3824 3825
    char* pTblName = childTblName;
    while((row = taos_fetch_row(res)) != NULL) {
        int32_t* len = taos_fetch_lengths(res);

        if (0 == strlen((char *)row[0])) {
3826
            errorPrint2("%s() LN%d, No.%"PRId64" table return empty name\n",
3827
                    __func__, __LINE__, count);
3828
            exit(EXIT_FAILURE);
3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846
        }

        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);
3847
                errorPrint2("%s() LN%d, realloc fail for save child table name of %s.%s\n",
3848
                        __func__, __LINE__, dbName, stbName);
3849
                exit(EXIT_FAILURE);
3850 3851 3852
            }
        }
        pTblName = childTblName + count * TSDB_TABLE_NAME_LEN;
3853
    }
3854

3855 3856
    *childTblCountOfSuperTbl = count;
    *childTblNameOfSuperTbl  = childTblName;
3857

3858 3859
    taos_free_result(res);
    return 0;
3860 3861
}

3862
static int getAllChildNameOfSuperTable(TAOS * taos, char* dbName,
3863
        char* stbName, char** childTblNameOfSuperTbl,
3864
        int64_t* childTblCountOfSuperTbl) {
3865

3866
    return getChildNameOfSuperTableWithLimitAndOffset(taos, dbName, stbName,
3867
            childTblNameOfSuperTbl, childTblCountOfSuperTbl,
3868
            -1, 0, false);
3869 3870
}

3871
static int getSuperTableFromServer(TAOS * taos, char* dbName,
3872
        SSuperTable*  superTbls) {
3873

3874
    char command[1024] = "\0";
3875 3876 3877 3878 3879
    TAOS_RES * res;
    TAOS_ROW row = NULL;
    int count = 0;

    //get schema use cmd: describe superTblName;
3880
    snprintf(command, 1024, "describe %s.%s", dbName, superTbls->stbName);
3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901
    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);
3902
            if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3903 3904
                        "INT", strlen("INT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_INT;
3905
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3906 3907
                        "TINYINT", strlen("TINYINT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_TINYINT;
3908
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3909 3910
                        "SMALLINT", strlen("SMALLINT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_SMALLINT;
3911
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3912 3913
                        "BIGINT", strlen("BIGINT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_BIGINT;
3914
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3915 3916
                        "FLOAT", strlen("FLOAT"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_FLOAT;
3917
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3918 3919
                        "DOUBLE", strlen("DOUBLE"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_DOUBLE;
3920
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3921 3922
                        "BINARY", strlen("BINARY"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_BINARY;
3923
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3924 3925
                        "NCHAR", strlen("NCHAR"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_NCHAR;
3926
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3927 3928
                        "BOOL", strlen("BOOL"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_BOOL;
3929
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3930 3931
                        "TIMESTAMP", strlen("TIMESTAMP"))) {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_TIMESTAMP;
3932 3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955
            } 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);
3956 3957 3958
            } else {
                superTbls->tags[tagIndex].data_type = TSDB_DATA_TYPE_NULL;
            }
3959 3960 3961 3962
            superTbls->tags[tagIndex].dataLen =
                *((int *)row[TSDB_DESCRIBE_METRIC_LENGTH_INDEX]);
            tstrncpy(superTbls->tags[tagIndex].note,
                    (char *)row[TSDB_DESCRIBE_METRIC_NOTE_INDEX],
3963 3964
                    min(NOTE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_NOTE_INDEX].bytes) + 1);
3965 3966 3967 3968 3969 3970 3971
            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);
            }
3972 3973 3974 3975 3976
            tagIndex++;
        } else {
            tstrncpy(superTbls->columns[columnIndex].field,
                    (char *)row[TSDB_DESCRIBE_METRIC_FIELD_INDEX],
                    fields[TSDB_DESCRIBE_METRIC_FIELD_INDEX].bytes);
3977

3978

3979
            if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3980
                        "INT", strlen("INT")) &&
3981
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3982
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_INT;
3983
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3984
                        "TINYINT", strlen("TINYINT")) &&
3985
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3986
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_TINYINT;
3987
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3988
                        "SMALLINT", strlen("SMALLINT")) &&
3989
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3990
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_SMALLINT;
3991
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3992
                        "BIGINT", strlen("BIGINT")) &&
3993
                        strstr((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX], "UNSIGNED") == NULL) {
3994
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_BIGINT;
3995
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3996 3997
                        "FLOAT", strlen("FLOAT"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_FLOAT;
3998
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
3999 4000
                        "DOUBLE", strlen("DOUBLE"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_DOUBLE;
4001
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
4002 4003
                        "BINARY", strlen("BINARY"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_BINARY;
4004
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
4005 4006
                        "NCHAR", strlen("NCHAR"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_NCHAR;
4007
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
4008 4009
                        "BOOL", strlen("BOOL"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_BOOL;
4010
            } else if (0 == strncasecmp((char *)row[TSDB_DESCRIBE_METRIC_TYPE_INDEX],
4011 4012
                        "TIMESTAMP", strlen("TIMESTAMP"))) {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_TIMESTAMP;
4013 4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036
            } 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);
4037 4038 4039
            } else {
                superTbls->columns[columnIndex].data_type = TSDB_DATA_TYPE_NULL;
            }
4040 4041 4042 4043
            superTbls->columns[columnIndex].dataLen =
                *((int *)row[TSDB_DESCRIBE_METRIC_LENGTH_INDEX]);
            tstrncpy(superTbls->columns[columnIndex].note,
                    (char *)row[TSDB_DESCRIBE_METRIC_NOTE_INDEX],
4044 4045
                    min(NOTE_BUFF_LEN,
                        fields[TSDB_DESCRIBE_METRIC_NOTE_INDEX].bytes) + 1);
4046

4047 4048 4049 4050 4051 4052
            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);
            }
4053

4054 4055 4056 4057 4058 4059 4060
            columnIndex++;
        }
        count++;
    }

    superTbls->columnCount = columnIndex;
    superTbls->tagCount    = tagIndex;
4061
    taos_free_result(res);
4062

4063 4064 4065 4066
    calcRowLen(superTbls);

    /*
       if (TBL_ALREADY_EXISTS == superTbls->childTblExists) {
4067
    //get all child table name use cmd: select tbname from superTblName;
4068 4069 4070
    int childTblCount = 10000;
    superTbls->childTblName = (char*)calloc(1, childTblCount * TSDB_TABLE_NAME_LEN);
    if (superTbls->childTblName == NULL) {
4071
    errorPrint2("%s() LN%d, alloc memory failed!\n", __func__, __LINE__);
4072
    return -1;
4073
    }
4074
    getAllChildNameOfSuperTable(taos, dbName,
4075
    superTbls->stbName,
4076 4077 4078 4079 4080
    &superTbls->childTblName,
    &superTbls->childTblCount);
    }
    */
    return 0;
4081 4082
}

H
Haojun Liao 已提交
4083
static int createSuperTable(
4084 4085
        TAOS * taos, char* dbName,
        SSuperTable*  superTbl) {
H
Haojun Liao 已提交
4086

4087 4088
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);
4089

4090
    char cols[COL_BUFFER_LEN] = "\0";
4091
    int len = 0;
4092

4093
    int  lenOfOneRow = 0;
4094

4095
    if (superTbl->columnCount == 0) {
4096
        errorPrint2("%s() LN%d, super table column count is %d\n",
4097
                __func__, __LINE__, superTbl->columnCount);
4098
        free(command);
4099 4100
        return -1;
    }
4101

4102
    for (int colIndex = 0; colIndex < superTbl->columnCount; colIndex++) {
4103

4104 4105
        switch(superTbl->columns[colIndex].data_type) {
            case TSDB_DATA_TYPE_BINARY:
4106
                len += snprintf(cols + len, COL_BUFFER_LEN - len,
4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124
                        ",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");
4125
                }
4126 4127
                lenOfOneRow += INT_BUFF_LEN;
                break;
4128

4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176
            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;

4177 4178 4179 4180 4181 4182 4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199 4200
            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;

4201 4202 4203 4204 4205 4206
            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);
4207
        }
4208 4209
    }

4210
    superTbl->lenOfOneRow = lenOfOneRow + 20; // timestamp
4211

4212 4213 4214 4215
    // save for creating child table
    superTbl->colsOfCreateChildTable = (char*)calloc(len+20, 1);
    if (NULL == superTbl->colsOfCreateChildTable) {
        taos_close(taos);
4216
        free(command);
4217
        errorPrint2("%s() LN%d, Failed when calloc, size:%d",
4218 4219
                __func__, __LINE__, len+1);
        exit(EXIT_FAILURE);
4220
    }
4221

4222 4223 4224
    snprintf(superTbl->colsOfCreateChildTable, len+20, "(ts timestamp%s)", cols);
    verbosePrint("%s() LN%d: %s\n",
            __func__, __LINE__, superTbl->colsOfCreateChildTable);
4225

4226
    if (superTbl->tagCount == 0) {
4227
        errorPrint2("%s() LN%d, super table tag count is %d\n",
4228
                __func__, __LINE__, superTbl->tagCount);
4229
        free(command);
4230 4231
        return -1;
    }
4232

4233
    char tags[TSDB_MAX_TAGS_LEN] = "\0";
4234 4235
    int tagIndex;
    len = 0;
4236

4237
    int lenOfTagOfOneRow = 0;
4238
    len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len, "(");
4239 4240 4241 4242
    for (tagIndex = 0; tagIndex < superTbl->tagCount; tagIndex++) {
        char* dataType = superTbl->tags[tagIndex].dataType;

        if (strcasecmp(dataType, "BINARY") == 0) {
4243
            if ((g_args.demo_mode) && (tagIndex == 1)) {
4244
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4245
                        "location BINARY(%d),",
4246 4247
                        superTbl->tags[tagIndex].dataLen);
            } else {
4248
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4249
                        "T%d %s(%d),", tagIndex, "BINARY",
4250
                        superTbl->tags[tagIndex].dataLen);
4251
            }
4252 4253
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + 3;
        } else if (strcasecmp(dataType, "NCHAR") == 0) {
4254
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4255
                    "T%d %s(%d),", tagIndex,
4256 4257 4258
                    "NCHAR", superTbl->tags[tagIndex].dataLen);
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + 3;
        } else if (strcasecmp(dataType, "INT") == 0)  {
4259
            if ((g_args.demo_mode) && (tagIndex == 0)) {
4260 4261
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                        "groupId INT, ");
4262
            } else {
4263
                len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4264
                        "T%d %s,", tagIndex, "INT");
4265
            }
4266
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + INT_BUFF_LEN;
4267
        } else if (strcasecmp(dataType, "BIGINT") == 0)  {
4268
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4269
                    "T%d %s,", tagIndex, "BIGINT");
4270
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + BIGINT_BUFF_LEN;
4271
        } else if (strcasecmp(dataType, "SMALLINT") == 0)  {
4272
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4273
                    "T%d %s,", tagIndex, "SMALLINT");
4274
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + SMALLINT_BUFF_LEN;
4275
        } else if (strcasecmp(dataType, "TINYINT") == 0)  {
4276
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4277
                    "T%d %s,", tagIndex, "TINYINT");
4278
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + TINYINT_BUFF_LEN;
4279
        } else if (strcasecmp(dataType, "BOOL") == 0)  {
4280
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4281
                    "T%d %s,", tagIndex, "BOOL");
4282
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + BOOL_BUFF_LEN;
4283
        } else if (strcasecmp(dataType, "FLOAT") == 0) {
4284
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4285
                    "T%d %s,", tagIndex, "FLOAT");
4286
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + FLOAT_BUFF_LEN;
4287
        } else if (strcasecmp(dataType, "DOUBLE") == 0) {
4288
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
4289
                    "T%d %s,", tagIndex, "DOUBLE");
4290
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + DOUBLE_BUFF_LEN;
4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306
        } 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;
4307 4308 4309 4310
        } else if (strcasecmp(dataType, "TIMESTAMP") == 0) {
            len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len,
                    "T%d %s,", tagIndex, "TIMESTAMP");
            lenOfTagOfOneRow += superTbl->tags[tagIndex].dataLen + TIMESTAMP_BUFF_LEN;
4311 4312
        } else {
            taos_close(taos);
4313
            free(command);
4314
            errorPrint2("%s() LN%d, config error tag type : %s\n",
4315
                    __func__, __LINE__, dataType);
4316
            exit(EXIT_FAILURE);
4317
        }
4318
    }
4319

4320
    len -= 1;
4321
    len += snprintf(tags + len, TSDB_MAX_TAGS_LEN - len, ")");
4322

4323
    superTbl->lenOfTagOfOneRow = lenOfTagOfOneRow;
4324

4325
    
4326
    snprintf(command, BUFFER_SIZE,
4327 4328 4329 4330
        superTbl->escapeChar ?
        "CREATE TABLE IF NOT EXISTS %s.`%s` (ts TIMESTAMP%s) TAGS %s":
        "CREATE TABLE IF NOT EXISTS %s.%s (ts TIMESTAMP%s) TAGS %s",
        dbName, superTbl->stbName, cols, tags);
4331
    if (0 != queryDbExec(taos, command, NO_INSERT_TYPE, false)) {
4332
        errorPrint2("create supertable %s failed!\n\n",
4333
                superTbl->stbName);
4334
        free(command);
4335 4336
        return -1;
    }
4337

4338
    debugPrint("create supertable %s success!\n\n", superTbl->stbName);
4339
    free(command);
4340
    return 0;
4341 4342
}

4343
int createDatabasesAndStables(char *command) {
4344 4345 4346 4347
    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) {
4348
        errorPrint2("Failed to connect to TDengine, reason:%s\n", taos_errstr(NULL));
4349
        return -1;
4350
    }
4351

4352 4353 4354 4355 4356 4357 4358
    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;
            }
4359

4360 4361
            int dataLen = 0;
            dataLen += snprintf(command + dataLen,
4362
                    BUFFER_SIZE - dataLen, "CREATE DATABASE IF NOT EXISTS %s",
4363
                    g_Dbs.db[i].dbName);
4364

4365 4366
            if (g_Dbs.db[i].dbCfg.blocks > 0) {
                dataLen += snprintf(command + dataLen,
4367
                        BUFFER_SIZE - dataLen, " BLOCKS %d",
4368
                        g_Dbs.db[i].dbCfg.blocks);
4369 4370 4371
            }
            if (g_Dbs.db[i].dbCfg.cache > 0) {
                dataLen += snprintf(command + dataLen,
4372
                        BUFFER_SIZE - dataLen, " CACHE %d",
4373
                        g_Dbs.db[i].dbCfg.cache);
4374 4375 4376
            }
            if (g_Dbs.db[i].dbCfg.days > 0) {
                dataLen += snprintf(command + dataLen,
4377
                        BUFFER_SIZE - dataLen, " DAYS %d",
4378
                        g_Dbs.db[i].dbCfg.days);
4379 4380 4381
            }
            if (g_Dbs.db[i].dbCfg.keep > 0) {
                dataLen += snprintf(command + dataLen,
4382
                        BUFFER_SIZE - dataLen, " KEEP %d",
4383
                        g_Dbs.db[i].dbCfg.keep);
4384 4385 4386
            }
            if (g_Dbs.db[i].dbCfg.quorum > 1) {
                dataLen += snprintf(command + dataLen,
4387
                        BUFFER_SIZE - dataLen, " QUORUM %d",
4388
                        g_Dbs.db[i].dbCfg.quorum);
4389 4390 4391
            }
            if (g_Dbs.db[i].dbCfg.replica > 0) {
                dataLen += snprintf(command + dataLen,
4392
                        BUFFER_SIZE - dataLen, " REPLICA %d",
4393
                        g_Dbs.db[i].dbCfg.replica);
4394 4395 4396
            }
            if (g_Dbs.db[i].dbCfg.update > 0) {
                dataLen += snprintf(command + dataLen,
4397
                        BUFFER_SIZE - dataLen, " UPDATE %d",
4398
                        g_Dbs.db[i].dbCfg.update);
4399 4400 4401 4402 4403 4404 4405
            }
            //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,
4406
                        BUFFER_SIZE - dataLen, " MINROWS %d",
4407
                        g_Dbs.db[i].dbCfg.minRows);
4408 4409 4410
            }
            if (g_Dbs.db[i].dbCfg.maxRows > 0) {
                dataLen += snprintf(command + dataLen,
4411
                        BUFFER_SIZE - dataLen, " MAXROWS %d",
4412
                        g_Dbs.db[i].dbCfg.maxRows);
4413 4414 4415
            }
            if (g_Dbs.db[i].dbCfg.comp > 0) {
                dataLen += snprintf(command + dataLen,
4416
                        BUFFER_SIZE - dataLen, " COMP %d",
4417
                        g_Dbs.db[i].dbCfg.comp);
4418 4419 4420
            }
            if (g_Dbs.db[i].dbCfg.walLevel > 0) {
                dataLen += snprintf(command + dataLen,
4421 4422
                        BUFFER_SIZE - dataLen, " wal %d",
                        g_Dbs.db[i].dbCfg.walLevel);
4423 4424 4425
            }
            if (g_Dbs.db[i].dbCfg.cacheLast > 0) {
                dataLen += snprintf(command + dataLen,
4426
                        BUFFER_SIZE - dataLen, " CACHELAST %d",
4427
                        g_Dbs.db[i].dbCfg.cacheLast);
4428 4429 4430
            }
            if (g_Dbs.db[i].dbCfg.fsync > 0) {
                dataLen += snprintf(command + dataLen, BUFFER_SIZE - dataLen,
4431
                        " FSYNC %d", g_Dbs.db[i].dbCfg.fsync);
4432
            }
4433
            if ((0 == strncasecmp(g_Dbs.db[i].dbCfg.precision, "ms", 2))
4434
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision,
4435
                            "ns", 2))
4436
                    || (0 == strncasecmp(g_Dbs.db[i].dbCfg.precision,
4437
                            "us", 2))) {
4438 4439 4440
                dataLen += snprintf(command + dataLen, BUFFER_SIZE - dataLen,
                        " precision \'%s\';", g_Dbs.db[i].dbCfg.precision);
            }
4441

4442 4443
            if (0 != queryDbExec(taos, command, NO_INSERT_TYPE, false)) {
                taos_close(taos);
4444
                errorPrint("\ncreate database %s failed!\n\n",
4445
                        g_Dbs.db[i].dbName);
4446 4447 4448 4449
                return -1;
            }
            printf("\ncreate database %s success!\n\n", g_Dbs.db[i].dbName);
        }
4450

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

4454
        int validStbCount = 0;
4455

4456 4457
        for (uint64_t j = 0; j < g_Dbs.db[i].superTblCount; j++) {
            sprintf(command, "describe %s.%s;", g_Dbs.db[i].dbName,
4458
                    g_Dbs.db[i].superTbls[j].stbName);
4459
            ret = queryDbExec(taos, command, NO_INSERT_TYPE, true);
4460

4461 4462 4463
            if ((ret != 0) || (g_Dbs.db[i].drop)) {
                ret = createSuperTable(taos, g_Dbs.db[i].dbName,
                        &g_Dbs.db[i].superTbls[j]);
4464

4465 4466 4467 4468
                if (0 != ret) {
                    errorPrint("create super table %"PRIu64" failed!\n\n", j);
                    continue;
                }
4469 4470
            } else {
                ret = getSuperTableFromServer(taos, g_Dbs.db[i].dbName,
4471
                    &g_Dbs.db[i].superTbls[j]);
4472 4473 4474 4475 4476
                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;
                }
4477 4478 4479 4480 4481 4482 4483 4484
            }
            validStbCount ++;
        }
        g_Dbs.db[i].superTblCount = validStbCount;
    }

    taos_close(taos);
    return 0;
4485 4486
}

4487 4488
static void* createTable(void *sarg)
{
4489
    threadInfo *pThreadInfo = (threadInfo *)sarg;
4490
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
4491

4492 4493
    setThreadName("createTable");

4494
    uint64_t  lastPrintTime = taosGetTimestampMs();
4495

4496
    int buff_len = BUFFER_SIZE;
4497

4498 4499
    pThreadInfo->buffer = calloc(buff_len, 1);
    if (pThreadInfo->buffer == NULL) {
4500
        errorPrint2("%s() LN%d, Memory allocated failed!\n", __func__, __LINE__);
4501
        exit(EXIT_FAILURE);
4502 4503 4504 4505 4506 4507 4508 4509 4510 4511 4512 4513 4514
    }

    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,
4515 4516
                    g_args.escapeChar ? 
                    "CREATE TABLE IF NOT EXISTS %s.`%s%"PRIu64"` %s;" :
4517
                    "CREATE TABLE IF NOT EXISTS %s.%s%"PRIu64" %s;",
4518 4519 4520
                    pThreadInfo->db_name,
                    g_args.tb_prefix, i,
                    pThreadInfo->cols);
4521
            batchNum ++;
4522
        } else {
4523 4524
            if (stbInfo == NULL) {
                free(pThreadInfo->buffer);
4525
                errorPrint2("%s() LN%d, use metric, but super table info is NULL\n",
4526
                        __func__, __LINE__);
4527
                exit(EXIT_FAILURE);
4528 4529 4530 4531 4532
            } else {
                if (0 == len) {
                    batchNum = 0;
                    memset(pThreadInfo->buffer, 0, buff_len);
                    len += snprintf(pThreadInfo->buffer + len,
4533
                            buff_len - len, "CREATE TABLE ");
4534
                }
4535

4536
                char* tagsValBuf = NULL;
4537 4538
                if (0 == stbInfo->tagSource) {
                    tagsValBuf = generateTagValuesForStb(stbInfo, i);
4539
                } else {
4540 4541 4542 4543
                    if (0 == stbInfo->tagSampleCount) {
                        free(pThreadInfo->buffer);
                        ERROR_EXIT("use sample file for tag, but has no content!\n");
                    }
4544
                    tagsValBuf = getTagValueFromTagSample(
4545 4546
                            stbInfo,
                            i % stbInfo->tagSampleCount);
4547
                }
4548

4549 4550
                if (NULL == tagsValBuf) {
                    free(pThreadInfo->buffer);
4551
                    ERROR_EXIT("use metric, but tag buffer is NULL\n");
4552 4553
                }
                len += snprintf(pThreadInfo->buffer + len,
4554 4555
                        buff_len - len, stbInfo->escapeChar ?
                        "if not exists %s.`%s%"PRIu64"` using %s.`%s` tags %s " :
4556
                        "if not exists %s.%s%"PRIu64" using %s.%s tags %s ",
4557
                        pThreadInfo->db_name, stbInfo->childTblPrefix,
4558
                        i, pThreadInfo->db_name,
4559
                        stbInfo->stbName, tagsValBuf);
4560 4561
                free(tagsValBuf);
                batchNum++;
4562
                if ((batchNum < stbInfo->batchCreateTableNum)
4563
                        && ((buff_len - len)
4564
                            >= (stbInfo->lenOfTagOfOneRow + 256))) {
4565 4566 4567
                    continue;
                }
            }
4568
        }
4569

4570
        len = 0;
S
Shengliang Guan 已提交
4571

4572
        if (0 != queryDbExec(pThreadInfo->taos, pThreadInfo->buffer,
4573
                    NO_INSERT_TYPE, false)) {
4574
            errorPrint2("queryDbExec() failed. buffer:\n%s\n", pThreadInfo->buffer);
4575 4576 4577
            free(pThreadInfo->buffer);
            return NULL;
        }
4578 4579
        pThreadInfo->tables_created += batchNum;
        uint64_t currentPrintTime = taosGetTimestampMs();
4580 4581 4582 4583 4584
        if (currentPrintTime - lastPrintTime > 30*1000) {
            printf("thread[%d] already create %"PRIu64" - %"PRIu64" tables\n",
                    pThreadInfo->threadID, pThreadInfo->start_table_from, i);
            lastPrintTime = currentPrintTime;
        }
4585
    }
4586

4587 4588 4589
    if (0 != len) {
        if (0 != queryDbExec(pThreadInfo->taos, pThreadInfo->buffer,
                    NO_INSERT_TYPE, false)) {
4590
            errorPrint2("queryDbExec() failed. buffer:\n%s\n", pThreadInfo->buffer);
4591
        }
4592
        pThreadInfo->tables_created += batchNum;
4593
    }
4594 4595
    free(pThreadInfo->buffer);
    return NULL;
4596 4597
}

4598
static int startMultiThreadCreateChildTable(
4599
        char* cols, int threads, uint64_t tableFrom, int64_t ntables,
4600
        char* db_name, SSuperTable* stbInfo) {
4601

4602 4603
    pthread_t *pids = calloc(1, threads * sizeof(pthread_t));
    threadInfo *infos = calloc(1, threads * sizeof(threadInfo));
4604

4605
    if ((NULL == pids) || (NULL == infos)) {
4606
        ERROR_EXIT("createChildTable malloc failed\n");
4607
    }
4608

4609 4610 4611
    if (threads < 1) {
        threads = 1;
    }
4612

4613 4614 4615 4616
    int64_t a = ntables / threads;
    if (a < 1) {
        threads = ntables;
        a = 1;
4617
    }
4618

4619 4620
    int64_t b = 0;
    b = ntables % threads;
4621

4622 4623 4624 4625
    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);
4626
        pThreadInfo->stbInfo = stbInfo;
4627 4628 4629 4630 4631 4632 4633 4634
        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) {
4635
            errorPrint2("%s() LN%d, Failed to connect to TDengine, reason:%s\n",
4636 4637 4638 4639 4640
                    __func__, __LINE__, taos_errstr(NULL));
            free(pids);
            free(infos);
            return -1;
        }
4641

4642 4643 4644 4645 4646 4647 4648
        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;
4649
        pThreadInfo->tables_created = 0;
4650 4651
        pthread_create(pids + i, NULL, createTable, pThreadInfo);
    }
4652

4653 4654 4655
    for (int i = 0; i < threads; i++) {
        pthread_join(pids[i], NULL);
    }
4656

4657 4658 4659
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
        taos_close(pThreadInfo->taos);
4660 4661

        g_actualChildTables += pThreadInfo->tables_created;
4662 4663 4664 4665 4666 4667
    }

    free(pids);
    free(infos);

    return 0;
4668 4669
}

4670
static void createChildTables() {
4671
    char tblColsBuf[TSDB_MAX_BYTES_PER_ROW];
4672
    int len;
4673

4674 4675 4676 4677 4678
    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++) {
4679 4680 4681 4682
                    if ((AUTO_CREATE_SUBTBL
                                == g_Dbs.db[i].superTbls[j].autoCreateTable)
                            || (TBL_ALREADY_EXISTS
                                == g_Dbs.db[i].superTbls[j].childTblExists)) {
4683 4684 4685
                        continue;
                    }
                    verbosePrint("%s() LN%d: %s\n", __func__, __LINE__,
4686
                            g_Dbs.db[i].superTbls[j].colsOfCreateChildTable);
4687 4688 4689 4690 4691 4692 4693
                    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,
4694
                            g_Dbs.threadCountForCreateTbl,
4695 4696 4697 4698 4699 4700 4701
                            startFrom,
                            g_Dbs.db[i].superTbls[j].childTblCount,
                            g_Dbs.db[i].dbName, &(g_Dbs.db[i].superTbls[j]));
                }
            }
        } else {
            // normal table
4702
            len = snprintf(tblColsBuf, TSDB_MAX_BYTES_PER_ROW, "(TS TIMESTAMP");
4703 4704 4705
            for (int j = 0; j < g_args.columnCount; j++) {
                if ((strncasecmp(g_args.dataType[j], "BINARY", strlen("BINARY")) == 0)
                        || (strncasecmp(g_args.dataType[j],
4706
                                "NCHAR", strlen("NCHAR")) == 0)) {
4707
                    snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len,
4708
                            ",C%d %s(%d)", j, g_args.dataType[j], g_args.binwidth);
4709
                } else {
4710
                    snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len,
4711
                            ",C%d %s", j, g_args.dataType[j]);
4712 4713 4714
                }
                len = strlen(tblColsBuf);
            }
4715

4716
            snprintf(tblColsBuf + len, TSDB_MAX_BYTES_PER_ROW - len, ")");
4717 4718 4719

            verbosePrint("%s() LN%d: dbName: %s num of tb: %"PRId64" schema: %s\n",
                    __func__, __LINE__,
4720
                    g_Dbs.db[i].dbName, g_args.ntables, tblColsBuf);
4721 4722
            startMultiThreadCreateChildTable(
                    tblColsBuf,
4723
                    g_Dbs.threadCountForCreateTbl,
4724
                    0,
4725
                    g_args.ntables,
4726 4727
                    g_Dbs.db[i].dbName,
                    NULL);
4728
        }
4729
    }
4730 4731 4732
}

/*
4733 4734
   Read 10000 lines at most. If more than 10000 lines, continue to read after using
   */
4735
static int readTagFromCsvFileToMem(SSuperTable  * stbInfo) {
4736 4737 4738
    size_t  n = 0;
    ssize_t readLen = 0;
    char *  line = NULL;
4739

4740
    FILE *fp = fopen(stbInfo->tagsFile, "r");
4741 4742
    if (fp == NULL) {
        printf("Failed to open tags file: %s, reason:%s\n",
4743
                stbInfo->tagsFile, strerror(errno));
4744
        return -1;
4745 4746
    }

4747 4748 4749
    if (stbInfo->tagDataBuf) {
        free(stbInfo->tagDataBuf);
        stbInfo->tagDataBuf = NULL;
4750 4751
    }

4752 4753
    int tagCount = 10000;
    int count = 0;
4754
    char* tagDataBuf = calloc(1, stbInfo->lenOfTagOfOneRow * tagCount);
4755 4756
    if (tagDataBuf == NULL) {
        printf("Failed to calloc, reason:%s\n", strerror(errno));
4757 4758 4759 4760
        fclose(fp);
        return -1;
    }

4761 4762 4763 4764 4765 4766 4767 4768 4769
    while((readLen = tgetline(&line, &n, fp)) != -1) {
        if (('\r' == line[readLen - 1]) || ('\n' == line[readLen - 1])) {
            line[--readLen] = 0;
        }

        if (readLen == 0) {
            continue;
        }

4770
        memcpy(tagDataBuf + count * stbInfo->lenOfTagOfOneRow, line, readLen);
4771 4772 4773 4774
        count++;

        if (count >= tagCount - 1) {
            char *tmp = realloc(tagDataBuf,
4775
                    (size_t)tagCount*1.5*stbInfo->lenOfTagOfOneRow);
4776 4777 4778
            if (tmp != NULL) {
                tagDataBuf = tmp;
                tagCount = (int)(tagCount*1.5);
4779 4780
                memset(tagDataBuf + count*stbInfo->lenOfTagOfOneRow,
                        0, (size_t)((tagCount-count)*stbInfo->lenOfTagOfOneRow));
4781 4782
            } else {
                // exit, if allocate more memory failed
4783
                printf("realloc fail for save tag val from %s\n", stbInfo->tagsFile);
4784 4785 4786 4787 4788 4789 4790
                tmfree(tagDataBuf);
                free(line);
                fclose(fp);
                return -1;
            }
        }
    }
4791

4792 4793
    stbInfo->tagDataBuf = tagDataBuf;
    stbInfo->tagSampleCount = count;
4794 4795 4796 4797

    free(line);
    fclose(fp);
    return 0;
4798 4799
}

4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813 4814 4815 4816
static void getAndSetRowsFromCsvFile(SSuperTable *stbInfo) {
    FILE *fp = fopen(stbInfo->sampleFile, "r");
    int line_count = 0;
    if (fp == NULL) {
        errorPrint("Failed to open sample file: %s, reason:%s\n",
                stbInfo->sampleFile, strerror(errno));
        exit(EXIT_FAILURE);
    }
    char *buf = calloc(1, stbInfo->maxSqlLen);
    while (fgets(buf, stbInfo->maxSqlLen, fp)) {
        line_count++;
    }
    fclose(fp);
    tmfree(buf);
    stbInfo->insertRows = line_count;
}

4817
/*
4818 4819
   Read 10000 lines at most. If more than 10000 lines, continue to read after using
   */
4820
static int generateSampleFromCsvForStb(
4821
        SSuperTable* stbInfo) {
4822 4823 4824 4825 4826
    size_t  n = 0;
    ssize_t readLen = 0;
    char *  line = NULL;
    int getRows = 0;

4827
    FILE*  fp = fopen(stbInfo->sampleFile, "r");
4828
    if (fp == NULL) {
4829
        errorPrint("Failed to open sample file: %s, reason:%s\n",
4830
                stbInfo->sampleFile, strerror(errno));
4831 4832
        return -1;
    }
4833

4834 4835
    assert(stbInfo->sampleDataBuf);
    memset(stbInfo->sampleDataBuf, 0,
4836
            MAX_SAMPLES * stbInfo->lenOfOneRow);
4837 4838 4839 4840
    while(1) {
        readLen = tgetline(&line, &n, fp);
        if (-1 == readLen) {
            if(0 != fseek(fp, 0, SEEK_SET)) {
4841
                errorPrint("Failed to fseek file: %s, reason:%s\n",
4842
                        stbInfo->sampleFile, strerror(errno));
4843 4844 4845 4846 4847
                fclose(fp);
                return -1;
            }
            continue;
        }
4848

4849 4850 4851
        if (('\r' == line[readLen - 1]) || ('\n' == line[readLen - 1])) {
            line[--readLen] = 0;
        }
4852

4853 4854 4855
        if (readLen == 0) {
            continue;
        }
4856

4857
        if (readLen > stbInfo->lenOfOneRow) {
4858
            printf("sample row len[%d] overflow define schema len[%"PRIu64"], so discard this row\n",
4859
                    (int32_t)readLen, stbInfo->lenOfOneRow);
4860 4861
            continue;
        }
4862

4863
        memcpy(stbInfo->sampleDataBuf + getRows * stbInfo->lenOfOneRow,
4864 4865 4866
                line, readLen);
        getRows++;

4867
        if (getRows == MAX_SAMPLES) {
4868 4869
            break;
        }
4870 4871
    }

4872 4873 4874
    fclose(fp);
    tmfree(line);
    return 0;
4875 4876
}

4877
static bool getColumnAndTagTypeFromInsertJsonFile(
4878
        cJSON* stbInfo, SSuperTable* superTbls) {
4879 4880 4881 4882 4883
    bool  ret = false;

    // columns
    cJSON *columns = cJSON_GetObjectItem(stbInfo, "columns");
    if (columns && columns->type != cJSON_Array) {
4884
        errorPrint("%s", "failed to read json, columns not found\n");
4885 4886 4887 4888 4889 4890
        goto PARSE_OVER;
    } else if (NULL == columns) {
        superTbls->columnCount = 0;
        superTbls->tagCount    = 0;
        return true;
    }
4891

4892 4893
    int columnSize = cJSON_GetArraySize(columns);
    if ((columnSize + 1/* ts */) > TSDB_MAX_COLUMNS) {
4894 4895
        errorPrint("failed to read json, column size overflow, max column size is %d\n",
                TSDB_MAX_COLUMNS);
4896 4897
        goto PARSE_OVER;
    }
4898

4899 4900 4901
    int count = 1;
    int index = 0;
    StrColumn    columnCase;
4902

4903 4904 4905 4906
    //superTbls->columnCount = columnSize;
    for (int k = 0; k < columnSize; ++k) {
        cJSON* column = cJSON_GetArrayItem(columns, k);
        if (column == NULL) continue;
4907

4908 4909 4910 4911 4912
        count = 1;
        cJSON* countObj = cJSON_GetObjectItem(column, "count");
        if (countObj && countObj->type == cJSON_Number) {
            count = countObj->valueint;
        } else if (countObj && countObj->type != cJSON_Number) {
4913
            errorPrint("%s", "failed to read json, column count not found\n");
4914 4915 4916 4917 4918 4919 4920 4921 4922 4923
            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) {
4924
            errorPrint("%s", "failed to read json, column type not found\n");
4925 4926
            goto PARSE_OVER;
        }
4927 4928 4929
        //tstrncpy(superTbls->columns[k].dataType, dataType->valuestring, DATATYPE_BUFF_LEN);
        tstrncpy(columnCase.dataType, dataType->valuestring,
                min(DATATYPE_BUFF_LEN, strlen(dataType->valuestring) + 1));
4930 4931 4932 4933 4934 4935 4936 4937 4938

        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 {
4939
            columnCase.dataLen = SMALL_BUFF_LEN;
4940 4941 4942 4943
        }

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

4947 4948 4949 4950 4951 4952
            superTbls->columns[index].dataLen = columnCase.dataLen;
            index++;
        }
    }

    if ((index + 1 /* ts */) > MAX_NUM_COLUMNS) {
4953 4954
        errorPrint("failed to read json, column size overflow, allowed max column size is %d\n",
                MAX_NUM_COLUMNS);
4955 4956 4957 4958
        goto PARSE_OVER;
    }

    superTbls->columnCount = index;
4959

4960 4961 4962 4963 4964 4965 4966 4967 4968 4969 4970 4971 4972 4973 4974 4975 4976 4977 4978 4979 4980 4981 4982 4983 4984 4985 4986 4987 4988 4989 4990
    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;
4991 4992 4993 4994 4995 4996 4997 4998 4999 5000 5001 5002
        } 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;
5003 5004 5005 5006 5007
        } else {
            superTbls->columns[c].data_type = TSDB_DATA_TYPE_NULL;
        }
    }

5008
    count = 1;
5009 5010 5011 5012
    index = 0;
    // tags
    cJSON *tags = cJSON_GetObjectItem(stbInfo, "tags");
    if (!tags || tags->type != cJSON_Array) {
5013
        errorPrint("%s", "failed to read json, tags not found\n");
5014 5015 5016 5017 5018
        goto PARSE_OVER;
    }

    int tagSize = cJSON_GetArraySize(tags);
    if (tagSize > TSDB_MAX_TAGS) {
5019 5020
        errorPrint("failed to read json, tags size overflow, max tag size is %d\n",
                TSDB_MAX_TAGS);
5021
        goto PARSE_OVER;
5022 5023
    }

5024 5025 5026 5027 5028 5029 5030 5031 5032 5033
    //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) {
5034
            errorPrint("%s", "failed to read json, column count not found\n");
5035 5036 5037 5038 5039 5040 5041 5042 5043 5044
            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) {
5045
            errorPrint("%s", "failed to read json, tag type not found\n");
5046 5047
            goto PARSE_OVER;
        }
5048 5049
        tstrncpy(columnCase.dataType, dataType->valuestring,
                min(DATATYPE_BUFF_LEN, strlen(dataType->valuestring) + 1));
5050 5051 5052 5053 5054

        cJSON* dataLen = cJSON_GetObjectItem(tag, "len");
        if (dataLen && dataLen->type == cJSON_Number) {
            columnCase.dataLen = dataLen->valueint;
        } else if (dataLen && dataLen->type != cJSON_Number) {
5055
            errorPrint("%s", "failed to read json, column len not found\n");
5056 5057 5058 5059 5060 5061 5062
            goto PARSE_OVER;
        } else {
            columnCase.dataLen = 0;
        }

        for (int n = 0; n < count; ++n) {
            tstrncpy(superTbls->tags[index].dataType, columnCase.dataType,
5063
                    min(DATATYPE_BUFF_LEN, strlen(columnCase.dataType) + 1));
5064 5065 5066
            superTbls->tags[index].dataLen = columnCase.dataLen;
            index++;
        }
5067
    }
5068

5069
    if (index > TSDB_MAX_TAGS) {
5070 5071
        errorPrint("failed to read json, tags size overflow, allowed max tag count is %d\n",
                TSDB_MAX_TAGS);
5072 5073
        goto PARSE_OVER;
    }
5074

5075
    superTbls->tagCount = index;
5076

5077 5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090 5091 5092 5093 5094 5095 5096 5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107
    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;
5108 5109 5110 5111 5112 5113 5114 5115 5116 5117 5118 5119
        } 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;
5120 5121 5122 5123 5124
        } else {
            superTbls->tags[t].data_type = TSDB_DATA_TYPE_NULL;
        }
    }

5125
    if ((superTbls->columnCount + superTbls->tagCount + 1 /* ts */) > TSDB_MAX_COLUMNS) {
5126 5127
        errorPrint("columns + tags is more than allowed max columns count: %d\n",
                TSDB_MAX_COLUMNS);
5128 5129 5130
        goto PARSE_OVER;
    }
    ret = true;
5131

5132
PARSE_OVER:
5133
    return ret;
5134 5135 5136
}

static bool getMetaFromInsertJsonFile(cJSON* root) {
5137
    bool  ret = false;
5138

5139 5140 5141
    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);
5142
    }
5143

5144 5145 5146 5147 5148
    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);
5149
    } else {
5150
        errorPrint("%s", "failed to read json, host not found\n");
5151
        goto PARSE_OVER;
5152 5153
    }

5154 5155 5156 5157 5158
    cJSON* port = cJSON_GetObjectItem(root, "port");
    if (port && port->type == cJSON_Number) {
        g_Dbs.port = port->valueint;
    } else if (!port) {
        g_Dbs.port = 6030;
5159
    }
5160

5161 5162 5163 5164 5165
    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);
5166 5167
    }

5168 5169
    cJSON* password = cJSON_GetObjectItem(root, "password");
    if (password && password->type == cJSON_String && password->valuestring != NULL) {
5170
        tstrncpy(g_Dbs.password, password->valuestring, SHELL_MAX_PASSWORD_LEN);
5171
    } else if (!password) {
5172
        tstrncpy(g_Dbs.password, "taosdata", SHELL_MAX_PASSWORD_LEN);
5173 5174
    }

5175 5176 5177 5178 5179
    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);
5180 5181
    }

5182 5183 5184 5185 5186
    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;
5187
    } else {
5188
        errorPrint("%s", "failed to read json, threads not found\n");
5189
        goto PARSE_OVER;
5190
    }
5191

5192 5193
    cJSON* threads2 = cJSON_GetObjectItem(root, "thread_count_create_tbl");
    if (threads2 && threads2->type == cJSON_Number) {
5194
        g_Dbs.threadCountForCreateTbl = threads2->valueint;
5195
    } else if (!threads2) {
5196
        g_Dbs.threadCountForCreateTbl = 1;
5197
    } else {
5198
        errorPrint("%s", "failed to read json, threads2 not found\n");
5199
        goto PARSE_OVER;
5200
    }
5201

5202 5203 5204
    cJSON* gInsertInterval = cJSON_GetObjectItem(root, "insert_interval");
    if (gInsertInterval && gInsertInterval->type == cJSON_Number) {
        if (gInsertInterval->valueint <0) {
5205
            errorPrint("%s", "failed to read json, insert interval input mistake\n");
5206 5207 5208 5209 5210
            goto PARSE_OVER;
        }
        g_args.insert_interval = gInsertInterval->valueint;
    } else if (!gInsertInterval) {
        g_args.insert_interval = 0;
5211
    } else {
5212
        errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5213
        goto PARSE_OVER;
5214
    }
5215

5216 5217 5218
    cJSON* interlaceRows = cJSON_GetObjectItem(root, "interlace_rows");
    if (interlaceRows && interlaceRows->type == cJSON_Number) {
        if (interlaceRows->valueint < 0) {
5219
            errorPrint("%s", "failed to read json, interlaceRows input mistake\n");
5220
            goto PARSE_OVER;
5221

5222
        }
5223
        g_args.interlaceRows = interlaceRows->valueint;
5224
    } else if (!interlaceRows) {
5225
        g_args.interlaceRows = 0; // 0 means progressive mode, > 0 mean interlace mode. max value is less or equ num_of_records_per_req
5226
    } else {
5227
        errorPrint("%s", "failed to read json, interlaceRows input mistake\n");
5228
        goto PARSE_OVER;
5229 5230
    }

5231 5232 5233 5234 5235 5236 5237 5238 5239 5240
    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);
5241
    } else {
5242 5243 5244
        errorPrint("%s() LN%d, failed to read json, max_sql_len input mistake\n",
                __func__, __LINE__);
        goto PARSE_OVER;
5245 5246
    }

5247 5248 5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260
    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;
        }
5261
        g_args.reqPerReq = numRecPerReq->valueint;
5262
    } else if (!numRecPerReq) {
5263
        g_args.reqPerReq = MAX_RECORDS_PER_REQ;
5264
    } else {
5265 5266 5267
        errorPrint("%s() LN%d, failed to read json, num_of_records_per_req not found\n",
                __func__, __LINE__);
        goto PARSE_OVER;
5268 5269
    }

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

5288
    // rows per table need be less than insert batch
5289
    if (g_args.interlaceRows > g_args.reqPerReq) {
5290
        printf("NOTICE: interlace rows value %u > num_of_records_per_req %u\n\n",
5291
                g_args.interlaceRows, g_args.reqPerReq);
5292
        printf("        interlace rows value will be set to num_of_records_per_req %u\n\n",
5293
                g_args.reqPerReq);
5294
        prompt();
5295
        g_args.interlaceRows = g_args.reqPerReq;
5296 5297
    }

5298 5299
    cJSON* dbs = cJSON_GetObjectItem(root, "databases");
    if (!dbs || dbs->type != cJSON_Array) {
5300
        errorPrint("%s", "failed to read json, databases not found\n");
5301
        goto PARSE_OVER;
5302
    }
5303

5304 5305 5306
    int dbSize = cJSON_GetArraySize(dbs);
    if (dbSize > MAX_DB_COUNT) {
        errorPrint(
5307
                "failed to read json, databases size overflow, max database is %d\n",
5308 5309
                MAX_DB_COUNT);
        goto PARSE_OVER;
5310 5311
    }

5312 5313 5314 5315
    g_Dbs.dbCount = dbSize;
    for (int i = 0; i < dbSize; ++i) {
        cJSON* dbinfos = cJSON_GetArrayItem(dbs, i);
        if (dbinfos == NULL) continue;
5316

5317 5318 5319
        // dbinfo
        cJSON *dbinfo = cJSON_GetObjectItem(dbinfos, "dbinfo");
        if (!dbinfo || dbinfo->type != cJSON_Object) {
5320
            errorPrint("%s", "failed to read json, dbinfo not found\n");
5321 5322
            goto PARSE_OVER;
        }
5323

5324 5325
        cJSON *dbName = cJSON_GetObjectItem(dbinfo, "name");
        if (!dbName || dbName->type != cJSON_String || dbName->valuestring == NULL) {
5326
            errorPrint("%s", "failed to read json, db name not found\n");
5327 5328 5329
            goto PARSE_OVER;
        }
        tstrncpy(g_Dbs.db[i].dbName, dbName->valuestring, TSDB_DB_NAME_LEN);
5330

5331 5332 5333 5334 5335 5336 5337 5338 5339 5340
        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 {
5341
            errorPrint("%s", "failed to read json, drop input mistake\n");
5342 5343
            goto PARSE_OVER;
        }
5344

5345 5346 5347 5348
        cJSON *precision = cJSON_GetObjectItem(dbinfo, "precision");
        if (precision && precision->type == cJSON_String
                && precision->valuestring != NULL) {
            tstrncpy(g_Dbs.db[i].dbCfg.precision, precision->valuestring,
5349
                    SMALL_BUFF_LEN);
5350
        } else if (!precision) {
5351
            memset(g_Dbs.db[i].dbCfg.precision, 0, SMALL_BUFF_LEN);
5352
        } else {
5353
            errorPrint("%s", "failed to read json, precision not found\n");
5354 5355
            goto PARSE_OVER;
        }
5356

5357 5358 5359 5360 5361
        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;
5362
        } else {
5363
            errorPrint("%s", "failed to read json, update not found\n");
5364
            goto PARSE_OVER;
5365
        }
5366

5367 5368 5369 5370 5371 5372
        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 {
5373
            errorPrint("%s", "failed to read json, replica not found\n");
5374 5375
            goto PARSE_OVER;
        }
5376

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

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

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

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

5417 5418 5419 5420 5421 5422 5423 5424 5425 5426 5427 5428 5429 5430 5431 5432
        //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 {
5433
            errorPrint("%s", "failed to read json, minRows not found\n");
5434 5435
            goto PARSE_OVER;
        }
5436

5437 5438 5439 5440 5441 5442
        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 {
5443
            errorPrint("%s", "failed to read json, maxRows not found\n");
5444 5445
            goto PARSE_OVER;
        }
5446

5447 5448 5449 5450 5451 5452
        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 {
5453
            errorPrint("%s", "failed to read json, comp not found\n");
5454 5455
            goto PARSE_OVER;
        }
5456

5457 5458 5459 5460 5461 5462
        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 {
5463
            errorPrint("%s", "failed to read json, walLevel not found\n");
5464 5465
            goto PARSE_OVER;
        }
5466

5467 5468 5469 5470 5471
        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;
5472
        } else {
5473
            errorPrint("%s", "failed to read json, cacheLast not found\n");
5474
            goto PARSE_OVER;
5475
        }
5476

5477 5478 5479 5480 5481
        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;
5482
        } else {
5483 5484
            printf("failed to read json, quorum input mistake");
            goto PARSE_OVER;
5485
        }
5486 5487 5488 5489 5490 5491 5492

        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 {
5493
            errorPrint("%s", "failed to read json, fsync input mistake\n");
5494
            goto PARSE_OVER;
5495 5496
        }

5497
        // super_tables
5498 5499
        cJSON *stables = cJSON_GetObjectItem(dbinfos, "super_tables");
        if (!stables || stables->type != cJSON_Array) {
5500
            errorPrint("%s", "failed to read json, super_tables not found\n");
5501
            goto PARSE_OVER;
5502
        }
5503

5504 5505 5506
        int stbSize = cJSON_GetArraySize(stables);
        if (stbSize > MAX_SUPER_TABLE_COUNT) {
            errorPrint(
5507 5508
                    "failed to read json, supertable size overflow, max supertable is %d\n",
                    MAX_SUPER_TABLE_COUNT);
5509
            goto PARSE_OVER;
5510
        }
5511

5512 5513 5514 5515
        g_Dbs.db[i].superTblCount = stbSize;
        for (int j = 0; j < stbSize; ++j) {
            cJSON* stbInfo = cJSON_GetArrayItem(stables, j);
            if (stbInfo == NULL) continue;
5516

5517 5518 5519 5520
            // dbinfo
            cJSON *stbName = cJSON_GetObjectItem(stbInfo, "name");
            if (!stbName || stbName->type != cJSON_String
                    || stbName->valuestring == NULL) {
5521
                errorPrint("%s", "failed to read json, stb name not found\n");
5522 5523
                goto PARSE_OVER;
            }
5524
            tstrncpy(g_Dbs.db[i].superTbls[j].stbName, stbName->valuestring,
5525
                    TSDB_TABLE_NAME_LEN);
5526

5527 5528
            cJSON *prefix = cJSON_GetObjectItem(stbInfo, "childtable_prefix");
            if (!prefix || prefix->type != cJSON_String || prefix->valuestring == NULL) {
5529
                errorPrint("%s", "failed to read json, childtable_prefix not found\n");
5530 5531 5532
                goto PARSE_OVER;
            }
            tstrncpy(g_Dbs.db[i].superTbls[j].childTblPrefix, prefix->valuestring,
5533
                    TBNAME_PREFIX_LEN);
5534

5535 5536 5537 5538 5539 5540 5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551 5552
            cJSON *escapeChar = cJSON_GetObjectItem(stbInfo, "escape_character");
            if (escapeChar
                    && escapeChar->type == cJSON_String
                    && escapeChar->valuestring != NULL) {
                if ((0 == strncasecmp(escapeChar->valuestring, "yes", 3))) {
                    g_Dbs.db[i].superTbls[j].escapeChar = true;
                } else if (0 == strncasecmp(escapeChar->valuestring, "no", 2)) {
                    g_Dbs.db[i].superTbls[j].escapeChar = false;
                } else {
                    g_Dbs.db[i].superTbls[j].escapeChar = false;
                }
            } else if (!escapeChar) {
                g_Dbs.db[i].superTbls[j].escapeChar = false;
            } else {
                errorPrint("%s", "failed to read json, escape_character not found\n");
                goto PARSE_OVER;
            }

5553 5554 5555 5556 5557 5558 5559 5560 5561 5562 5563 5564 5565 5566 5567
            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 {
5568
                errorPrint("%s", "failed to read json, auto_create_table not found\n");
5569 5570
                goto PARSE_OVER;
            }
5571

5572 5573 5574 5575
            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) {
5576
                g_Dbs.db[i].superTbls[j].batchCreateTableNum = 10;
5577
            } else {
5578
                errorPrint("%s", "failed to read json, batch_create_tbl_num not found\n");
5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594 5595 5596 5597
                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 {
5598 5599
                errorPrint("%s",
                        "failed to read json, child_table_exists not found\n");
5600 5601 5602 5603 5604 5605 5606 5607 5608
                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) {
5609 5610
                errorPrint("%s",
                        "failed to read json, childtable_count input mistake\n");
5611 5612 5613
                goto PARSE_OVER;
            }
            g_Dbs.db[i].superTbls[j].childTblCount = count->valueint;
5614
            g_totalChildTables += g_Dbs.db[i].superTbls[j].childTblCount;
5615 5616 5617 5618 5619

            cJSON *dataSource = cJSON_GetObjectItem(stbInfo, "data_source");
            if (dataSource && dataSource->type == cJSON_String
                    && dataSource->valuestring != NULL) {
                tstrncpy(g_Dbs.db[i].superTbls[j].dataSource,
5620 5621
                        dataSource->valuestring,
                        min(SMALL_BUFF_LEN, strlen(dataSource->valuestring) + 1));
5622
            } else if (!dataSource) {
5623 5624
                tstrncpy(g_Dbs.db[i].superTbls[j].dataSource, "rand",
                        min(SMALL_BUFF_LEN, strlen("rand") + 1));
5625
            } else {
5626
                errorPrint("%s", "failed to read json, data_source not found\n");
5627 5628 5629 5630 5631 5632 5633 5634 5635 5636 5637 5638 5639
                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 {
5640 5641
                    errorPrint("failed to read json, insert_mode %s not recognized\n",
                            stbIface->valuestring);
5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654
                    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) {
5655
                    errorPrint("%s", "failed to read json, childtable_limit\n");
5656 5657 5658 5659 5660 5661 5662 5663 5664 5665 5666 5667
                    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)) {
5668
                    errorPrint("%s", "failed to read json, childtable_offset\n");
5669 5670 5671 5672 5673 5674 5675 5676 5677 5678 5679 5680 5681 5682 5683
                    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 {
5684
                errorPrint("%s", "failed to read json, start_timestamp not found\n");
5685 5686 5687 5688 5689 5690 5691
                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) {
5692
                g_Dbs.db[i].superTbls[j].timeStampStep = g_args.timestamp_step;
5693
            } else {
5694
                errorPrint("%s", "failed to read json, timestamp_step not found\n");
5695 5696 5697 5698 5699 5700 5701
                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,
5702 5703 5704
                        sampleFormat->valuestring,
                        min(SMALL_BUFF_LEN,
                            strlen(sampleFormat->valuestring) + 1));
5705
            } else if (!sampleFormat) {
5706 5707
                tstrncpy(g_Dbs.db[i].superTbls[j].sampleFormat, "csv",
                        SMALL_BUFF_LEN);
5708
            } else {
5709
                errorPrint("%s", "failed to read json, sample_format not found\n");
5710 5711 5712 5713 5714 5715 5716
                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,
5717 5718 5719
                        sampleFile->valuestring,
                        min(MAX_FILE_NAME_LEN,
                            strlen(sampleFile->valuestring) + 1));
5720
            } else if (!sampleFile) {
5721 5722
                memset(g_Dbs.db[i].superTbls[j].sampleFile, 0,
                        MAX_FILE_NAME_LEN);
5723
            } else {
5724
                errorPrint("%s", "failed to read json, sample_file not found\n");
5725 5726 5727
                goto PARSE_OVER;
            }

5728 5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742 5743 5744
            cJSON *useSampleTs = cJSON_GetObjectItem(stbInfo, "use_sample_ts");
            if (useSampleTs && useSampleTs->type == cJSON_String
                    && useSampleTs->valuestring != NULL) {
                if (0 == strncasecmp(useSampleTs->valuestring, "yes", 3)) {
                    g_Dbs.db[i].superTbls[j].useSampleTs = true;
                } else if (0 == strncasecmp(useSampleTs->valuestring, "no", 2)){
                    g_Dbs.db[i].superTbls[j].useSampleTs = false;
                } else {
                    g_Dbs.db[i].superTbls[j].useSampleTs = false;
                }
            } else if (!useSampleTs) {
                g_Dbs.db[i].superTbls[j].useSampleTs = false;
            } else {
                errorPrint("%s", "failed to read json, use_sample_ts not found\n");
                goto PARSE_OVER;
            }

5745 5746 5747 5748 5749 5750 5751 5752 5753 5754 5755 5756 5757 5758
            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 {
5759
                errorPrint("%s", "failed to read json, tags_file not found\n");
5760 5761 5762 5763 5764 5765 5766 5767 5768 5769 5770 5771 5772 5773 5774
                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 {
5775
                errorPrint("%s", "failed to read json, stbMaxSqlLen input mistake\n");
5776 5777 5778 5779 5780 5781 5782 5783 5784 5785 5786 5787 5788 5789 5790 5791
                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 {
5792
               errorPrint("%s", "failed to read json, multiThreadWriteOneTbl not found\n");
5793 5794 5795 5796 5797 5798
               goto PARSE_OVER;
               }
               */
            cJSON* insertRows = cJSON_GetObjectItem(stbInfo, "insert_rows");
            if (insertRows && insertRows->type == cJSON_Number) {
                if (insertRows->valueint < 0) {
5799
                    errorPrint("%s", "failed to read json, insert_rows input mistake\n");
5800 5801 5802 5803 5804 5805
                    goto PARSE_OVER;
                }
                g_Dbs.db[i].superTbls[j].insertRows = insertRows->valueint;
            } else if (!insertRows) {
                g_Dbs.db[i].superTbls[j].insertRows = 0x7FFFFFFFFFFFFFFF;
            } else {
5806
                errorPrint("%s", "failed to read json, insert_rows input mistake\n");
5807 5808 5809 5810 5811 5812
                goto PARSE_OVER;
            }

            cJSON* stbInterlaceRows = cJSON_GetObjectItem(stbInfo, "interlace_rows");
            if (stbInterlaceRows && stbInterlaceRows->type == cJSON_Number) {
                if (stbInterlaceRows->valueint < 0) {
5813
                    errorPrint("%s", "failed to read json, interlace rows input mistake\n");
5814 5815 5816 5817 5818 5819 5820 5821 5822 5823 5824 5825 5826 5827
                    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) {
5828
                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
5829 5830
            } else {
                errorPrint(
5831
                        "%s", "failed to read json, interlace rows input mistake\n");
5832 5833 5834 5835 5836 5837 5838 5839 5840 5841 5842 5843 5844 5845 5846
                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 {
5847
                errorPrint("%s", "failed to read json, disorderRatio not found\n");
5848 5849 5850 5851 5852 5853 5854 5855 5856
                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 {
5857
                errorPrint("%s", "failed to read json, disorderRange not found\n");
5858 5859 5860 5861 5862 5863 5864
                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) {
5865
                    errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5866 5867 5868
                    goto PARSE_OVER;
                }
            } else if (!insertInterval) {
5869
                verbosePrint("%s() LN%d: stable insert interval be overrode by global %"PRIu64".\n",
5870 5871 5872
                        __func__, __LINE__, g_args.insert_interval);
                g_Dbs.db[i].superTbls[j].insertInterval = g_args.insert_interval;
            } else {
5873
                errorPrint("%s", "failed to read json, insert_interval input mistake\n");
5874 5875
                goto PARSE_OVER;
            }
5876

5877 5878 5879 5880 5881
            int retVal = getColumnAndTagTypeFromInsertJsonFile(
                    stbInfo, &g_Dbs.db[i].superTbls[j]);
            if (false == retVal) {
                goto PARSE_OVER;
            }
5882
        }
5883
    }
5884

5885
    ret = true;
5886

5887
PARSE_OVER:
5888
    return ret;
5889 5890 5891
}

static bool getMetaFromQueryJsonFile(cJSON* root) {
5892
    bool  ret = false;
5893

5894 5895 5896 5897
    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 已提交
5898

5899 5900 5901 5902 5903
    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 已提交
5904
    } else {
5905
        errorPrint("%s", "failed to read json, host not found\n");
5906
        goto PARSE_OVER;
H
hzcheng 已提交
5907
    }
5908

5909 5910 5911 5912 5913
    cJSON* port = cJSON_GetObjectItem(root, "port");
    if (port && port->type == cJSON_Number) {
        g_queryInfo.port = port->valueint;
    } else if (!port) {
        g_queryInfo.port = 6030;
5914
    }
5915

5916 5917 5918 5919 5920
    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); ;
5921 5922
    }

5923 5924
    cJSON* password = cJSON_GetObjectItem(root, "password");
    if (password && password->type == cJSON_String && password->valuestring != NULL) {
5925
        tstrncpy(g_queryInfo.password, password->valuestring, SHELL_MAX_PASSWORD_LEN);
5926
    } else if (!password) {
5927
        tstrncpy(g_queryInfo.password, "taosdata", SHELL_MAX_PASSWORD_LEN);;
5928
    }
5929

5930 5931 5932 5933 5934 5935 5936 5937 5938 5939 5940 5941
    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;
5942
    } else {
5943
        errorPrint("%s", "failed to read json, confirm_parameter_prompt not found\n");
5944
        goto PARSE_OVER;
5945
    }
5946

5947 5948 5949
    cJSON* gQueryTimes = cJSON_GetObjectItem(root, "query_times");
    if (gQueryTimes && gQueryTimes->type == cJSON_Number) {
        if (gQueryTimes->valueint <= 0) {
5950
            errorPrint("%s()", "failed to read json, query_times input mistake\n");
5951 5952 5953 5954 5955
            goto PARSE_OVER;
        }
        g_args.query_times = gQueryTimes->valueint;
    } else if (!gQueryTimes) {
        g_args.query_times = 1;
5956
    } else {
5957
        errorPrint("%s", "failed to read json, query_times input mistake\n");
5958
        goto PARSE_OVER;
5959
    }
5960

5961 5962 5963 5964
    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) {
5965
        errorPrint("%s", "failed to read json, databases not found\n");
5966
        goto PARSE_OVER;
5967
    }
5968

5969
    cJSON* queryMode = cJSON_GetObjectItem(root, "query_mode");
5970 5971 5972 5973 5974
    if (queryMode
            && queryMode->type == cJSON_String
            && queryMode->valuestring != NULL) {
        tstrncpy(g_queryInfo.queryMode, queryMode->valuestring,
                min(SMALL_BUFF_LEN, strlen(queryMode->valuestring) + 1));
5975
    } else if (!queryMode) {
5976 5977
        tstrncpy(g_queryInfo.queryMode, "taosc",
                min(SMALL_BUFF_LEN, strlen("taosc") + 1));
5978
    } else {
5979
        errorPrint("%s", "failed to read json, query_mode not found\n");
5980
        goto PARSE_OVER;
5981
    }
5982

5983 5984 5985 5986 5987 5988
    // 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) {
5989
        errorPrint("%s", "failed to read json, super_table_query not found\n");
5990 5991
        goto PARSE_OVER;
    } else {
5992 5993 5994 5995 5996 5997
        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;
        }
5998

5999 6000 6001 6002 6003
        cJSON* specifiedQueryTimes = cJSON_GetObjectItem(specifiedQuery,
                "query_times");
        if (specifiedQueryTimes && specifiedQueryTimes->type == cJSON_Number) {
            if (specifiedQueryTimes->valueint <= 0) {
                errorPrint(
6004 6005
                        "failed to read json, query_times: %"PRId64", need be a valid (>0) number\n",
                        specifiedQueryTimes->valueint);
6006
                goto PARSE_OVER;
6007

6008 6009 6010 6011
            }
            g_queryInfo.specifiedQueryInfo.queryTimes = specifiedQueryTimes->valueint;
        } else if (!specifiedQueryTimes) {
            g_queryInfo.specifiedQueryInfo.queryTimes = g_args.query_times;
6012
        } else {
6013 6014 6015
            errorPrint("%s() LN%d, failed to read json, query_times input mistake\n",
                    __func__, __LINE__);
            goto PARSE_OVER;
6016
        }
6017

6018 6019 6020 6021
        cJSON* concurrent = cJSON_GetObjectItem(specifiedQuery, "concurrent");
        if (concurrent && concurrent->type == cJSON_Number) {
            if (concurrent->valueint <= 0) {
                errorPrint(
6022
                        "query sqlCount %d or concurrent %d is not correct.\n",
6023 6024 6025 6026 6027 6028 6029 6030
                        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;
        }
6031

6032 6033 6034 6035 6036 6037 6038 6039
        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 {
6040
                errorPrint("%s", "failed to read json, async mode input error\n");
6041 6042 6043 6044 6045
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.asyncMode = SYNC_MODE;
        }
6046

6047 6048 6049 6050 6051 6052 6053 6054
        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;
        }
6055

6056 6057 6058 6059 6060 6061 6062
        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 {
6063
                errorPrint("%s", "failed to read json, subscribe restart error\n");
6064 6065 6066 6067 6068
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.subscribeRestart = true;
        }
6069

6070 6071 6072 6073 6074 6075 6076 6077 6078
        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 {
6079
                errorPrint("%s", "failed to read json, subscribe keepProgress error\n");
6080 6081 6082 6083 6084
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.specifiedQueryInfo.subscribeKeepProgress = 0;
        }
6085

6086 6087 6088 6089 6090
        // sqls
        cJSON* specifiedSqls = cJSON_GetObjectItem(specifiedQuery, "sqls");
        if (!specifiedSqls) {
            g_queryInfo.specifiedQueryInfo.sqlCount = 0;
        } else if (specifiedSqls->type != cJSON_Array) {
6091
            errorPrint("%s", "failed to read json, super sqls not found\n");
6092 6093 6094 6095 6096
            goto PARSE_OVER;
        } else {
            int superSqlSize = cJSON_GetArraySize(specifiedSqls);
            if (superSqlSize * g_queryInfo.specifiedQueryInfo.concurrent
                    > MAX_QUERY_SQL_COUNT) {
6097
                errorPrint("failed to read json, query sql(%d) * concurrent(%d) overflow, max is %d\n",
6098 6099 6100 6101 6102
                        superSqlSize,
                        g_queryInfo.specifiedQueryInfo.concurrent,
                        MAX_QUERY_SQL_COUNT);
                goto PARSE_OVER;
            }
6103

6104 6105 6106 6107 6108 6109 6110
            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) {
6111
                    errorPrint("%s", "failed to read json, sql not found\n");
6112 6113 6114
                    goto PARSE_OVER;
                }
                tstrncpy(g_queryInfo.specifiedQueryInfo.sql[j],
6115
                        sqlStr->valuestring, BUFFER_SIZE);
6116 6117 6118 6119 6120 6121 6122 6123 6124 6125 6126 6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137 6138 6139 6140 6141 6142 6143 6144 6145 6146 6147 6148 6149 6150

                // 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 {
6151 6152
                    errorPrint("%s",
                            "failed to read json, super query result file not found\n");
6153 6154 6155 6156
                    goto PARSE_OVER;
                }
            }
        }
6157
    }
6158

6159 6160 6161 6162 6163 6164
    // 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) {
6165
        errorPrint("%s", "failed to read json, sub_table_query not found\n");
6166
        ret = true;
6167 6168
        goto PARSE_OVER;
    } else {
6169 6170 6171 6172 6173 6174 6175 6176 6177 6178
        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) {
6179 6180
                errorPrint("failed to read json, query_times: %"PRId64", need be a valid (>0) number\n",
                        superQueryTimes->valueint);
6181 6182 6183 6184 6185 6186
                goto PARSE_OVER;
            }
            g_queryInfo.superQueryInfo.queryTimes = superQueryTimes->valueint;
        } else if (!superQueryTimes) {
            g_queryInfo.superQueryInfo.queryTimes = g_args.query_times;
        } else {
6187
            errorPrint("%s", "failed to read json, query_times input mistake\n");
6188 6189
            goto PARSE_OVER;
        }
6190

6191 6192 6193
        cJSON* threads = cJSON_GetObjectItem(superQuery, "threads");
        if (threads && threads->type == cJSON_Number) {
            if (threads->valueint <= 0) {
6194
                errorPrint("%s", "failed to read json, threads input mistake\n");
6195 6196 6197 6198 6199 6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212
                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) {
6213
            tstrncpy(g_queryInfo.superQueryInfo.stbName, stblname->valuestring,
6214 6215
                    TSDB_TABLE_NAME_LEN);
        } else {
6216
            errorPrint("%s", "failed to read json, super table name input error\n");
6217 6218 6219 6220 6221 6222 6223 6224 6225 6226 6227
            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 {
6228
                errorPrint("%s", "failed to read json, async mode input error\n");
6229 6230 6231 6232 6233 6234 6235 6236 6237
                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) {
6238
                errorPrint("%s", "failed to read json, interval input mistake\n");
6239 6240 6241 6242 6243 6244 6245 6246 6247 6248 6249 6250 6251 6252 6253 6254 6255
                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 {
6256
                errorPrint("%s", "failed to read json, subscribe restart error\n");
6257 6258 6259 6260 6261 6262 6263 6264 6265 6266 6267 6268 6269 6270 6271
                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 {
6272 6273
                errorPrint("%s",
                        "failed to read json, subscribe super table keepProgress error\n");
6274 6275 6276 6277 6278 6279 6280
                goto PARSE_OVER;
            }
        } else {
            g_queryInfo.superQueryInfo.subscribeKeepProgress = 0;
        }

        // default value is -1, which mean do not resub
6281
        g_queryInfo.superQueryInfo.endAfterConsume = -1;
6282 6283 6284 6285 6286 6287 6288 6289 6290
        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;
6291

6292
        // default value is -1, which mean do not resub
6293
        g_queryInfo.superQueryInfo.resubAfterConsume = -1;
6294 6295 6296 6297 6298 6299 6300 6301 6302 6303 6304 6305 6306 6307 6308 6309
        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) {
6310
            errorPrint("%s", "failed to read json, super sqls not found\n");
6311 6312 6313 6314
            goto PARSE_OVER;
        } else {
            int superSqlSize = cJSON_GetArraySize(superSqls);
            if (superSqlSize > MAX_QUERY_SQL_COUNT) {
6315 6316
                errorPrint("failed to read json, query sql size overflow, max is %d\n",
                        MAX_QUERY_SQL_COUNT);
6317 6318
                goto PARSE_OVER;
            }
6319

6320 6321 6322 6323
            g_queryInfo.superQueryInfo.sqlCount = superSqlSize;
            for (int j = 0; j < superSqlSize; ++j) {
                cJSON* sql = cJSON_GetArrayItem(superSqls, j);
                if (sql == NULL) continue;
6324

6325 6326 6327
                cJSON *sqlStr = cJSON_GetObjectItem(sql, "sql");
                if (!sqlStr || sqlStr->type != cJSON_String
                        || sqlStr->valuestring == NULL) {
6328
                    errorPrint("%s", "failed to read json, sql not found\n");
6329 6330 6331
                    goto PARSE_OVER;
                }
                tstrncpy(g_queryInfo.superQueryInfo.sql[j], sqlStr->valuestring,
6332
                        BUFFER_SIZE);
6333 6334 6335

                cJSON *result = cJSON_GetObjectItem(sql, "result");
                if (result != NULL && result->type == cJSON_String
6336
                        && result->valuestring != NULL) {
6337 6338 6339 6340 6341
                    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 {
6342
                    errorPrint("%s", "failed to read json, sub query result file not found\n");
6343 6344 6345
                    goto PARSE_OVER;
                }
            }
6346
        }
6347
    }
H
hzcheng 已提交
6348

6349
    ret = true;
H
hzcheng 已提交
6350

6351
PARSE_OVER:
6352
    return ret;
6353
}
H
hzcheng 已提交
6354

6355
static bool getInfoFromJsonFile(char* file) {
6356
    debugPrint("%s %d %s\n", __func__, __LINE__, file);
6357

6358 6359
    FILE *fp = fopen(file, "r");
    if (!fp) {
6360
        errorPrint("failed to read %s, reason:%s\n", file, strerror(errno));
6361 6362
        return false;
    }
H
Hui Li 已提交
6363

6364 6365 6366 6367 6368 6369 6370
    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);
6371
        errorPrint("failed to read %s, content is null", file);
6372 6373
        return false;
    }
H
Hui Li 已提交
6374

6375 6376 6377
    content[len] = 0;
    cJSON* root = cJSON_Parse(content);
    if (root == NULL) {
6378
        errorPrint("failed to cjson parse %s, invalid json format\n", file);
6379 6380
        goto PARSE_OVER;
    }
H
Hui Li 已提交
6381

6382 6383 6384 6385 6386 6387 6388 6389 6390
    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 {
6391
            errorPrint("%s", "failed to read json, filetype not support\n");
6392 6393 6394 6395
            goto PARSE_OVER;
        }
    } else if (!filetype) {
        g_args.test_mode = INSERT_TEST;
6396
    } else {
6397
        errorPrint("%s", "failed to read json, filetype not found\n");
6398
        goto PARSE_OVER;
6399
    }
H
hzcheng 已提交
6400

6401 6402 6403 6404 6405 6406
    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 {
6407 6408
        errorPrint("%s",
                "input json file type error! please input correct file type: insert or query or subscribe\n");
6409 6410
        goto PARSE_OVER;
    }
6411

6412
PARSE_OVER:
6413 6414 6415 6416
    free(content);
    cJSON_Delete(root);
    fclose(fp);
    return ret;
6417
}
H
hzcheng 已提交
6418

6419
static int prepareSampleData() {
6420 6421 6422 6423 6424 6425 6426
    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;
                }
            }
6427
        }
6428
    }
6429

6430
    return 0;
6431 6432
}

6433
static void postFreeResource() {
6434
    tmfclose(g_fpOfInsertResult);
6435

6436 6437 6438
    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) {
6439
                tmfree(g_Dbs.db[i].superTbls[j].colsOfCreateChildTable);
6440 6441 6442
                g_Dbs.db[i].superTbls[j].colsOfCreateChildTable = NULL;
            }
            if (0 != g_Dbs.db[i].superTbls[j].sampleDataBuf) {
6443
                tmfree(g_Dbs.db[i].superTbls[j].sampleDataBuf);
6444 6445
                g_Dbs.db[i].superTbls[j].sampleDataBuf = NULL;
            }
6446

6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459
#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
6460
            if (0 != g_Dbs.db[i].superTbls[j].tagDataBuf) {
6461
                tmfree(g_Dbs.db[i].superTbls[j].tagDataBuf);
6462 6463 6464
                g_Dbs.db[i].superTbls[j].tagDataBuf = NULL;
            }
            if (0 != g_Dbs.db[i].superTbls[j].childTblName) {
6465
                tmfree(g_Dbs.db[i].superTbls[j].childTblName);
6466 6467 6468
                g_Dbs.db[i].superTbls[j].childTblName = NULL;
            }
        }
S
Shuaiqiang Chang 已提交
6469
    }
6470 6471 6472 6473 6474 6475 6476 6477 6478 6479

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

6481 6482 6483 6484 6485 6486 6487 6488 6489 6490 6491 6492 6493
    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
6494
}
S
Shuaiqiang Chang 已提交
6495

6496 6497
static int getRowDataFromSample(
        char* dataBuf, int64_t maxLen, int64_t timestamp,
6498
        SSuperTable* stbInfo, int64_t* sampleUsePos)
6499
{
6500
    if ((*sampleUsePos) == MAX_SAMPLES) {
6501 6502 6503 6504
        *sampleUsePos = 0;
    }

    int    dataLen = 0;
6505 6506 6507 6508 6509 6510 6511
    if(stbInfo->useSampleTs) {
        dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
            "(%s",
            stbInfo->sampleDataBuf
            + stbInfo->lenOfOneRow * (*sampleUsePos));
    } else {
        dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
6512
            "(%" PRId64 ", ", timestamp);
6513
        dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen,
6514
            "%s",
6515 6516
            stbInfo->sampleDataBuf
            + stbInfo->lenOfOneRow * (*sampleUsePos));
6517 6518
    }
    
6519 6520 6521 6522 6523
    dataLen += snprintf(dataBuf + dataLen, maxLen - dataLen, ")");

    (*sampleUsePos)++;

    return dataLen;
6524
}
S
Shuaiqiang Chang 已提交
6525

6526 6527
static int64_t generateStbRowData(
        SSuperTable* stbInfo,
6528 6529 6530
        char* recBuf,
        int64_t remainderBufLen,
        int64_t timestamp)
6531
{
6532 6533 6534
    int64_t   dataLen = 0;
    char  *pstr = recBuf;
    int64_t maxLen = MAX_DATA_SIZE;
6535
    int tmpLen;
6536 6537

    dataLen += snprintf(pstr + dataLen, maxLen - dataLen,
6538
            "(%" PRId64 "", timestamp);
6539 6540

    for (int i = 0; i < stbInfo->columnCount; i++) {
6541 6542 6543 6544 6545
        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)) {
6546
            if (stbInfo->columns[i].dataLen > TSDB_MAX_BINARY_LEN) {
6547
                errorPrint2("binary or nchar length overflow, max size:%u\n",
6548 6549 6550
                        (uint32_t)TSDB_MAX_BINARY_LEN);
                return -1;
            }
6551

6552 6553 6554 6555 6556
            if ((stbInfo->columns[i].dataLen + 1) >
                    /* need count 3 extra chars \', \', and , */
                    (remainderBufLen - dataLen - 3)) {
                return 0;
            }
6557 6558
            char* buf = (char*)calloc(stbInfo->columns[i].dataLen+1, 1);
            if (NULL == buf) {
6559
                errorPrint2("calloc failed! size:%d\n", stbInfo->columns[i].dataLen);
6560 6561 6562
                return -1;
            }
            rand_string(buf, stbInfo->columns[i].dataLen);
6563
            dataLen += snprintf(pstr + dataLen, maxLen - dataLen, "\'%s\'", buf);
6564
            tmfree(buf);
6565

6566
        } else {
6567 6568 6569 6570 6571 6572 6573 6574 6575 6576 6577
            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;
6578

6579 6580 6581 6582 6583 6584
                case TSDB_DATA_TYPE_UINT:
                    tmp = rand_uint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, INT_BUFF_LEN));
                    break;

6585 6586
                case TSDB_DATA_TYPE_BIGINT:
                    tmp = rand_bigint_str();
6587 6588 6589
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
                    break;
6590

6591 6592 6593 6594
                case TSDB_DATA_TYPE_UBIGINT:
                    tmp = rand_ubigint_str();
                    tmpLen = strlen(tmp);
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
6595 6596 6597 6598 6599 6600 6601 6602 6603
                    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();
                        }
6604
                    } else {
6605
                        tmp = rand_float_str();
6606
                    }
6607
                    tmpLen = strlen(tmp);
6608
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, FLOAT_BUFF_LEN));
6609 6610 6611 6612 6613
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    tmp = rand_double_str();
                    tmpLen = strlen(tmp);
6614
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, DOUBLE_BUFF_LEN));
6615 6616 6617 6618 6619
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
                    tmp = rand_smallint_str();
                    tmpLen = strlen(tmp);
6620 6621 6622 6623 6624 6625 6626
                    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));
6627 6628 6629 6630 6631
                    break;

                case TSDB_DATA_TYPE_TINYINT:
                    tmp = rand_tinyint_str();
                    tmpLen = strlen(tmp);
6632 6633 6634 6635 6636 6637 6638
                    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));
6639 6640 6641 6642 6643
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    tmp = rand_bool_str();
                    tmpLen = strlen(tmp);
6644
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BOOL_BUFF_LEN));
6645 6646 6647 6648 6649
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    tmp = rand_bigint_str();
                    tmpLen = strlen(tmp);
6650
                    tstrncpy(pstr + dataLen, tmp, min(tmpLen + 1, BIGINT_BUFF_LEN));
6651 6652 6653 6654 6655 6656 6657 6658 6659
                    break;

                case TSDB_DATA_TYPE_NULL:
                    break;

                default:
                    errorPrint2("Not support data type: %s\n",
                            stbInfo->columns[i].dataType);
                    exit(EXIT_FAILURE);
6660
            }
6661
            if (tmp) {
6662
                dataLen += tmpLen;
6663
            }
6664
        }
6665

6666
        if (dataLen > (remainderBufLen - (128)))
6667
            return 0;
6668
    }
6669

6670
    dataLen += snprintf(pstr + dataLen, 2, ")");
6671

6672 6673
    verbosePrint("%s() LN%d, dataLen:%"PRId64"\n", __func__, __LINE__, dataLen);
    verbosePrint("%s() LN%d, recBuf:\n\t%s\n", __func__, __LINE__, recBuf);
6674

6675
    return strlen(recBuf);
6676
}
S
Shuaiqiang Chang 已提交
6677

6678
static int64_t generateData(char *recBuf, char *data_type,
6679
        int64_t timestamp, int lenOfBinary) {
6680 6681
    memset(recBuf, 0, MAX_DATA_SIZE);
    char *pstr = recBuf;
6682
    pstr += sprintf(pstr, "(%"PRId64"", timestamp);
6683

6684
    int columnCount = g_args.columnCount;
6685

6686 6687
    bool b;
    char *s;
6688
    for (int i = 0; i < columnCount; i++) {
6689 6690 6691 6692 6693 6694 6695 6696 6697 6698 6699 6700 6701 6702 6703 6704 6705 6706 6707 6708 6709 6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722 6723 6724 6725 6726 6727 6728 6729 6730 6731 6732 6733 6734 6735 6736 6737 6738 6739 6740 6741 6742 6743 6744 6745 6746
        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;

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

6763 6764 6765 6766 6767 6768 6769
            case TSDB_DATA_TYPE_NULL:
                break;

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

6773
        if (strlen(recBuf) > MAX_DATA_SIZE) {
6774
            ERROR_EXIT("column length too long, abort");
6775
        }
6776 6777
    }

6778
    pstr += sprintf(pstr, ")");
6779

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

6782
    return (int32_t)strlen(recBuf);
6783 6784
}

6785 6786 6787 6788 6789 6790
static int generateSampleFromRand(
        char *sampleDataBuf,
        uint64_t lenOfOneRow,
        int columnCount,
        StrColumn *columns
        )
6791 6792 6793 6794
{
    char data[MAX_DATA_SIZE];
    memset(data, 0, MAX_DATA_SIZE);

6795
    char *buff = malloc(lenOfOneRow);
6796
    if (NULL == buff) {
6797 6798
        errorPrint2("%s() LN%d, memory allocation %"PRIu64" bytes failed\n",
                __func__, __LINE__, lenOfOneRow);
6799 6800 6801
        exit(EXIT_FAILURE);
    }

6802
    for (int i=0; i < MAX_SAMPLES; i++) {
6803
        uint64_t pos = 0;
6804
        memset(buff, 0, lenOfOneRow);
6805

6806 6807 6808 6809 6810 6811 6812 6813 6814 6815 6816 6817 6818 6819 6820
        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;
6821
                    rand_string(data, dataLen - 1);
6822 6823 6824 6825 6826 6827
                    pos += sprintf(buff + pos, "%s,", data);
                    break;

                case TSDB_DATA_TYPE_INT:
                    if ((g_args.demo_mode) && (c == 1)) {
                        tmp = demo_voltage_int_str();
6828
                    } else {
6829
                        tmp = rand_int_str();
6830
                    }
6831 6832 6833
                    pos += sprintf(buff + pos, "%s,", tmp);
                    break;

6834 6835 6836 6837
                case TSDB_DATA_TYPE_UINT:
                    pos += sprintf(buff + pos, "%s,", rand_uint_str());
                    break;

6838 6839 6840 6841
                case TSDB_DATA_TYPE_BIGINT:
                    pos += sprintf(buff + pos, "%s,", rand_bigint_str());
                    break;

6842 6843 6844 6845
                case TSDB_DATA_TYPE_UBIGINT:
                    pos += sprintf(buff + pos, "%s,", rand_ubigint_str());
                    break;

6846 6847 6848 6849 6850 6851 6852 6853 6854 6855 6856 6857 6858 6859 6860 6861 6862 6863 6864 6865 6866
                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;

6867 6868 6869 6870
                case TSDB_DATA_TYPE_USMALLINT:
                    pos += sprintf(buff + pos, "%s,", rand_usmallint_str());
                    break;

6871 6872 6873 6874
                case TSDB_DATA_TYPE_TINYINT:
                    pos += sprintf(buff + pos, "%s,", rand_tinyint_str());
                    break;

6875 6876 6877 6878
                case TSDB_DATA_TYPE_UTINYINT:
                    pos += sprintf(buff + pos, "%s,", rand_utinyint_str());
                    break;

6879 6880 6881 6882 6883 6884 6885 6886 6887 6888 6889 6890 6891 6892 6893 6894
                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);
6895 6896
            }
        }
6897

6898
        *(buff + pos - 1) = 0;
6899
        memcpy(sampleDataBuf + i * lenOfOneRow, buff, pos);
6900 6901 6902 6903 6904 6905
    }

    free(buff);
    return 0;
}

6906 6907 6908 6909 6910 6911 6912 6913 6914 6915 6916 6917 6918 6919 6920 6921 6922 6923 6924
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() {
6925
    g_sampleDataBuf = calloc(g_args.lenOfOneRow * MAX_SAMPLES, 1);
6926 6927 6928
    if (NULL == g_sampleDataBuf) {
        errorPrint2("%s() LN%d, Failed to calloc %"PRIu64" Bytes, reason:%s\n",
                __func__, __LINE__,
6929
                g_args.lenOfOneRow * MAX_SAMPLES,
6930 6931 6932 6933 6934 6935 6936 6937
                strerror(errno));
        return -1;
    }

    return generateSampleFromRandForNtb();
}

static int prepareSampleForStb(SSuperTable *stbInfo) {
6938

6939
    stbInfo->sampleDataBuf = calloc(
6940
            stbInfo->lenOfOneRow * MAX_SAMPLES, 1);
6941
    if (NULL == stbInfo->sampleDataBuf) {
6942
        errorPrint2("%s() LN%d, Failed to calloc %"PRIu64" Bytes, reason:%s\n",
6943
                __func__, __LINE__,
6944
                stbInfo->lenOfOneRow * MAX_SAMPLES,
6945 6946 6947
                strerror(errno));
        return -1;
    }
6948

6949
    int ret;
6950
    if (0 == strncasecmp(stbInfo->dataSource, "sample", strlen("sample"))) {
6951 6952 6953
        if(stbInfo->useSampleTs) {
            getAndSetRowsFromCsvFile(stbInfo);
        }
6954 6955 6956 6957
        ret = generateSampleFromCsvForStb(stbInfo);
    } else {
        ret = generateSampleFromRandForStb(stbInfo);
    }
6958

6959
    if (0 != ret) {
6960
        errorPrint2("%s() LN%d, read sample from csv file failed.\n",
6961
                __func__, __LINE__);
6962
        tmfree(stbInfo->sampleDataBuf);
6963
        stbInfo->sampleDataBuf = NULL;
6964 6965
        return -1;
    }
6966

6967
    return 0;
6968 6969
}

6970
static int32_t execInsert(threadInfo *pThreadInfo, uint32_t k)
6971
{
6972
    int32_t affectedRows;
6973
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
6974

6975
    uint16_t iface;
6976 6977
    if (stbInfo)
        iface = stbInfo->iface;
6978 6979 6980 6981 6982 6983
    else {
        if (g_args.iface == INTERFACE_BUT)
            iface = TAOSC_IFACE;
        else
            iface = g_args.iface;
    }
6984 6985 6986

    debugPrint("[%d] %s() LN%d %s\n", pThreadInfo->threadID,
            __func__, __LINE__,
6987 6988
            (iface==TAOSC_IFACE)?
            "taosc":(iface==REST_IFACE)?"rest":"stmt");
6989 6990 6991

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

6995 6996 6997 6998 6999 7000
            affectedRows = queryDbExec(
                    pThreadInfo->taos,
                    pThreadInfo->buffer, INSERT_TYPE, false);
            break;

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

7004
            if (0 != postProceSql(g_Dbs.host, g_Dbs.port,
7005
                        pThreadInfo->buffer, pThreadInfo)) {
7006 7007 7008 7009 7010 7011 7012 7013 7014
                affectedRows = -1;
                printf("========restful return fail, threadID[%d]\n",
                        pThreadInfo->threadID);
            } else {
                affectedRows = k;
            }
            break;

        case STMT_IFACE:
7015 7016
            debugPrint("%s() LN%d, stmt=%p",
                    __func__, __LINE__, pThreadInfo->stmt);
7017
            if (0 != taos_stmt_execute(pThreadInfo->stmt)) {
7018
                errorPrint2("%s() LN%d, failied to execute insert statement. reason: %s\n",
7019
                        __func__, __LINE__, taos_stmt_errstr(pThreadInfo->stmt));
7020 7021

                fprintf(stderr, "\n\033[31m === Please reduce batch number if WAL size exceeds limit. ===\033[0m\n\n");
7022
                exit(EXIT_FAILURE);
7023 7024 7025 7026 7027
            }
            affectedRows = k;
            break;

        default:
7028
            errorPrint2("%s() LN%d: unknown insert mode: %d\n",
7029
                    __func__, __LINE__, stbInfo->iface);
7030
            affectedRows = 0;
7031
    }
7032

7033
    return affectedRows;
7034 7035
}

7036 7037
static void getTableName(char *pTblName,
        threadInfo* pThreadInfo, uint64_t tableSeq)
7038
{
7039 7040 7041 7042
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    if (stbInfo) {
        if (AUTO_CREATE_SUBTBL != stbInfo->autoCreateTable) {
            if (stbInfo->childTblLimit > 0) {
7043 7044
                snprintf(pTblName, TSDB_TABLE_NAME_LEN, 
                        stbInfo->escapeChar ? "`%s`" : "%s",
7045 7046
                        stbInfo->childTblName +
                        (tableSeq - stbInfo->childTblOffset) * TSDB_TABLE_NAME_LEN);
7047 7048 7049 7050 7051
            } else {
                verbosePrint("[%d] %s() LN%d: from=%"PRIu64" count=%"PRId64" seq=%"PRIu64"\n",
                        pThreadInfo->threadID, __func__, __LINE__,
                        pThreadInfo->start_table_from,
                        pThreadInfo->ntables, tableSeq);
7052
                snprintf(pTblName, TSDB_TABLE_NAME_LEN, stbInfo->escapeChar ? "`%s`" : "%s",
7053
                        stbInfo->childTblName + tableSeq * TSDB_TABLE_NAME_LEN);
7054 7055
            }
        } else {
7056 7057
            snprintf(pTblName, TSDB_TABLE_NAME_LEN, 
            stbInfo->escapeChar ? "`%s%"PRIu64"`" : "%s%"PRIu64"",
7058
                    stbInfo->childTblPrefix, tableSeq);
7059
        }
7060
    } else {
7061 7062
        snprintf(pTblName, TSDB_TABLE_NAME_LEN, 
        g_args.escapeChar ? "`%s%"PRIu64"`" : "%s%"PRIu64"",
7063
                g_args.tb_prefix, tableSeq);
7064 7065 7066
    }
}

7067
static int32_t generateDataTailWithoutStb(
7068
        uint32_t batch, char* buffer,
7069
        int64_t remainderBufLen, int64_t insertRows,
7070
        uint64_t recordFrom, int64_t startTime,
7071
        /* int64_t *pSamplePos, */int64_t *dataLen) {
7072

7073 7074
    uint64_t len = 0;
    char *pstr = buffer;
7075

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

7078 7079
    int32_t k = 0;
    for (k = 0; k < batch;) {
7080
        char *data = pstr;
7081
        memset(data, 0, MAX_DATA_SIZE);
7082

7083
        int64_t retLen = 0;
7084

7085
        char *data_type = g_args.data_type;
7086
        int lenOfBinary = g_args.binwidth;
7087

7088 7089 7090
        if (g_args.disorderRatio) {
            retLen = generateData(data, data_type,
                    startTime + getTSRandTail(
7091
                        g_args.timestamp_step, k,
7092 7093 7094 7095 7096
                        g_args.disorderRatio,
                        g_args.disorderRange),
                    lenOfBinary);
        } else {
            retLen = generateData(data, data_type,
7097
                    startTime + g_args.timestamp_step * k,
7098 7099
                    lenOfBinary);
        }
7100

7101 7102
        if (len > remainderBufLen)
            break;
7103

7104
        pstr += retLen;
7105 7106 7107
        k++;
        len += retLen;
        remainderBufLen -= retLen;
7108

7109 7110
        verbosePrint("%s() LN%d len=%"PRIu64" k=%d \nbuffer=%s\n",
                __func__, __LINE__, len, k, buffer);
7111

7112
        recordFrom ++;
7113

7114 7115 7116
        if (recordFrom >= insertRows) {
            break;
        }
7117
    }
7118

7119 7120
    *dataLen = len;
    return k;
7121 7122
}

7123 7124 7125 7126 7127 7128 7129 7130 7131 7132 7133 7134 7135 7136 7137 7138 7139
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(
7140
        SSuperTable* stbInfo,
7141
        uint32_t batch, char* buffer,
7142
        int64_t remainderBufLen, int64_t insertRows,
7143
        uint64_t recordFrom, int64_t startTime,
7144
        int64_t *pSamplePos, int64_t *dataLen) {
7145
    uint64_t len = 0;
7146

7147
    char *pstr = buffer;
7148

7149
    bool tsRand;
7150
    if (0 == strncasecmp(stbInfo->dataSource, "rand", strlen("rand"))) {
7151 7152 7153 7154 7155 7156
        tsRand = true;
    } else {
        tsRand = false;
    }
    verbosePrint("%s() LN%d batch=%u buflen=%"PRId64"\n",
            __func__, __LINE__, batch, remainderBufLen);
7157

7158 7159
    int32_t k;
    for (k = 0; k < batch;) {
7160
        char *data = pstr;
7161

7162
        int64_t lenOfRow = 0;
7163

7164
        if (tsRand) {
7165 7166
            if (stbInfo->disorderRatio > 0) {
                lenOfRow = generateStbRowData(stbInfo, data,
7167
                        remainderBufLen,
7168
                        startTime + getTSRandTail(
7169 7170 7171
                            stbInfo->timeStampStep, k,
                            stbInfo->disorderRatio,
                            stbInfo->disorderRange)
7172 7173
                        );
            } else {
7174
                lenOfRow = generateStbRowData(stbInfo, data,
7175
                        remainderBufLen,
7176
                        startTime + stbInfo->timeStampStep * k
7177 7178 7179 7180 7181 7182
                        );
            }
        } else {
            lenOfRow = getRowDataFromSample(
                    data,
                    (remainderBufLen < MAX_DATA_SIZE)?remainderBufLen:MAX_DATA_SIZE,
7183 7184
                    startTime + stbInfo->timeStampStep * k,
                    stbInfo,
7185 7186
                    pSamplePos);
        }
7187

7188 7189 7190 7191
        if (lenOfRow == 0) {
            data[0] = '\0';
            break;
        }
7192 7193 7194
        if ((lenOfRow + 1) > remainderBufLen) {
            break;
        }
7195

7196
        pstr += lenOfRow;
7197 7198 7199
        k++;
        len += lenOfRow;
        remainderBufLen -= lenOfRow;
7200

7201 7202
        verbosePrint("%s() LN%d len=%"PRIu64" k=%u \nbuffer=%s\n",
                __func__, __LINE__, len, k, buffer);
7203

7204
        recordFrom ++;
7205

7206 7207 7208
        if (recordFrom >= insertRows) {
            break;
        }
7209
    }
7210

7211 7212
    *dataLen = len;
    return k;
7213
}
7214

7215 7216 7217

static int generateSQLHeadWithoutStb(char *tableName,
        char *dbName,
7218
        char *buffer, int remainderBufLen)
7219
{
7220
    int len;
7221

7222
    char headBuf[HEAD_BUFF_LEN];
7223

7224 7225 7226 7227 7228 7229
    len = snprintf(
            headBuf,
            HEAD_BUFF_LEN,
            "%s.%s values",
            dbName,
            tableName);
7230

7231 7232
    if (len > remainderBufLen)
        return -1;
7233

7234
    tstrncpy(buffer, headBuf, len + 1);
7235

7236
    return len;
7237 7238 7239
}

static int generateStbSQLHead(
7240
        SSuperTable* stbInfo,
7241
        char *tableName, int64_t tableSeq,
7242 7243 7244
        char *dbName,
        char *buffer, int remainderBufLen)
{
7245
    int len;
7246

7247
    char headBuf[HEAD_BUFF_LEN];
7248

7249
    if (AUTO_CREATE_SUBTBL == stbInfo->autoCreateTable) {
7250
        char* tagsValBuf = NULL;
7251 7252
        if (0 == stbInfo->tagSource) {
            tagsValBuf = generateTagValuesForStb(stbInfo, tableSeq);
7253
        } else {
7254
            tagsValBuf = getTagValueFromTagSample(
7255 7256
                    stbInfo,
                    tableSeq % stbInfo->tagSampleCount);
7257 7258
        }
        if (NULL == tagsValBuf) {
7259
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
7260 7261 7262
                    __func__, __LINE__);
            return -1;
        }
7263

7264 7265 7266 7267 7268 7269 7270
        len = snprintf(
                headBuf,
                HEAD_BUFF_LEN,
                "%s.%s using %s.%s TAGS%s values",
                dbName,
                tableName,
                dbName,
7271
                stbInfo->stbName,
7272 7273
                tagsValBuf);
        tmfree(tagsValBuf);
7274
    } else if (TBL_ALREADY_EXISTS == stbInfo->childTblExists) {
7275 7276 7277 7278 7279 7280
        len = snprintf(
                headBuf,
                HEAD_BUFF_LEN,
                "%s.%s values",
                dbName,
                tableName);
7281
    } else {
7282 7283 7284 7285 7286 7287 7288
        len = snprintf(
                headBuf,
                HEAD_BUFF_LEN,
                "%s.%s values",
                dbName,
                tableName);
    }
7289

7290 7291
    if (len > remainderBufLen)
        return -1;
7292

7293
    tstrncpy(buffer, headBuf, len + 1);
7294

7295
    return len;
7296 7297
}

7298
static int32_t generateStbInterlaceData(
7299
        threadInfo *pThreadInfo,
7300 7301 7302
        char *tableName, uint32_t batchPerTbl,
        uint64_t i,
        uint32_t batchPerTblTimes,
7303
        uint64_t tableSeq,
7304
        char *buffer,
7305
        int64_t insertRows,
7306
        int64_t startTime,
7307
        uint64_t *pRemainderBufLen)
7308
{
7309 7310
    assert(buffer);
    char *pstr = buffer;
7311

7312
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
7313
    int headLen = generateStbSQLHead(
7314
            stbInfo,
7315 7316
            tableName, tableSeq, pThreadInfo->db_name,
            pstr, *pRemainderBufLen);
7317

7318 7319 7320 7321 7322
    if (headLen <= 0) {
        return 0;
    }
    // generate data buffer
    verbosePrint("[%d] %s() LN%d i=%"PRIu64" buffer:\n%s\n",
7323
            pThreadInfo->threadID, __func__, __LINE__, i, buffer);
7324

7325 7326
    pstr += headLen;
    *pRemainderBufLen -= headLen;
7327

7328
    int64_t dataLen = 0;
7329

7330
    verbosePrint("[%d] %s() LN%d i=%"PRIu64" batchPerTblTimes=%u batchPerTbl = %u\n",
7331 7332
            pThreadInfo->threadID, __func__, __LINE__,
            i, batchPerTblTimes, batchPerTbl);
7333

7334
    if (0 == strncasecmp(stbInfo->startTimestamp, "now", 3)) {
7335 7336
        startTime = taosGetTimestamp(pThreadInfo->time_precision);
    }
7337

7338
    int32_t k = generateStbDataTail(
7339
            stbInfo,
7340 7341 7342
            batchPerTbl, pstr, *pRemainderBufLen, insertRows, 0,
            startTime,
            &(pThreadInfo->samplePos), &dataLen);
7343

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

7355
    return k;
7356 7357 7358
}

static int64_t generateInterlaceDataWithoutStb(
7359
        char *tableName, uint32_t batch,
7360 7361 7362
        uint64_t tableSeq,
        char *dbName, char *buffer,
        int64_t insertRows,
7363
        int64_t startTime,
7364 7365
        uint64_t *pRemainderBufLen)
{
7366 7367
    assert(buffer);
    char *pstr = buffer;
7368

7369 7370
    int headLen = generateSQLHeadWithoutStb(
            tableName, dbName,
7371 7372
            pstr, *pRemainderBufLen);

7373 7374 7375
    if (headLen <= 0) {
        return 0;
    }
7376

7377 7378
    pstr += headLen;
    *pRemainderBufLen -= headLen;
7379

7380
    int64_t dataLen = 0;
7381

7382
    int32_t k = generateDataTailWithoutStb(
7383
            batch, pstr, *pRemainderBufLen, insertRows, 0,
7384
            startTime,
7385
            &dataLen);
7386

7387 7388 7389 7390 7391 7392 7393 7394 7395 7396 7397 7398
    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;
7399 7400
}

7401 7402
static int32_t prepareStmtBindArrayByType(
        TAOS_BIND *bind,
7403
        char data_type, int32_t dataLen,
7404 7405 7406
        int32_t timePrec,
        char *value)
{
7407
    int32_t *bind_int;
7408
    uint32_t *bind_uint;
7409
    int64_t *bind_bigint;
7410
    uint64_t *bind_ubigint;
7411 7412 7413 7414 7415
    float   *bind_float;
    double  *bind_double;
    int8_t  *bind_bool;
    int64_t *bind_ts2;
    int16_t *bind_smallint;
7416
    uint16_t *bind_usmallint;
7417
    int8_t  *bind_tinyint;
7418
    uint8_t  *bind_utinyint;
7419 7420 7421 7422 7423 7424 7425 7426 7427

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

7429 7430 7431 7432 7433 7434 7435 7436 7437 7438
            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;
            }
7439

7440 7441 7442 7443
            bind->length = &bind->buffer_length;
            bind->buffer = bind_binary;
            bind->is_null = NULL;
            break;
7444

7445 7446 7447 7448 7449 7450 7451
        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;
7452

7453 7454 7455 7456 7457 7458 7459 7460
            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);
            }
7461

7462 7463 7464 7465 7466
            bind->buffer_length = strlen(bind_nchar);
            bind->buffer = bind_nchar;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
            break;
7467

7468 7469 7470
        case TSDB_DATA_TYPE_INT:
            bind_int = malloc(sizeof(int32_t));
            assert(bind_int);
7471

7472 7473 7474 7475 7476 7477 7478 7479 7480 7481 7482
            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;
7483

7484 7485 7486 7487 7488 7489 7490 7491 7492 7493 7494 7495 7496 7497 7498
        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;
7499

7500 7501 7502
        case TSDB_DATA_TYPE_BIGINT:
            bind_bigint = malloc(sizeof(int64_t));
            assert(bind_bigint);
7503

7504 7505 7506 7507 7508 7509 7510 7511 7512 7513 7514
            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;
7515

7516 7517 7518 7519 7520 7521 7522 7523 7524 7525 7526 7527 7528 7529 7530 7531
        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;

7532 7533 7534 7535 7536 7537
        case TSDB_DATA_TYPE_FLOAT:
            bind_float = malloc(sizeof(float));
            assert(bind_float);

            if (value) {
                *bind_float = (float)atof(value);
7538
            } else {
7539
                *bind_float = rand_float();
7540
            }
7541 7542 7543 7544 7545 7546 7547 7548 7549 7550 7551 7552 7553 7554 7555 7556 7557 7558 7559 7560 7561 7562 7563 7564 7565 7566 7567 7568 7569 7570 7571 7572 7573 7574 7575 7576 7577 7578 7579
            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;

7580 7581 7582 7583 7584 7585 7586 7587 7588 7589 7590 7591 7592 7593 7594 7595
        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;

7596 7597 7598 7599 7600 7601 7602 7603 7604 7605 7606 7607 7608 7609 7610 7611
        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;

7612 7613 7614 7615 7616 7617 7618 7619 7620 7621 7622 7623 7624 7625 7626 7627
        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;

7628 7629 7630 7631 7632 7633 7634 7635 7636 7637 7638 7639 7640 7641 7642 7643 7644 7645 7646
        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;
7647

7648 7649 7650 7651 7652 7653 7654 7655 7656 7657 7658 7659
        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++;
7660
                    }
7661 7662 7663 7664 7665 7666 7667 7668 7669 7670 7671
                    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);
7672 7673
                }
            } else {
7674
                *bind_ts2 = rand_bigint();
7675
            }
7676 7677 7678 7679 7680 7681 7682 7683 7684 7685 7686 7687 7688
            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);
7689 7690 7691 7692 7693 7694 7695
    }

    return 0;
}

static int32_t prepareStmtBindArrayByTypeForRand(
        TAOS_BIND *bind,
7696
        char data_type, int32_t dataLen,
7697 7698 7699
        int32_t timePrec,
        char **ptr,
        char *value)
7700
{
7701
    int32_t *bind_int;
7702
    uint32_t *bind_uint;
7703
    int64_t *bind_bigint;
7704
    uint64_t *bind_ubigint;
7705 7706 7707
    float   *bind_float;
    double  *bind_double;
    int16_t *bind_smallint;
7708
    uint16_t *bind_usmallint;
7709
    int8_t  *bind_tinyint;
7710
    uint8_t  *bind_utinyint;
7711 7712 7713 7714 7715
    int8_t  *bind_bool;
    int64_t *bind_ts2;

    switch(data_type) {
        case TSDB_DATA_TYPE_BINARY:
7716

7717 7718 7719 7720 7721 7722
            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;
7723

7724 7725 7726 7727 7728 7729 7730 7731
            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;
            }
7732

7733 7734 7735
            bind->length = &bind->buffer_length;
            bind->buffer = bind_binary;
            bind->is_null = NULL;
7736

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

7740 7741 7742 7743 7744 7745 7746
        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;
7747

7748 7749 7750 7751 7752 7753
            bind->buffer_type = TSDB_DATA_TYPE_NCHAR;
            if (value) {
                strncpy(bind_nchar, value, strlen(value));
            } else {
                rand_string(bind_nchar, dataLen);
            }
7754

7755 7756 7757 7758
            bind->buffer_length = strlen(bind_nchar);
            bind->buffer = bind_nchar;
            bind->length = &bind->buffer_length;
            bind->is_null = NULL;
7759

7760 7761
            *ptr += bind->buffer_length;
            break;
7762

7763 7764
        case TSDB_DATA_TYPE_INT:
            bind_int = (int32_t *)*ptr;
7765

7766 7767 7768 7769 7770 7771 7772 7773 7774 7775
            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;
7776

7777 7778
            *ptr += bind->buffer_length;
            break;
7779

7780 7781 7782 7783 7784 7785 7786 7787 7788 7789 7790 7791 7792 7793 7794 7795 7796
        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;

7797 7798
        case TSDB_DATA_TYPE_BIGINT:
            bind_bigint = (int64_t *)*ptr;
7799

7800 7801 7802 7803 7804 7805 7806 7807 7808 7809
            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;
7810

7811 7812
            *ptr += bind->buffer_length;
            break;
7813

7814 7815 7816 7817 7818 7819 7820 7821 7822 7823 7824 7825 7826 7827 7828 7829 7830
        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;

7831 7832
        case TSDB_DATA_TYPE_FLOAT:
            bind_float = (float *)*ptr;
7833

7834 7835 7836 7837 7838 7839 7840 7841 7842 7843
            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;
7844

7845 7846
            *ptr += bind->buffer_length;
            break;
7847

7848 7849
        case TSDB_DATA_TYPE_DOUBLE:
            bind_double = (double *)*ptr;
7850

7851 7852
            if (value) {
                *bind_double = atof(value);
7853
            } else {
7854
                *bind_double = rand_double();
7855
            }
7856 7857 7858 7859 7860 7861 7862 7863 7864 7865 7866 7867 7868 7869 7870 7871 7872 7873 7874 7875 7876 7877 7878 7879 7880 7881
            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;

7882 7883 7884 7885 7886 7887 7888 7889 7890 7891 7892 7893 7894 7895 7896 7897 7898
        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;

7899 7900 7901 7902 7903 7904 7905 7906 7907 7908 7909 7910 7911 7912 7913 7914 7915
        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;

7916 7917 7918 7919 7920 7921 7922 7923 7924 7925 7926 7927 7928 7929 7930 7931 7932
        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;

7933 7934 7935 7936 7937 7938 7939 7940
        case TSDB_DATA_TYPE_BOOL:
            bind_bool = (int8_t *)*ptr;

            if (value) {
                if (strncasecmp(value, "true", 4)) {
                    *bind_bool = true;
                } else {
                    *bind_bool = false;
7941
                }
7942 7943 7944 7945 7946 7947 7948 7949 7950 7951 7952 7953 7954 7955 7956 7957 7958 7959 7960 7961 7962 7963 7964 7965 7966 7967 7968 7969 7970 7971 7972 7973 7974 7975
            } 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);
7976 7977
                }
            } else {
7978
                *bind_ts2 = rand_bigint();
7979
            }
7980 7981 7982 7983 7984
            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;
7985

7986 7987 7988 7989 7990 7991
            *ptr += bind->buffer_length;
            break;

        default:
            errorPrint2("No support data type: %d\n", data_type);
            return -1;
7992 7993 7994 7995 7996 7997
    }

    return 0;
}

static int32_t prepareStmtWithoutStb(
7998
        threadInfo *pThreadInfo,
7999 8000 8001 8002 8003
        char *tableName,
        uint32_t batch,
        int64_t insertRows,
        int64_t recordFrom,
        int64_t startTime)
8004
{
8005
    TAOS_STMT *stmt = pThreadInfo->stmt;
8006 8007
    int ret = taos_stmt_set_tbname(stmt, tableName);
    if (ret != 0) {
8008
        errorPrint2("failed to execute taos_stmt_set_tbname(%s). return 0x%x. reason: %s\n",
8009
                tableName, ret, taos_stmt_errstr(stmt));
8010 8011 8012
        return ret;
    }

8013
    char *data_type = g_args.data_type;
8014

8015
    char *bindArray = malloc(sizeof(TAOS_BIND) * (g_args.columnCount + 1));
8016
    if (bindArray == NULL) {
8017
        errorPrint2("Failed to allocate %d bind params\n",
8018
                (g_args.columnCount + 1));
8019
        return -1;
8020 8021
    }

8022 8023 8024 8025 8026 8027
    int32_t k = 0;
    for (k = 0; k < batch;) {
        /* columnCount + 1 (ts) */

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

8028
        int64_t *bind_ts = pThreadInfo->bind_ts;
8029 8030

        bind->buffer_type = TSDB_DATA_TYPE_TIMESTAMP;
8031 8032 8033

        if (g_args.disorderRatio) {
            *bind_ts = startTime + getTSRandTail(
8034
                    g_args.timestamp_step, k,
8035 8036 8037
                    g_args.disorderRatio,
                    g_args.disorderRange);
        } else {
8038
            *bind_ts = startTime + g_args.timestamp_step * k;
8039
        }
8040 8041 8042 8043 8044
        bind->buffer_length = sizeof(int64_t);
        bind->buffer = bind_ts;
        bind->length = &bind->buffer_length;
        bind->is_null = NULL;

8045
        for (int i = 0; i < g_args.columnCount; i ++) {
8046 8047
            bind = (TAOS_BIND *)((char *)bindArray
                    + (sizeof(TAOS_BIND) * (i + 1)));
8048 8049 8050
            if ( -1 == prepareStmtBindArrayByType(
                        bind,
                        data_type[i],
8051
                        g_args.binwidth,
8052 8053
                        pThreadInfo->time_precision,
                        NULL)) {
8054
                free(bindArray);
8055 8056 8057
                return -1;
            }
        }
8058
        if (0 != taos_stmt_bind_param(stmt, (TAOS_BIND *)bindArray)) {
8059
            errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
8060 8061 8062
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            break;
        }
8063
        // if msg > 3MB, break
8064
        if (0 != taos_stmt_add_batch(stmt)) {
8065
            errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
8066 8067 8068
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            break;
        }
8069 8070 8071 8072 8073 8074 8075 8076

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

8077
    free(bindArray);
8078 8079 8080
    return k;
}

8081 8082 8083 8084 8085 8086 8087 8088 8089 8090 8091
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,
8092
                    stbInfo->tags[t].data_type,
8093 8094 8095 8096 8097 8098 8099 8100 8101 8102 8103 8104 8105
                    stbInfo->tags[t].dataLen,
                    timePrec,
                    NULL)) {
            return -1;
        }
    }

    return 0;
}

static int32_t prepareStbStmtBindRand(
        int64_t *ts,
        char *bindArray, SSuperTable *stbInfo,
8106
        int64_t startTime, int32_t recSeq,
8107
        int32_t timePrec)
8108 8109 8110 8111 8112 8113 8114
{
    char data[MAX_DATA_SIZE];
    memset(data, 0, MAX_DATA_SIZE);
    char *ptr = data;

    TAOS_BIND *bind;

8115 8116
    for (int i = 0; i < stbInfo->columnCount + 1; i ++) {
        bind = (TAOS_BIND *)((char *)bindArray + (sizeof(TAOS_BIND) * i));
8117

8118 8119
        if (i == 0) {
            int64_t *bind_ts = ts;
8120

8121 8122 8123 8124 8125 8126 8127 8128 8129 8130 8131 8132 8133 8134 8135 8136 8137
            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,
8138
                    stbInfo->columns[i-1].data_type,
8139 8140 8141 8142 8143 8144 8145
                    stbInfo->columns[i-1].dataLen,
                    timePrec,
                    &ptr,
                    NULL)) {
            return -1;
        }
    }
8146

8147 8148 8149 8150 8151 8152 8153 8154 8155 8156 8157 8158 8159 8160 8161 8162 8163 8164 8165 8166 8167 8168 8169 8170 8171 8172 8173 8174 8175 8176 8177 8178 8179 8180 8181 8182 8183 8184 8185 8186 8187 8188 8189 8190 8191 8192 8193 8194 8195 8196 8197 8198 8199 8200 8201 8202 8203 8204 8205 8206 8207 8208 8209 8210 8211 8212 8213 8214 8215 8216 8217 8218 8219 8220 8221 8222 8223 8224 8225 8226 8227 8228 8229 8230 8231 8232 8233 8234 8235 8236 8237 8238 8239 8240 8241 8242 8243 8244 8245 8246 8247 8248 8249 8250 8251 8252 8253 8254 8255 8256 8257 8258 8259 8260 8261 8262
    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
8263
static int execStbBindParamBatch(
8264 8265 8266 8267 8268 8269 8270 8271 8272 8273 8274 8275 8276
        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;
8277 8278 8279
    assert(stbInfo);

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

8281
    uint32_t thisBatch = MAX_SAMPLES - (*pSamplePos);
8282 8283 8284 8285 8286 8287 8288 8289 8290 8291 8292 8293 8294 8295 8296 8297 8298 8299 8300 8301 8302 8303

    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 {
8304
            data_type = stbInfo->columns[c-1].data_type;
8305 8306 8307 8308 8309

            char *tmpP;

            switch(data_type) {
                case TSDB_DATA_TYPE_BINARY:
8310 8311 8312 8313 8314 8315 8316 8317 8318 8319 8320 8321 8322 8323
                    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;

8324 8325
                case TSDB_DATA_TYPE_NCHAR:
                    param->buffer_length =
8326
                        stbInfo->columns[c-1].dataLen;
8327 8328 8329 8330 8331

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

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

8336
                    param->buffer = (void *)(tmpP + *pSamplePos * param->buffer_length);
8337 8338 8339
                    break;

                case TSDB_DATA_TYPE_INT:
8340
                case TSDB_DATA_TYPE_UINT:
8341
                    param->buffer_length = sizeof(int32_t);
8342
                    param->buffer =
8343
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8344
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8345 8346 8347
                    break;

                case TSDB_DATA_TYPE_TINYINT:
8348
                case TSDB_DATA_TYPE_UTINYINT:
8349
                    param->buffer_length = sizeof(int8_t);
8350
                    param->buffer =
8351 8352 8353
                        (void *)((uintptr_t)*(uintptr_t*)(
                                    stbInfo->sampleBindBatchArray
                                    +sizeof(char*)*(c-1))
8354
                                + stbInfo->columns[c-1].dataLen*(*pSamplePos));
8355 8356 8357
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
8358
                case TSDB_DATA_TYPE_USMALLINT:
8359
                    param->buffer_length = sizeof(int16_t);
8360
                    param->buffer =
8361
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8362
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8363 8364 8365
                    break;

                case TSDB_DATA_TYPE_BIGINT:
8366
                case TSDB_DATA_TYPE_UBIGINT:
8367
                    param->buffer_length = sizeof(int64_t);
8368
                    param->buffer =
8369
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8370
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8371 8372 8373 8374
                    break;

                case TSDB_DATA_TYPE_BOOL:
                    param->buffer_length = sizeof(int8_t);
8375
                    param->buffer =
8376
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8377
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8378 8379 8380 8381
                    break;

                case TSDB_DATA_TYPE_FLOAT:
                    param->buffer_length = sizeof(float);
8382
                    param->buffer =
8383
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8384
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8385 8386 8387 8388
                    break;

                case TSDB_DATA_TYPE_DOUBLE:
                    param->buffer_length = sizeof(double);
8389
                    param->buffer =
8390
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8391
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8392 8393 8394 8395
                    break;

                case TSDB_DATA_TYPE_TIMESTAMP:
                    param->buffer_length = sizeof(int64_t);
8396
                    param->buffer =
8397
                        (void *)((uintptr_t)*(uintptr_t*)(stbInfo->sampleBindBatchArray+sizeof(char*)*(c-1))
8398
                                        + stbInfo->columns[c-1].dataLen * (*pSamplePos));
8399 8400 8401 8402 8403 8404 8405 8406 8407 8408 8409 8410 8411 8412 8413 8414 8415 8416 8417 8418
                    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 *
8419
                         stbInfo->columns[c].dataLen
8420 8421 8422 8423 8424 8425 8426 8427 8428 8429 8430 8431 8432 8433 8434 8435 8436 8437 8438 8439 8440 8441 8442 8443 8444 8445 8446 8447
                        );
            } 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) ++;
8448
        if ((*pSamplePos) == MAX_SAMPLES) {
8449 8450 8451 8452 8453 8454 8455 8456 8457 8458 8459 8460 8461 8462 8463 8464 8465 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 8496 8497 8498 8499 8500 8501 8502
            *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:
8503
            case TSDB_DATA_TYPE_UINT:
8504
                tmpP = calloc(1, sizeof(int) * MAX_SAMPLES);
8505 8506 8507 8508 8509
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_TINYINT:
8510
            case TSDB_DATA_TYPE_UTINYINT:
8511
                tmpP = calloc(1, sizeof(int8_t) * MAX_SAMPLES);
8512 8513 8514 8515 8516
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_SMALLINT:
8517
            case TSDB_DATA_TYPE_USMALLINT:
8518
                tmpP = calloc(1, sizeof(int16_t) * MAX_SAMPLES);
8519 8520 8521 8522 8523
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BIGINT:
8524
            case TSDB_DATA_TYPE_UBIGINT:
8525
                tmpP = calloc(1, sizeof(int64_t) * MAX_SAMPLES);
8526 8527 8528 8529 8530
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BOOL:
8531
                tmpP = calloc(1, sizeof(int8_t) * MAX_SAMPLES);
8532 8533 8534 8535 8536
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_FLOAT:
8537
                tmpP = calloc(1, sizeof(float) * MAX_SAMPLES);
8538 8539 8540 8541 8542
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_DOUBLE:
8543
                tmpP = calloc(1, sizeof(double) * MAX_SAMPLES);
8544 8545 8546 8547 8548 8549
                assert(tmpP);
                *(uintptr_t*)(sampleBindBatchArray+ sizeof(uintptr_t*)*c) = (uintptr_t)tmpP;
                break;

            case TSDB_DATA_TYPE_BINARY:
            case TSDB_DATA_TYPE_NCHAR:
8550
                tmpP = calloc(1, MAX_SAMPLES *
8551 8552 8553 8554 8555 8556
                        (((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:
8557
                tmpP = calloc(1, sizeof(int64_t) * MAX_SAMPLES);
8558 8559 8560 8561 8562 8563 8564 8565 8566 8567 8568 8569 8570 8571
                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;

8572
    for (int i=0; i < MAX_SAMPLES; i++) {
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
        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:
8603
                case TSDB_DATA_TYPE_UINT:
8604 8605 8606 8607 8608 8609 8610 8611 8612 8613 8614 8615 8616 8617 8618 8619 8620 8621
                    *((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:
8622
                case TSDB_DATA_TYPE_UTINYINT:
8623 8624 8625 8626 8627 8628
                    *((int8_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int8_t)*i)) =
                        (int8_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_SMALLINT:
8629
                case TSDB_DATA_TYPE_USMALLINT:
8630 8631 8632 8633 8634 8635
                    *((int16_t*)((uintptr_t)*(uintptr_t*)(sampleBindBatchArray
                                    +sizeof(char*)*c)+sizeof(int16_t)*i)) =
                        (int16_t)atoi(tmpStr);
                    break;

                case TSDB_DATA_TYPE_BIGINT:
8636
                case TSDB_DATA_TYPE_UBIGINT:
8637 8638 8639 8640 8641 8642 8643 8644 8645 8646 8647 8648 8649 8650 8651 8652 8653 8654 8655 8656 8657 8658 8659 8660 8661 8662 8663 8664 8665 8666 8667 8668 8669 8670 8671 8672 8673 8674 8675 8676 8677 8678 8679 8680 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 8707 8708 8709 8710 8711 8712 8713 8714 8715
                    *((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 =
8716
        (char *)calloc(1, sizeof(char *) * MAX_SAMPLES);
8717 8718 8719
    if (pThreadInfo->sampleBindArray == NULL) {
        errorPrint2("%s() LN%d, Failed to allocate %"PRIu64" bind array buffer\n",
                __func__, __LINE__,
8720
                (uint64_t)sizeof(char *) * MAX_SAMPLES);
8721 8722 8723 8724 8725 8726 8727
        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;

8728
    for (int i=0; i < MAX_SAMPLES; i++) {
8729 8730 8731 8732 8733 8734 8735 8736 8737 8738 8739 8740 8741 8742 8743 8744 8745 8746 8747 8748 8749 8750 8751 8752 8753 8754 8755 8756 8757 8758 8759 8760 8761 8762 8763 8764 8765 8766 8767 8768 8769 8770 8771 8772 8773 8774 8775 8776 8777 8778 8779 8780 8781 8782 8783
        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);
8784
                    free(bindArray);
8785 8786 8787 8788 8789 8790 8791 8792 8793 8794 8795 8796 8797 8798 8799 8800 8801 8802 8803 8804 8805 8806 8807 8808 8809 8810 8811 8812 8813
                    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);
8814
}
8815

8816 8817
static int32_t prepareStbStmtBindStartTime(
        char *tableName,
8818 8819
        int64_t *ts,
        char *bindArray, SSuperTable *stbInfo,
8820
        int64_t startTime, int32_t recSeq)
8821 8822
{
    TAOS_BIND *bind;
8823

8824
    bind = (TAOS_BIND *)bindArray;
8825

8826
    int64_t *bind_ts = ts;
8827

8828 8829 8830 8831 8832 8833 8834 8835
    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;
8836
    }
8837 8838 8839 8840

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

8841 8842 8843 8844
    bind->buffer_length = sizeof(int64_t);
    bind->buffer = bind_ts;
    bind->length = &bind->buffer_length;
    bind->is_null = NULL;
8845 8846 8847 8848

    return 0;
}

8849
static uint32_t execBindParam(
8850
        threadInfo *pThreadInfo,
8851 8852 8853
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
8854 8855
        uint64_t insertRows,
        uint64_t recordFrom,
8856 8857
        int64_t startTime,
        int64_t *pSamplePos)
8858
{
8859
    int ret;
8860 8861
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;
8862

8863 8864
    uint32_t k;
    for (k = 0; k < batch;) {
8865 8866
        char *bindArray = (char *)(*((uintptr_t *)
                    (pThreadInfo->sampleBindArray + (sizeof(char *)) * (*pSamplePos))));
8867
        /* columnCount + 1 (ts) */
8868 8869
        if (-1 == prepareStbStmtBindStartTime(
                    tableName,
8870 8871
                    pThreadInfo->bind_ts,
                    bindArray, stbInfo,
8872
                    startTime, k
8873
                    /* is column */)) {
8874
            return -1;
8875
        }
8876 8877
        ret = taos_stmt_bind_param(stmt, (TAOS_BIND *)bindArray);
        if (0 != ret) {
8878
            errorPrint2("%s() LN%d, stmt_bind_param() failed! reason: %s\n",
8879 8880 8881
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
8882
        // if msg > 3MB, break
8883 8884
        ret = taos_stmt_add_batch(stmt);
        if (0 != ret) {
8885
            errorPrint2("%s() LN%d, stmt_add_batch() failed! reason: %s\n",
8886 8887 8888
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
8889

8890 8891
        k++;
        recordFrom ++;
8892

8893
        (*pSamplePos) ++;
8894
        if ((*pSamplePos) == MAX_SAMPLES) {
8895 8896 8897
            *pSamplePos = 0;
        }

8898 8899 8900 8901
        if (recordFrom >= insertRows) {
            break;
        }
    }
8902 8903 8904

    return k;
}
8905
#endif
8906

8907
static int32_t prepareStbStmt(
8908
        threadInfo *pThreadInfo,
8909 8910 8911
        char *tableName,
        int64_t tableSeq,
        uint32_t batch,
8912 8913 8914 8915 8916
        uint64_t insertRows,
        uint64_t recordFrom,
        int64_t startTime,
        int64_t *pSamplePos)
{
8917 8918 8919
    int ret;
    SSuperTable *stbInfo = pThreadInfo->stbInfo;
    TAOS_STMT *stmt = pThreadInfo->stmt;
8920

8921 8922 8923 8924 8925 8926 8927 8928 8929 8930 8931 8932
    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) {
8933
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
8934 8935 8936 8937 8938 8939 8940
                    __func__, __LINE__);
            return -1;
        }

        char *tagsArray = calloc(1, sizeof(TAOS_BIND) * stbInfo->tagCount);
        if (NULL == tagsArray) {
            tmfree(tagsValBuf);
8941
            errorPrint2("%s() LN%d, tag buf failed to allocate  memory\n",
8942 8943 8944 8945 8946 8947 8948 8949 8950 8951 8952 8953 8954 8955 8956 8957 8958 8959
                    __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) {
8960
            errorPrint2("%s() LN%d, stmt_set_tbname_tags() failed! reason: %s\n",
8961 8962 8963 8964 8965 8966
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    } else {
        ret = taos_stmt_set_tbname(stmt, tableName);
        if (0 != ret) {
8967
            errorPrint2("%s() LN%d, stmt_set_tbname() failed! reason: %s\n",
8968 8969 8970 8971 8972
                    __func__, __LINE__, taos_stmt_errstr(stmt));
            return -1;
        }
    }

8973
#if STMT_BIND_PARAM_BATCH == 1
8974
    return execStbBindParamBatch(
8975 8976 8977 8978 8979 8980 8981 8982 8983 8984 8985 8986 8987 8988 8989 8990 8991 8992 8993
        pThreadInfo,
        tableName,
        tableSeq,
        batch,
        insertRows,
        recordFrom,
        startTime,
        pSamplePos);
#else
    return execBindParam(
        pThreadInfo,
        tableName,
        tableSeq,
        batch,
        insertRows,
        recordFrom,
        startTime,
        pSamplePos);
#endif
8994
}
8995 8996

static int32_t generateStbProgressiveData(
8997
        SSuperTable *stbInfo,
8998
        char *tableName,
8999
        int64_t tableSeq,
9000
        char *dbName, char *buffer,
9001
        int64_t insertRows,
9002
        uint64_t recordFrom, int64_t startTime, int64_t *pSamplePos,
9003
        int64_t *pRemainderBufLen)
9004
{
9005 9006
    assert(buffer != NULL);
    char *pstr = buffer;
9007

9008
    memset(pstr, 0, *pRemainderBufLen);
9009

9010
    int64_t headLen = generateStbSQLHead(
9011
            stbInfo,
9012 9013
            tableName, tableSeq, dbName,
            buffer, *pRemainderBufLen);
9014

9015 9016 9017 9018 9019
    if (headLen <= 0) {
        return 0;
    }
    pstr += headLen;
    *pRemainderBufLen -= headLen;
9020

9021
    int64_t dataLen;
9022

9023
    return generateStbDataTail(stbInfo,
9024
            g_args.reqPerReq, pstr, *pRemainderBufLen,
9025 9026 9027
            insertRows, recordFrom,
            startTime,
            pSamplePos, &dataLen);
9028 9029
}

9030
static int32_t generateProgressiveDataWithoutStb(
9031
        char *tableName,
9032
        /* int64_t tableSeq, */
9033 9034
        threadInfo *pThreadInfo, char *buffer,
        int64_t insertRows,
9035
        uint64_t recordFrom, int64_t startTime, /*int64_t *pSamplePos, */
9036 9037
        int64_t *pRemainderBufLen)
{
9038 9039
    assert(buffer != NULL);
    char *pstr = buffer;
9040

9041
    memset(buffer, 0, *pRemainderBufLen);
9042

9043 9044 9045
    int64_t headLen = generateSQLHeadWithoutStb(
            tableName, pThreadInfo->db_name,
            buffer, *pRemainderBufLen);
9046

9047 9048 9049 9050 9051
    if (headLen <= 0) {
        return 0;
    }
    pstr += headLen;
    *pRemainderBufLen -= headLen;
9052

9053
    int64_t dataLen;
9054

9055
    return generateDataTailWithoutStb(
9056
            g_args.reqPerReq, pstr, *pRemainderBufLen, insertRows, recordFrom,
9057 9058
            startTime,
            /*pSamplePos, */&dataLen);
9059
}
9060

9061 9062
static void printStatPerThread(threadInfo *pThreadInfo)
{
9063 9064 9065
    if (0 == pThreadInfo->totalDelay)
        pThreadInfo->totalDelay = 1;

9066
    fprintf(stderr, "====thread[%d] completed total inserted rows: %"PRIu64 ", total affected rows: %"PRIu64". %.2f records/second====\n",
9067 9068 9069
            pThreadInfo->threadID,
            pThreadInfo->totalInsertRows,
            pThreadInfo->totalAffectedRows,
9070 9071
            (double)(pThreadInfo->totalAffectedRows/((double)pThreadInfo->totalDelay/1000000.0))
            );
9072 9073
}

9074 9075 9076 9077
#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",
9078
            pThreadInfo->threadID, __func__, __LINE__);
9079

9080
    int64_t insertRows;
9081
    int64_t timeStampStep;
9082
    uint64_t insert_interval;
9083

9084
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
9085

9086 9087
    if (stbInfo) {
        insertRows = stbInfo->insertRows;
9088
        timeStampStep = stbInfo->timeStampStep;
9089 9090 9091
        insert_interval = stbInfo->insertInterval;
    } else {
        insertRows = g_args.insertRows;
9092
        timeStampStep = g_args.timestamp_step;
9093 9094
        insert_interval = g_args.insert_interval;
    }
9095

9096 9097 9098 9099 9100
    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);

9101 9102
    uint64_t timesInterlace = (insertRows / interlaceRows) + 1;
    uint32_t precalcBatch = interlaceRows;
9103

9104 9105
    if (precalcBatch > g_args.reqPerReq)
        precalcBatch = g_args.reqPerReq;
9106

9107 9108
    if (precalcBatch > MAX_SAMPLES)
        precalcBatch = MAX_SAMPLES;
9109 9110 9111 9112 9113 9114 9115 9116 9117 9118 9119 9120

    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;
9121
    int64_t startTime;
9122 9123 9124 9125 9126 9127

    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;

    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;
9128
    pThreadInfo->samplePos = 0;
9129

9130 9131
    for (int64_t interlace = 0;
            interlace < timesInterlace; interlace ++) {
9132 9133 9134 9135 9136
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }

9137 9138
        int64_t generated = 0;
        int64_t samplePos;
9139

9140
        for (; tableSeq < pThreadInfo->start_table_from + pThreadInfo->ntables; tableSeq ++) {
9141 9142 9143 9144 9145 9146 9147 9148
            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;
            }

9149 9150 9151 9152 9153 9154
            samplePos = pThreadInfo->samplePos;
            startTime = pThreadInfo->start_time
                + interlace * interlaceRows * timeStampStep;
            uint64_t remainRecPerTbl =
                    insertRows - interlaceRows * interlace;
            uint64_t recPerTbl = 0;
9155

9156 9157 9158 9159 9160
            uint64_t remainPerInterlace;
            if (remainRecPerTbl > interlaceRows) {
                remainPerInterlace = interlaceRows;
            } else {
                remainPerInterlace = remainRecPerTbl;
9161 9162
            }

9163
            while(remainPerInterlace > 0) {
9164

9165 9166 9167 9168 9169 9170 9171 9172 9173 9174
                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);
9175

9176 9177 9178 9179 9180 9181 9182 9183 9184 9185 9186 9187 9188 9189 9190 9191 9192 9193
                if (stbInfo) {
                    generated = prepareStbStmt(
                            pThreadInfo,
                            tableName,
                            tableSeq,
                            batch,
                            insertRows, 0,
                            startTime,
                            &samplePos);
                } else {
                    generated = prepareStmtWithoutStb(
                            pThreadInfo,
                            tableName,
                            batch,
                            insertRows,
                            interlaceRows * interlace + recPerTbl,
                            startTime);
                }
9194

9195 9196 9197 9198 9199 9200 9201
                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) {
9202
                    break;
9203
                }
9204

9205 9206 9207
                recPerTbl += generated;
                remainPerInterlace -= generated;
                pThreadInfo->totalInsertRows += generated;
9208

9209 9210 9211
                verbosePrint("[%d] %s() LN%d totalInsertRows=%"PRIu64"\n",
                        pThreadInfo->threadID, __func__, __LINE__,
                        pThreadInfo->totalInsertRows);
9212

9213
                startTs = taosGetTimestampUs();
9214

9215
                int64_t affectedRows = execInsert(pThreadInfo, generated);
9216

9217 9218 9219 9220 9221 9222 9223
                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);
9224

9225 9226 9227 9228
                if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
                if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
                pThreadInfo->cntDelay++;
                pThreadInfo->totalDelay += delay;
9229

9230 9231 9232 9233 9234 9235
                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;
                }
9236

9237
                pThreadInfo->totalAffectedRows += affectedRows;
9238

9239 9240 9241 9242 9243 9244 9245 9246 9247 9248 9249 9250 9251
                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;
                }
9252

9253 9254
                startTime += (generated * timeStampStep);
            }
9255
        }
9256
        pThreadInfo->samplePos = samplePos;
9257

9258 9259 9260 9261
        if (tableSeq == pThreadInfo->start_table_from
                + pThreadInfo->ntables) {
            // turn to first table
            tableSeq = pThreadInfo->start_table_from;
9262

9263
            flagSleep = true;
9264 9265 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
        }

        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;
9293
    int64_t timeStampStep;
9294 9295 9296 9297 9298 9299
    uint64_t insert_interval;

    SSuperTable* stbInfo = pThreadInfo->stbInfo;

    if (stbInfo) {
        insertRows = stbInfo->insertRows;
9300
        maxSqlLen = stbInfo->maxSqlLen;
9301
        timeStampStep = stbInfo->timeStampStep;
9302
        insert_interval = stbInfo->insertInterval;
9303
    } else {
9304
        insertRows = g_args.insertRows;
9305
        maxSqlLen = g_args.max_sql_len;
9306
        timeStampStep = g_args.timestamp_step;
9307
        insert_interval = g_args.insert_interval;
9308
    }
9309

9310 9311 9312 9313
    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);
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
    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;
            }

9367 9368 9369 9370
            debugPrint("[%d] %s() LN%d, tableName:%s, batch:%d startTime:%"PRId64"\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__,
                    tableName, batchPerTbl, startTime);
9371
            if (stbInfo) {
9372
                generated = prepareStbStmt(
9373 9374 9375 9376 9377 9378 9379 9380 9381 9382 9383 9384 9385 9386 9387 9388 9389 9390 9391 9392 9393 9394 9395 9396 9397 9398 9399 9400 9401 9402 9403 9404 9405 9406 9407 9408 9409 9410 9411 9412 9413
                        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
9414
                    + generatedRecPerTbl * timeStampStep;
9415 9416 9417 9418 9419 9420 9421 9422 9423 9424 9425 9426 9427 9428 9429 9430 9431 9432 9433 9434 9435 9436 9437 9438 9439 9440 9441 9442 9443 9444 9445 9446 9447 9448 9449 9450 9451 9452 9453 9454 9455 9456 9457 9458 9459 9460 9461 9462 9463 9464 9465 9466 9467 9468 9469 9470 9471 9472 9473 9474 9475 9476 9477 9478 9479 9480 9481 9482 9483 9484 9485 9486 9487 9488 9489 9490 9491 9492 9493 9494 9495 9496 9497 9498 9499 9500 9501 9502 9503 9504 9505 9506 9507 9508 9509 9510

                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
9511

9512 9513 9514 9515 9516 9517 9518
// 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;
9519
    int64_t timeStampStep;
9520 9521 9522 9523 9524 9525 9526
    uint64_t insert_interval;

    SSuperTable* stbInfo = pThreadInfo->stbInfo;

    if (stbInfo) {
        insertRows = stbInfo->insertRows;
        maxSqlLen = stbInfo->maxSqlLen;
9527
        timeStampStep = stbInfo->timeStampStep;
9528 9529 9530 9531
        insert_interval = stbInfo->insertInterval;
    } else {
        insertRows = g_args.insertRows;
        maxSqlLen = g_args.max_sql_len;
9532
        timeStampStep = g_args.timestamp_step;
9533 9534 9535 9536 9537 9538 9539 9540
        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
9541 9542
    if (interlaceRows > g_args.reqPerReq)
        interlaceRows = g_args.reqPerReq;
9543

9544 9545
    uint32_t batchPerTbl = interlaceRows;
    uint32_t batchPerTblTimes;
9546

9547 9548
    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
9549
            g_args.reqPerReq / interlaceRows;
9550 9551 9552
    } else {
        batchPerTblTimes = 1;
    }
9553 9554 9555 9556 9557 9558 9559 9560
#else
    uint32_t batchPerTbl;
    if (interlaceRows > g_args.reqPerReq)
        batchPerTbl = g_args.reqPerReq;
    else
        batchPerTbl = interlaceRows;

    uint32_t batchPerTblTimes;
9561

9562 9563 9564 9565 9566 9567 9568
    if ((interlaceRows > 0) && (pThreadInfo->ntables > 1)) {
        batchPerTblTimes =
            interlaceRows / batchPerTbl;
    } else {
        batchPerTblTimes = 1;
    }
#endif
9569 9570
    pThreadInfo->buffer = calloc(maxSqlLen, 1);
    if (NULL == pThreadInfo->buffer) {
9571
        errorPrint2( "%s() LN%d, Failed to alloc %"PRIu64" Bytes, reason:%s\n",
9572 9573 9574
                __func__, __LINE__, maxSqlLen, strerror(errno));
        return NULL;
    }
9575

9576 9577
    pThreadInfo->totalInsertRows = 0;
    pThreadInfo->totalAffectedRows = 0;
9578

9579 9580
    uint64_t st = 0;
    uint64_t et = UINT64_MAX;
9581

9582 9583 9584
    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
    uint64_t endTs;
9585

9586 9587
    uint64_t tableSeq = pThreadInfo->start_table_from;
    int64_t startTime = pThreadInfo->start_time;
9588

9589 9590 9591
    uint64_t generatedRecPerTbl = 0;
    bool flagSleep = true;
    uint64_t sleepTimeTotal = 0;
9592

9593 9594 9595
    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

9596 9597 9598 9599 9600
    while(pThreadInfo->totalInsertRows < pThreadInfo->ntables * insertRows) {
        if ((flagSleep) && (insert_interval)) {
            st = taosGetTimestampMs();
            flagSleep = false;
        }
9601

9602 9603 9604
        // generate data
        memset(pThreadInfo->buffer, 0, maxSqlLen);
        uint64_t remainderBufLen = maxSqlLen;
9605

9606
        char *pstr = pThreadInfo->buffer;
9607

9608 9609 9610 9611
        int len = snprintf(pstr,
                strlen(STR_INSERT_INTO) + 1, "%s", STR_INSERT_INTO);
        pstr += len;
        remainderBufLen -= len;
9612

9613
        uint32_t recOfBatch = 0;
9614

9615
        int32_t generated;
9616 9617
        for (uint64_t i = 0; i < batchPerTblTimes; i ++) {
            char tableName[TSDB_TABLE_NAME_LEN];
9618

9619 9620
            getTableName(tableName, pThreadInfo, tableSeq);
            if (0 == strlen(tableName)) {
9621
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
9622 9623 9624 9625
                        pThreadInfo->threadID, __func__, __LINE__);
                free(pThreadInfo->buffer);
                return NULL;
            }
9626

9627
            uint64_t oldRemainderLen = remainderBufLen;
9628

9629
            if (stbInfo) {
9630 9631 9632 9633 9634 9635 9636 9637 9638
                generated = generateStbInterlaceData(
                        pThreadInfo,
                        tableName, batchPerTbl, i,
                        batchPerTblTimes,
                        tableSeq,
                        pstr,
                        insertRows,
                        startTime,
                        &remainderBufLen);
9639
            } else {
9640 9641 9642 9643 9644 9645 9646
                generated = generateInterlaceDataWithoutStb(
                        tableName, batchPerTbl,
                        tableSeq,
                        pThreadInfo->db_name, pstr,
                        insertRows,
                        startTime,
                        &remainderBufLen);
9647
            }
9648

9649 9650 9651
            debugPrint("[%d] %s() LN%d, generated records is %d\n",
                    pThreadInfo->threadID, __func__, __LINE__, generated);
            if (generated < 0) {
9652
                errorPrint2("[%d] %s() LN%d, generated records is %d\n",
9653 9654 9655 9656 9657
                        pThreadInfo->threadID, __func__, __LINE__, generated);
                goto free_of_interlace;
            } else if (generated == 0) {
                break;
            }
9658

9659 9660
            tableSeq ++;
            recOfBatch += batchPerTbl;
9661

9662 9663
            pstr += (oldRemainderLen - remainderBufLen);
            pThreadInfo->totalInsertRows += batchPerTbl;
9664

9665 9666 9667
            verbosePrint("[%d] %s() LN%d batchPerTbl=%d recOfBatch=%d\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    batchPerTbl, recOfBatch);
9668

9669 9670 9671 9672
            if (tableSeq == pThreadInfo->start_table_from + pThreadInfo->ntables) {
                // turn to first table
                tableSeq = pThreadInfo->start_table_from;
                generatedRecPerTbl += batchPerTbl;
9673

9674
                startTime = pThreadInfo->start_time
9675
                    + generatedRecPerTbl * timeStampStep;
9676

9677 9678 9679
                flagSleep = true;
                if (generatedRecPerTbl >= insertRows)
                    break;
9680

9681 9682 9683
                int64_t remainRows = insertRows - generatedRecPerTbl;
                if ((remainRows > 0) && (batchPerTbl > remainRows))
                    batchPerTbl = remainRows;
9684

9685
                if (pThreadInfo->ntables * batchPerTbl < g_args.reqPerReq)
9686 9687
                    break;
            }
9688

9689 9690 9691
            verbosePrint("[%d] %s() LN%d generatedRecPerTbl=%"PRId64" insertRows=%"PRId64"\n",
                    pThreadInfo->threadID, __func__, __LINE__,
                    generatedRecPerTbl, insertRows);
9692

9693
            if ((g_args.reqPerReq - recOfBatch) < batchPerTbl)
9694
                break;
9695
        }
9696

9697 9698 9699 9700 9701 9702
        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);

9703
        startTs = taosGetTimestampUs();
9704 9705

        if (recOfBatch == 0) {
9706
            errorPrint2("[%d] %s() LN%d Failed to insert records of batch %d\n",
9707 9708 9709 9710 9711 9712 9713 9714 9715 9716 9717 9718
                    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);

9719
        endTs = taosGetTimestampUs();
9720
        uint64_t delay = endTs - startTs;
9721 9722
        performancePrint("%s() LN%d, insert execution time is %10.2f ms\n",
                __func__, __LINE__, delay / 1000.0);
9723 9724 9725
        verbosePrint("[%d] %s() LN%d affectedRows=%"PRId64"\n",
                pThreadInfo->threadID,
                __func__, __LINE__, affectedRows);
9726

9727 9728 9729 9730
        if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
        if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
        pThreadInfo->cntDelay++;
        pThreadInfo->totalDelay += delay;
9731

9732
        if (recOfBatch != affectedRows) {
9733
            errorPrint2("[%d] %s() LN%d execInsert insert %d, affected rows: %"PRId64"\n%s\n",
9734 9735 9736 9737
                    pThreadInfo->threadID, __func__, __LINE__,
                    recOfBatch, affectedRows, pThreadInfo->buffer);
            goto free_of_interlace;
        }
9738

9739
        pThreadInfo->totalAffectedRows += affectedRows;
9740

9741 9742 9743 9744 9745
        int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
        if (currentPercent > percentComplete ) {
            printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
            percentComplete = currentPercent;
        }
9746 9747 9748
        int64_t  currentPrintTime = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
            printf("thread[%d] has currently inserted rows: %"PRIu64 ", affected rows: %"PRIu64 "\n",
9749 9750 9751
                    pThreadInfo->threadID,
                    pThreadInfo->totalInsertRows,
                    pThreadInfo->totalAffectedRows);
9752 9753
            lastPrintTime = currentPrintTime;
        }
9754

9755 9756
        if ((insert_interval) && flagSleep) {
            et = taosGetTimestampMs();
9757

9758 9759 9760 9761 9762 9763 9764 9765
            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;
            }
        }
9766
    }
9767 9768
    if (percentComplete < 100)
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
9769

9770
free_of_interlace:
9771 9772 9773
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
9774 9775
}

9776 9777 9778 9779 9780 9781 9782 9783 9784 9785 9786 9787 9788 9789 9790 9791 9792 9793 9794 9795 9796 9797 9798 9799 9800 9801 9802 9803 9804 9805 9806 9807 9808 9809 9810 9811 9812 9813 9814 9815 9816 9817 9818 9819 9820 9821 9822 9823 9824 9825 9826 9827 9828 9829 9830 9831 9832 9833 9834 9835 9836 9837 9838 9839 9840 9841 9842 9843 9844 9845 9846 9847 9848 9849 9850 9851 9852 9853 9854 9855 9856 9857 9858 9859 9860 9861 9862 9863 9864 9865 9866 9867 9868 9869 9870 9871 9872 9873 9874 9875 9876 9877 9878 9879 9880 9881 9882 9883 9884 9885 9886 9887 9888 9889 9890 9891 9892 9893 9894 9895 9896 9897 9898 9899 9900 9901 9902 9903 9904 9905 9906 9907 9908 9909 9910 9911 9912 9913
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;
}
9914
// sync insertion progressive data
9915
static void* syncWriteProgressive(threadInfo *pThreadInfo) {
9916
    debugPrint("%s() LN%d: ### progressive write\n", __func__, __LINE__);
9917

9918 9919
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
    uint64_t maxSqlLen = stbInfo?stbInfo->maxSqlLen:g_args.max_sql_len;
9920
    int64_t timeStampStep =
9921
        stbInfo?stbInfo->timeStampStep:g_args.timestamp_step;
9922
    int64_t insertRows =
9923
        (stbInfo)?stbInfo->insertRows:g_args.insertRows;
9924
    verbosePrint("%s() LN%d insertRows=%"PRId64"\n",
9925
            __func__, __LINE__, insertRows);
9926

9927 9928
    pThreadInfo->buffer = calloc(maxSqlLen, 1);
    if (NULL == pThreadInfo->buffer) {
9929
        errorPrint2("Failed to alloc %"PRIu64" bytes, reason:%s\n",
9930 9931
                maxSqlLen,
                strerror(errno));
9932
        return NULL;
9933 9934 9935 9936 9937 9938 9939 9940 9941 9942 9943
    }

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

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

    pThreadInfo->samplePos = 0;

9944 9945 9946
    int percentComplete = 0;
    int64_t totalRows = insertRows * pThreadInfo->ntables;

9947 9948 9949 9950 9951 9952 9953 9954 9955 9956 9957 9958
    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)) {
9959
                errorPrint2("[%d] %s() LN%d, getTableName return null\n",
9960 9961 9962 9963
                        pThreadInfo->threadID, __func__, __LINE__);
                free(pThreadInfo->buffer);
                return NULL;
            }
9964

9965
            int64_t remainderBufLen = maxSqlLen - 2000;
9966
            char *pstr = pThreadInfo->buffer;
9967

9968 9969
            int len = snprintf(pstr,
                    strlen(STR_INSERT_INTO) + 1, "%s", STR_INSERT_INTO);
9970

9971 9972
            pstr += len;
            remainderBufLen -= len;
9973

9974 9975 9976
            // measure prepare + insert
            startTs = taosGetTimestampUs();

9977
            int32_t generated;
9978 9979
            if (stbInfo) {
                if (stbInfo->iface == STMT_IFACE) {
9980
                    generated = prepareStbStmt(
9981 9982 9983
                            pThreadInfo,
                            tableName,
                            tableSeq,
9984 9985 9986
                            (g_args.reqPerReq>stbInfo->insertRows)?
                                stbInfo->insertRows:
                                g_args.reqPerReq,
9987 9988
                            insertRows, i, start_time,
                            &(pThreadInfo->samplePos));
9989 9990
                } else {
                    generated = generateStbProgressiveData(
9991 9992 9993
                            stbInfo,
                            tableName, tableSeq,
                            pThreadInfo->db_name, pstr,
9994 9995 9996 9997 9998 9999 10000
                            insertRows, i, start_time,
                            &(pThreadInfo->samplePos),
                            &remainderBufLen);
                }
            } else {
                if (g_args.iface == STMT_IFACE) {
                    generated = prepareStmtWithoutStb(
10001
                            pThreadInfo,
10002
                            tableName,
10003
                            g_args.reqPerReq,
10004 10005 10006 10007 10008 10009 10010 10011 10012 10013 10014 10015
                            insertRows, i,
                            start_time);
                } else {
                    generated = generateProgressiveDataWithoutStb(
                            tableName,
                            /*  tableSeq, */
                            pThreadInfo, pstr, insertRows,
                            i, start_time,
                            /* &(pThreadInfo->samplePos), */
                            &remainderBufLen);
                }
            }
10016 10017 10018 10019 10020

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

10021 10022 10023 10024
            if (generated > 0)
                i += generated;
            else
                goto free_of_progressive;
10025

10026 10027
            start_time +=  generated * timeStampStep;
            pThreadInfo->totalInsertRows += generated;
10028

10029 10030
            // only measure insert
            // startTs = taosGetTimestampUs();
10031

10032
            int32_t affectedRows = execInsert(pThreadInfo, generated);
10033

10034
            endTs = taosGetTimestampUs();
10035
            uint64_t delay = endTs - startTs;
10036 10037
            performancePrint("%s() LN%d, insert execution time is %10.f ms\n",
                    __func__, __LINE__, delay/1000.0);
10038 10039 10040
            verbosePrint("[%d] %s() LN%d affectedRows=%d\n",
                    pThreadInfo->threadID,
                    __func__, __LINE__, affectedRows);
10041

10042 10043 10044 10045
            if (delay > pThreadInfo->maxDelay) pThreadInfo->maxDelay = delay;
            if (delay < pThreadInfo->minDelay) pThreadInfo->minDelay = delay;
            pThreadInfo->cntDelay++;
            pThreadInfo->totalDelay += delay;
10046

10047
            if (affectedRows < 0) {
10048
                errorPrint2("%s() LN%d, affected rows: %d\n",
10049 10050 10051
                        __func__, __LINE__, affectedRows);
                goto free_of_progressive;
            }
10052

10053
            pThreadInfo->totalAffectedRows += affectedRows;
10054

10055 10056 10057 10058 10059
            int currentPercent = pThreadInfo->totalAffectedRows * 100 / totalRows;
            if (currentPercent > percentComplete ) {
                printf("[%d]:%d%%\n", pThreadInfo->threadID, currentPercent);
                percentComplete = currentPercent;
            }
10060 10061 10062 10063 10064 10065 10066 10067
            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;
            }
10068

10069 10070
            if (i >= insertRows)
                break;
10071
        }   // insertRows
10072

10073
        if ((g_args.verbose_print) &&
10074
                (tableSeq == pThreadInfo->ntables - 1) && (stbInfo)
10075
                && (0 == strncasecmp(
10076
                        stbInfo->dataSource,
10077
                        "sample", strlen("sample")))) {
10078 10079 10080 10081
            verbosePrint("%s() LN%d samplePos=%"PRId64"\n",
                    __func__, __LINE__, pThreadInfo->samplePos);
        }
    } // tableSeq
10082 10083

    if (percentComplete < 100) {
10084
        printf("[%d]:%d%%\n", pThreadInfo->threadID, percentComplete);
10085
    }
10086

10087
free_of_progressive:
10088 10089 10090
    tmfree(pThreadInfo->buffer);
    printStatPerThread(pThreadInfo);
    return NULL;
H
Hui Li 已提交
10091 10092
}

10093 10094
static void* syncWrite(void *sarg) {

10095
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10096
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10097

10098 10099
    setThreadName("syncWrite");

10100
    uint32_t interlaceRows = 0;
10101

10102
    if (stbInfo) {
10103
        if (stbInfo->interlaceRows < stbInfo->insertRows)
10104
            interlaceRows = stbInfo->interlaceRows;
10105
    } else {
10106 10107
        if (g_args.interlaceRows < g_args.insertRows)
            interlaceRows = g_args.interlaceRows;
10108
    }
10109

10110 10111
    if (interlaceRows > 0) {
        // interlace mode
10112 10113
        if (stbInfo) {
            if (STMT_IFACE == stbInfo->iface) {
10114
#if STMT_BIND_PARAM_BATCH == 1
10115
                return syncWriteInterlaceStmtBatch(pThreadInfo, interlaceRows);
10116
#else
10117
                return syncWriteInterlaceStmt(pThreadInfo, interlaceRows);
10118
#endif
10119 10120 10121
            } else {
                return syncWriteInterlace(pThreadInfo, interlaceRows);
            }
10122
        }
10123 10124 10125 10126 10127 10128 10129 10130
    } else {
      // progressive mode
      if (((stbInfo) && (STMT_IFACE == stbInfo->iface))
              || (STMT_IFACE == g_args.iface)) {
          return syncWriteProgressiveStmt(pThreadInfo);
      } else {
          return syncWriteProgressive(pThreadInfo);
      }
10131
    }
10132 10133

    return NULL;
10134 10135
}

10136
static void callBack(void *param, TAOS_RES *res, int code) {
10137
    threadInfo* pThreadInfo = (threadInfo*)param;
10138
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10139 10140

    int insert_interval =
10141
        stbInfo?stbInfo->insertInterval:g_args.insert_interval;
10142 10143 10144 10145 10146
    if (insert_interval) {
        pThreadInfo->et = taosGetTimestampMs();
        if ((pThreadInfo->et - pThreadInfo->st) < insert_interval) {
            taosMsleep(insert_interval - (pThreadInfo->et - pThreadInfo->st)); // ms
        }
H
Hui Li 已提交
10147 10148
    }

10149
    char *buffer = calloc(1, pThreadInfo->stbInfo->maxSqlLen);
10150 10151
    char data[MAX_DATA_SIZE];
    char *pstr = buffer;
10152
    pstr += sprintf(pstr, "INSERT INTO %s.%s%"PRId64" VALUES",
10153 10154
            pThreadInfo->db_name, pThreadInfo->tb_prefix,
            pThreadInfo->start_table_from);
10155
    //  if (pThreadInfo->counter >= pThreadInfo->stbInfo->insertRows) {
10156
    if (pThreadInfo->counter >= g_args.reqPerReq) {
10157 10158 10159 10160 10161 10162 10163 10164
        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 已提交
10165
    }
10166

10167
    for (int i = 0; i < g_args.reqPerReq; i++) {
10168
        int rand_num = taosRandom() % 100;
10169 10170
        if (0 != pThreadInfo->stbInfo->disorderRatio
                && rand_num < pThreadInfo->stbInfo->disorderRatio) {
10171
            int64_t d = pThreadInfo->lastTs
10172 10173
                - (taosRandom() % pThreadInfo->stbInfo->disorderRange + 1);
            generateStbRowData(pThreadInfo->stbInfo, data,
10174 10175
                    MAX_DATA_SIZE,
                    d);
10176
        } else {
10177
            generateStbRowData(pThreadInfo->stbInfo,
10178 10179 10180
                    data,
                    MAX_DATA_SIZE,
                    pThreadInfo->lastTs += 1000);
10181 10182 10183
        }
        pstr += sprintf(pstr, "%s", data);
        pThreadInfo->counter++;
H
Hui Li 已提交
10184

10185
        if (pThreadInfo->counter >= pThreadInfo->stbInfo->insertRows) {
10186 10187 10188 10189 10190 10191 10192 10193 10194 10195 10196
            break;
        }
    }

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

    taos_free_result(res);
H
Hui Li 已提交
10197 10198
}

10199
static void *asyncWrite(void *sarg) {
10200
    threadInfo *pThreadInfo = (threadInfo *)sarg;
10201
    SSuperTable* stbInfo = pThreadInfo->stbInfo;
10202

10203
    setThreadName("asyncWrite");
10204

10205 10206 10207
    pThreadInfo->st = 0;
    pThreadInfo->et = 0;
    pThreadInfo->lastTs = pThreadInfo->start_time;
10208

10209
    int insert_interval =
10210
        stbInfo?stbInfo->insertInterval:g_args.insert_interval;
10211 10212 10213 10214
    if (insert_interval) {
        pThreadInfo->st = taosGetTimestampMs();
    }
    taos_query_a(pThreadInfo->taos, "show databases", callBack, pThreadInfo);
10215

10216
    tsem_wait(&(pThreadInfo->lock_sem));
H
Hui Li 已提交
10217

10218
    return NULL;
H
Hui Li 已提交
10219 10220
}

10221 10222
static int convertHostToServAddr(char *host, uint16_t port, struct sockaddr_in *serv_addr)
{
10223 10224 10225
    uint16_t rest_port = port + TSDB_PORT_HTTP;
    struct hostent *server = gethostbyname(host);
    if ((server == NULL) || (server->h_addr == NULL)) {
10226
        errorPrint2("%s", "no such host");
10227 10228
        return -1;
    }
10229

10230
    debugPrint("h_name: %s\nh_addr=%p\nh_addretype: %s\nh_length: %d\n",
10231 10232 10233 10234 10235
            server->h_name,
            server->h_addr,
            (server->h_addrtype == AF_INET)?"ipv4":"ipv6",
            server->h_length);

10236 10237 10238
    memset(serv_addr, 0, sizeof(struct sockaddr_in));
    serv_addr->sin_family = AF_INET;
    serv_addr->sin_port = htons(rest_port);
10239
#ifdef WINDOWS
10240
    serv_addr->sin_addr.s_addr = inet_addr(host);
10241
#else
10242
    memcpy(&(serv_addr->sin_addr.s_addr), server->h_addr, server->h_length);
10243
#endif
10244
    return 0;
10245 10246
}

10247
static void startMultiThreadInsertData(int threads, char* db_name,
10248
        char* precision, SSuperTable* stbInfo) {
10249

10250 10251 10252 10253 10254 10255 10256 10257 10258
    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 {
10259
            errorPrint2("Not support precision: %s\n", precision);
10260
            exit(EXIT_FAILURE);
10261
        }
10262 10263
    }

10264
    int64_t startTime;
10265 10266
    if (stbInfo) {
        if (0 == strncasecmp(stbInfo->startTimestamp, "now", 3)) {
10267
            startTime = taosGetTimestamp(timePrec);
10268 10269
        } else {
            if (TSDB_CODE_SUCCESS != taosParseTime(
10270
                        stbInfo->startTimestamp,
10271
                        &startTime,
10272
                        strlen(stbInfo->startTimestamp),
10273 10274 10275 10276
                        timePrec, 0)) {
                ERROR_EXIT("failed to parse time!\n");
            }
        }
10277
    } else {
10278
        startTime = DEFAULT_START_TIME;
10279
    }
10280 10281
    debugPrint("%s() LN%d, startTime= %"PRId64"\n",
            __func__, __LINE__, startTime);
10282

10283
    // read sample data from file first
10284
    int ret;
10285
    if (stbInfo) {
10286 10287 10288 10289 10290 10291 10292 10293 10294
        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);
10295 10296
    }

10297 10298 10299 10300
    TAOS* taos0 = taos_connect(
            g_Dbs.host, g_Dbs.user,
            g_Dbs.password, db_name, g_Dbs.port);
    if (NULL == taos0) {
10301
        errorPrint2("%s() LN%d, connect to server fail , reason: %s\n",
10302
                __func__, __LINE__, taos_errstr(NULL));
10303
        exit(EXIT_FAILURE);
10304
    }
10305

10306 10307
    int64_t ntables = 0;
    uint64_t tableFrom;
10308

10309
    if (stbInfo) {
10310 10311
        int64_t limit;
        uint64_t offset;
10312

10313
        if ((NULL != g_args.sqlFile)
10314 10315 10316
                && (stbInfo->childTblExists == TBL_NO_EXISTS)
                && ((stbInfo->childTblOffset != 0)
                    || (stbInfo->childTblLimit >= 0))) {
10317 10318
            printf("WARNING: offset and limit will not be used since the child tables not exists!\n");
        }
10319

10320 10321 10322 10323 10324
        if (stbInfo->childTblExists == TBL_ALREADY_EXISTS) {
            if ((stbInfo->childTblLimit < 0)
                    || ((stbInfo->childTblOffset
                            + stbInfo->childTblLimit)
                        > (stbInfo->childTblCount))) {
10325 10326 10327 10328 10329 10330

                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;
                }
10331 10332
                stbInfo->childTblLimit =
                    stbInfo->childTblCount - stbInfo->childTblOffset;
10333
            }
10334

10335 10336
            offset = stbInfo->childTblOffset;
            limit = stbInfo->childTblLimit;
10337
        } else {
10338
            limit = stbInfo->childTblCount;
10339 10340
            offset = 0;
        }
10341

10342 10343
        ntables = limit;
        tableFrom = offset;
10344

10345 10346 10347
        if ((stbInfo->childTblExists != TBL_NO_EXISTS)
                && ((stbInfo->childTblOffset + stbInfo->childTblLimit)
                    > stbInfo->childTblCount)) {
10348 10349 10350
            printf("WARNING: specified offset + limit > child table count!\n");
            prompt();
        }
10351

10352 10353
        if ((stbInfo->childTblExists != TBL_NO_EXISTS)
                && (0 == stbInfo->childTblLimit)) {
10354 10355 10356
            printf("WARNING: specified limit = 0, which cannot find table name to insert or query! \n");
            prompt();
        }
10357

10358
        stbInfo->childTblName = (char*)calloc(1,
10359
                limit * TSDB_TABLE_NAME_LEN);
10360
        if (stbInfo->childTblName == NULL) {
10361
            taos_close(taos0);
10362
            errorPrint2("%s() LN%d, alloc memory failed!\n", __func__, __LINE__);
10363
            exit(EXIT_FAILURE);
10364
        }
10365

10366 10367 10368
        int64_t childTblCount;
        getChildNameOfSuperTableWithLimitAndOffset(
                taos0,
10369
                db_name, stbInfo->stbName,
10370
                &stbInfo->childTblName, &childTblCount,
10371
                limit,
10372
                offset, stbInfo->escapeChar);
10373
        ntables = childTblCount;
10374
    } else {
10375
        ntables = g_args.ntables;
10376 10377
        tableFrom = 0;
    }
10378

10379
    taos_close(taos0);
10380

10381 10382 10383 10384 10385
    int64_t a = ntables / threads;
    if (a < 1) {
        threads = ntables;
        a = 1;
    }
10386

10387 10388 10389 10390
    int64_t b = 0;
    if (threads != 0) {
        b = ntables % threads;
    }
10391

10392
    if (g_args.iface == REST_IFACE || ((stbInfo) && (stbInfo->iface == REST_IFACE))) {
10393 10394
        if (convertHostToServAddr(
                    g_Dbs.host, g_Dbs.port, &(g_Dbs.serv_addr)) != 0) {
10395
            ERROR_EXIT("convert host to server address");
10396 10397
        }
    }
10398

10399 10400
    pthread_t *pids = calloc(1, threads * sizeof(pthread_t));
    threadInfo *infos = calloc(1, threads * sizeof(threadInfo));
10401
    assert(pids != NULL);
10402
    assert(infos != NULL);
10403

10404 10405
    char *stmtBuffer = calloc(1, BUFFER_SIZE);
    assert(stmtBuffer);
10406 10407

#if STMT_BIND_PARAM_BATCH == 1
10408
    uint32_t interlaceRows = 0;
10409 10410 10411
    uint32_t batch;

    if (stbInfo) {
10412
        if (stbInfo->interlaceRows < stbInfo->insertRows)
10413 10414
            interlaceRows = stbInfo->interlaceRows;
    } else {
10415 10416
        if (g_args.interlaceRows < g_args.insertRows)
            interlaceRows = g_args.interlaceRows;
10417 10418 10419 10420 10421 10422 10423 10424 10425 10426 10427
    }

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

#endif

10428 10429 10430 10431 10432 10433 10434 10435 10436
    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(?",
10437
                    stbInfo->stbName);
10438 10439 10440 10441 10442 10443 10444 10445 10446
            for (int tag = 0; tag < (stbInfo->tagCount - 1);
                    tag ++ ) {
                pstr += sprintf(pstr, ",?");
            }
            pstr += sprintf(pstr, ") VALUES(?");
        } else {
            pstr += sprintf(pstr, "INSERT INTO ? VALUES(?");
        }

10447 10448 10449
        int columnCount = (stbInfo)?
            stbInfo->columnCount:
            g_args.columnCount;
10450 10451 10452 10453 10454 10455 10456

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

        debugPrint("%s() LN%d, stmtBuffer: %s", __func__, __LINE__, stmtBuffer);
10457 10458 10459
#if STMT_BIND_PARAM_BATCH == 1
        parseSamplefileToStmtBatch(stbInfo);
#endif
10460 10461
    }

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

10466 10467
        tstrncpy(pThreadInfo->db_name, db_name, TSDB_DB_NAME_LEN);
        pThreadInfo->time_precision = timePrec;
10468
        pThreadInfo->stbInfo = stbInfo;
10469

10470
        pThreadInfo->start_time = startTime;
10471 10472
        pThreadInfo->minDelay = UINT64_MAX;

10473 10474
        if ((NULL == stbInfo) ||
                (stbInfo->iface != REST_IFACE)) {
10475 10476 10477 10478 10479
            //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) {
10480
                free(infos);
10481
                errorPrint2(
10482 10483 10484
                        "%s() LN%d, connect to server fail from insert sub thread, reason: %s\n",
                        __func__, __LINE__,
                        taos_errstr(NULL));
10485
                exit(EXIT_FAILURE);
10486
            }
10487

10488
            if ((g_args.iface == STMT_IFACE)
10489 10490
                    || ((stbInfo)
                        && (stbInfo->iface == STMT_IFACE))) {
10491

10492 10493
                pThreadInfo->stmt = taos_stmt_init(pThreadInfo->taos);
                if (NULL == pThreadInfo->stmt) {
10494 10495
                    free(pids);
                    free(infos);
10496
                    errorPrint2(
10497 10498 10499
                            "%s() LN%d, failed init stmt, reason: %s\n",
                            __func__, __LINE__,
                            taos_errstr(NULL));
10500
                    exit(EXIT_FAILURE);
10501
                }
10502

10503
                if (0 != taos_stmt_prepare(pThreadInfo->stmt, stmtBuffer, 0)) {
10504 10505
                    free(pids);
                    free(infos);
10506
                    free(stmtBuffer);
10507
                    errorPrint2("failed to execute taos_stmt_prepare. return 0x%x. reason: %s\n",
10508 10509
                            ret, taos_stmt_errstr(pThreadInfo->stmt));
                    exit(EXIT_FAILURE);
10510
                }
10511
                pThreadInfo->bind_ts = malloc(sizeof(int64_t));
10512 10513

                if (stbInfo) {
10514 10515 10516 10517 10518 10519 10520 10521 10522 10523 10524 10525 10526
#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
10527
                }
10528 10529
            }
        } else {
10530
            pThreadInfo->taos = NULL;
10531
        }
10532

10533 10534
        /*    if ((NULL == stbInfo)
              || (0 == stbInfo->multiThreadWriteOneTbl)) {
10535 10536 10537 10538 10539 10540 10541
              */
        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;
10542
              pThreadInfo->ntables = stbInfo->childTblCount;
10543 10544 10545
              pThreadInfo->start_time = pThreadInfo->start_time + rand_int() % 10000 - rand_tinyint();
              }
              */
10546 10547 10548 10549 10550 10551 10552 10553 10554 10555 10556 10557 10558 10559 10560 10561 10562 10563 10564 10565 10566 10567 10568 10569 10570 10571 10572
        
        if (g_args.iface == REST_IFACE || ((stbInfo) && (stbInfo->iface == REST_IFACE))) {
#ifdef WINDOWS
            WSADATA wsaData;
            WSAStartup(MAKEWORD(2, 2), &wsaData);
            SOCKET sockfd;
#else
            int sockfd;
#endif
            sockfd = socket(AF_INET, SOCK_STREAM, 0);
            if (sockfd < 0) {
#ifdef WINDOWS
                errorPrint( "Could not create socket : %d" , WSAGetLastError());
#endif
                debugPrint("%s() LN%d, sockfd=%d\n", __func__, __LINE__, sockfd);
                ERROR_EXIT("opening socket");
            }

            int retConn = connect(sockfd, (struct sockaddr *)&(g_Dbs.serv_addr), sizeof(struct sockaddr));
            debugPrint("%s() LN%d connect() return %d\n", __func__, __LINE__, retConn);
            if (retConn < 0) {
                ERROR_EXIT("connecting");
            }
            pThreadInfo->sockfd = sockfd;
        }
        

10573 10574 10575 10576 10577
        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);
10578
        }
10579 10580
    }

10581 10582
    free(stmtBuffer);

10583
    int64_t start = taosGetTimestampUs();
10584

10585 10586
    for (int i = 0; i < threads; i++) {
        pthread_join(pids[i], NULL);
10587
    }
H
Hui Li 已提交
10588

10589 10590 10591 10592 10593
    uint64_t totalDelay = 0;
    uint64_t maxDelay = 0;
    uint64_t minDelay = UINT64_MAX;
    uint64_t cntDelay = 1;
    double  avgDelay = 0;
10594

10595 10596
    for (int i = 0; i < threads; i++) {
        threadInfo *pThreadInfo = infos + i;
10597

10598 10599 10600
        tsem_destroy(&(pThreadInfo->lock_sem));
        taos_close(pThreadInfo->taos);

10601 10602 10603
        if (pThreadInfo->stmt) {
            taos_stmt_close(pThreadInfo->stmt);
        }
10604

10605
        tmfree((char *)pThreadInfo->bind_ts);
10606
#if STMT_BIND_PARAM_BATCH == 1
10607 10608 10609
        tmfree((char *)pThreadInfo->bind_ts_array);
        tmfree(pThreadInfo->bindParams);
        tmfree(pThreadInfo->is_null);
10610 10611 10612 10613 10614 10615 10616 10617
        if (g_args.iface == REST_IFACE || ((stbInfo) && (stbInfo->iface == REST_IFACE))) {
#ifdef WINDOWS
            closesocket(pThreadInfo->sockfd);
            WSACleanup();
#else
            close(pThreadInfo->sockfd);
#endif
        }
10618
#else
10619
        if (pThreadInfo->sampleBindArray) {
10620
            for (int k = 0; k < MAX_SAMPLES; k++) {
10621 10622 10623
                uintptr_t *tmp = (uintptr_t *)(*(uintptr_t *)(
                            pThreadInfo->sampleBindArray
                            + sizeof(uintptr_t *) * k));
10624 10625 10626 10627
                int columnCount = (pThreadInfo->stbInfo)?
                    pThreadInfo->stbInfo->columnCount:
                    g_args.columnCount;
                for (int c = 1; c < columnCount + 1; c++) {
10628 10629 10630 10631 10632 10633 10634 10635
                    TAOS_BIND *bind = (TAOS_BIND *)((char *)tmp + (sizeof(TAOS_BIND) * c));
                    if (bind)
                        tmfree(bind->buffer);
                }
                tmfree((char *)tmp);
            }
            tmfree(pThreadInfo->sampleBindArray);
        }
10636
#endif
10637

10638 10639 10640 10641
        debugPrint("%s() LN%d, [%d] totalInsert=%"PRIu64" totalAffected=%"PRIu64"\n",
                __func__, __LINE__,
                pThreadInfo->threadID, pThreadInfo->totalInsertRows,
                pThreadInfo->totalAffectedRows);
10642 10643 10644
        if (stbInfo) {
            stbInfo->totalAffectedRows += pThreadInfo->totalAffectedRows;
            stbInfo->totalInsertRows += pThreadInfo->totalInsertRows;
10645 10646 10647 10648
        } else {
            g_args.totalAffectedRows += pThreadInfo->totalAffectedRows;
            g_args.totalInsertRows += pThreadInfo->totalInsertRows;
        }
H
Hui Li 已提交
10649

10650 10651 10652 10653 10654
        totalDelay  += pThreadInfo->totalDelay;
        cntDelay   += pThreadInfo->cntDelay;
        if (pThreadInfo->maxDelay > maxDelay) maxDelay = pThreadInfo->maxDelay;
        if (pThreadInfo->minDelay < minDelay) minDelay = pThreadInfo->minDelay;
    }
10655

10656 10657
    if (cntDelay == 0)    cntDelay = 1;
    avgDelay = (double)totalDelay / cntDelay;
10658

10659
    int64_t end = taosGetTimestampUs();
10660
    int64_t t = end - start;
10661
    if (0 == t) t = 1;
10662

10663
    double tInMs = (double) t / 1000000.0;
10664

10665
    if (stbInfo) {
10666
        fprintf(stderr, "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s.%s. %.2f records/second\n\n",
10667 10668
                tInMs, stbInfo->totalInsertRows,
                stbInfo->totalAffectedRows,
10669
                threads, db_name, stbInfo->stbName,
10670
                (double)(stbInfo->totalInsertRows/tInMs));
10671 10672 10673

        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10674
                    "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s.%s. %.2f records/second\n\n",
10675 10676
                    tInMs, stbInfo->totalInsertRows,
                    stbInfo->totalAffectedRows,
10677
                    threads, db_name, stbInfo->stbName,
10678
                    (double)(stbInfo->totalInsertRows/tInMs));
10679 10680
        }
    } else {
10681
        fprintf(stderr, "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s %.2f records/second\n\n",
10682 10683 10684
                tInMs, g_args.totalInsertRows,
                g_args.totalAffectedRows,
                threads, db_name,
10685
                (double)(g_args.totalInsertRows/tInMs));
10686 10687
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10688
                    "Spent %.4f seconds to insert rows: %"PRIu64", affected rows: %"PRIu64" with %d thread(s) into %s %.2f records/second\n\n",
10689 10690 10691
                    tInMs, g_args.totalInsertRows,
                    g_args.totalAffectedRows,
                    threads, db_name,
10692
                    (double)(g_args.totalInsertRows/tInMs));
10693
        }
10694
    }
10695

10696 10697 10698 10699 10700 10701 10702 10703 10704 10705 10706 10707
    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);
        }
10708
    }
10709

10710
    //taos_close(taos);
10711

10712 10713
    free(pids);
    free(infos);
H
Hui Li 已提交
10714 10715
}

10716
static void *queryNtableAggrFunc(void *sarg) {
10717 10718
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    TAOS *taos = pThreadInfo->taos;
10719
    setThreadName("queryNtableAggrFunc");
10720 10721 10722
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);

10723
    uint64_t startTime = pThreadInfo->start_time;
10724 10725 10726
    char *tb_prefix = pThreadInfo->tb_prefix;
    FILE *fp = fopen(pThreadInfo->filePath, "a");
    if (NULL == fp) {
10727
        errorPrint2("fopen %s fail, reason:%s.\n", pThreadInfo->filePath, strerror(errno));
10728
        free(command);
10729
        return NULL;
10730
    }
H
hzcheng 已提交
10731

10732
    int64_t insertRows;
10733
    /*  if (pThreadInfo->stbInfo) {
10734
        insertRows = pThreadInfo->stbInfo->insertRows; //  nrecords_per_table;
10735 10736
        } else {
        */
10737
    insertRows = g_args.insertRows;
10738 10739
    //  }

10740
    int64_t ntables = pThreadInfo->ntables; // pThreadInfo->end_table_to - pThreadInfo->start_table_from + 1;
10741
    int64_t totalData = insertRows * ntables;
10742 10743 10744 10745
    bool aggr_func = g_Dbs.aggr_func;

    char **aggreFunc;
    int n;
10746

10747 10748 10749 10750 10751 10752 10753 10754 10755
    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) {
10756 10757 10758 10759 10760 10761 10762 10763
        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;
10764 10765
        for (int64_t i = 0; i < ntables; i++) {
            sprintf(command, "SELECT %s FROM %s%"PRId64" WHERE ts>= %" PRIu64,
10766
                    aggreFunc[j], tb_prefix, i, startTime);
10767

10768 10769 10770
            double t = taosGetTimestampUs();
            debugPrint("%s() LN%d, sql command: %s\n",
                    __func__, __LINE__, command);
10771 10772 10773 10774
            TAOS_RES *pSql = taos_query(taos, command);
            int32_t code = taos_errno(pSql);

            if (code != 0) {
10775
                errorPrint2("Failed to query:%s\n", taos_errstr(pSql));
10776 10777 10778
                taos_free_result(pSql);
                taos_close(taos);
                fclose(fp);
10779
                free(command);
10780 10781
                return NULL;
            }
H
hzcheng 已提交
10782

10783 10784 10785
            while(taos_fetch_row(pSql) != NULL) {
                count++;
            }
H
hzcheng 已提交
10786

10787
            t = taosGetTimestampUs() - t;
10788 10789 10790 10791
            totalT += t;

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

10793
        fprintf(fp, "|%10s  |   %"PRId64"   |  %12.2f   |   %10.2f  |\n",
10794 10795 10796
                aggreFunc[j][0] == '*' ? "   *   " : aggreFunc[j], totalData,
                (double)(ntables * insertRows) / totalT, totalT / 1000000);
        printf("select %10s took %.6f second(s)\n", aggreFunc[j], totalT / 1000000);
10797 10798 10799
    }
    fprintf(fp, "\n");
    fclose(fp);
10800
    free(command);
10801
    return NULL;
H
hzcheng 已提交
10802 10803
}

10804
static void *queryStableAggrFunc(void *sarg) {
10805 10806
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    TAOS *taos = pThreadInfo->taos;
10807
    setThreadName("queryStableAggrFunc");
10808 10809 10810
    char *command = calloc(1, BUFFER_SIZE);
    assert(command);

10811 10812 10813
    FILE *fp = fopen(pThreadInfo->filePath, "a");
    if (NULL == fp) {
        printf("fopen %s fail, reason:%s.\n", pThreadInfo->filePath, strerror(errno));
10814
        free(command);
10815 10816
        return NULL;
    }
10817

10818
    int64_t insertRows = pThreadInfo->stbInfo->insertRows;
10819
    int64_t ntables = pThreadInfo->ntables; // pThreadInfo->end_table_to - pThreadInfo->start_table_from + 1;
10820
    int64_t totalData = insertRows * ntables;
10821 10822 10823 10824
    bool aggr_func = g_Dbs.aggr_func;

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

10826 10827 10828 10829 10830 10831 10832 10833 10834
    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) {
10835 10836
        printf("\nThe first field is either Binary or Bool. Aggregation functions are not supported.\n");
    }
10837

10838 10839
    printf("%"PRId64" records:\n", totalData);
    fprintf(fp, "Querying On %"PRId64" records:\n", totalData);
H
hzcheng 已提交
10840

10841 10842 10843
    for (int j = 0; j < n; j++) {
        char condition[COND_BUF_LEN] = "\0";
        char tempS[64] = "\0";
H
hzcheng 已提交
10844

10845
        int64_t m = 10 < ntables ? 10 : ntables;
H
hzcheng 已提交
10846

10847 10848
        for (int64_t i = 1; i <= m; i++) {
            if (i == 1) {
10849 10850 10851 10852 10853
                if (g_args.demo_mode) {
                    sprintf(tempS, "groupid = %"PRId64"", i);
                } else {
                    sprintf(tempS, "t0 = %"PRId64"", i);
                }
10854
            } else {
10855 10856 10857 10858 10859
                if (g_args.demo_mode) {
                    sprintf(tempS, " or groupid = %"PRId64" ", i);
                } else {
                    sprintf(tempS, " or t0 = %"PRId64" ", i);
                }
10860 10861
            }
            strncat(condition, tempS, COND_BUF_LEN - 1);
H
hzcheng 已提交
10862

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

10865
            printf("Where condition: %s\n", condition);
10866 10867 10868

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

10871
            double t = taosGetTimestampUs();
H
hzcheng 已提交
10872

10873 10874
            TAOS_RES *pSql = taos_query(taos, command);
            int32_t code = taos_errno(pSql);
S
Shuaiqiang Chang 已提交
10875

10876
            if (code != 0) {
10877
                errorPrint2("Failed to query:%s\n", taos_errstr(pSql));
10878 10879 10880
                taos_free_result(pSql);
                taos_close(taos);
                fclose(fp);
10881
                free(command);
10882 10883 10884 10885 10886 10887
                return NULL;
            }
            int count = 0;
            while(taos_fetch_row(pSql) != NULL) {
                count++;
            }
10888
            t = taosGetTimestampUs() - t;
H
hzcheng 已提交
10889

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

10894 10895 10896
            taos_free_result(pSql);
        }
        fprintf(fp, "\n");
H
hzcheng 已提交
10897
    }
10898
    fclose(fp);
10899
    free(command);
10900

10901
    return NULL;
H
hzcheng 已提交
10902 10903
}

10904 10905
static void prompt()
{
10906 10907 10908 10909
    if (!g_args.answer_yes) {
        printf("         Press enter key to continue or Ctrl-C to stop\n\n");
        (void)getchar();
    }
10910
}
H
Hui Li 已提交
10911

10912
static int insertTestProcess() {
10913

10914 10915 10916
    setupForAnsiEscape();
    int ret = printfInsertMeta();
    resetAfterAnsiEscape();
10917

10918 10919
    if (ret == -1)
        exit(EXIT_FAILURE);
10920

10921 10922 10923
    debugPrint("%d result file: %s\n", __LINE__, g_Dbs.resultFile);
    g_fpOfInsertResult = fopen(g_Dbs.resultFile, "a");
    if (NULL == g_fpOfInsertResult) {
10924
        errorPrint("Failed to open %s for save result\n", g_Dbs.resultFile);
10925 10926
        return -1;
    }
10927

10928 10929
    if (g_fpOfInsertResult)
        printfInsertMetaToFile(g_fpOfInsertResult);
10930

10931
    prompt();
10932

10933
    init_rand_data();
10934

10935
    // create database and super tables
10936 10937 10938 10939
    char *cmdBuffer = calloc(1, BUFFER_SIZE);
    assert(cmdBuffer);

    if(createDatabasesAndStables(cmdBuffer) != 0) {
10940 10941
        if (g_fpOfInsertResult)
            fclose(g_fpOfInsertResult);
10942
        free(cmdBuffer);
10943 10944
        return -1;
    }
10945
    free(cmdBuffer);
10946

10947
    // pretreatment
10948 10949 10950 10951 10952
    if (prepareSampleData() != 0) {
        if (g_fpOfInsertResult)
            fclose(g_fpOfInsertResult);
        return -1;
    }
10953

10954 10955
    double start;
    double end;
10956

10957
    if (g_totalChildTables > 0) {
10958 10959
        fprintf(stderr,
                "creating %"PRId64" table(s) with %d thread(s)\n\n",
10960
                g_totalChildTables, g_Dbs.threadCountForCreateTbl);
10961 10962 10963
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
                "creating %"PRId64" table(s) with %d thread(s)\n\n",
10964
                g_totalChildTables, g_Dbs.threadCountForCreateTbl);
10965 10966 10967 10968 10969 10970 10971 10972
        }

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

        fprintf(stderr,
10973
                "\nSpent %.4f seconds to create %"PRId64" table(s) with %d thread(s), actual %"PRId64" table(s) created\n\n",
10974
                (end - start)/1000.0, g_totalChildTables,
10975
                g_Dbs.threadCountForCreateTbl, g_actualChildTables);
10976 10977
        if (g_fpOfInsertResult) {
            fprintf(g_fpOfInsertResult,
10978
                "\nSpent %.4f seconds to create %"PRId64" table(s) with %d thread(s), actual %"PRId64" table(s) created\n\n",
10979
                (end - start)/1000.0, g_totalChildTables,
10980
                g_Dbs.threadCountForCreateTbl, g_actualChildTables);
10981
        }
10982
    }
10983

10984 10985 10986 10987 10988 10989
    // 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++) {
10990

10991
                    SSuperTable* stbInfo = &g_Dbs.db[i].superTbls[j];
10992

10993
                    if (stbInfo && (stbInfo->insertRows > 0)) {
10994 10995 10996 10997
                        startMultiThreadInsertData(
                                g_Dbs.threadCount,
                                g_Dbs.db[i].dbName,
                                g_Dbs.db[i].dbCfg.precision,
10998
                                stbInfo);
10999 11000 11001 11002
                    }
                }
            }
        } else {
11003
            startMultiThreadInsertData(
11004 11005 11006 11007
                    g_Dbs.threadCount,
                    g_Dbs.db[i].dbName,
                    g_Dbs.db[i].dbCfg.precision,
                    NULL);
11008
        }
H
Hui Li 已提交
11009
    }
11010
    //end = taosGetTimestampMs();
11011

11012 11013 11014 11015 11016 11017 11018 11019 11020 11021 11022
    //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;
11023 11024
}

11025
static void *specifiedTableQuery(void *sarg) {
11026
    threadInfo *pThreadInfo = (threadInfo *)sarg;
11027

11028
    setThreadName("specTableQuery");
11029

11030 11031 11032 11033 11034 11035 11036 11037
    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) {
11038
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11039 11040 11041 11042 11043
                    pThreadInfo->threadID, taos_errstr(NULL));
            return NULL;
        } else {
            pThreadInfo->taos = taos;
        }
11044 11045
    }

11046 11047 11048 11049
    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);
11050
        errorPrint("use database %s failed!\n\n",
11051
                g_queryInfo.dbName);
11052 11053
        return NULL;
    }
11054

11055 11056
    uint64_t st = 0;
    uint64_t et = 0;
11057

11058
    uint64_t queryTimes = g_queryInfo.specifiedQueryInfo.queryTimes;
11059

11060 11061 11062
    uint64_t totalQueried = 0;
    uint64_t lastPrintTime = taosGetTimestampMs();
    uint64_t startTs = taosGetTimestampMs();
11063

11064 11065
    if (g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq][0] != '\0') {
        sprintf(pThreadInfo->filePath, "%s-%d",
11066 11067
                g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                pThreadInfo->threadID);
11068 11069
    }

11070 11071 11072 11073 11074 11075 11076
    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();
11077

11078 11079
        selectAndGetResult(pThreadInfo,
                g_queryInfo.specifiedQueryInfo.sql[pThreadInfo->querySeq]);
11080

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

11085 11086
        totalQueried ++;
        g_queryInfo.specifiedQueryInfo.totalQueried ++;
11087

11088 11089 11090
        uint64_t  currentPrintTime = taosGetTimestampMs();
        uint64_t  endTs = taosGetTimestampMs();
        if (currentPrintTime - lastPrintTime > 30*1000) {
11091
            debugPrint("%s() LN%d, endTs=%"PRIu64" ms, startTs=%"PRIu64" ms\n",
11092 11093
                    __func__, __LINE__, endTs, startTs);
            printf("thread[%d] has currently completed queries: %"PRIu64", QPS: %10.6f\n",
11094 11095
                    pThreadInfo->threadID,
                    totalQueried,
11096
                    (double)(totalQueried/((endTs-startTs)/1000.0)));
11097 11098
            lastPrintTime = currentPrintTime;
        }
11099
    }
11100
    return NULL;
11101
}
H
Hui Li 已提交
11102

11103
static void replaceChildTblName(char* inSql, char* outSql, int tblIndex) {
11104
    char sourceString[32] = "xxxx";
11105
    char subTblName[TSDB_TABLE_NAME_LEN];
11106 11107 11108
    sprintf(subTblName, "%s.%s",
            g_queryInfo.dbName,
            g_queryInfo.superQueryInfo.childTblName + tblIndex*TSDB_TABLE_NAME_LEN);
11109

11110
    //printf("inSql: %s\n", inSql);
11111

11112 11113 11114 11115 11116 11117 11118
    char* pos = strstr(inSql, sourceString);
    if (0 == pos) {
        return;
    }

    tstrncpy(outSql, inSql, pos - inSql + 1);
    //printf("1: %s\n", outSql);
11119
    strncat(outSql, subTblName, BUFFER_SIZE - 1);
11120
    //printf("2: %s\n", outSql);
11121
    strncat(outSql, pos+strlen(sourceString), BUFFER_SIZE - 1);
11122
    //printf("3: %s\n", outSql);
H
Hui Li 已提交
11123 11124
}

11125
static void *superTableQuery(void *sarg) {
11126 11127 11128
    char *sqlstr = calloc(1, BUFFER_SIZE);
    assert(sqlstr);

11129
    threadInfo *pThreadInfo = (threadInfo *)sarg;
11130

11131
    setThreadName("superTableQuery");
11132

11133 11134 11135 11136 11137 11138 11139 11140 11141 11142
    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));
11143
            free(sqlstr);
11144 11145 11146 11147
            return NULL;
        } else {
            pThreadInfo->taos = taos;
        }
11148
    }
11149

11150 11151
    uint64_t st = 0;
    uint64_t et = (int64_t)g_queryInfo.superQueryInfo.queryInterval;
11152

11153 11154 11155
    uint64_t queryTimes = g_queryInfo.superQueryInfo.queryTimes;
    uint64_t totalQueried = 0;
    uint64_t  startTs = taosGetTimestampMs();
H
Hui Li 已提交
11156

11157 11158 11159 11160 11161 11162 11163 11164 11165 11166 11167
    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++) {
11168
                memset(sqlstr, 0, BUFFER_SIZE);
11169 11170 11171 11172 11173 11174 11175
                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);
11176

11177 11178
                totalQueried++;
                g_queryInfo.superQueryInfo.totalQueried ++;
11179

11180 11181 11182 11183 11184 11185 11186 11187 11188 11189
                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;
                }
            }
11190
        }
11191 11192 11193 11194 11195 11196
        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 已提交
11197
    }
11198

11199
    free(sqlstr);
11200
    return NULL;
11201 11202
}

11203
static int queryTestProcess() {
11204

11205 11206 11207
    setupForAnsiEscape();
    printfQueryMeta();
    resetAfterAnsiEscape();
11208

11209 11210 11211 11212 11213 11214 11215
    TAOS * taos = NULL;
    taos = taos_connect(g_queryInfo.host,
            g_queryInfo.user,
            g_queryInfo.password,
            NULL,
            g_queryInfo.port);
    if (taos == NULL) {
11216
        errorPrint("Failed to connect to TDengine, reason:%s\n",
11217
                taos_errstr(NULL));
11218
        exit(EXIT_FAILURE);
11219
    }
11220

11221 11222 11223
    if (0 != g_queryInfo.superQueryInfo.sqlCount) {
        getAllChildNameOfSuperTable(taos,
                g_queryInfo.dbName,
11224
                g_queryInfo.superQueryInfo.stbName,
11225 11226 11227
                &g_queryInfo.superQueryInfo.childTblName,
                &g_queryInfo.superQueryInfo.childTblCount);
    }
11228

11229
    prompt();
11230

11231 11232 11233
    if (g_args.debug_print || g_args.verbose_print) {
        printfQuerySystemInfo(taos);
    }
11234

11235 11236 11237
    if (0 == strncasecmp(g_queryInfo.queryMode, "rest", strlen("rest"))) {
        if (convertHostToServAddr(
                    g_queryInfo.host, g_queryInfo.port, &g_queryInfo.serv_addr) != 0)
11238
            ERROR_EXIT("convert host to server address");
11239
    }
11240

11241 11242 11243 11244 11245
    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;
11246

11247
    uint64_t startTs = taosGetTimestampMs();
11248

11249
    if ((nSqlCount > 0) && (nConcurrent > 0)) {
11250

11251 11252
        pids  = calloc(1, nConcurrent * nSqlCount * sizeof(pthread_t));
        infos = calloc(1, nConcurrent * nSqlCount * sizeof(threadInfo));
11253

11254 11255 11256 11257
        if ((NULL == pids) || (NULL == infos)) {
            taos_close(taos);
            ERROR_EXIT("memory allocation failed for create threads\n");
        }
11258

11259 11260 11261 11262 11263 11264 11265 11266 11267
        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)) {

11268 11269
                    char sqlStr[TSDB_DB_NAME_LEN + 5];
                    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11270 11271 11272 11273
                    if (0 != queryDbExec(taos, sqlStr, NO_INSERT_TYPE, false)) {
                        taos_close(taos);
                        free(infos);
                        free(pids);
11274
                        errorPrint2("use database %s failed!\n\n",
11275 11276 11277
                                g_queryInfo.dbName);
                        return -1;
                    }
11278
                }
11279

11280 11281 11282 11283 11284 11285 11286 11287 11288 11289 11290 11291 11292 11293 11294 11295 11296 11297 11298 11299 11300 11301 11302 11303 11304
                if (0 == strncasecmp(g_queryInfo.queryMode, "rest", 4)) {
#ifdef WINDOWS
                    WSADATA wsaData;
                    WSAStartup(MAKEWORD(2, 2), &wsaData);
                    SOCKET sockfd;
#else
                    int sockfd;
#endif
                    sockfd = socket(AF_INET, SOCK_STREAM, 0);
                    if (sockfd < 0) {
#ifdef WINDOWS
                        errorPrint( "Could not create socket : %d" , WSAGetLastError());
#endif
                        debugPrint("%s() LN%d, sockfd=%d\n", __func__, __LINE__, sockfd);
                        ERROR_EXIT("opening socket");
                    }

                    int retConn = connect(sockfd, (struct sockaddr *)&(g_queryInfo.serv_addr),
                         sizeof(struct sockaddr));
                    debugPrint("%s() LN%d connect() return %d\n", __func__, __LINE__, retConn);
                    if (retConn < 0) {
                        ERROR_EXIT("connecting");
                    }
                    pThreadInfo->sockfd = sockfd;
                }
11305
                pThreadInfo->taos = NULL;// workaround to use separate taos connection;
11306

11307 11308 11309
                pthread_create(pids + seq, NULL, specifiedTableQuery,
                        pThreadInfo);
            }
11310
        }
11311 11312
    } else {
        g_queryInfo.specifiedQueryInfo.concurrent = 0;
11313 11314
    }

11315
    taos_close(taos);
11316

11317 11318 11319 11320 11321 11322 11323
    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));
11324

11325 11326 11327
        if ((NULL == pidsOfSub) || (NULL == infosOfSub)) {
            free(infos);
            free(pids);
11328

11329 11330
            ERROR_EXIT("memory allocation failed for create threads\n");
        }
11331

11332 11333
        int64_t ntables = g_queryInfo.superQueryInfo.childTblCount;
        int threads = g_queryInfo.superQueryInfo.threadCnt;
11334

11335 11336 11337 11338 11339
        int64_t a = ntables / threads;
        if (a < 1) {
            threads = ntables;
            a = 1;
        }
11340

11341 11342 11343 11344
        int64_t b = 0;
        if (threads != 0) {
            b = ntables % threads;
        }
11345

11346 11347 11348 11349
        uint64_t tableFrom = 0;
        for (int i = 0; i < threads; i++) {
            threadInfo *pThreadInfo = infosOfSub + i;
            pThreadInfo->threadID = i;
11350

11351 11352 11353 11354
            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;
11355
            pThreadInfo->taos = NULL; // workaround to use separate taos connection;
11356 11357 11358 11359 11360 11361 11362 11363 11364 11365 11366 11367 11368 11369 11370 11371 11372 11373 11374 11375 11376 11377 11378 11379 11380
            if (0 == strncasecmp(g_queryInfo.queryMode, "rest", 4)) {
#ifdef WINDOWS
                WSADATA wsaData;
                WSAStartup(MAKEWORD(2, 2), &wsaData);
                SOCKET sockfd;
#else
                int sockfd;
#endif
                sockfd = socket(AF_INET, SOCK_STREAM, 0);
                if (sockfd < 0) {
#ifdef WINDOWS
                    errorPrint( "Could not create socket : %d" , WSAGetLastError());
#endif
                    debugPrint("%s() LN%d, sockfd=%d\n", __func__, __LINE__, sockfd);
                    ERROR_EXIT("opening socket");
                }

                int retConn = connect(sockfd, (struct sockaddr *)&(g_queryInfo.serv_addr),
                        sizeof(struct sockaddr));
                debugPrint("%s() LN%d connect() return %d\n", __func__, __LINE__, retConn);
                if (retConn < 0) {
                    ERROR_EXIT("connecting");
                }
                pThreadInfo->sockfd = sockfd;
            }
11381 11382
            pthread_create(pidsOfSub + i, NULL, superTableQuery, pThreadInfo);
        }
11383

11384 11385 11386 11387
        g_queryInfo.superQueryInfo.threadCnt = threads;
    } else {
        g_queryInfo.superQueryInfo.threadCnt = 0;
    }
11388

11389 11390 11391 11392
    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);
11393 11394 11395 11396 11397 11398 11399 11400 11401
                if (0 == strncasecmp(g_queryInfo.queryMode, "rest", 4)) {
                    threadInfo *pThreadInfo = infos + i * nSqlCount + j;
#ifdef WINDOWS
                    closesocket(pThreadInfo->sockfd);
                    WSACleanup();
#else
                    close(pThreadInfo->sockfd);
#endif
                }
11402 11403
            }
        }
11404
    }
S
Shuaiqiang Chang 已提交
11405

11406 11407
    tmfree((char*)pids);
    tmfree((char*)infos);
11408

11409 11410
    for (int i = 0; i < g_queryInfo.superQueryInfo.threadCnt; i++) {
        pthread_join(pidsOfSub[i], NULL);
11411 11412 11413 11414 11415 11416 11417 11418 11419
        if (0 == strncasecmp(g_queryInfo.queryMode, "rest", 4)) {
            threadInfo *pThreadInfo = infosOfSub + i;
#ifdef WINDOWS
            closesocket(pThreadInfo->sockfd);
            WSACleanup();
#else
            close(pThreadInfo->sockfd);
#endif
        }
11420
    }
H
Hui Li 已提交
11421

11422 11423
    tmfree((char*)pidsOfSub);
    tmfree((char*)infosOfSub);
11424

11425
    //  taos_close(taos);// workaround to use separate taos connection;
11426
    uint64_t endTs = taosGetTimestampMs();
11427

11428 11429
    uint64_t totalQueried = g_queryInfo.specifiedQueryInfo.totalQueried +
        g_queryInfo.superQueryInfo.totalQueried;
11430

11431 11432 11433 11434
    fprintf(stderr, "==== completed total queries: %"PRIu64", the QPS of all threads: %10.3f====\n",
            totalQueried,
            (double)(totalQueried/((endTs-startTs)/1000.0)));
    return 0;
11435 11436
}

11437 11438
static void stable_sub_callback(
        TAOS_SUB* tsub, TAOS_RES *res, void* param, int code) {
11439
    if (res == NULL || taos_errno(res) != 0) {
11440
        errorPrint2("%s() LN%d, failed to subscribe result, code:%d, reason:%s\n",
11441 11442 11443
                __func__, __LINE__, code, taos_errstr(res));
        return;
    }
11444

11445 11446
    if (param)
        fetchResult(res, (threadInfo *)param);
11447
    // tao_unsubscribe() will free result.
11448 11449 11450 11451
}

static void specified_sub_callback(
        TAOS_SUB* tsub, TAOS_RES *res, void* param, int code) {
11452
    if (res == NULL || taos_errno(res) != 0) {
11453
        errorPrint2("%s() LN%d, failed to subscribe result, code:%d, reason:%s\n",
11454 11455 11456
                __func__, __LINE__, code, taos_errstr(res));
        return;
    }
11457

11458 11459
    if (param)
        fetchResult(res, (threadInfo *)param);
11460
    // tao_unsubscribe() will free result.
H
hzcheng 已提交
11461 11462
}

11463
static TAOS_SUB* subscribeImpl(
11464 11465 11466 11467
        QUERY_CLASS class,
        threadInfo *pThreadInfo,
        char *sql, char* topic, bool restart, uint64_t interval)
{
11468 11469 11470 11471 11472 11473 11474 11475 11476 11477 11478 11479 11480 11481 11482 11483 11484 11485 11486 11487 11488 11489 11490 11491
    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) {
11492
        errorPrint2("failed to create subscription. topic:%s, sql:%s\n", topic, sql);
11493 11494 11495 11496
        return NULL;
    }

    return tsub;
11497
}
H
hzcheng 已提交
11498

11499
static void *superSubscribe(void *sarg) {
11500
    threadInfo *pThreadInfo = (threadInfo *)sarg;
11501 11502 11503
    char *subSqlStr = calloc(1, BUFFER_SIZE);
    assert(subSqlStr);

11504 11505 11506
    TAOS_SUB*    tsub[MAX_QUERY_SQL_COUNT] = {0};
    uint64_t tsubSeq;

11507 11508
    setThreadName("superSub");

11509
    if (pThreadInfo->ntables > MAX_QUERY_SQL_COUNT) {
11510
        free(subSqlStr);
11511 11512
        errorPrint("The table number(%"PRId64") of the thread is more than max query sql count: %d\n",
                pThreadInfo->ntables, MAX_QUERY_SQL_COUNT);
11513
        exit(EXIT_FAILURE);
11514
    }
11515

11516
    if (pThreadInfo->taos == NULL) {
11517 11518 11519 11520 11521 11522
        pThreadInfo->taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                g_queryInfo.dbName,
                g_queryInfo.port);
        if (pThreadInfo->taos == NULL) {
11523
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11524
                    pThreadInfo->threadID, taos_errstr(NULL));
11525
            free(subSqlStr);
11526 11527
            return NULL;
        }
11528 11529
    }

11530 11531
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11532 11533
    if (0 != queryDbExec(pThreadInfo->taos, sqlStr, NO_INSERT_TYPE, false)) {
        taos_close(pThreadInfo->taos);
11534
        errorPrint2("use database %s failed!\n\n",
11535
                g_queryInfo.dbName);
11536
        free(subSqlStr);
11537 11538
        return NULL;
    }
11539

11540 11541 11542 11543 11544 11545 11546 11547 11548 11549 11550
    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);
11551
        memset(subSqlStr, 0, BUFFER_SIZE);
11552 11553
        replaceChildTblName(
                g_queryInfo.superQueryInfo.sql[pThreadInfo->querySeq],
11554
                subSqlStr, i);
11555 11556 11557 11558 11559
        if (g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq][0] != 0) {
            sprintf(pThreadInfo->filePath, "%s-%d",
                    g_queryInfo.superQueryInfo.result[pThreadInfo->querySeq],
                    pThreadInfo->threadID);
        }
11560

11561 11562
        verbosePrint("%s() LN%d, [%d] subSqlStr: %s\n",
                __func__, __LINE__, pThreadInfo->threadID, subSqlStr);
11563 11564
        tsub[tsubSeq] = subscribeImpl(
                STABLE_CLASS,
11565
                pThreadInfo, subSqlStr, topic,
11566 11567 11568 11569
                g_queryInfo.superQueryInfo.subscribeRestart,
                g_queryInfo.superQueryInfo.subscribeInterval);
        if (NULL == tsub[tsubSeq]) {
            taos_close(pThreadInfo->taos);
11570
            free(subSqlStr);
11571 11572 11573
            return NULL;
        }
    }
11574

11575 11576 11577 11578 11579 11580
    // 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;
11581

11582
    uint64_t st = 0, et = 0;
11583

11584 11585 11586 11587
    while ((g_queryInfo.superQueryInfo.endAfterConsume == -1)
            || (g_queryInfo.superQueryInfo.endAfterConsume >
                consumed[pThreadInfo->end_table_to
                - pThreadInfo->start_table_from])) {
11588

11589 11590 11591 11592 11593 11594 11595 11596 11597 11598 11599 11600 11601 11602 11603 11604 11605 11606 11607 11608 11609 11610 11611 11612 11613 11614 11615 11616 11617 11618 11619 11620 11621 11622 11623 11624 11625 11626
        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,
11627
                            pThreadInfo, subSqlStr, topic,
11628 11629 11630 11631 11632
                            g_queryInfo.superQueryInfo.subscribeRestart,
                            g_queryInfo.superQueryInfo.subscribeInterval
                            );
                    if (NULL == tsub[tsubSeq]) {
                        taos_close(pThreadInfo->taos);
11633
                        free(subSqlStr);
11634 11635 11636 11637 11638 11639 11640 11641 11642 11643 11644
                        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);
11645

11646 11647
    for (uint64_t i = pThreadInfo->start_table_from;
            i <= pThreadInfo->end_table_to; i++) {
11648 11649
        tsubSeq = i - pThreadInfo->start_table_from;
        taos_unsubscribe(tsub[tsubSeq], 0);
H
hzcheng 已提交
11650
    }
11651

11652
    taos_close(pThreadInfo->taos);
11653
    free(subSqlStr);
11654
    return NULL;
H
hzcheng 已提交
11655 11656
}

11657
static void *specifiedSubscribe(void *sarg) {
11658 11659
    threadInfo *pThreadInfo = (threadInfo *)sarg;
    //  TAOS_SUB*  tsub = NULL;
11660

11661
    setThreadName("specSub");
11662

11663
    if (pThreadInfo->taos == NULL) {
11664 11665 11666 11667 11668 11669
        pThreadInfo->taos = taos_connect(g_queryInfo.host,
                g_queryInfo.user,
                g_queryInfo.password,
                g_queryInfo.dbName,
                g_queryInfo.port);
        if (pThreadInfo->taos == NULL) {
11670
            errorPrint2("[%d] Failed to connect to TDengine, reason:%s\n",
11671 11672 11673
                    pThreadInfo->threadID, taos_errstr(NULL));
            return NULL;
        }
11674 11675
    }

11676 11677
    char sqlStr[TSDB_DB_NAME_LEN + 5];
    sprintf(sqlStr, "USE %s", g_queryInfo.dbName);
11678 11679 11680 11681 11682 11683 11684 11685 11686 11687 11688
    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",
11689 11690
                g_queryInfo.specifiedQueryInfo.result[pThreadInfo->querySeq],
                pThreadInfo->threadID);
11691 11692 11693 11694 11695 11696 11697 11698 11699 11700 11701 11702 11703
    }
    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
11704

11705 11706 11707 11708 11709 11710 11711 11712 11713 11714 11715 11716 11717 11718 11719 11720 11721 11722 11723 11724 11725 11726 11727 11728 11729 11730 11731 11732 11733 11734 11735 11736 11737 11738 11739 11740 11741 11742 11743 11744 11745 11746 11747 11748 11749 11750 11751 11752 11753 11754 11755 11756 11757 11758 11759 11760
    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 已提交
11761 11762
}

11763
static int subscribeTestProcess() {
11764 11765 11766 11767 11768 11769 11770 11771 11772 11773
    setupForAnsiEscape();
    printfQueryMeta();
    resetAfterAnsiEscape();

    prompt();

    TAOS * taos = NULL;
    taos = taos_connect(g_queryInfo.host,
            g_queryInfo.user,
            g_queryInfo.password,
11774
            g_queryInfo.dbName,
11775 11776
            g_queryInfo.port);
    if (taos == NULL) {
11777
        errorPrint2("Failed to connect to TDengine, reason:%s\n",
11778
                taos_errstr(NULL));
11779
        exit(EXIT_FAILURE);
S
Shengliang Guan 已提交
11780
    }
11781

11782 11783 11784
    if (0 != g_queryInfo.superQueryInfo.sqlCount) {
        getAllChildNameOfSuperTable(taos,
                g_queryInfo.dbName,
11785
                g_queryInfo.superQueryInfo.stbName,
11786 11787
                &g_queryInfo.superQueryInfo.childTblName,
                &g_queryInfo.superQueryInfo.childTblCount);
S
Shengliang Guan 已提交
11788
    }
11789

11790
    taos_close(taos); // workaround to use separate taos connection;
11791 11792 11793 11794 11795 11796 11797 11798 11799

    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) {
11800
        debugPrint("%s() LN%d, specified query sqlCount %d.\n",
11801 11802 11803 11804
                __func__, __LINE__,
                g_queryInfo.specifiedQueryInfo.sqlCount);
    } else {
        if (g_queryInfo.specifiedQueryInfo.concurrent <= 0) {
11805
            errorPrint2("%s() LN%d, specified query sqlCount %d.\n",
11806 11807
                    __func__, __LINE__,
                    g_queryInfo.specifiedQueryInfo.sqlCount);
11808
            exit(EXIT_FAILURE);
11809
        }
11810

11811
        pids  = calloc(
11812
                1,
11813 11814 11815 11816
                g_queryInfo.specifiedQueryInfo.sqlCount *
                g_queryInfo.specifiedQueryInfo.concurrent *
                sizeof(pthread_t));
        infos = calloc(
11817
                1,
11818 11819 11820 11821
                g_queryInfo.specifiedQueryInfo.sqlCount *
                g_queryInfo.specifiedQueryInfo.concurrent *
                sizeof(threadInfo));
        if ((NULL == pids) || (NULL == infos)) {
11822
            errorPrint2("%s() LN%d, malloc failed for create threads\n", __func__, __LINE__);
11823
            exit(EXIT_FAILURE);
11824
        }
11825

11826 11827 11828 11829 11830 11831
        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;
11832
                pThreadInfo->taos = NULL;  // workaround to use separate taos connection;
11833 11834
                pthread_create(pids + seq, NULL, specifiedSubscribe, pThreadInfo);
            }
11835
        }
11836
    }
11837

11838 11839 11840 11841 11842 11843 11844 11845 11846 11847 11848 11849 11850 11851 11852 11853 11854 11855 11856
    //==== 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)) {
11857
                errorPrint2("%s() LN%d, malloc failed for create threads\n",
11858 11859
                        __func__, __LINE__);
                // taos_close(taos);
11860
                exit(EXIT_FAILURE);
11861
            }
11862

11863 11864
            int64_t ntables = g_queryInfo.superQueryInfo.childTblCount;
            int threads = g_queryInfo.superQueryInfo.threadCnt;
11865

11866 11867 11868 11869
            int64_t a = ntables / threads;
            if (a < 1) {
                threads = ntables;
                a = 1;
11870
            }
11871

11872 11873 11874 11875 11876 11877 11878 11879 11880 11881 11882 11883 11884 11885 11886 11887 11888
            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;
11889
                    pThreadInfo->taos = NULL; // workaround to use separate taos connection;
11890 11891 11892 11893
                    pthread_create(pidsOfStable + seq,
                            NULL, superSubscribe, pThreadInfo);
                }
            }
11894

11895 11896 11897 11898 11899 11900 11901
            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);
                }
11902 11903
            }
        }
11904
    }
11905

11906 11907 11908 11909 11910
    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);
        }
11911
    }
H
hzcheng 已提交
11912

11913 11914
    tmfree((char*)pids);
    tmfree((char*)infos);
H
hzcheng 已提交
11915

11916 11917 11918 11919
    tmfree((char*)pidsOfStable);
    tmfree((char*)infosOfStable);
    //   taos_close(taos);
    return 0;
H
hzcheng 已提交
11920 11921
}

11922
static void initOfInsertMeta() {
11923
    memset(&g_Dbs, 0, sizeof(SDbs));
11924

11925 11926 11927 11928
    // 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);
11929
    tstrncpy(g_Dbs.password, TSDB_DEFAULT_PASS, SHELL_MAX_PASSWORD_LEN);
11930
    g_Dbs.threadCount = 2;
11931

11932
    g_Dbs.use_metric = g_args.use_metric;
L
Liu Tao 已提交
11933 11934
}

11935
static void initOfQueryMeta() {
11936
    memset(&g_queryInfo, 0, sizeof(SQueryMetaInfo));
11937

11938 11939 11940 11941
    // 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);
11942
    tstrncpy(g_queryInfo.password, TSDB_DEFAULT_PASS, SHELL_MAX_PASSWORD_LEN);
L
Liu Tao 已提交
11943 11944
}

11945
static void setParaFromArg() {
11946 11947
    char type[20];
    char length[20];
11948 11949 11950 11951 11952
    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);
    }
11953

11954 11955 11956
    if (g_args.user) {
        tstrncpy(g_Dbs.user, g_args.user, MAX_USERNAME_SIZE);
    }
11957

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

11960 11961 11962
    if (g_args.port) {
        g_Dbs.port = g_args.port;
    }
11963

11964 11965
    g_Dbs.threadCount = g_args.nthreads;
    g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11966

11967 11968
    g_Dbs.dbCount = 1;
    g_Dbs.db[0].drop = true;
L
Liu Tao 已提交
11969

11970 11971
    tstrncpy(g_Dbs.db[0].dbName, g_args.database, TSDB_DB_NAME_LEN);
    g_Dbs.db[0].dbCfg.replica = g_args.replica;
11972
    tstrncpy(g_Dbs.db[0].dbCfg.precision, "ms", SMALL_BUFF_LEN);
11973

11974
    tstrncpy(g_Dbs.resultFile, g_args.output_file, MAX_FILE_NAME_LEN);
11975

11976
    g_Dbs.use_metric = g_args.use_metric;
L
Liu Tao 已提交
11977

11978
    g_Dbs.aggr_func = g_args.aggr_func;
11979

11980
    char dataString[TSDB_MAX_BYTES_PER_ROW];
11981 11982
    char *data_type = g_args.data_type;
    char **dataType = g_args.dataType;
L
Liu Tao 已提交
11983

11984
    memset(dataString, 0, TSDB_MAX_BYTES_PER_ROW);
H
hzcheng 已提交
11985

11986 11987 11988
    if ((data_type[0] == TSDB_DATA_TYPE_BINARY)
            || (data_type[0] == TSDB_DATA_TYPE_BOOL)
            || (data_type[0] == TSDB_DATA_TYPE_NCHAR)) {
11989
        g_Dbs.aggr_func = false;
11990
    }
11991

11992 11993
    if (g_args.use_metric) {
        g_Dbs.db[0].superTblCount = 1;
11994 11995
        tstrncpy(g_Dbs.db[0].superTbls[0].stbName, "meters", TSDB_TABLE_NAME_LEN);
        g_Dbs.db[0].superTbls[0].childTblCount = g_args.ntables;
11996
        g_Dbs.db[0].superTbls[0].escapeChar = g_args.escapeChar;
11997 11998
        g_Dbs.threadCount = g_args.nthreads;
        g_Dbs.threadCountForCreateTbl = g_args.nthreads;
11999
        g_Dbs.asyncMode = g_args.async_mode;
12000

12001 12002 12003 12004 12005
        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,
12006 12007
                g_args.tb_prefix, TBNAME_PREFIX_LEN);
        tstrncpy(g_Dbs.db[0].superTbls[0].dataSource, "rand", SMALL_BUFF_LEN);
12008

12009 12010 12011 12012 12013 12014 12015
        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);
12016
        g_Dbs.db[0].superTbls[0].timeStampStep = g_args.timestamp_step;
12017

12018
        g_Dbs.db[0].superTbls[0].insertRows = g_args.insertRows;
12019
        g_Dbs.db[0].superTbls[0].maxSqlLen = g_args.max_sql_len;
12020

12021 12022
        g_Dbs.db[0].superTbls[0].columnCount = 0;
        for (int i = 0; i < MAX_NUM_COLUMNS; i++) {
12023
            if (data_type[i] == TSDB_DATA_TYPE_NULL) {
12024 12025
                break;
            }
12026

12027
            g_Dbs.db[0].superTbls[0].columns[i].data_type = data_type[i];
12028
            tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
12029
                    dataType[i], min(DATATYPE_BUFF_LEN, strlen(dataType[i]) + 1));
12030
            if (1 == regexMatch(dataType[i], "^(NCHAR|BINARY)(\\([1-9][0-9]*\\))$", REG_ICASE |
12031 12032 12033 12034 12035 12036 12037 12038
                    REG_EXTENDED)) {
                sscanf(dataType[i], "%[^(](%[^)]", type, length);
                g_Dbs.db[0].superTbls[0].columns[i].dataLen = atoi(length);
                tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
                    type, min(DATATYPE_BUFF_LEN, strlen(type) + 1));
            } else {
                g_Dbs.db[0].superTbls[0].columns[i].dataLen = g_args.binwidth;
                tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
12039
                    dataType[i], min(DATATYPE_BUFF_LEN, strlen(dataType[i]) + 1));
12040
            }
12041 12042 12043
            g_Dbs.db[0].superTbls[0].columnCount++;
        }

12044 12045
        if (g_Dbs.db[0].superTbls[0].columnCount > g_args.columnCount) {
            g_Dbs.db[0].superTbls[0].columnCount = g_args.columnCount;
12046 12047
        } else {
            for (int i = g_Dbs.db[0].superTbls[0].columnCount;
12048 12049
                    i < g_args.columnCount; i++) {
                g_Dbs.db[0].superTbls[0].columns[i].data_type = TSDB_DATA_TYPE_INT;
12050
                tstrncpy(g_Dbs.db[0].superTbls[0].columns[i].dataType,
12051
                        "INT", min(DATATYPE_BUFF_LEN, strlen("INT") + 1));
12052 12053 12054 12055 12056 12057
                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,
12058
                "INT", min(DATATYPE_BUFF_LEN, strlen("INT") + 1));
12059 12060 12061
        g_Dbs.db[0].superTbls[0].tags[0].dataLen = 0;

        tstrncpy(g_Dbs.db[0].superTbls[0].tags[1].dataType,
12062
                "BINARY", min(DATATYPE_BUFF_LEN, strlen("BINARY") + 1));
12063
        g_Dbs.db[0].superTbls[0].tags[1].dataLen = g_args.binwidth;
12064 12065
        g_Dbs.db[0].superTbls[0].tagCount = 2;
    } else {
12066
        g_Dbs.threadCountForCreateTbl = g_args.nthreads;
12067 12068
        g_Dbs.db[0].superTbls[0].tagCount = 0;
    }
12069 12070 12071 12072
}

/* Function to do regular expression check */
static int regexMatch(const char *s, const char *reg, int cflags) {
12073 12074 12075 12076 12077
    regex_t regex;
    char    msgbuf[100] = {0};

    /* Compile regular expression */
    if (regcomp(&regex, reg, cflags) != 0) {
12078
        ERROR_EXIT("Fail to compile regex\n");
12079 12080 12081 12082 12083 12084 12085 12086 12087 12088 12089 12090 12091
    }

    /* 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);
12092 12093
        printf("Regex match failed: %s\n", msgbuf);
        exit(EXIT_FAILURE);
12094
    }
12095 12096 12097 12098
    return 0;
}

static int isCommentLine(char *line) {
12099
    if (line == NULL) return 1;
12100

12101
    return regexMatch(line, "^\\s*#.*", REG_EXTENDED);
12102 12103
}

12104
static void querySqlFile(TAOS* taos, char* sqlFile)
12105
{
12106 12107 12108 12109 12110
    FILE *fp = fopen(sqlFile, "r");
    if (fp == NULL) {
        printf("failed to open file %s, reason:%s\n", sqlFile, strerror(errno));
        return;
    }
12111

12112 12113 12114 12115 12116
    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;
12117

12118
    double t = taosGetTimestampMs();
12119

12120 12121 12122
    while((read_len = tgetline(&line, &line_len, fp)) != -1) {
        if (read_len >= TSDB_MAX_BYTES_PER_ROW) continue;
        line[--read_len] = '\0';
12123

12124 12125 12126
        if (read_len == 0 || isCommentLine(line)) {  // line starts with #
            continue;
        }
12127

12128 12129 12130 12131 12132 12133
        if (line[read_len - 1] == '\\') {
            line[read_len - 1] = ' ';
            memcpy(cmd + cmd_len, line, read_len);
            cmd_len += read_len;
            continue;
        }
12134

12135 12136
        memcpy(cmd + cmd_len, line, read_len);
        if (0 != queryDbExec(taos, cmd, NO_INSERT_TYPE, false)) {
12137
            errorPrint2("%s() LN%d, queryDbExec %s failed!\n",
12138 12139 12140 12141 12142 12143 12144 12145
                    __func__, __LINE__, cmd);
            tmfree(cmd);
            tmfree(line);
            tmfclose(fp);
            return;
        }
        memset(cmd, 0, TSDB_MAX_BYTES_PER_ROW);
        cmd_len = 0;
12146
    }
H
hzcheng 已提交
12147

12148 12149
    t = taosGetTimestampMs() - t;
    printf("run %s took %.6f second(s)\n\n", sqlFile, t);
12150

12151 12152 12153 12154
    tmfree(cmd);
    tmfree(line);
    tmfclose(fp);
    return;
H
hzcheng 已提交
12155 12156
}

12157
static void testMetaFile() {
12158
    if (INSERT_TEST == g_args.test_mode) {
12159 12160
        if (g_Dbs.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_Dbs.cfgDir);
12161

12162
        insertTestProcess();
12163

12164
    } else if (QUERY_TEST == g_args.test_mode) {
12165 12166
        if (g_queryInfo.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_queryInfo.cfgDir);
12167

12168
        queryTestProcess();
12169

12170
    } else if (SUBSCRIBE_TEST == g_args.test_mode) {
12171 12172
        if (g_queryInfo.cfgDir[0])
            taos_options(TSDB_OPTION_CONFIGDIR, g_queryInfo.cfgDir);
12173

12174
        subscribeTestProcess();
12175

12176
    }  else {
12177
        ;
12178
    }
12179
}
12180

12181
static void queryAggrFunc() {
12182 12183 12184 12185 12186
    // query data

    pthread_t read_id;
    threadInfo *pThreadInfo = calloc(1, sizeof(threadInfo));
    assert(pThreadInfo);
12187
    pThreadInfo->start_time = DEFAULT_START_TIME;  // 2017-07-14 10:40:00.000
12188 12189 12190 12191 12192
    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;
12193
        pThreadInfo->stbInfo = &g_Dbs.db[0].superTbls[0];
12194
        tstrncpy(pThreadInfo->tb_prefix,
12195
                g_Dbs.db[0].superTbls[0].childTblPrefix, TBNAME_PREFIX_LEN);
12196
    } else {
12197 12198
        pThreadInfo->ntables = g_args.ntables;
        pThreadInfo->end_table_to = g_args.ntables -1;
12199 12200 12201 12202 12203 12204 12205 12206 12207 12208
        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) {
12209
        free(pThreadInfo);
12210
        errorPrint2("Failed to connect to TDengine, reason:%s\n",
12211
                taos_errstr(NULL));
12212
        exit(EXIT_FAILURE);
12213 12214 12215 12216 12217
    }

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

    if (!g_Dbs.use_metric) {
12218
        pthread_create(&read_id, NULL, queryNtableAggrFunc, pThreadInfo);
12219
    } else {
12220
        pthread_create(&read_id, NULL, queryStableAggrFunc, pThreadInfo);
12221 12222 12223
    }
    pthread_join(read_id, NULL);
    taos_close(pThreadInfo->taos);
12224
    free(pThreadInfo);
12225 12226
}

12227 12228
static void testCmdLine() {

12229 12230 12231
    if (strlen(configDir)) {
        wordexp_t full_path;
        if (wordexp(configDir, &full_path, 0) != 0) {
12232
            errorPrint("Invalid path %s\n", configDir);
12233 12234 12235 12236
            return;
        }
        taos_options(TSDB_OPTION_CONFIGDIR, full_path.we_wordv[0]);
        wordfree(&full_path);
12237 12238
    }

12239 12240
    g_args.test_mode = INSERT_TEST;
    insertTestProcess();
12241

12242 12243 12244
    if (g_Dbs.aggr_func) {
        queryAggrFunc();
    }
12245 12246
}

12247
int main(int argc, char *argv[]) {
12248
    parse_args(argc, argv, &g_args);
12249

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

12252
    if (g_args.metaFile) {
12253
        g_totalChildTables = 0;
12254 12255
        initOfInsertMeta();
        initOfQueryMeta();
12256

12257 12258 12259 12260
        if (false == getInfoFromJsonFile(g_args.metaFile)) {
            printf("Failed to read %s\n", g_args.metaFile);
            return 1;
        }
12261

12262
        testMetaFile();
12263
    } else {
12264 12265 12266 12267 12268 12269 12270 12271 12272 12273 12274 12275
        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);
12276

12277 12278 12279
        } else {
            testCmdLine();
        }
12280

12281 12282 12283 12284 12285
        if (g_dupstr)
            free(g_dupstr);
    }

    return 0;
H
hzcheng 已提交
12286
}