nayota-energy-ms-sdk 0.0.1

package/README.md

@@ -0,0 +1,10 @@
+# Nayota服务端微服务SDK
+```
+yarn add nayota-energy-ms-sdk
+```
+## 使用方法
+```javascript
+import NayotaEnergyMsSdk from 'nayota-energy-ms-sdk'
+
+NayotaEnergyMsSdk.init(seneca, redis, role, version)
+```

package/config.js

@@ -0,0 +1,11 @@
+
+module.exports = {
+  setSeneca: function(seneca) {
+    this.seneca = seneca
+  },
+  setRedis: function(redis) {
+    this.redis = redis
+  },
+  role: '',
+  version: ''
+}

package/distribute.js

@@ -0,0 +1,9 @@
+/**
+ * @author lifeng
+ * @date 20-10-23
+ * @Description: 应用与服务全局的监听对象
+*/
+const EventEmitter = require('events').EventEmitter
+const listenBroker = new EventEmitter()
+listenBroker.setMaxListeners(50) // 增大单个监听的监听上限
+module.exports = new EventEmitter()

package/index.js

@@ -0,0 +1,41 @@
+const config = require('./config')
+const checks = require('./seneca/checks.seneca')
+const EnergyRecords = require('./redis/energyRecord.consumer')
+const distribute = require('./distribute')
+const redisObj = {}
+
+const sdk = {
+  init: function(seneca, redis, role, version) {
+    config.role = role
+    config.version = version
+    if (seneca)config.setSeneca(seneca)
+    if (redis) {
+      config.setRedis(redis)
+    }
+  },
+  seneca: { // 开放seneca接口
+    checks
+  },
+  on: (topic, cb) => {
+    switch (topic) {
+      case 'energyRecords':
+        if (!redisObj.energyRecords) {
+          redisObj.energyRecords = new EnergyRecords()
+          redisObj.energyRecords.start()
+        }
+        break
+      default:
+        break
+    }
+    distribute.on(topic, cb)
+  },
+  close: () => {
+    for (const key in redisObj) {
+      if (redisObj[key].stop) {
+        redisObj[key].stop()
+      }
+    }
+  }
+}
+
+module.exports = sdk

package/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "nayota-energy-ms-sdk",
+  "version": "0.0.1",
+  "description": "nayota-energy-server后台微服务",
+  "main": "index.js",
+  "author": "zhutianxiang",
+  "license": "MIT",
+  "dependencies": {
+    "bluebird": "^3.7.2"
+  }
+}

package/redis/base.js

@@ -0,0 +1,127 @@
+/**
+ * @author lifeng
+ * @date 23-11-17
+ * @Description: 消息队列流接口
+*/
+const config = require('../config')
+const em = require('../distribute')
+
+class Base {
+  constructor() {
+    this.stop = true
+    this.key = 'server:base'
+  }
+
+  async start() {
+    if (!config.redis) return
+    const client = config.redis
+    try {
+      await client.xGroupCreate(this.key, config.role, '0', { // 在一个流数据上建立一个分组  第一个参数是流的key,第二个参数是分组名 第三个参数，是从流的哪个ID开始取值，0是从开头，$是从结尾
+        MKSTREAM: true // 添加这个参数 则会判定如果流不存在 自动创建对应流 默认流不存在创建分组会报错
+      })
+      console.log(this.key, 'Created consumer group.')
+    } catch (e) {
+      console.log(this.key, 'Consumer group already exists, skipped creation.')
+    }
+    // console.log(`Starting consumer ${config.version}.`)
+    await this.xPending()
+    this.stop = false
+    while (!this.stop) {
+      try {
+        const response = await client.xReadGroup(
+          client.commandOptions({
+            isolated: true
+          }),
+          config.role, // 组名
+          config.version, // 消费者名 每一个消费者都应该有自己唯一的消费者名
+          [
+            {
+              key: this.key,
+              id: '>' // Id有两种类型，一种为>符号,代表只读取新消息（从未读取过的）,任何其他Id,会读取到已经被读取过但未确认处理的
+            }
+          ], {
+            // Read 1 entry at a time, block for 5 seconds if there are none.
+            COUNT: 1, // 一次最多读取几条消息
+            BLOCK: 5000 // 超时时间,超时返回空回复
+            // NOACK: true  // 读取到就确认  BLOCK NOACK 这两个参数只对id为>有效
+          }
+        )
+
+        if (response) {
+          if (response[0].messages[0]) {
+            const entryId = response[0].messages[0].id
+            await this.handle(response[0].messages[0].message) // 处理消息
+            await client.xAck(this.key, config.role, entryId) // 提交 消息消费成功 消息消费后必须提交，不然消息将处于已读取待处理状态 可通过XPENDING命令查看是否有待处理命令，然后提取，再次处理 也可以在读取消息的时候标记NOACK 表示此命令读取到就已处理，这样可能会造成偶尔的数据丢失
+          }
+        }
+      } catch (err) {
+        console.error(err)
+      }
+    }
+  }
+
+  // 检查是否有已读取过未处理消息
+  async xPending() {
+    if (!config.client) return
+    const client = config.redis
+    const response = await client.xPending( // 查询 是否有已读未确认消息。一般用于消息有问题或者当接受端的服务突然崩溃时，重新获取消息再处理
+      client.commandOptions({
+        isolated: true
+      }),
+      this.key, // 流key
+      config.role, // 分组名
+      config.version // 消费者名
+    )
+    if (response && response.pending > 0) {
+      this.getPending(response.firstId, response.lastId)
+    }
+  }
+
+  // 获取1条已读未确认消息
+  async getPending(firstId, lastId) {
+    if (!config.redis) return
+    console.log(firstId, lastId)
+    const client = config.redis
+    let currentId = null
+    while (currentId !== lastId) {
+      try {
+        const response = await client.xReadGroup(
+          client.commandOptions({
+            isolated: true
+          }),
+          config.role,
+          config.version,
+          [
+            {
+              key: this.key,
+              id: firstId
+            }
+          ], {
+            COUNT: 1
+          }
+        )
+        if (response && response[0].messages[0]) {
+          currentId = response[0].messages[0].id
+          await this.handle(response[0].messages[0].message) // 处理消息
+          await client.xAck(this.key, config.role, currentId)
+        } else {
+          break
+        }
+      } catch (err) {
+        console.error(err)
+        await client.xAck(this.key, config.role, firstId)
+      }
+    }
+  }
+
+  // 处理消息
+  async handle(message) {
+    em.emit(this.key.split(':').slice(-1)[0], message)
+  }
+
+  stop() {
+    this.stop = true
+  }
+}
+
+module.exports = Base

package/redis/energyRecord.consumer.js

@@ -0,0 +1,10 @@
+const Base = require('./base')
+
+class EnergyRecordConsumer extends Base {
+  constructor() {
+    super()
+    this.key = 'nayota-opoec-server:energyRecords'
+  }
+}
+
+module.exports = EnergyRecordConsumer

package/seneca/checks.seneca.js

@@ -0,0 +1,16 @@
+const seneca = require('../config')
+const { SUCCESS } = require('../seneca.error')
+const Promise = require('bluebird')
+
+module.exports = {
+  list
+}
+async function list(option) {
+  const act = Promise.promisify(seneca.seneca.act, { context: seneca.seneca })
+  const result = await act({ role: 'nayota-config-server-prod', cmd: 'check-all-user-data', data: option })
+  if (result.code !== SUCCESS) {
+    throw new Error(result.message)
+  }
+  return result.data
+}
+

package/seneca.error.js

@@ -0,0 +1,6 @@
+// ERROR CODE
+
+module.exports = {
+  SUCCESS: 0,
+  ERROR: 1
+}