@abtnode/core 1.15.17 → 1.16.0-beta-8ee536d7

package/lib/api/node.js

@@ -6,97 +6,45 @@ const isDocker = require('@abtnode/util/lib/is-docker');
 const isGitpod = require('@abtnode/util/lib/is-gitpod');
 const getFolderSize = require('@abtnode/util/lib/get-folder-size');
 const canPackageReadWrite = require('@abtnode/util/lib/can-pkg-rw');
-const { NODE_PACKAGE_NAME, NODE_COMMAND_NAME } = require('@abtnode/constant');
+const { toDelegateAddress } = require('@arcblock/did-util');
 
 const logger = require('@abtnode/logger')('@abtnode/core:api:node');
 
 const IP = require('../util/ip');
-const { validateNodeInfo } = require('../validators/node');
-const BlockletRegistry = require('../blocklet/registry');
+const { validateNodeInfo, validateUpdateGateway } = require('../validators/node');
 const { getAll } = require('../blocklet/manager/engine');
+const { getDelegateState } = require('../util');
+const { NodeRuntimeMonitor } = require('../monitor/node-runtime-monitor');
+const getHistoryList = require('../monitor/get-history-list');
 
-const sanitizeUrl = (url) => {
-  if (!url) {
-    throw new Error('Registry URL should not be empty');
-  }
+// 10s 上报统计一次
+const MONITOR_RECORD_INTERVAL_SEC = 10;
 
-  return url.trim();
-};
+// 保存当天数据, 每天上报 8640 次
+const MONITOR_HISTORY_LENGTH = 86400 / MONITOR_RECORD_INTERVAL_SEC;
 
 class NodeAPI {
   /**
    *
    * @param {object} state Node StateDB
    */
-  constructor(state, blockletRegistry) {
+  constructor(state) {
     assert.notStrictEqual(state, undefined, 'argument state can not be undefined');
     assert.notStrictEqual(state, null, 'argument state can not be null');
 
-    this.state = state;
-    this.blockletRegistry = blockletRegistry;
-  }
-
-  // eslint-disable-next-line no-unused-vars
-  async addRegistry({ url }, context) {
-    logger.info('add registry', { url });
-    const sanitized = sanitizeUrl(url);
-    const info = await this.state.read();
-    const exist = info.blockletRegistryList.find((x) => x.url === sanitized);
-    if (exist) {
-      throw new Error(`Blocklet registry already exist: ${sanitized}`);
-    }
-
-    await BlockletRegistry.validateRegistryURL(sanitized);
-
-    const newBlockletRegistry = await BlockletRegistry.getRegistryMeta(sanitized);
-    const newBlockletRegistryList = info.blockletRegistryList.map((x) => ({ ...x, selected: false }));
-    newBlockletRegistryList.push({ ...newBlockletRegistry, url: sanitized, selected: true, protected: false });
-
-    this.blockletRegistry.clearCache();
-    return this.state.updateNodeInfo({ blockletRegistryList: newBlockletRegistryList });
-  }
-
-  // eslint-disable-next-line no-unused-vars
-  async deleteRegistry({ url }, context) {
-    logger.info('delete registry', { url });
-    const sanitized = sanitizeUrl(url);
-    const info = await this.state.read();
-    const exist = info.blockletRegistryList.find((x) => x.url === sanitized);
-    if (!exist) {
-      throw new Error(`Blocklet registry does not exist: ${sanitized}`);
-    }
-
-    this.blockletRegistry.clearCache();
-
-    const blockletRegistryList = info.blockletRegistryList.filter((x) => x.url !== sanitized);
-    if (!blockletRegistryList.find((x) => x.selected)) {
-      blockletRegistryList[0].selected = true;
-    }
-    return this.state.updateNodeInfo({ blockletRegistryList });
-  }
-
-  // eslint-disable-next-line no-unused-vars
-  async selectRegistry({ url }, context) {
-    logger.info('select registry', { url });
-    const sanitized = sanitizeUrl(url);
-    const info = await this.state.read();
-    const exist = info.blockletRegistryList.find((x) => x.url === sanitized);
-    if (!exist) {
-      throw new Error(`Blocklet registry does not exist: ${sanitized}`);
-    }
-
-    this.blockletRegistry.clearCache();
-    return this.state.updateNodeInfo({
-      blockletRegistryList: info.blockletRegistryList.map((x) => ({ ...x, selected: x.url === sanitized })),
+    this.runtimeMonitor = new NodeRuntimeMonitor({
+      historyLength: MONITOR_HISTORY_LENGTH,
     });
+
+    this.state = state;
   }
 
   async updateNodeInfo(entity = {}, context) {
     await validateNodeInfo(entity, context);
 
-    if (entity.autoUpgrade) {
+    if (entity.autoUpgrade && process.env.NODE_ENV !== 'development') {
       try {
-        canPackageReadWrite(NODE_COMMAND_NAME, NODE_PACKAGE_NAME);
+        canPackageReadWrite(process.env.ABT_NODE_BINARY_NAME, process.env.ABT_NODE_PACKAGE_NAME);
       } catch (err) {
         throw new Error(`Auto upgrade is not supported for this node due to: ${err.message}`);
       }
@@ -115,7 +63,7 @@ class NodeAPI {
     const info = await this.state.read();
     const env = await this.state.getEnvironments();
     info.environments = Object.keys(env).map((x) => ({ key: x, value: env[x] }));
-    info.uptime = +new Date() - process.uptime() * 1000;
+    info.uptime = process.uptime() * 1000;
 
     return info;
   }
@@ -155,6 +103,56 @@ class NodeAPI {
 
     return res;
   }
+
+  async updateGateway(entity, context) {
+    const data = await validateUpdateGateway(entity, context);
+
+    return this.state.updateGateway(data);
+  }
+
+  async getDelegationState() {
+    const info = await this.state.read();
+
+    const ownerNft = info.ownerNft || {};
+    if (!ownerNft.did) {
+      throw new Error('Invalid owner NFT');
+    }
+
+    const address = toDelegateAddress(ownerNft.holder, info.did);
+    const state = await getDelegateState(info.launcher.chainHost, address);
+    if (!state) {
+      return { delegated: false };
+    }
+
+    const transferV2Delegation = (state.ops || []).find(({ key }) => key === 'fg:t:transfer_v2');
+    return { delegated: !!transferV2Delegation };
+  }
+
+  // eslint-disable-next-line no-unused-vars
+  getHistory({ hours = 1 } = {}, context) {
+    const history = this.runtimeMonitor.getHistory();
+
+    return getHistoryList({
+      history,
+      hours,
+      recordIntervalSec: MONITOR_RECORD_INTERVAL_SEC,
+      props: ['date', 'cpu', 'mem', 'daemonMem', 'serviceMem', 'dbMem'],
+    });
+  }
+
+  getRealtimeData() {
+    return this.runtimeMonitor.getRealtimeData();
+  }
+
+  getCrons() {
+    return [
+      {
+        name: 'record-node-runtime-info',
+        time: `*/${MONITOR_RECORD_INTERVAL_SEC} * * * * *`,
+        fn: () => this.runtimeMonitor.monit(),
+      },
+    ];
+  }
 }
 
 module.exports = NodeAPI;