@abtnode/core 1.17.7-beta-20251227-001958-ea2ba3f5 → 1.17.7-beta-20251229-223813-e1e6c5e3

package/lib/states/blocklet-child.js

@@ -1,5 +1,6 @@
 /* eslint-disable no-underscore-dangle */
-const { BlockletStatus } = require('@blocklet/constant');
+const { Sequelize } = require('sequelize');
+const { BlockletStatus, BLOCKLET_INTERFACE_TYPE_SERVICE } = require('@blocklet/constant');
 const BaseState = require('./base');
 
 /**
@@ -247,6 +248,97 @@ class BlockletChildState extends BaseState {
     const [, [updated]] = await this.update({ parentBlockletId, childDid }, { $set: updates });
     return updated;
   }
+
+  /**
+   * Get aggregated status counts using efficient SQL GROUP BY
+   * This avoids loading all children into memory for counting
+   * @returns {Promise<{total: number, counts: Object}>} - Total count and counts by status
+   */
+  async getStatusCounts() {
+    try {
+      // Use raw SQL for efficient GROUP BY COUNT aggregation
+      // This is O(1) memory vs O(n) for loading all records
+      const results = await this.query(`
+        SELECT status, COUNT(*) as count
+        FROM blocklet_children
+        GROUP BY status
+      `);
+
+      const counts = {};
+      let total = 0;
+
+      for (const row of results) {
+        const { status, count: countStr } = row;
+        const count = parseInt(countStr, 10);
+        counts[status] = count;
+        total += count;
+      }
+
+      return { total, counts };
+    } catch (error) {
+      // Fallback: return empty counts on error
+      return { total: 0, counts: {} };
+    }
+  }
+
+  /**
+   * Get count of children by status value
+   * @param {number} status - The status value to count
+   * @returns {Promise<number>} - Count of children with the given status
+   */
+  countByStatus(status) {
+    return this.count({ status });
+  }
+
+  /**
+   * Get all port-related data from all children (optimized for _getOccupiedPorts)
+   * Only fetches necessary fields to minimize memory usage
+   * @returns {Promise<Array<{ports: Object, greenPorts: Object, meta: Object}>>}
+   */
+  async getAllPorts() {
+    const results = await this.model.findAll({
+      attributes: ['ports', 'greenPorts', 'meta'],
+      raw: true,
+    });
+    return results;
+  }
+
+  /**
+   * Get children that have SERVICE type interfaces (optimized for getServices)
+   * Uses database-level JSON filtering to reduce data transfer
+   * Works with both SQLite and PostgreSQL
+   * @returns {Promise<Array<{meta: Object, ports: Object, greenPorts: Object, greenStatus: number}>>}
+   */
+  async getChildrenWithServiceInterfaces() {
+    const dialect = this.model.sequelize.getDialect();
+
+    let condition;
+
+    if (dialect === 'postgres') {
+      // PostgreSQL: Use jsonb_array_elements to check interface types
+      condition = Sequelize.literal(`
+        EXISTS (
+          SELECT 1 FROM jsonb_array_elements(meta->'interfaces') AS iface
+          WHERE iface->>'type' = '${BLOCKLET_INTERFACE_TYPE_SERVICE}'
+        )
+      `);
+    } else {
+      // SQLite: Use json_each and json_extract for interface type check
+      condition = Sequelize.literal(`
+        EXISTS (
+          SELECT 1 FROM json_each(json_extract(meta, '$.interfaces'))
+          WHERE json_extract(value, '$.type') = '${BLOCKLET_INTERFACE_TYPE_SERVICE}'
+        )
+      `);
+    }
+
+    const results = await this.model.findAll({
+      attributes: ['meta', 'ports', 'greenPorts', 'greenStatus'],
+      where: condition,
+    });
+
+    return results.map((x) => x.toJSON());
+  }
 }
 
 module.exports = BlockletChildState;

package/lib/states/blocklet-extras.js

@@ -349,7 +349,7 @@ class BlockletExtrasState extends BaseState {
   }
 
   getWafDisabledBlocklets() {
-    return super.find({ where: { 'settings.gateway.wafPolicy.enabled': false } });
+    return super.find({ where: { 'settings.gateway.wafPolicy.enabled': false } }, { did: 1 });
   }
 }