feat(orap): add params for TaskFlow and add TaskFlowParams type for event flow

Author: murongg

packages/orap/src/flow/event.ts

@@ -1,7 +1,8 @@
 import type { AutoCrossCheckParam, Providers } from '@ora-io/reku'
 import type { Context } from '../task'
+import type { TaskFlowParams } from '../flow/task'
 import { TaskFlow } from '../flow/task'
-import type { StoreManager } from '../store'
+import { StoreManager } from '../store'
 import type { EventSignalRegisterParams } from '../signal'
 import { EventVerse } from '../verse/event'
 import type { TaskVerse } from '../verse/task'
@@ -37,13 +38,19 @@ export class EventFlow implements Flow {
     return this
   }
 
-  // task(store: Store, context?: Context): TaskFlow {
-  task(sm?: StoreManager, context?: Context): TaskFlow {
-    const tf = new TaskFlow(this)
-    if (sm)
-      tf.cache(sm)
-    if (context)
-      tf.context(context)
+  task(params: TaskFlowParams): TaskFlow
+  task(sm?: StoreManager | TaskFlowParams, context?: Context): TaskFlow {
+    let tf: TaskFlow
+    if (sm instanceof StoreManager) {
+      tf = new TaskFlow(this)
+      if (sm)
+        tf.cache(sm)
+      if (context)
+        tf.context(context)
+    }
+    else {
+      tf = new TaskFlow(this, sm)
+    }
     this._taskFlows.push(tf)
     return tf
   }

packages/orap/src/flow/task.ts

@@ -18,13 +18,26 @@ const defaultHandleFn: HandleFn = () => {
 const defaultToKeyFn: ToKeyFn = _ => randomStr(8, alphabetHex)
 export interface TaskFlowTTL { taskTtl: Milliseconds; doneTtl: Milliseconds }
 
+export interface TaskFlowParams {
+  context?: Context
+  taskPrefix?: Prefix
+  donePrefix?: Prefix
+  taskTtl?: Milliseconds
+  doneTtl?: Milliseconds
+  cache?: StoreManager
+  toKeyFn?: ToKeyFn
+  handleFn?: HandleFn
+  successFn?: HandleResultFn
+  failFn?: HandleResultFn
+}
+
 // TODO: add 'Failed-Task:' ?
 export class TaskFlow implements Flow {
   sm: StoreManager = new StoreManager(memoryStore())
   taskPrefix: Prefix = 'Task:'
   donePrefix: Prefix = 'Done-Task:'
-  taskTtl?: Milliseconds
-  doneTtl?: Milliseconds
+  taskTtl: Milliseconds = 60 * 1000
+  doneTtl: Milliseconds = 60 * 1000
   toKeyFn: ToKeyFn = defaultToKeyFn
   handleFn: HandleFn = defaultHandleFn
   successFn: HandleResultFn = defaultSuccessFn
@@ -39,7 +52,21 @@ export class TaskFlow implements Flow {
 
   constructor(
     private parentFlow: EventFlow,
-  ) { }
+    params?: TaskFlowParams,
+  ) {
+    params?.context && this.context(params?.context)
+    const taskPrefix = params?.taskPrefix ?? this.taskPrefix
+    const donePrefix = params?.donePrefix ?? this.donePrefix
+    this.prefix(taskPrefix, donePrefix)
+    const taskTtl = params?.taskTtl ?? this.taskTtl
+    const doneTtl = params?.doneTtl ?? this.doneTtl
+    this.ttl(taskTtl, doneTtl)
+    params?.cache && this.cache(params?.cache)
+    params?.toKeyFn && this.key(params?.toKeyFn)
+    params?.handleFn && this.handle(params?.handleFn)
+    params?.successFn && this.success(params?.successFn)
+    params?.failFn && this.fail(params?.failFn)
+  }
 
   get middlewares() {
     return this._middlewares