simExe.c 28.9 KB
Newer Older
S
[TD-73]  
slguan 已提交
1 2
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
S
slguan 已提交
3
 *
S
[TD-73]  
slguan 已提交
4 5 6
 * 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.
S
slguan 已提交
7
 *
S
[TD-73]  
slguan 已提交
8 9 10 11 12 13 14 15
 * 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
#define _DEFAULT_SOURCE
S
slguan 已提交
17 18 19 20
#include "os.h"
#include "sim.h"
#include "taos.h"
#include "taoserror.h"
S
slguan 已提交
21
#include "tglobal.h"
S
slguan 已提交
22 23
#include "tutil.h"
#include "cJSON.h"
S
Shengliang Guan 已提交
24
#undef TAOS_MEM_CHECK
S
slguan 已提交
25

S
Shengliang Guan 已提交
26
void simLogSql(char *sql, bool useSharp) {
S
slguan 已提交
27 28
  static FILE *fp = NULL;
  char filename[256];
S
slguan 已提交
29
  sprintf(filename, "%s/sim.sql", tsScriptDir);
S
slguan 已提交
30 31 32 33 34 35 36
  if (fp == NULL) {
    fp = fopen(filename, "w");
    if (fp == NULL) {
      fprintf(stderr, "ERROR: failed to open file: %s\n", filename);
      return;
    }
  }
S
Shengliang Guan 已提交
37 38 39 40 41
  if (useSharp) {
    fprintf(fp, "# %s;\n", sql);
  } else {
    fprintf(fp, "%s;\n", sql);
  }
42

S
slguan 已提交
43 44 45
  fflush(fp);
}

H
Hui Li 已提交
46
char *simParseArbitratorName(char *varName);
S
slguan 已提交
47
char *simParseHostName(char *varName);
48
char *simGetVariable(SScript *script, char *varName, int32_t varLen) {
S
slguan 已提交
49 50 51 52
  if (strncmp(varName, "hostname", 8) == 0) {
    return simParseHostName(varName);
  }

H
Hui Li 已提交
53
  if (strncmp(varName, "arbitrator", 10) == 0) {
54
    return simParseArbitratorName(varName);
H
Hui Li 已提交
55 56
  }

S
slguan 已提交
57 58 59 60
  if (strncmp(varName, "error", varLen) == 0) return script->error;

  if (strncmp(varName, "rows", varLen) == 0) return script->rows;

61
  if (strncmp(varName, "system_exit", varLen) == 0) return script->system_exit_code;
S
slguan 已提交
62

63
  if (strncmp(varName, "system_content", varLen) == 0) return script->system_ret_content;
S
slguan 已提交
64 65 66 67 68 69 70 71

  // variable like data2_192.168.0.1
  if (strncmp(varName, "data", 4) == 0) {
    if (varLen < 6) {
      return "null";
    }

    if (varName[5] == '_') {
72
      int32_t col = varName[4] - '0';
S
slguan 已提交
73 74 75 76
      if (col < 0 || col >= MAX_QUERY_COL_NUM) {
        return "null";
      }

77 78
      char *  keyName;
      int32_t keyLen;
S
slguan 已提交
79 80
      paGetToken(varName + 6, &keyName, &keyLen);

81
      for (int32_t i = 0; i < MAX_QUERY_ROW_NUM; ++i) {
S
scripts  
Shengliang Guan 已提交
82
        if (strncmp(keyName, script->data[i][0], keyLen) == 0) {
S
Shengliang Guan 已提交
83
          simDebug("script:%s, keyName:%s, keyValue:%s", script->fileName, script->data[i][0], script->data[i][col]);
S
scripts  
Shengliang Guan 已提交
84 85 86 87 88
          return script->data[i][col];
        }
      }
      return "null";
    } else if (varName[6] == '_') {
89
      int32_t col = (varName[4] - '0') * 10 + (varName[5] - '0');
S
scripts  
Shengliang Guan 已提交
90 91 92 93
      if (col < 0 || col >= MAX_QUERY_COL_NUM) {
        return "null";
      }

94 95
      char *  keyName;
      int32_t keyLen;
S
scripts  
Shengliang Guan 已提交
96 97
      paGetToken(varName + 7, &keyName, &keyLen);

98
      for (int32_t i = 0; i < MAX_QUERY_ROW_NUM; ++i) {
S
slguan 已提交
99 100 101 102 103 104 105
        if (strncmp(keyName, script->data[i][0], keyLen) == 0) {
          simTrace("script:%s, keyName:%s, keyValue:%s", script->fileName, script->data[i][0], script->data[i][col]);
          return script->data[i][col];
        }
      }
      return "null";
    } else {
106 107
      int32_t row = varName[4] - '0';
      int32_t col = varName[5] - '0';
S
slguan 已提交
108 109 110 111 112 113 114
      if (row < 0 || row >= MAX_QUERY_ROW_NUM) {
        return "null";
      }
      if (col < 0 || col >= MAX_QUERY_COL_NUM) {
        return "null";
      }

S
Shengliang Guan 已提交
115
      simDebug("script:%s, data[%d][%d]=%s", script->fileName, row, col, script->data[row][col]);
S
slguan 已提交
116 117 118 119
      return script->data[row][col];
    }
  }

120
  for (int32_t i = 0; i < script->varLen; ++i) {
S
slguan 已提交
121 122 123 124 125 126
    SVariable *var = &script->variables[i];
    if (var->varNameLen != varLen) {
      continue;
    }
    if (strncmp(varName, var->varName, varLen) == 0) {
      // if (strlen(var->varValue) != 0)
S
Shengliang Guan 已提交
127
      //  simDebug("script:%s, var:%s, value:%s", script->fileName,
S
slguan 已提交
128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
      //  var->varName, var->varValue);
      return var->varValue;
    }
  }

  if (script->varLen >= MAX_VAR_LEN) {
    simError("script:%s, too many varialbes:%d", script->fileName, script->varLen);
    exit(0);
  }

  SVariable *var = &script->variables[script->varLen];
  script->varLen++;
  strncpy(var->varName, varName, varLen);
  var->varNameLen = varLen;
  var->varValue[0] = 0;
  return var->varValue;
}

146 147 148 149 150
int32_t simExecuteExpression(SScript *script, char *exp) {
  char *  op1, *op2, *var1, *var2, *var3, *rest;
  int32_t op1Len, op2Len, var1Len, var2Len, var3Len, val0, val1;
  char    t0[512], t1[512], t2[512], t3[1024];
  int32_t result;
S
slguan 已提交
151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235

  rest = paGetToken(exp, &var1, &var1Len);
  rest = paGetToken(rest, &op1, &op1Len);
  rest = paGetToken(rest, &var2, &var2Len);
  rest = paGetToken(rest, &op2, &op2Len);

  if (var1[0] == '$')
    strcpy(t0, simGetVariable(script, var1 + 1, var1Len - 1));
  else {
    memcpy(t0, var1, var1Len);
    t0[var1Len] = 0;
  }

  if (var2[0] == '$')
    strcpy(t1, simGetVariable(script, var2 + 1, var2Len - 1));
  else {
    memcpy(t1, var2, var2Len);
    t1[var2Len] = 0;
  }

  if (op2Len != 0) {
    rest = paGetToken(rest, &var3, &var3Len);

    if (var3[0] == '$')
      strcpy(t2, simGetVariable(script, var3 + 1, var3Len - 1));
    else {
      memcpy(t2, var3, var3Len);
      t2[var3Len] = 0;
    }

    if (op2[0] == '+') {
      sprintf(t3, "%lld", atoll(t1) + atoll(t2));
    } else if (op2[0] == '-') {
      sprintf(t3, "%lld", atoll(t1) - atoll(t2));
    } else if (op2[0] == '*') {
      sprintf(t3, "%lld", atoll(t1) * atoll(t2));
    } else if (op2[0] == '/') {
      sprintf(t3, "%lld", atoll(t1) / atoll(t2));
    } else if (op2[0] == '.') {
      sprintf(t3, "%s%s", t1, t2);
    }
  } else {
    strcpy(t3, t1);
  }

  result = 0;

  if (op1Len == 1) {
    if (op1[0] == '=') {
      strcpy(simGetVariable(script, var1 + 1, var1Len - 1), t3);
    } else if (op1[0] == '<') {
      val0 = atoi(t0);
      val1 = atoi(t3);
      if (val0 >= val1) result = -1;
    } else if (op1[0] == '>') {
      val0 = atoi(t0);
      val1 = atoi(t3);
      if (val0 <= val1) result = -1;
    }
  } else {
    if (op1[0] == '=' && op1[1] == '=') {
      if (strcmp(t0, t3) != 0) result = -1;
    } else if (op1[0] == '!' && op1[1] == '=') {
      if (strcmp(t0, t3) == 0) result = -1;
    } else if (op1[0] == '<' && op1[1] == '=') {
      val0 = atoi(t0);
      val1 = atoi(t3);
      if (val0 > val1) result = -1;
    } else if (op1[0] == '>' && op1[1] == '=') {
      val0 = atoi(t0);
      val1 = atoi(t3);
      if (val0 < val1) result = -1;
    }
  }

  return result;
}

bool simExecuteExpCmd(SScript *script, char *option) {
  simExecuteExpression(script, option);
  script->linePos++;
  return true;
}

bool simExecuteTestCmd(SScript *script, char *option) {
236
  int32_t result;
S
slguan 已提交
237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264
  result = simExecuteExpression(script, option);

  if (result >= 0)
    script->linePos++;
  else
    script->linePos = script->lines[script->linePos].jump;

  return true;
}

bool simExecuteGotoCmd(SScript *script, char *option) {
  script->linePos = script->lines[script->linePos].jump;
  return true;
}

bool simExecuteRunCmd(SScript *script, char *option) {
  char *fileName = option;
  if (fileName == NULL || strlen(fileName) == 0) {
    sprintf(script->error, "lineNum:%d. script file is null", script->lines[script->linePos].lineNum);
    return false;
  }

  SScript *newScript = simParseScript(option);
  if (newScript == NULL) {
    sprintf(script->error, "lineNum:%d. parse file:%s error", script->lines[script->linePos].lineNum, fileName);
    return false;
  }

S
Shengliang Guan 已提交
265
  simInfo("script:%s, start to execute", newScript->fileName);
S
slguan 已提交
266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289

  newScript->type = SIM_SCRIPT_TYPE_MAIN;
  simScriptPos++;
  simScriptList[simScriptPos] = newScript;

  script->linePos++;
  return true;
}

bool simExecuteRunBackCmd(SScript *script, char *option) {
  char *fileName = option;
  if (fileName == NULL || strlen(fileName) == 0) {
    sprintf(script->error, "lineNum:%d. script file is null", script->lines[script->linePos].lineNum);
    return false;
  }

  SScript *newScript = simParseScript(option);
  if (newScript == NULL) {
    sprintf(script->error, "lineNum:%d. parse file:%s error", script->lines[script->linePos].lineNum, fileName);
    return false;
  }

  newScript->type = SIM_SCRIPT_TYPE_BACKGROUND;
  script->bgScripts[script->bgScriptLen++] = newScript;
290
  simInfo("script:%s, start to execute in background,", newScript->fileName);
S
slguan 已提交
291

292
  if (pthread_create(&newScript->bgPid, NULL, simExecuteScript, (void *)newScript) != 0) {
S
slguan 已提交
293 294 295 296 297 298 299 300 301 302 303
    sprintf(script->error, "lineNum:%d. create background thread failed", script->lines[script->linePos].lineNum);
    return false;
  }

  script->linePos++;
  return true;
}

bool simExecuteSystemCmd(SScript *script, char *option) {
  char buf[4096] = {0};

S
slguan 已提交
304
  sprintf(buf, "cd %s; ", tsScriptDir);
S
slguan 已提交
305 306
  simVisuallizeOption(script, option, buf + strlen(buf));

S
Shengliang Guan 已提交
307
  simLogSql(buf, true);
308 309
  int32_t code = system(buf);
  int32_t repeatTimes = 0;
S
slguan 已提交
310
  while (code < 0) {
311 312
    simError("script:%s, failed to execute %s , code %d, errno:%d %s, repeatTimes:%d", script->fileName, buf, code,
             errno, strerror(errno), repeatTimes);
S
slguan 已提交
313
    taosMsleep(1000);
314
#ifdef LINUX
S
slguan 已提交
315
    signal(SIGCHLD, SIG_DFL);
S
TD-1057  
Shengliang Guan 已提交
316
#endif
S
slguan 已提交
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341
    if (repeatTimes++ >= 10) {
      exit(0);
    }
  }

  sprintf(script->system_exit_code, "%d", code);
  script->linePos++;
  return true;
}

void simStoreSystemContentResult(SScript *script, char *filename) {
  memset(script->system_ret_content, 0, MAX_SYSTEM_RESULT_LEN);

  FILE *fd;
  if ((fd = fopen(filename, "r")) != NULL) {
    fread(script->system_ret_content, 1, MAX_SYSTEM_RESULT_LEN - 1, fd);
    fclose(fd);
    char rmCmd[MAX_FILE_NAME_LEN] = {0};
    sprintf(rmCmd, "rm -f %s", filename);
    system(rmCmd);
  }
}

bool simExecuteSystemContentCmd(SScript *script, char *option) {
  char buf[4096] = {0};
sangshuduo's avatar
sangshuduo 已提交
342
  char buf1[4096 + 512] = {0};
S
slguan 已提交
343
  char filename[400] = {0};
S
slguan 已提交
344
  sprintf(filename, "%s/%s.tmp", tsScriptDir, script->fileName);
S
slguan 已提交
345

S
slguan 已提交
346
  sprintf(buf, "cd %s; ", tsScriptDir);
S
slguan 已提交
347
  simVisuallizeOption(script, option, buf + strlen(buf));
sangshuduo's avatar
sangshuduo 已提交
348
  sprintf(buf1, "%s > %s 2>/dev/null", buf, filename);
S
slguan 已提交
349

sangshuduo's avatar
sangshuduo 已提交
350
  sprintf(script->system_exit_code, "%d", system(buf1));
S
slguan 已提交
351 352 353 354 355 356 357 358 359 360 361 362
  simStoreSystemContentResult(script, filename);

  script->linePos++;
  return true;
}

bool simExecutePrintCmd(SScript *script, char *rest) {
  char buf[65536];

  simVisuallizeOption(script, rest, buf);
  rest = buf;

S
Shengliang Guan 已提交
363
  simInfo("script:%s, %s", script->fileName, rest);
S
slguan 已提交
364 365 366 367 368
  script->linePos++;
  return true;
}

bool simExecuteSleepCmd(SScript *script, char *option) {
369 370
  int32_t delta;
  char    buf[1024];
S
slguan 已提交
371 372 373 374 375 376 377

  simVisuallizeOption(script, option, buf);
  option = buf;

  delta = atoi(option);
  if (delta <= 0) delta = 5;

S
Shengliang Guan 已提交
378
  simInfo("script:%s, sleep %dms begin", script->fileName, delta);
S
slguan 已提交
379
  taosMsleep(delta);
S
Shengliang Guan 已提交
380
  simInfo("script:%s, sleep %dms finished", script->fileName, delta);
S
slguan 已提交
381

S
Shengliang Guan 已提交
382 383 384 385
  char sleepStr[32] = {0};
  sprintf(sleepStr, "sleep %d", delta);
  simLogSql(sleepStr, true);

S
slguan 已提交
386 387 388 389 390 391 392 393 394 395
  script->linePos++;
  return true;
}

bool simExecuteReturnCmd(SScript *script, char *option) {
  char buf[1024];

  simVisuallizeOption(script, option, buf);
  option = buf;

396
  int32_t ret = 1;
S
slguan 已提交
397 398 399 400 401 402
  if (option && option[0] != 0) ret = atoi(option);

  if (ret < 0) {
    sprintf(script->error, "lineNum:%d. error return %s", script->lines[script->linePos].lineNum, option);
    return false;
  } else {
S
Shengliang Guan 已提交
403
    simInfo("script:%s, return cmd execute with:%d", script->fileName, ret);
S
slguan 已提交
404 405 406 407 408 409 410 411
    script->linePos = script->numOfLines;
  }

  script->linePos++;
  return true;
}

void simVisuallizeOption(SScript *script, char *src, char *dst) {
412 413
  char *  var, *token, *value;
  int32_t dstLen, srcLen, tokenLen;
S
slguan 已提交
414 415 416 417 418 419 420

  dst[0] = 0, dstLen = 0;

  while (1) {
    var = strchr(src, '$');
    if (var == NULL) break;
    if (var && ((var - src - 1) > 0) && *(var - 1) == '\\') {
421
      srcLen = (int32_t)(var - src - 1);
S
slguan 已提交
422 423 424 425 426 427
      memcpy(dst + dstLen, src, srcLen);
      dstLen += srcLen;
      src = var;
      break;
    }

428
    srcLen = (int32_t)(var - src);
S
slguan 已提交
429 430 431 432 433 434 435
    memcpy(dst + dstLen, src, srcLen);
    dstLen += srcLen;

    src = paGetToken(var + 1, &token, &tokenLen);
    value = simGetVariable(script, token, tokenLen);

    strcpy(dst + dstLen, value);
436
    dstLen += (int32_t)strlen(value);
S
slguan 已提交
437 438 439 440 441
  }

  strcpy(dst + dstLen, src);
}

442
void simCloseRestFulConnect(SScript *script) { 
S
slguan 已提交
443 444 445 446 447 448
  memset(script->auth, 0, sizeof(script->auth));
}

void simCloseNativeConnect(SScript *script) {
  if (script->taos == NULL) return;

S
Shengliang Guan 已提交
449
  simDebug("script:%s, taos:%p closed", script->fileName, script->taos);
S
slguan 已提交
450 451 452 453 454 455 456
  taos_close(script->taos);

  script->taos = NULL;
}

void simCloseTaosdConnect(SScript *script) {
  if (simAsyncQuery) {
S
TD-1057  
Shengliang Guan 已提交
457
    simCloseRestFulConnect(script);
S
slguan 已提交
458
  } else {
S
TD-1057  
Shengliang Guan 已提交
459
    simCloseNativeConnect(script);
S
slguan 已提交
460 461 462 463 464
  }
}
//  {"status":"succ","code":0,"desc":"/KfeAzX/f9na8qdtNZmtONryp201ma04bEl8LcvLUd7a8qdtNZmtONryp201ma04"}
//  {"status":"succ","head":["affected_rows"],"data":[[1]],"rows":1}
//  {"status":"succ","head":["ts","i"],"data":[["2017-12-25 21:28:41.022",1],["2017-12-25 21:28:42.022",2],["2017-12-25 21:28:43.022",3],["2017-12-25 21:28:44.022",4],["2017-12-25 21:28:45.022",5],["2017-12-25 21:28:46.022",6],["2017-12-25 21:28:47.022",7],["2017-12-25 21:28:48.022",8],["2017-12-25 21:28:49.022",9],["2017-12-25 21:28:50.022",10]],"rows":10}
465
int32_t simParseHttpCommandResult(SScript *script, char *command) {
S
slguan 已提交
466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
  cJSON* root = cJSON_Parse(command);
  if (root == NULL) {
    simError("script:%s, failed to parse json, response:%s", script->fileName, command);
    return -1;
  }

  cJSON *status = cJSON_GetObjectItem(root, "status");
  if (status == NULL) {
    simError("script:%s, failed to parse json, status is null, response:%s", script->fileName, command);
    cJSON_Delete(root);
    return -1;
  }

  if (status->valuestring == NULL || strlen(status->valuestring) == 0) {
    simError("script:%s, failed to parse json, status value is null, response:%s", script->fileName, command);
    cJSON_Delete(root);
    return -1;
  }

  if (strcmp(status->valuestring, "succ") != 0) {
    cJSON *code = cJSON_GetObjectItem(root, "code");
    if (code == NULL) {
      simError("script:%s, failed to parse json, code is null, response:%s", script->fileName, command);
      cJSON_Delete(root);
      return -1;
    }
492
    int32_t retcode = (int32_t)code->valueint;
S
slguan 已提交
493
    if (retcode != 1017) {
494 495
      simError("script:%s, json:status:%s not equal to succ, response:%s", script->fileName, status->valuestring,
               command);
S
slguan 已提交
496 497 498
      cJSON_Delete(root);
      return retcode;
    } else {
S
Shengliang Guan 已提交
499
      simDebug("script:%s, json:status:%s not equal to succ, but code is %d, response:%s", script->fileName,
500
               status->valuestring, retcode, command);
S
slguan 已提交
501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524
      cJSON_Delete(root);
      return 0;
    }
  }

  cJSON *desc = cJSON_GetObjectItem(root, "desc");
  if (desc != NULL) {
    if (desc->valuestring == NULL || strlen(desc->valuestring) == 0) {
      simError("script:%s, failed to parse json, desc value is null, response:%s", script->fileName, command);
      cJSON_Delete(root);
      return -1;
    }
    strcpy(script->auth, desc->valuestring);
    cJSON_Delete(root);
    return 0;
  }

  cJSON *data = cJSON_GetObjectItem(root, "data");
  if (data == NULL) {
    simError("script:%s, failed to parse json, data is null, response:%s", script->fileName, command);
    cJSON_Delete(root);
    return -1;
  }

525
  int32_t rowsize = cJSON_GetArraySize(data);
S
slguan 已提交
526 527 528 529 530 531
  if (rowsize < 0) {
    simError("script:%s, failed to parse json:data, data size %d, response:%s", script->fileName, rowsize, command);
    cJSON_Delete(root);
    return -1;
  }

532
  int32_t rowIndex = 0;
S
slguan 已提交
533
  sprintf(script->rows, "%d", rowsize);
534
  for (int32_t r = 0; r < rowsize; ++r) {
S
slguan 已提交
535 536 537 538
    cJSON *row = cJSON_GetArrayItem(data, r);
    if (row == NULL) continue;
    if (rowIndex++ >= 10) break;

539
    int32_t colsize = cJSON_GetArraySize(row);
S
slguan 已提交
540 541 542 543 544
    if (colsize < 0) {
      break;
    }

    colsize = MIN(10, colsize);
545
    for (int32_t c = 0; c < colsize; ++c) {
S
slguan 已提交
546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561
      cJSON *col = cJSON_GetArrayItem(row, c);
      if (col->valuestring != NULL) {
        strcpy(script->data[r][c], col->valuestring);
      } else {
        if (col->numberstring[0] == 0) {
          strcpy(script->data[r][c], "null");
        } else {
          strcpy(script->data[r][c], col->numberstring);
        }
      }
    }
  }

  return 0;
}

562
int32_t simExecuteRestFulCommand(SScript *script, char *command) {
S
slguan 已提交
563 564 565 566 567 568
  char buf[5000] = {0};
  sprintf(buf, "%s 2>/dev/null", command);

  FILE *fp = popen(buf, "r");
  if (fp == NULL) {
    simError("failed to execute %s", buf);
569
    return -1;
S
slguan 已提交
570 571
  }

572 573 574
  int32_t mallocSize = 2000;
  int32_t alreadyReadSize = 0;
  char *  content = malloc(mallocSize);
S
slguan 已提交
575 576

  while (!feof(fp)) {
577 578
    int32_t availSize = mallocSize - alreadyReadSize;
    int32_t len = (int32_t)fread(content + alreadyReadSize, 1, availSize, fp);
S
slguan 已提交
579 580 581 582 583 584 585 586 587 588 589 590 591 592
    if (len >= availSize) {
      alreadyReadSize += len;
      mallocSize *= 2;
      content = realloc(content, mallocSize);
    }
  }

  pclose(fp);

  return simParseHttpCommandResult(script, content);
}

bool simCreateRestFulConnect(SScript *script, char *user, char *pass) {
  char command[4096];
S
Shengliang Guan 已提交
593
  sprintf(command, "curl 127.0.0.1:6041/rest/login/%s/%s", user, pass);
S
slguan 已提交
594 595

  bool success = false;
596
  for (int32_t attempt = 0; attempt < 10; ++attempt) {
S
slguan 已提交
597 598
    success = simExecuteRestFulCommand(script, command) == 0;
    if (!success) {
599 600
      simDebug("script:%s, user:%s connect taosd failed:%s, attempt:%d", script->fileName, user, taos_errstr(NULL),
               attempt);
S
slguan 已提交
601 602
      taosMsleep(1000);
    } else {
S
Shengliang Guan 已提交
603
      simDebug("script:%s, user:%s connect taosd successed, attempt:%d", script->fileName, user, attempt);
S
slguan 已提交
604 605 606 607 608
      break;
    }
  }

  if (!success) {
609 610
    sprintf(script->error, "lineNum:%d. connect taosd failed:%s", script->lines[script->linePos].lineNum,
            taos_errstr(NULL));
S
slguan 已提交
611 612 613
    return false;
  }

S
Shengliang Guan 已提交
614
  simDebug("script:%s, connect taosd successed, auth:%p", script->fileName, script->auth);
S
slguan 已提交
615 616 617 618 619 620
  return true;
}

bool simCreateNativeConnect(SScript *script, char *user, char *pass) {
  simCloseTaosdConnect(script);
  void *taos = NULL;
S
slguan 已提交
621
  taosMsleep(2000);
622
  for (int32_t attempt = 0; attempt < 10; ++attempt) {
J
jtao1735 已提交
623
    taos = taos_connect(NULL, user, pass, NULL, tsDnodeShellPort);
S
slguan 已提交
624
    if (taos == NULL) {
625 626
      simDebug("script:%s, user:%s connect taosd failed:%s, attempt:%d", script->fileName, user, taos_errstr(NULL),
               attempt);
S
slguan 已提交
627 628
      taosMsleep(1000);
    } else {
S
Shengliang Guan 已提交
629
      simDebug("script:%s, user:%s connect taosd successed, attempt:%d", script->fileName, user, attempt);
S
slguan 已提交
630 631 632 633 634
      break;
    }
  }

  if (taos == NULL) {
635 636
    sprintf(script->error, "lineNum:%d. connect taosd failed:%s", script->lines[script->linePos].lineNum,
            taos_errstr(NULL));
S
slguan 已提交
637 638 639 640
    return false;
  }

  script->taos = taos;
S
Shengliang Guan 已提交
641
  simDebug("script:%s, connect taosd successed, taos:%p", script->fileName, taos);
S
slguan 已提交
642 643 644 645 646

  return true;
}

bool simCreateTaosdConnect(SScript *script, char *rest) {
647 648 649
  char *  user = TSDB_DEFAULT_USER;
  char *  token;
  int32_t tokenLen;
S
slguan 已提交
650 651 652 653 654 655 656
  rest = paGetToken(rest, &token, &tokenLen);
  rest = paGetToken(rest, &token, &tokenLen);
  if (tokenLen != 0) {
    user = token;
  }

  if (simAsyncQuery) {
657
    return simCreateRestFulConnect(script, user, TSDB_DEFAULT_PASS);
S
slguan 已提交
658
  } else {
659
    return simCreateNativeConnect(script, user, TSDB_DEFAULT_PASS);
S
slguan 已提交
660 661 662 663
  }
}

bool simExecuteNativeSqlCommand(SScript *script, char *rest, bool isSlow) {
664 665
  char       timeStr[30] = {0};
  time_t     tt;
S
slguan 已提交
666
  struct tm *tp;
667 668 669 670
  SCmdLine * line = &script->lines[script->linePos];
  int32_t    ret = -1;

  TAOS_RES *pSql = NULL;
S
slguan 已提交
671

672
  for (int32_t attempt = 0; attempt < 10; ++attempt) {
S
Shengliang Guan 已提交
673
    simLogSql(rest, false);
H
Haojun Liao 已提交
674
    pSql = taos_query(script->taos, rest);
H
Haojun Liao 已提交
675
    ret = taos_errno(pSql);
676

677
    if (ret == TSDB_CODE_MND_TABLE_ALREADY_EXIST || ret == TSDB_CODE_MND_DB_ALREADY_EXIST) {
678 679
      simDebug("script:%s, taos:%p, %s success, ret:%d:%s", script->fileName, script->taos, rest, ret & 0XFFFF,
               tstrerror(ret));
S
slguan 已提交
680 681 682
      ret = 0;
      break;
    } else if (ret != 0) {
683 684
      simDebug("script:%s, taos:%p, %s failed, ret:%d:%s, error:%s", script->fileName, script->taos, rest, ret & 0XFFFF,
               tstrerror(ret), taos_errstr(pSql));
S
slguan 已提交
685 686 687

      if (line->errorJump == SQL_JUMP_TRUE) {
        script->linePos = line->jump;
688
        taos_free_result(pSql);
S
slguan 已提交
689 690 691 692 693 694
        return true;
      }
      taosMsleep(1000);
    } else {
      break;
    }
695

H
Haojun Liao 已提交
696
    taos_free_result(pSql);
S
slguan 已提交
697 698 699
  }

  if (ret) {
S
Shengliang Guan 已提交
700
    sprintf(script->error, "lineNum:%d. sql:%s failed, ret:%d:%s", line->lineNum, rest, ret & 0XFFFF, tstrerror(ret));
S
slguan 已提交
701 702 703
    return false;
  }

704 705
  int32_t numOfRows = 0;
  int32_t num_fields = taos_field_count(pSql);
S
slguan 已提交
706
  if (num_fields != 0) {
H
Haojun Liao 已提交
707
    if (pSql == NULL) {
S
Shengliang Guan 已提交
708
      simDebug("script:%s, taos:%p, %s failed, result is null", script->fileName, script->taos, rest);
S
slguan 已提交
709 710 711 712 713 714 715 716 717 718 719
      if (line->errorJump == SQL_JUMP_TRUE) {
        script->linePos = line->jump;
        return true;
      }

      sprintf(script->error, "lineNum:%d. result set null, sql:%s", line->lineNum, rest);
      return false;
    }

    TAOS_ROW row;

H
Haojun Liao 已提交
720
    while ((row = taos_fetch_row(pSql))) {
S
slguan 已提交
721
      if (numOfRows < MAX_QUERY_ROW_NUM) {
H
Haojun Liao 已提交
722
        TAOS_FIELD *fields = taos_fetch_fields(pSql);
723 724 725
        int32_t *   length = taos_fetch_lengths(pSql);

        for (int32_t i = 0; i < num_fields; i++) {
S
slguan 已提交
726 727 728 729 730 731 732 733 734
          char *value = NULL;
          if (i < MAX_QUERY_COL_NUM) {
            value = script->data[numOfRows][i];
          }
          if (value == NULL) {
            continue;
          }

          if (row[i] == 0) {
H
hjxilinx 已提交
735
            strcpy(value, TSDB_DATA_NULL_STR);
S
slguan 已提交
736 737 738 739 740
            continue;
          }

          switch (fields[i].type) {
            case TSDB_DATA_TYPE_BOOL:
741
              sprintf(value, "%s", ((((int32_t)(*((char *)row[i]))) == 1) ? "1" : "0"));
S
slguan 已提交
742 743
              break;
            case TSDB_DATA_TYPE_TINYINT:
S
TD-1530  
Shengliang Guan 已提交
744
              sprintf(value, "%d", *((int8_t *)row[i]));
S
slguan 已提交
745 746
              break;
            case TSDB_DATA_TYPE_SMALLINT:
S
TD-1530  
Shengliang Guan 已提交
747
              sprintf(value, "%d", *((int16_t *)row[i]));
S
slguan 已提交
748 749
              break;
            case TSDB_DATA_TYPE_INT:
S
TD-1530  
Shengliang Guan 已提交
750
              sprintf(value, "%d", *((int32_t *)row[i]));
S
slguan 已提交
751 752
              break;
            case TSDB_DATA_TYPE_BIGINT:
753
              sprintf(value, "%" PRId64, *((int64_t *)row[i]));
S
slguan 已提交
754
              break;
S
TD-1530  
Shengliang Guan 已提交
755 756
            case TSDB_DATA_TYPE_FLOAT:
              sprintf(value, "%.5f", GET_FLOAT_VAL(row[i]));
S
slguan 已提交
757
              break;
S
TD-1530  
Shengliang Guan 已提交
758 759
            case TSDB_DATA_TYPE_DOUBLE:
              sprintf(value, "%.9lf", GET_DOUBLE_VAL(row[i]));
S
slguan 已提交
760 761 762
              break;
            case TSDB_DATA_TYPE_BINARY:
            case TSDB_DATA_TYPE_NCHAR:
S
scripts  
slguan 已提交
763
              memset(value, 0, MAX_QUERY_VALUE_LEN);
H
hjxilinx 已提交
764 765
              memcpy(value, row[i], length[i]);
              value[length[i]] = 0;
S
slguan 已提交
766 767 768 769
              // snprintf(value, fields[i].bytes, "%s", (char *)row[i]);
              break;
            case TSDB_DATA_TYPE_TIMESTAMP:
              tt = *(int64_t *)row[i] / 1000;
B
Bomin Zhang 已提交
770 771 772
              /* 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.
773 774 775
              if (tt < 0) {
                tt = 0;
              }
B
Bomin Zhang 已提交
776
              */
777

778 779 780 781
#ifdef WINDOWS
              if (tt < 0) tt = 0;
#endif

S
slguan 已提交
782 783
              tp = localtime(&tt);
              strftime(timeStr, 64, "%y-%m-%d %H:%M:%S", tp);
784 785
              sprintf(value, "%s.%03d", timeStr, (int32_t)(*((int64_t *)row[i]) % 1000));

S
slguan 已提交
786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802
              break;
            default:
              break;
          }  // end of switch
        }    // end of for
      }      // end of if
      numOfRows++;
      if (isSlow && numOfRows % 100 == 0) {
        taosMsleep(200);
      }
      if (numOfRows > 2000000000) {
        simError("script:%s, too many rows return from query", script->fileName);
        break;
      }
    }

  } else {
H
Haojun Liao 已提交
803
    numOfRows = taos_affected_rows(pSql);
S
slguan 已提交
804 805
  }

806
  taos_free_result(pSql);
S
slguan 已提交
807 808 809 810 811 812 813 814 815
  sprintf(script->rows, "%d", numOfRows);

  script->linePos++;
  return true;
}

bool simExecuteRestFulSqlCommand(SScript *script, char *rest) {
  SCmdLine *line = &script->lines[script->linePos];
  char command[4096];
S
Shengliang Guan 已提交
816
  sprintf(command, "curl -H 'Authorization: Taosd %s' -d \"%s\" 127.0.0.1:6041/rest/sql", script->auth, rest);
S
slguan 已提交
817

818 819
  int32_t ret = -1;
  for (int32_t attempt = 0; attempt < 10; ++attempt) {
S
slguan 已提交
820
    ret = simExecuteRestFulCommand(script, command);
821 822 823
    if (ret == TSDB_CODE_MND_TABLE_ALREADY_EXIST || ret == TSDB_CODE_MND_DB_ALREADY_EXIST) {
      simDebug("script:%s, taos:%p, %s success, ret:%d:%s", script->fileName, script->taos, rest, ret & 0XFFFF,
               tstrerror(ret));
S
slguan 已提交
824 825 826
      ret = 0;
      break;
    } else if (ret != 0) {
827
      simDebug("script:%s, taos:%p, %s failed, ret:%d", script->fileName, script->taos, rest, ret);
S
slguan 已提交
828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854

      if (line->errorJump == SQL_JUMP_TRUE) {
        script->linePos = line->jump;
        return true;
      }
      taosMsleep(1000);
    } else {
      break;
    }
  }

  if (ret) {
    sprintf(script->error, "lineNum:%d. sql:%s failed, ret:%d", line->lineNum, rest, ret);
    return false;
  }

  script->linePos++;
  return true;
}

bool simExecuteSqlImpCmd(SScript *script, char *rest, bool isSlow) {
  char buf[3000];
  SCmdLine *line = &script->lines[script->linePos];

  simVisuallizeOption(script, rest, buf);
  rest = buf;

S
Shengliang Guan 已提交
855
  simDebug("script:%s, exec:%s", script->fileName, rest);
S
slguan 已提交
856
  strcpy(script->rows, "-1");
857 858
  for (int32_t row = 0; row < MAX_QUERY_ROW_NUM; ++row) {
    for (int32_t col = 0; col < MAX_QUERY_COL_NUM; ++col) {
S
slguan 已提交
859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903
      strcpy(script->data[row][col], "null");
    }
  }

  if (strncmp(rest, "connect", 7) == 0) {
    if (!simCreateTaosdConnect(script, rest)) {
      return false;
    }
    script->linePos++;
    return true;
  }

  if ((!simAsyncQuery && script->taos == NULL) || (simAsyncQuery && script->auth[0] == 0)) {
    if (!simCreateTaosdConnect(script, "connect root")) {
      if (line->errorJump == SQL_JUMP_TRUE) {
        script->linePos = line->jump;
        return true;
      }
      return false;
    }
  }

  if (strncmp(rest, "close", 5) == 0) {
    simCloseTaosdConnect(script);
    script->linePos++;
    return true;
  }

  if (simAsyncQuery) {
    return simExecuteRestFulSqlCommand(script, rest);
  } else {
    return simExecuteNativeSqlCommand(script, rest, isSlow);
  }
}

bool simExecuteSqlCmd(SScript *script, char *rest) {
  bool isSlow = false;
  return simExecuteSqlImpCmd(script, rest, isSlow);
}

bool simExecuteSqlSlowCmd(SScript *script, char *rest) {
  bool isSlow = true;
  return simExecuteSqlImpCmd(script, rest, isSlow);
}

S
TD-1225  
Shengliang Guan 已提交
904 905
bool simExecuteRestfulCmd(SScript *script, char *rest) {
  FILE *fp = NULL;
906 907
  char  filename[256];
  sprintf(filename, "%s/tmp.sql", tsScriptDir);
S
TD-1225  
Shengliang Guan 已提交
908 909 910 911 912 913
  fp = fopen(filename, "w");
  if (fp == NULL) {
    fprintf(stderr, "ERROR: failed to open file: %s\n", filename);
    return false;
  }

914 915 916
  char    db[64] = {0};
  char    tb[64] = {0};
  char    gzip[32] = {0};
S
TD-1225  
Shengliang Guan 已提交
917 918 919
  int32_t ts;
  int32_t times;
  sscanf(rest, "%s %s %d %d %s", db, tb, &ts, &times, gzip);
920

S
TD-1225  
Shengliang Guan 已提交
921
  fprintf(fp, "insert into %s.%s values ", db, tb);
922
  for (int32_t i = 0; i < times; ++i) {
S
TD-1225  
Shengliang Guan 已提交
923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944
    fprintf(fp, "(%d000, %d)", ts + i, ts);
  }
  fprintf(fp, "  \n");
  fflush(fp);
  fclose(fp);

  char cmd[1024] = {0};
  if (strcmp(gzip, "gzip") == 0) {
    sprintf(cmd,
            "curl -H 'Authorization: Taosd /KfeAzX/f9na8qdtNZmtONryp201ma04bEl8LcvLUd7a8qdtNZmtONryp201ma04' --header "
            "--compressed --data-ascii @%s 127.0.0.1:7111/rest/sql",
            filename);
  } else {
    sprintf(cmd,
            "curl -H 'Authorization: Taosd /KfeAzX/f9na8qdtNZmtONryp201ma04bEl8LcvLUd7a8qdtNZmtONryp201ma04' --header "
            "'Transfer-Encoding: chunked' --data-ascii @%s 127.0.0.1:7111/rest/sql",
            filename);
  }

  return simExecuteSystemCmd(script, cmd);
}

S
slguan 已提交
945 946 947 948 949 950 951
bool simExecuteSqlErrorCmd(SScript *script, char *rest) {
  char buf[3000];
  SCmdLine *line = &script->lines[script->linePos];

  simVisuallizeOption(script, rest, buf);
  rest = buf;

S
Shengliang Guan 已提交
952
  simDebug("script:%s, exec:%s", script->fileName, rest);
S
slguan 已提交
953
  strcpy(script->rows, "-1");
954 955
  for (int32_t row = 0; row < MAX_QUERY_ROW_NUM; ++row) {
    for (int32_t col = 0; col < MAX_QUERY_COL_NUM; ++col) {
S
slguan 已提交
956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983
      strcpy(script->data[row][col], "null");
    }
  }

  if (strncmp(rest, "connect", 7) == 0) {
    if (!simCreateTaosdConnect(script, rest)) {
      return false;
    }
    script->linePos++;
    return true;
  }

  if ((!simAsyncQuery && script->taos == NULL) || (simAsyncQuery && script->auth[0] == 0)) {
    if (!simCreateTaosdConnect(script, "connect root")) {
      if (line->errorJump == SQL_JUMP_TRUE) {
        script->linePos = line->jump;
        return true;
      }
      return false;
    }
  }

  if (strncmp(rest, "close", 5) == 0) {
    simCloseTaosdConnect(script);
    script->linePos++;
    return true;
  }

984 985
  int32_t   ret;
  TAOS_RES *pSql = NULL;
S
slguan 已提交
986 987
  if (simAsyncQuery) {
    char command[4096];
S
Shengliang Guan 已提交
988
    sprintf(command, "curl -H 'Authorization: Taosd %s' -d '%s' 127.0.0.1:6041/rest/sql", script->auth, rest);
S
slguan 已提交
989
    ret = simExecuteRestFulCommand(script, command);
990
  } else {
H
Haojun Liao 已提交
991
    pSql = taos_query(script->taos, rest);
H
Haojun Liao 已提交
992
    ret = taos_errno(pSql);
H
Haojun Liao 已提交
993
    taos_free_result(pSql);
S
slguan 已提交
994 995 996
  }

  if (ret != TSDB_CODE_SUCCESS) {
997 998
    simDebug("script:%s, taos:%p, %s execute, expect failed, so success, ret:%d:%s", script->fileName, script->taos,
             rest, ret & 0XFFFF, tstrerror(ret));
S
slguan 已提交
999 1000 1001
    script->linePos++;
    return true;
  }
1002 1003 1004

  sprintf(script->error, "lineNum:%d. sql:%s expect failed, but success, ret:%d:%s", line->lineNum, rest, ret & 0XFFFF,
          tstrerror(ret));
S
slguan 已提交
1005 1006 1007

  return false;
}