dmMain.c 7.6 KB
Newer Older
H
refact  
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * 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/>.
 */
S
Shengliang Guan 已提交
15 16

#define _DEFAULT_SOURCE
S
Shengliang Guan 已提交
17
#include "dmMgmt.h"
S
Shengliang Guan 已提交
18
#include "mnode.h"
H
Hongze Cheng 已提交
19
#include "tconfig.h"
20
#include "tglobal.h"
H
refact  
Hongze Cheng 已提交
21

H
Hongze Cheng 已提交
22
// clang-format off
23 24 25
#define DM_APOLLO_URL    "The apollo string to use when configuring the server, such as: -a 'jsonFile:./tests/cfg.json', cfg.json text can be '{\"fqdn\":\"td1\"}'."
#define DM_CFG_DIR       "Configuration directory."
#define DM_DMP_CFG       "Dump configuration."
S
Shengliang Guan 已提交
26
#define DM_SDB_INFO      "Dump sdb info."
27 28 29 30 31
#define DM_ENV_CMD       "The env cmd variable string to use when configuring the server, such as: -e 'TAOS_FQDN=td1'."
#define DM_ENV_FILE      "The env variable file path to use when configuring the server, default is './.env', .env text can be 'TAOS_FQDN=td1'."
#define DM_MACHINE_CODE  "Get machine code."
#define DM_VERSION       "Print program version."
#define DM_EMAIL         "<support@taosdata.com>"
H
Hongze Cheng 已提交
32
// clang-format on
S
Shengliang Guan 已提交
33
static struct {
wafwerar's avatar
wafwerar 已提交
34
#ifdef WINDOWS
H
Hongze Cheng 已提交
35
  bool winServiceMode;
wafwerar's avatar
wafwerar 已提交
36
#endif
S
Shengliang Guan 已提交
37
  bool         dumpConfig;
S
Shengliang Guan 已提交
38
  bool         dumpSdb;
S
Shengliang Guan 已提交
39 40 41
  bool         generateGrant;
  bool         printAuth;
  bool         printVersion;
42
  bool         printHelp;
S
Shengliang Guan 已提交
43 44
  char         envFile[PATH_MAX];
  char         apolloUrl[PATH_MAX];
wafwerar's avatar
wafwerar 已提交
45
  const char **envCmd;
S
Shengliang Guan 已提交
46
  SArray      *pArgs;  // SConfigPair
S
shm  
Shengliang Guan 已提交
47
} global = {0};
S
Shengliang Guan 已提交
48

49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
static void dmSetDebugFlag(int32_t signum, void *sigInfo, void *context) { taosSetAllDebugFlag(143, true); }
static void dmSetAssert(int32_t signum, void *sigInfo, void *context) { tsAssert = 1; }

static void dmStopDnode(int signum, void *sigInfo, void *context) {
  // taosIgnSignal(SIGUSR1);
  // taosIgnSignal(SIGUSR2);
  taosIgnSignal(SIGTERM);
  taosIgnSignal(SIGHUP);
  taosIgnSignal(SIGINT);
  taosIgnSignal(SIGABRT);
  taosIgnSignal(SIGBREAK);

  dInfo("shut down signal is %d", signum);
#ifndef WINDOWS
  dInfo("sender PID:%d cmdline:%s", ((siginfo_t *)sigInfo)->si_pid,
        taosGetCmdlineByPID(((siginfo_t *)sigInfo)->si_pid));
#endif

  dmStop();
}
S
Shengliang Guan 已提交
69

S
Shengliang Guan 已提交
70
static void dmSetSignalHandle() {
71 72
  taosSetSignal(SIGUSR1, dmSetDebugFlag);
  taosSetSignal(SIGUSR2, dmSetAssert);
S
Shengliang Guan 已提交
73 74 75 76 77
  taosSetSignal(SIGTERM, dmStopDnode);
  taosSetSignal(SIGHUP, dmStopDnode);
  taosSetSignal(SIGINT, dmStopDnode);
  taosSetSignal(SIGABRT, dmStopDnode);
  taosSetSignal(SIGBREAK, dmStopDnode);
wafwerar's avatar
wafwerar 已提交
78 79
#ifndef WINDOWS
  taosSetSignal(SIGTSTP, dmStopDnode);
S
Shengliang Guan 已提交
80
  taosSetSignal(SIGQUIT, dmStopDnode);
wafwerar's avatar
wafwerar 已提交
81
#endif
S
Shengliang Guan 已提交
82 83
}

S
Shengliang Guan 已提交
84
static int32_t dmParseArgs(int32_t argc, char const *argv[]) {
wafwerar's avatar
wafwerar 已提交
85
  int32_t cmdEnvIndex = 0;
wafwerar's avatar
wafwerar 已提交
86
  if (argc < 2) return 0;
87 88
  global.envCmd = taosMemoryMalloc((argc - 1) * sizeof(char *));
  memset(global.envCmd, 0, (argc - 1) * sizeof(char *));
S
Shengliang Guan 已提交
89
  for (int32_t i = 1; i < argc; ++i) {
S
Shengliang Guan 已提交
90 91 92 93 94 95
    if (strcmp(argv[i], "-c") == 0) {
      if (i < argc - 1) {
        if (strlen(argv[++i]) >= PATH_MAX) {
          printf("config file path overflow");
          return -1;
        }
S
Shengliang Guan 已提交
96
        tstrncpy(configDir, argv[i], PATH_MAX);
S
Shengliang Guan 已提交
97
      } else {
98
        printf("'-c' requires a parameter, default is %s\n", configDir);
S
Shengliang Guan 已提交
99 100
        return -1;
      }
S
Shengliang Guan 已提交
101 102
    } else if (strcmp(argv[i], "-a") == 0) {
      tstrncpy(global.apolloUrl, argv[++i], PATH_MAX);
S
Shengliang Guan 已提交
103 104
    } else if (strcmp(argv[i], "-s") == 0) {
      global.dumpSdb = true;
wafwerar's avatar
wafwerar 已提交
105
    } else if (strcmp(argv[i], "-E") == 0) {
S
Shengliang Guan 已提交
106
      tstrncpy(global.envFile, argv[++i], PATH_MAX);
S
shm  
Shengliang Guan 已提交
107 108
    } else if (strcmp(argv[i], "-k") == 0) {
      global.generateGrant = true;
S
Shengliang Guan 已提交
109 110
    } else if (strcmp(argv[i], "-C") == 0) {
      global.dumpConfig = true;
S
Shengliang Guan 已提交
111
    } else if (strcmp(argv[i], "-V") == 0) {
S
shm  
Shengliang Guan 已提交
112
      global.printVersion = true;
H
Hongze Cheng 已提交
113
#ifdef WINDOWS
wafwerar's avatar
wafwerar 已提交
114 115
    } else if (strcmp(argv[i], "--win_service") == 0) {
      global.winServiceMode = true;
H
Hongze Cheng 已提交
116
#endif
wafwerar's avatar
wafwerar 已提交
117 118 119
    } else if (strcmp(argv[i], "-e") == 0) {
      global.envCmd[cmdEnvIndex] = argv[++i];
      cmdEnvIndex++;
120 121
    } else if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0 || strcmp(argv[i], "--usage") == 0 ||
               strcmp(argv[i], "-?")) {
122
      global.printHelp = true;
S
Shengliang Guan 已提交
123 124
    } else {
    }
S
Shengliang Guan 已提交
125 126
  }

S
Shengliang Guan 已提交
127 128 129
  return 0;
}

S
Shengliang Guan 已提交
130
static void dmGenerateGrant() { mndGenerateMachineCode(); }
S
Shengliang Guan 已提交
131

S
Shengliang Guan 已提交
132
static void dmPrintVersion() {
S
Shengliang Guan 已提交
133 134 135 136 137 138 139 140 141 142
#ifdef TD_ENTERPRISE
  char *releaseName = "enterprise";
#else
  char *releaseName = "community";
#endif
  printf("%s version: %s compatible_version: %s\n", releaseName, version, compatible_version);
  printf("gitinfo: %s\n", gitinfo);
  printf("buildInfo: %s\n", buildinfo);
}

143 144 145 146 147
static void dmPrintHelp() {
  char indent[] = "  ";
  printf("Usage: taosd [OPTION...] \n\n");
  printf("%s%s%s%s\n", indent, "-a,", indent, DM_APOLLO_URL);
  printf("%s%s%s%s\n", indent, "-c,", indent, DM_CFG_DIR);
S
Shengliang Guan 已提交
148
  printf("%s%s%s%s\n", indent, "-s,", indent, DM_SDB_INFO);
149 150 151 152 153 154 155 156 157
  printf("%s%s%s%s\n", indent, "-C,", indent, DM_DMP_CFG);
  printf("%s%s%s%s\n", indent, "-e,", indent, DM_ENV_CMD);
  printf("%s%s%s%s\n", indent, "-E,", indent, DM_ENV_FILE);
  printf("%s%s%s%s\n", indent, "-k,", indent, DM_MACHINE_CODE);
  printf("%s%s%s%s\n", indent, "-V,", indent, DM_VERSION);

  printf("\n\nReport bugs to %s.\n", DM_EMAIL);
}

S
Shengliang Guan 已提交
158
static void dmDumpCfg() {
S
Shengliang Guan 已提交
159
  SConfig *pCfg = taosGetCfg();
160
  cfgDumpCfg(pCfg, 0, true);
S
Shengliang Guan 已提交
161 162
}

S
Shengliang Guan 已提交
163
static int32_t dmInitLog() {
164
  return taosCreateLog("taosdlog", 1, configDir, global.envCmd, global.envFile, global.apolloUrl, global.pArgs, 0);
S
shm  
Shengliang Guan 已提交
165 166
}

wafwerar's avatar
wafwerar 已提交
167 168 169 170
static void taosCleanupArgs() {
  if (global.envCmd != NULL) taosMemoryFree(global.envCmd);
}

S
Shengliang Guan 已提交
171
int main(int argc, char const *argv[]) {
D
dapan1121 已提交
172 173
  if (!taosCheckSystemIsLittleEnd()) {
    printf("failed to start since on non-little-end machines\n");
174 175 176
    return -1;
  }

S
Shengliang Guan 已提交
177
  if (dmParseArgs(argc, argv) != 0) {
S
Shengliang Guan 已提交
178
    printf("failed to start since parse args error\n");
wafwerar's avatar
wafwerar 已提交
179
    taosCleanupArgs();
S
Shengliang Guan 已提交
180 181 182
    return -1;
  }

wafwerar's avatar
wafwerar 已提交
183
#ifdef WINDOWS
H
Hongze Cheng 已提交
184
  int mainWindows(int argc, char **argv);
wafwerar's avatar
wafwerar 已提交
185 186 187 188 189 190 191
  if (global.winServiceMode) {
    stratWindowsService(mainWindows);
  } else {
    return mainWindows(argc, argv);
  }
  return 0;
}
H
Hongze Cheng 已提交
192
int mainWindows(int argc, char **argv) {
wafwerar's avatar
wafwerar 已提交
193 194
#endif

S
shm  
Shengliang Guan 已提交
195
  if (global.generateGrant) {
S
Shengliang Guan 已提交
196
    dmGenerateGrant();
wafwerar's avatar
wafwerar 已提交
197
    taosCleanupArgs();
S
Shengliang Guan 已提交
198 199 200
    return 0;
  }

201 202 203 204 205 206
  if (global.printHelp) {
    dmPrintHelp();
    taosCleanupArgs();
    return 0;
  }

S
shm  
Shengliang Guan 已提交
207
  if (global.printVersion) {
S
Shengliang Guan 已提交
208
    dmPrintVersion();
wafwerar's avatar
wafwerar 已提交
209
    taosCleanupArgs();
S
Shengliang Guan 已提交
210
    return 0;
S
Shengliang Guan 已提交
211 212
  }

S
Shengliang Guan 已提交
213
  if (dmInitLog() != 0) {
S
Shengliang Guan 已提交
214
    printf("failed to start since init log error\n");
wafwerar's avatar
wafwerar 已提交
215
    taosCleanupArgs();
S
Shengliang Guan 已提交
216
    return -1;
S
Shengliang Guan 已提交
217 218
  }

wafwerar's avatar
wafwerar 已提交
219
  if (taosInitCfg(configDir, global.envCmd, global.envFile, global.apolloUrl, global.pArgs, 0) != 0) {
S
Shengliang Guan 已提交
220
    dError("failed to start since read config error");
S
Shengliang Guan 已提交
221
    taosCloseLog();
wafwerar's avatar
wafwerar 已提交
222
    taosCleanupArgs();
S
Shengliang Guan 已提交
223 224
    return -1;
  }
H
Hongze Cheng 已提交
225

S
Shengliang Guan 已提交
226 227 228 229 230 231
  if (taosConvInit() != 0) {
    dError("failed to init conv");
    taosCloseLog();
    taosCleanupArgs();
    return -1;
  }
S
Shengliang Guan 已提交
232

S
shm  
Shengliang Guan 已提交
233
  if (global.dumpConfig) {
S
Shengliang Guan 已提交
234
    dmDumpCfg();
S
Shengliang Guan 已提交
235
    taosCleanupCfg();
S
shm  
Shengliang Guan 已提交
236
    taosCloseLog();
wafwerar's avatar
wafwerar 已提交
237
    taosCleanupArgs();
238
    taosConvDestroy();
S
Shengliang Guan 已提交
239 240 241
    return 0;
  }

S
Shengliang Guan 已提交
242 243 244 245 246
  if (global.dumpSdb) {
    mndDumpSdb();
    taosCleanupCfg();
    taosCloseLog();
    taosCleanupArgs();
247
    taosConvDestroy();
S
Shengliang Guan 已提交
248 249 250
    return 0;
  }

251
  osSetProcPath(argc, (char **)argv);
wafwerar's avatar
wafwerar 已提交
252
  taosCleanupArgs();
253

254
  if (dmInit() != 0) {
255 256 257 258
    dError("failed to init dnode since %s", terrstr());
    return -1;
  }

S
Shengliang Guan 已提交
259
  dInfo("start to init service");
260 261 262 263 264 265
  dmSetSignalHandle();
  int32_t code = dmRun();
  dInfo("shutting down the service");

  dmCleanup();
  return code;
S
Shengliang Guan 已提交
266
}