diff --git a/tools/inc/pub.h b/tools/inc/pub.h index 7e96c7a324..9fa8d0a142 100644 --- a/tools/inc/pub.h +++ b/tools/inc/pub.h @@ -50,6 +50,7 @@ #define CONN_MODE_INVALID -1 #define CONN_MODE_NATIVE 0 #define CONN_MODE_WEBSOCKET 1 +#define CONN_MODE_DEFAULT CONN_MODE_NATIVE // set default mode // define error show module #define INIT_PHASE "init" diff --git a/tools/shell/src/shellMain.c b/tools/shell/src/shellMain.c index 677d2808b2..8f4b2c4448 100644 --- a/tools/shell/src/shellMain.c +++ b/tools/shell/src/shellMain.c @@ -56,6 +56,11 @@ void initArgument(SShellArgs *pArgs) { // set conn mode int32_t setConnMode(int8_t connMode) { + // default + if (connMode == CONN_MODE_INVALID) { + connMode = CONN_MODE_DEFAULT; + } + // set conn mode char * strMode = connMode == CONN_MODE_NATIVE ? STR_NATIVE : STR_WEBSOCKET; int32_t code = taos_options(TSDB_OPTION_DRIVER, strMode); diff --git a/tools/taos-tools/src/benchMain.c b/tools/taos-tools/src/benchMain.c index 564bc80bc6..c2df571a22 100644 --- a/tools/taos-tools/src/benchMain.c +++ b/tools/taos-tools/src/benchMain.c @@ -98,7 +98,12 @@ int32_t applyConfigDir(char * cfgDir){ return code; } -int32_t setConnMode(int8_t connMode, char *dsn) { +int32_t setConnMode(int8_t connMode) { + // default + if (connMode == CONN_MODE_INVALID) { + connMode = CONN_MODE_DEFAULT; + } + // set conn mode char * strMode = connMode == CONN_MODE_NATIVE ? STR_NATIVE : STR_WEBSOCKET; int32_t code = taos_options(TSDB_OPTION_DRIVER, strMode); @@ -181,7 +186,7 @@ int main(int argc, char* argv[]) { } // conn mode - if (setConnMode(g_arguments->connMode, g_arguments->dsn) != 0) { + if (setConnMode(g_arguments->connMode) != 0) { exitLog(); return -1; } diff --git a/tools/taos-tools/src/taosdump.c b/tools/taos-tools/src/taosdump.c index 6c14ca080a..ec2917940b 100644 --- a/tools/taos-tools/src/taosdump.c +++ b/tools/taos-tools/src/taosdump.c @@ -10857,6 +10857,11 @@ static int inspectAvroFiles(int argc, char *argv[]) { } int32_t setConnMode(int8_t connMode) { + // default + if (connMode == CONN_MODE_INVALID) { + connMode = CONN_MODE_DEFAULT; + } + // set conn mode char * strMode = connMode == CONN_MODE_NATIVE ? STR_NATIVE : STR_WEBSOCKET; int32_t code = taos_options(TSDB_OPTION_DRIVER, strMode);