oox 0.0.7 → 0.2.0

package/global.js

@@ -1,13 +1,118 @@
-
-if ( Error.stackTraceLimit < 20 ) Error.stackTraceLimit = 20
-
-const Global = require ( './global.class' )
-
-/**
- * @type {Global}
- */
-const oox = global.oox
-
-module.exports = oox || new Global
-
-global.oox = module.exports
+
+if ( Error.stackTraceLimit < 20 ) Error.stackTraceLimit = 20
+
+const { AsyncLocalStorage } = require ( 'async_hooks' )
+
+const Context = require ( './rpc/context.class' )
+
+const RPC = require ( './rpc/rpc.interface.class' )
+
+const Socket = require ( './socketio/socket.class' )
+
+const SetMap = require ( './setMap.class' )
+
+const Middleware = require ( './middleware' )
+
+
+
+const Global = {
+
+
+
+    asyncStore: new AsyncLocalStorage ( ),
+
+
+
+    md: Middleware.handler,
+
+
+
+    /**
+     * @type {[RPC]}
+     */
+    instances: [ ],
+
+
+
+    /**
+     * @type {Map<String,Socket>}
+     */
+    sockets: new Map ( ),
+
+
+
+    /**
+     * @type {SocketIOServer[]}
+     */
+    socketIOServers: [ ],
+
+
+
+    /**
+     * @type {Map<String,Set<Socket>>}
+     */
+    socketIORegistry: new SetMap,
+
+
+
+    /**
+     * 生成随机不重复id
+     * @returns {String}
+     */
+    genTraceId ( ) {
+
+        const uid = [ 
+            Math.floor ( Date.now ( ) / 1000 ).toString ( 16 ),
+            Math.floor ( Math.random ( ) * 0xffffffff ).toString ( 16 ).padStart ( 8, '0' )
+        ]
+
+        return uid.join ( '' )
+    },
+
+
+
+    /**
+     * 获取链路跟踪上下文
+     * @param {Context} param0 
+     * @returns {Context}
+     */
+    genContext ( { caller, callerId, traceId, ip, sourceIP } = { } ) {
+
+        const context = new Context ( )
+
+        if ( caller ) {
+
+            context.caller = caller
+        } else {
+
+            const primaryService = this.instances [ 0 ]
+
+            if ( primaryService ) context.caller = primaryService.name
+        }
+
+        context.traceId = traceId || this.genTraceId ( )
+        context.ip = ip
+        context.sourceIP = sourceIP || ip
+        context.callerId = callerId
+
+        return context
+    },
+
+
+
+    /**
+     * 
+     * @returns {Context}
+     */
+    getContext ( ) {
+
+        const context = this.asyncStore.getStore ( )
+
+        return context || this.genContext ( )
+    },
+}
+
+/**
+ * @type {Global}
+ */
+global.oox = module.exports = global.oox || Global

package/index.js

@@ -1,9 +1,6 @@
-
-const Service = require ( './service/service.class' )
-
-const Global = require ( './global' )
-
-module.exports = {
-    Service,
-    Global
+
+const Service = require ( './service/service.class' )
+
+module.exports = {
+    Service
 }

package/middleware.js

@@ -0,0 +1,167 @@
+
+
+
+/**
+ * Map<中间件名字,中间件函数>
+ * @type {Map<String,Function>}
+ */
+exports.middlewares = new Map ( )
+
+
+
+/**
+ * WeakMap<殼函數，原函數>
+ * @type {Map<String,Function>}
+ */
+exports.wrappedActions = new Map ( )
+
+
+
+/**
+ * Map<中间件名字,Set<接口函数>>
+ * @type {Map<String,Set<Function>>}
+ */
+exports.middlewareActions = new Map ( )
+
+
+
+/**
+ * WeakMap<接口函数,[中间件名字]>
+ * @type {WeakMap<Function,[String]>}
+ */
+exports.actionMiddlewares = new WeakMap ( )
+
+
+
+/**
+ * @type {Map<String,Proxy>}
+ */
+exports.middlewareProxys = new Map ( )
+
+
+
+/**
+ * 
+ * @param {String} key 
+ * @param {Function} middleware 
+ * @returns {(Function)=>{}}
+ */
+exports.define = ( key, middleware ) => {
+
+    if ( middleware ) {
+
+        exports.middlewares.set ( key, middleware )
+
+        exports.middlewareActions.set ( key, new Set ( ) )
+
+        const proxy = new Proxy ( middleware, {
+
+            has ( target, action ) {
+
+                const sourceMethod = exports.wrappedActions.get ( action )
+
+                return exports.middlewareActions.get ( key ).has ( sourceMethod )
+            }
+        } )
+
+        // save proxy
+        exports.middlewareProxys.set ( key, proxy )
+    } else {
+
+        return arg => exports.define ( key, arg )
+    }
+}
+
+
+
+/**
+ * 
+ * @param {String} key 
+ * @param {Function} action 
+ * @param {'start'|'end'} place 
+ */
+exports.use = ( key, action, place='end' ) => {
+
+    if ( !exports.middlewareActions.has ( key ) ) exports.middlewareActions.set ( key, new Set ( ) )
+
+    if ( !exports.actionMiddlewares.has ( action ) ) exports.actionMiddlewares.set ( action, [ ] )
+
+    // 双向绑定
+
+    exports.middlewareActions.get ( key ).add ( action )
+
+    if ( place === 'end' ) {
+
+        exports.actionMiddlewares.get ( action ).push ( key )
+    } else {
+
+        exports.actionMiddlewares.get ( action ).unshift ( key )
+    }
+}
+
+
+
+exports.delete = ( key ) => {
+
+    exports.middlewares.delete ( key )
+
+    if ( exports.middlewareActions.has ( key ) ) {
+
+        const actions = exports.middlewareActions.get ( key ).values ( )
+
+        // remove action bindings
+        for ( const action of actions ) {
+
+            const middlewareNames = exports.actionMiddlewares.get ( action ).filter ( value => value !== key )
+
+            exports.actionMiddlewares.set ( action, middlewareNames )
+        }
+    }
+
+    exports.middlewareActions.get ( key ).clear ( )
+
+    exports.middlewareActions.delete ( key )
+}
+
+
+
+exports.handler = new Proxy ( o=>o, {
+
+
+
+    get ( target, key ) {
+
+        return exports.middlewareProxys.get ( key )
+    },
+
+
+
+    set ( target, key, action ) {
+
+        if ( exports.middlewares.has ( key ) ) {
+
+            exports.use ( key, action, 'start' )
+        } else {
+
+            exports.define ( key, action )
+        }
+
+        return true
+    },
+
+
+
+    deleteProperty ( target, key ) {
+
+        exports.delete ( key )
+
+        return true
+    },
+
+
+
+    apply ( target, thisArg, args ) {
+
+        return exports.define.apply ( exports, args )
+    }
+} )

package/package.json

@@ -1,17 +1,17 @@
-{
-  "name": "oox",
-  "version": "0.0.7",
-  "description": "distributed api service",
-  "main": "index.js",
-  "bin": {
-    "oox": "bin/cli.js"
-  },
-  "author": "lipingruan",
-  "homepage": "https://github.com/lipingruan/oox",
-  "license": "MIT",
-  "dependencies": {
-    "chalk": "^4.1.0",
-    "socket.io": "^4.0.2",
-    "socket.io-client": "^4.0.2"
-  }
-}
+{
+  "name": "oox",
+  "version": "0.2.0",
+  "description": "distributed api service",
+  "main": "index.js",
+  "bin": {
+    "oox": "bin/cli.js"
+  },
+  "author": "lipingruan",
+  "homepage": "https://github.com/lipingruan/oox",
+  "license": "MIT",
+  "dependencies": {
+    "chalk": "^4.1.0",
+    "socket.io": "^4.4.0",
+    "socket.io-client": "^4.4.0"
+  }
+}

package/rpc/config.class.js

@@ -1,48 +1,48 @@
-
-module.exports = class Config {
-
-
-
-    /**
-     * 服务名称
-     */
-    name = ''
-
-
-
-    /**
-     * 主机地址列表
-     */
-    host = ''
-
-
-
-    /**
-     * 跨域处理
-     */
-    origin = null
-
-
-
-    /**
-     * 网关地址
-     */
-    gateway = {
-
-
-
-        /**
-         * http 网关地址
-         * @type {{port:Number,path:String}}
-         */
-        http: null,
-
-
-
-        /**
-         * socketio 网关地址
-         * @type {{port:Number,path:String}}
-         */
-        socketio: null
-    }
+
+module.exports = class Config {
+
+
+
+    /**
+     * 服务名称
+     */
+    name = ''
+
+
+
+    /**
+     * 主机地址列表
+     */
+    host = ''
+
+
+
+    /**
+     * 跨域处理
+     */
+    origin = null
+
+
+
+    /**
+     * 网关地址
+     */
+    gateway = {
+
+
+
+        /**
+         * http 网关地址
+         * @type {{port:Number,path:String}}
+         */
+        http: null,
+
+
+
+        /**
+         * socketio 网关地址
+         * @type {{port:Number,path:String}}
+         */
+        socketio: null
+    }
 }

package/rpc/context.class.js

@@ -1,15 +1,15 @@
-
-
-module.exports = class Context {
-
-    // 请求溯源IP
-    sourceIP = ''
-    // 请求者IP
-    ip = ''
-    // 请求溯源ID
-    traceId = ''
-    // 请求者名称
-    caller = ''
-    // 请求者ID (长连接专用)
-    callerId = ''
+
+
+module.exports = class Context {
+
+    // 请求溯源IP
+    sourceIP = ''
+    // 请求者IP
+    ip = ''
+    // 请求溯源ID
+    traceId = ''
+    // 请求者名称
+    caller = ''
+    // 请求者ID (长连接专用)
+    callerId = ''
 }