simMain.c 2.2 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
 * 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/>.
 */
S
slguan 已提交
15

16
#define _DEFAULT_SOURCE
S
slguan 已提交
17 18
#include "os.h"
#include "tglobal.h"
S
slguan 已提交
19
#include "sim.h"
S
Shengliang Guan 已提交
20
#undef TAOS_MEM_CHECK
S
slguan 已提交
21 22

bool simAsyncQuery = false;
S
TD-2808  
Shengliang Guan 已提交
23
bool simExecSuccess = false;
24
bool abortExecution = false;
S
slguan 已提交
25

S
TD-1207  
Shengliang Guan 已提交
26
void simHandleSignal(int32_t signo, void *sigInfo, void *context) {
S
slguan 已提交
27
  simSystemCleanUp();
28 29 30
  abortExecution = true;
//  runningScript->killed = true;
//  exit(1);
S
slguan 已提交
31 32
}

33
int32_t main(int32_t argc, char *argv[]) {
S
slguan 已提交
34 35
  char scriptFile[MAX_FILE_NAME_LEN] = "sim_main_test.sim";

36
  for (int32_t i = 1; i < argc; ++i) {
S
slguan 已提交
37
    if (strcmp(argv[i], "-c") == 0 && i < argc - 1) {
B
Bomin Zhang 已提交
38
      tstrncpy(configDir, argv[++i], MAX_FILE_NAME_LEN);
S
slguan 已提交
39 40 41 42 43 44
    } else if (strcmp(argv[i], "-f") == 0 && i < argc - 1) {
      strcpy(scriptFile, argv[++i]);
    } else if (strcmp(argv[i], "-a") == 0) {
      simAsyncQuery = true;
    } else {
      printf("usage: %s [options] \n", argv[0]);
45
      printf("       [-c config]: config directory, default is: %s\n", configDir);
S
slguan 已提交
46
      printf("       [-f script]: script filename\n");
47
      return 0;
S
slguan 已提交
48 49 50 51 52 53
    }
  }

  if (!simSystemInit()) {
    simError("failed to initialize the system");
    simSystemCleanUp();
54
    return -1;
S
slguan 已提交
55 56
  }

S
Shengliang Guan 已提交
57
  simInfo("simulator is running ...");
S
TD-1207  
Shengliang Guan 已提交
58
  taosSetSignal(SIGINT, simHandleSignal);
S
slguan 已提交
59 60 61 62

  SScript *script = simParseScript(scriptFile);
  if (script == NULL) {
    simError("parse script file:%s failed", scriptFile);
63
    return -1;
S
slguan 已提交
64 65
  }

66 67 68 69 70
  if (abortExecution) {
    simError("execute abort");
    return -1;
  }

S
slguan 已提交
71 72 73
  simScriptList[++simScriptPos] = script;
  simExecuteScript(script);

S
TD-2808  
Shengliang Guan 已提交
74
  int32_t ret = simExecSuccess ? 0 : -1;
S
TD-1207  
Shengliang Guan 已提交
75
  simInfo("execute result %d", ret);
S
TD-2808  
Shengliang Guan 已提交
76 77

  return ret;
S
slguan 已提交
78
}