feat: 增加跨云函数的全局配置对象；

labring · Jun 11, 2021 · 8f753b8 · 8f753b8
1 parent f0c04ab
commit 8f753b8
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 20 deletions.
diff --git a/src/lib/db.ts b/src/lib/db.ts
@@ -9,8 +9,12 @@ export const accessor = new MongoAccessor(Config.db.database, Config.db.uri, {
 })
 
 accessor.setLogger(getLogger('server:db', 'warning'))
-
 accessor.init()
 
 // 获取 db 对象
-export const db = getDb(accessor)
+export const db = getDb(accessor)
+
+// 创建新 db 实例
+export function createDb() {
+    return getDb(accessor)
+}
diff --git a/src/lib/faas/invoke.ts b/src/lib/faas/invoke.ts
@@ -1,27 +1,10 @@
 import { FunctionEngine, scheduler } from "."
-import { db } from '../../lib/db'
+import { createDb, db } from '../../lib/db'
 import { LocalFileStorage } from "../storage/local_file_storage"
 import request from 'axios'
 import Config from "../../config"
 import { CloudFunctionStruct, CloudSdkInterface, FunctionContext } from "./types"
 
-/**
- * 创建云函数 cloud sdk
- * @returns 
- */
-function createCloudSdk(): CloudSdkInterface {
-
-  const less: CloudSdkInterface = {
-    database: () => db,
-    storage: (namespace: string) => new LocalFileStorage(Config.LOCAL_STORAGE_ROOT_PATH, namespace),
-    fetch: request,
-    invoke: _invokeInFunction,
-    emit: (event: string, param: any) => scheduler.emit(event, param)
-  }
-
-  return less
-}
-
 /**
  * 调用云函数
  */
@@ -80,6 +63,28 @@ export async function getCloudFunctionById(func_id: string): Promise<CloudFuncti
 }
 
 
+
+// 跨请求、跨函数的全局配置对象，单例（in memory）
+const _shared_preference = new Map<string, any>()
+
+/**
+ * 创建云函数 cloud sdk
+ * @returns 
+ */
+function createCloudSdk(): CloudSdkInterface {
+
+  const less: CloudSdkInterface = {
+    database: () => createDb(),
+    storage: (namespace: string) => new LocalFileStorage(Config.LOCAL_STORAGE_ROOT_PATH, namespace),
+    fetch: request,
+    invoke: _invokeInFunction,
+    emit: (event: string, param: any) => scheduler.emit(event, param),
+    shared: _shared_preference
+  }
+
+  return less
+}
+
 /**
  * 在云函数中[调用云函数]的函数
  */

diff --git a/src/lib/faas/trigger.ts b/src/lib/faas/trigger.ts
@@ -100,6 +100,8 @@ export class TriggerScheduler {
     this._triggers = await this.loadTriggers()
 
     this.scheduleTimer()
+
+    this.emit('triggers.init')
   }
 
   public destroy() {

diff --git a/src/lib/faas/types.ts b/src/lib/faas/types.ts
@@ -16,6 +16,7 @@ export interface CloudSdkInterface {
   database(): Db,
   invoke: InvokeFunctionType
   emit: EmitFunctionType
+  shared: Map<string, any>
 }
 
 // vm run context (global)