minor code refactor

kuzudb · Nov 15, 2022 · bbfe0fe · bbfe0fe
1 parent 18f4943
commit bbfe0fe
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tools/shell/embedded_shell.cpp b/tools/shell/embedded_shell.cpp
@@ -175,6 +175,8 @@ EmbeddedShell::EmbeddedShell(
 
 void EmbeddedShell::run() {
     char* line;
+    string query;
+    stringstream ss;
     while ((line = linenoise(continueLine ? ALTPROMPT : PROMPT)) != nullptr) {
         auto lineStr = string(line);
         if (continueLine) {
@@ -207,14 +209,14 @@ void EmbeddedShell::run() {
         } else if (lineStr.rfind(shellCommand.SHOW_REL) == 0) {
             printRelSchema(lineStr.substr(shellCommand.SHOW_REL.length()));
         } else if (!lineStr.empty()) {
-            stringstream ss(lineStr);
-            string thisQuery;
-            while (getline(ss, thisQuery, ';')) {
+            ss.clear();
+            ss.str(lineStr);
+            while (getline(ss, query, ';')) {
                 if (ss.eof() && ss.peek() == -1) {
                     continueLine = true;
-                    currLine += thisQuery + " ";
+                    currLine += query + " ";
                 } else {
-                    auto queryResult = conn->query(thisQuery);
+                    auto queryResult = conn->query(query);
                     if (queryResult->isSuccess()) {
                         printExecutionResult(*queryResult);
                     } else {