mirror of
https://github.com/taosdata/TDengine
synced 2026-05-24 10:09:01 +00:00
55c5f8190d
* test: add python scripts for test windows cases replace the old bat scripts * refactor: update color_echo function to use log_file instead of error_file * ci: update workflow to use fixed branch for Windows test runs * chore: add empty line for better readability in tdengine-test.yml * test:fix from branch fix/xftan/taos-shell-configdir * fix: replace global variant configDir with local variant on taos * fix: remove check cfgdir exist * fix: restore no input msg * fix: move dumpConfig after taos_options * fix: dump Config on windows failed * fix: move configDirShell variant to shellArgument.c * fix: get last char pos is -1 * ci: update Windows test workflow to use main branch and enhance test output * ci: remove unnecessary line from tdengine-test workflow --------- Co-authored-by: Alex Duan <417921451@qq.com>
151 lines
3.4 KiB
C
151 lines
3.4 KiB
C
/*
|
|
* 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/>.
|
|
*/
|
|
|
|
#define __USE_XOPEN
|
|
#include "shellAuto.h"
|
|
#include "shellInt.h"
|
|
|
|
extern SShellObj shell;
|
|
|
|
void shellCrashHandler(int signum, void *sigInfo, void *context) {
|
|
taosIgnSignal(SIGTERM);
|
|
taosIgnSignal(SIGHUP);
|
|
taosIgnSignal(SIGINT);
|
|
taosIgnSignal(SIGBREAK);
|
|
taosIgnSignal(SIGABRT);
|
|
taosIgnSignal(SIGFPE);
|
|
taosIgnSignal(SIGSEGV);
|
|
#if !defined(WINDOWS)
|
|
taosIgnSignal(SIGBUS);
|
|
#endif
|
|
#ifdef USE_REPORT
|
|
taos_write_crashinfo(signum, sigInfo, context);
|
|
#endif
|
|
#ifdef _TD_DARWIN_64
|
|
exit(signum);
|
|
#elif defined(WINDOWS)
|
|
exit(signum);
|
|
#endif
|
|
}
|
|
|
|
// init arguments
|
|
void initArgument(SShellArgs *pArgs) {
|
|
pArgs->host = NULL;
|
|
pArgs->port = 0;
|
|
pArgs->user = NULL;
|
|
pArgs->database = NULL;
|
|
|
|
// conn mode
|
|
pArgs->dsn = NULL;
|
|
pArgs->connMode = CONN_MODE_INVALID;
|
|
|
|
pArgs->port_inputted = false;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
int code = 0;
|
|
#if !defined(WINDOWS)
|
|
taosSetSignal(SIGBUS, shellCrashHandler);
|
|
#endif
|
|
taosSetSignal(SIGABRT, shellCrashHandler);
|
|
taosSetSignal(SIGFPE, shellCrashHandler);
|
|
taosSetSignal(SIGSEGV, shellCrashHandler);
|
|
|
|
initArgument(&shell.args);
|
|
|
|
if (shellCheckIntSize() != 0) {
|
|
return -1;
|
|
}
|
|
|
|
if (shellParseArgs(argc, argv) != 0) {
|
|
return -1;
|
|
}
|
|
|
|
if (shell.args.is_version) {
|
|
shellPrintVersion();
|
|
return 0;
|
|
}
|
|
|
|
if (shell.args.is_gen_auth) {
|
|
shellGenerateAuth();
|
|
return 0;
|
|
}
|
|
|
|
if (shell.args.is_help) {
|
|
shellPrintHelp();
|
|
return 0;
|
|
}
|
|
|
|
if (shell.args.netrole != NULL) {
|
|
shellTestNetWork();
|
|
return 0;
|
|
}
|
|
|
|
if (getDsnEnv() != 0) {
|
|
return -1;
|
|
}
|
|
|
|
// first taos_option(TSDB_OPTION_DRIVER ...) no load driver
|
|
if (setConnMode(shell.args.connMode, shell.args.dsn, false)) {
|
|
return -1;
|
|
}
|
|
|
|
// second taos_option(TSDB_OPTION_CONFIGDIR ...) set configDir global
|
|
if (configDirShell[0] != 0) {
|
|
code = taos_options(TSDB_OPTION_CONFIGDIR, configDirShell);
|
|
if (code) {
|
|
fprintf(stderr, "failed to set config dir:%s code:[0x%08X]\r\n", configDirShell, code);
|
|
return -1;
|
|
}
|
|
//printf("Load with input config dir:%s\n", configDirShell);
|
|
}
|
|
|
|
#ifndef TD_ASTRA
|
|
// dump config
|
|
if (shell.args.is_dump_config) {
|
|
shellDumpConfig();
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
// taos_init
|
|
if (taos_init() != 0) {
|
|
fprintf(stderr, "failed to init shell since %s [0x%08X]\r\n", taos_errstr(NULL), taos_errno(NULL));
|
|
return -1;
|
|
}
|
|
|
|
// kill heart-beat thread when quit
|
|
taos_set_hb_quit(1);
|
|
|
|
#ifndef TD_ASTRA
|
|
if (shell.args.is_startup || shell.args.is_check) {
|
|
shellCheckServerStatus();
|
|
taos_cleanup();
|
|
return 0;
|
|
}
|
|
|
|
if (shell.args.netrole != NULL) {
|
|
shellTestNetWork();
|
|
taos_cleanup();
|
|
return 0;
|
|
}
|
|
#endif
|
|
// support port feature
|
|
shellAutoInit();
|
|
int32_t ret = shellExecute(argc, argv);
|
|
shellAutoExit();
|
|
|
|
return ret;
|
|
}
|