spice-couchbase 2.0.30 → 2.0.31

package/build/index.js

@@ -586,18 +586,54 @@ module.exports = class Database {
     })();
   }
 
+  optimize(query, exec_scope) {
+    return _asyncToGenerator(function* () {
+      try {
+        var _$last, _$first, _$first$advice, _$first$advice$advise, _$first$advice$advise2;
+
+        var options = {
+          adhoc: true
+        };
+        var advise_query = "ADVISE ".concat(query);
+        var response = yield exec_scope.query(advise_query, options);
+        var best_index = (_$last = _.last(((_$first = _.first((response === null || response === void 0 ? void 0 : response.rows) || [])) === null || _$first === void 0 ? void 0 : (_$first$advice = _$first.advice) === null || _$first$advice === void 0 ? void 0 : (_$first$advice$advise = _$first$advice.adviseinfo) === null || _$first$advice$advise === void 0 ? void 0 : (_$first$advice$advise2 = _$first$advice$advise.recommended_indexes) === null || _$first$advice$advise2 === void 0 ? void 0 : _$first$advice$advise2.indexes) || [])) === null || _$last === void 0 ? void 0 : _$last.index_statement;
+
+        if (best_index) {
+          yield exec_scope.query(best_index);
+        }
+      } catch (e) {}
+    })();
+  }
+
   query(query) {
     var _this18 = this;
 
     var query_scope = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : "cluster";
     return _asyncToGenerator(function* () {
       var scope = query_scope || _this18.query_scope;
+      var sanitizedQuery = query; //sanitization step to be implemented
+
+      var exec_scope = null;
 
-      if (scope == "cluster") {
-        return yield _this18.cluster.query(query);
+      if (scope === "cluster") {
+        exec_scope = _this18.cluster;
       } else {
-        return yield _this18.scope.query(query);
+        exec_scope = _this18.scope;
+      } // Start time
+
+
+      var startTime = process.hrtime.bigint();
+      var response = yield exec_scope.query(sanitizedQuery); // End time
+
+      var endTime = process.hrtime.bigint(); // Calculate duration in microseconds
+
+      var durationInMilliseconds = (endTime - startTime) / BigInt(1000000); // optimize if response time is greater than 1 second
+
+      if (durationInMilliseconds > 1000) {
+        yield _this18.optimize(query, exec_scope);
       }
+
+      return response;
     })();
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "spice-couchbase",
-  "version": "2.0.30",
+  "version": "2.0.31",
   "description": "",
   "main": "build/index.js",
   "scripts": {

package/src/index.js

@@ -531,25 +531,45 @@ module.exports = class Database {
     return response.rows;
   }
 
-  /*  async query(query, query_scope = "cluster") {
-    let scope = query_scope || this.query_scope;
-    if (scope == "cluster") {
-      return await this.cluster.query(query);
-    } else {
-      return await this.scope.query(query);
-    }
-  } */
+  async optimize(query, exec_scope) {
+    try {
+      const options = {
+        adhoc: true,
+      };
+      let advise_query = `ADVISE ${query}`;
+      let response = await exec_scope.query(advise_query, options);
+
+      let best_index = _.last(
+        _.first(response?.rows || [])?.advice?.adviseinfo?.recommended_indexes
+          ?.indexes || []
+      )?.index_statement;
+      if (best_index) {
+        await exec_scope.query(best_index);
+      }
+    } catch (e) {}
+  }
 
   async query(query, query_scope = "cluster") {
     let scope = query_scope || this.query_scope;
-
-    // Escape the query string
-    const sanitizedQuery = couchbase.N1qlQuery.fromString(query).toString();
+    const sanitizedQuery = query; //sanitization step to be implemented
+    let exec_scope = null;
 
     if (scope === "cluster") {
-      return await this.cluster.query(sanitizedQuery);
+      exec_scope = this.cluster;
     } else {
-      return await this.scope.query(sanitizedQuery);
+      exec_scope = this.scope;
+    }
+    // Start time
+    let startTime = process.hrtime.bigint();
+    let response = await exec_scope.query(sanitizedQuery);
+    // End time
+    let endTime = process.hrtime.bigint();
+    // Calculate duration in microseconds
+    let durationInMilliseconds = (endTime - startTime) / BigInt(1000000);
+    // optimize if response time is greater than 1 second
+    if (durationInMilliseconds > 1000) {
+      await this.optimize(query, exec_scope);
     }
+    return response;
   }
 };