@dbcube/core 3.0.5 → 3.0.6

package/dist/index.js

@@ -884,22 +884,30 @@ var QueryEngine = class {
         reject(new Error("TCP request timeout"));
       }, this.timeout);
       client.connect(this.tcpPort, "127.0.0.1", () => {
+        const dmlIndex = args.findIndex((arg) => arg === "--dml");
+        const dmlJson = dmlIndex !== -1 ? args[dmlIndex + 1] : null;
+        if (!dmlJson) {
+          client.destroy();
+          reject(new Error("No DML found in arguments"));
+          return;
+        }
         const command = {
           action: "execute",
-          args
+          dml: dmlJson
         };
-        client.write(JSON.stringify(command) + "\n");
+        client.write(JSON.stringify(command));
       });
       client.on("data", (data) => {
         responseBuffer += data.toString();
-        if (responseBuffer.includes("\n")) {
+        const match = responseBuffer.match(/PROCESS_RESPONSE:(\{.*\})/);
+        if (match) {
           clearTimeout(timeout);
           client.destroy();
           try {
-            const response = JSON.parse(responseBuffer.trim());
+            const response = JSON.parse(match[1]);
             resolve5({
-              status: response.status || 200,
-              message: response.message || "Success",
+              status: response.status,
+              message: response.message,
               data: response.data
             });
           } catch (error) {