xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: add env TDENGINE_DSN to taos-CLI

This commit is contained in:
Alex Duan 2023-10-25 16:46:32 +08:00
parent d58b964b25
commit 368c6d996b
5 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tools/shell/inc/shellInt.h
View File

@ -80,6 +80,7 @@ typedef struct {
#ifdef WEBSOCKET #ifdef WEBSOCKET
bool restful; bool restful;
bool cloud; bool cloud;
bool local;
char* dsn; char* dsn;
int32_t timeout; int32_t timeout;
#endif #endif

2
tools/shell/src/shellArguments.c
View File

@ -410,7 +410,7 @@ int32_t shellParseArgs(int32_t argc, char *argv[]) {
shellInitArgs(argc, argv); shellInitArgs(argc, argv);
shell.info.clientVersion = shell.info.clientVersion =
"Welcome to the %s Command Line Interface, Client Version:%s\r\n" "Welcome to the %s Command Line Interface, Client Version:%s\r\n"
"Copyright (c) 2022 by %s, all rights reserved.\r\n\r\n"; "Copyright (c) 2023 by %s, all rights reserved.\r\n\r\n";
#ifdef CUS_NAME #ifdef CUS_NAME
strcpy(shell.info.cusName, CUS_NAME); strcpy(shell.info.cusName, CUS_NAME);
#else #else

1
tools/shell/src/shellMain.c
View File

@ -47,6 +47,7 @@ int main(int argc, char *argv[]) {
#ifdef WEBSOCKET #ifdef WEBSOCKET
shell.args.timeout = SHELL_WS_TIMEOUT; shell.args.timeout = SHELL_WS_TIMEOUT;
shell.args.cloud = true; shell.args.cloud = true;
shell.args.local = false;
#endif #endif
#if !defined(WINDOWS) #if !defined(WINDOWS)

12
tools/shell/src/shellUtil.c
View File

@ -130,11 +130,21 @@ void shellCheckConnectMode() {
} }
if (shell.args.cloud) { if (shell.args.cloud) {
shell.args.dsn = getenv("TDENGINE_CLOUD_DSN"); shell.args.dsn = getenv("TDENGINE_CLOUD_DSN");
if (shell.args.dsn) { if (shell.args.cloud_dsn) {
shell.args.cloud = true; shell.args.cloud = true;
shell.args.local = false;
shell.args.restful = false; shell.args.restful = false;
return; return;
} }
shell.args.dsn = getenv("TDENGINE_DSN");
if (shell.args.dsn) {
shell.args.cloud = true;
shell.args.local = true;
shell.args.restful = false;
return;
}
if (shell.args.restful) { if (shell.args.restful) {
if (!shell.args.host) { if (!shell.args.host) {
shell.args.host = "localhost"; shell.args.host = "localhost";

12
tools/shell/src/shellWebsocket.c
View File

@ -59,7 +59,17 @@ int shell_conn_ws_server(bool first) {
fprintf(stdout, "successfully connected to %s\n\n", fprintf(stdout, "successfully connected to %s\n\n",
shell.args.dsn); shell.args.dsn);
} else if (first && shell.args.cloud) { } else if (first && shell.args.cloud) {
fprintf(stdout, "successfully connected to cloud service\n"); if(shell.args.local) {
const char* host = strstr(shell.args.dsn, "@");
if(host) {
host += 1;
} else {
host = shell.args.dsn;
}
fprintf(stdout, "successfully connected to %s\n", host);
} else {
fprintf(stdout, "successfully connected to cloud service\n");
}
} }
fflush(stdout); fflush(stdout);