milkio 1.0.0-alpha.63 → 1.0.0-alpha.65

package/context/index.ts

@@ -10,6 +10,7 @@ export interface $context {
   http: ContextHttp<Record<any, any>>
   config: Readonly<Awaited<ReturnType<$types['configSchema']['get']>>>
   call: <Module extends Promise<{ default: Action<any> }>>(module: Module, params: Parameters<Awaited<Module>['default']['handler']>[1]) => Promise<ReturnType<Awaited<Module>['default']['handler']>>
+  onFinally: (handler: () => void | Promise<void>) => void
 }
 
 export interface ContextHttp<ParamsParsed = any> {
@@ -24,4 +25,4 @@ export interface ContextHttp<ParamsParsed = any> {
   response: MilkioHttpResponse
 }
 
-export type ContextCreatedHandler = (context: $context) => Promise<void> | void
+export type ContextCreatedHandler = (context: $context) => Promise<void> | void

package/execute/index.ts

@@ -23,7 +23,7 @@ export function __initExecuter(generated: GeneratedInit, runtime: any) {
         paramsType: 'string'
       }
     ),
-  ): Promise<{ executeId: string, headers: Headers, params: Record<any, unknown>, results: Results<any>, context: $context, meta: Readonly<$meta>, type: 'action' | 'stream', emptyResult: boolean, resultsTypeSafety: boolean }> => {
+  ): Promise<{ executeId: string, headers: Headers, params: Record<any, unknown>, results: Results<any>, context: $context, meta: Readonly<$meta>, type: 'action' | 'stream', emptyResult: boolean, resultsTypeSafety: boolean, finales: Array<() => void | Promise<void>> }> => {
     const executeId: string = options.createdExecuteId
     let headers: Headers
     if (!(options.headers instanceof Headers)) {
@@ -37,6 +37,9 @@ export function __initExecuter(generated: GeneratedInit, runtime: any) {
     }
     if (!('toJSON' in headers)) (headers as any).toJSON = () => headersToJSON(headers)
 
+    const finales: Array<any> = []
+    const onFinally = (handler: any) => finales.unshift(handler)
+
     let params: Record<any, unknown>
     if (options.paramsType === 'raw') {
       params = options.params
@@ -74,6 +77,7 @@ export function __initExecuter(generated: GeneratedInit, runtime: any) {
       executeId: options.createdExecuteId,
       config: runtime.runtime.config,
       call: (module: any, options: any) => __call(context, module, options),
+      onFinally: onFinally,
       _: runtime
     } as unknown as $context
     const results: Results<any> = { value: undefined }
@@ -116,7 +120,7 @@ export function __initExecuter(generated: GeneratedInit, runtime: any) {
 
     await runtime.emit('milkio:executeAfter', { executeId: options.createdExecuteId, logger: options.createdLogger, path: options.path, meta, context, results })
 
-    return { executeId, headers, params, results, context, meta, type: module.$milkioType, emptyResult, resultsTypeSafety }
+    return { executeId, headers, params, results, context, meta, type: module.$milkioType, emptyResult, resultsTypeSafety, finales }
   }
 
   const __call = async (context: $context, module: { default: any }, params?: any): Promise<any> => {

package/flow/index.ts

@@ -0,0 +1,35 @@
+export type MilkioFlow<T> = {
+    emit: (flow: T) => void;
+    next: () => Promise<T>;
+}
+
+export function createFlow<T extends unknown>(): MilkioFlow<T> {
+    const flows: Array<{
+        blank: boolean;
+        promise: Promise<T>;
+        resolve: (value?: T | PromiseLike<T>) => void;
+        reject: (reason?: any) => void;
+    }> = [];
+
+    return {
+        emit: (flow: T) => {
+            if (flows.at(-1)?.blank === true) {
+                flows.at(-1)!.resolve(flow);
+                return;
+            }
+            const resolvers = Promise.withResolvers<T>();
+            resolvers.resolve(flow);
+            flows.push({ ...resolvers, blank: false });
+        },
+        next: async () => {
+            if (flows.length === 0) {
+                const resolvers = Promise.withResolvers<T>();
+                flows.push({ ...resolvers, blank: true });
+            }
+            const flow = flows.at(0)!;
+            const result = await flow.promise;
+            flows.shift();
+            return result;
+        },
+    }
+}

package/index.ts

@@ -5,6 +5,7 @@ export * from './config'
 export * from './execute'
 export * from './bootstrap'
 export * from './event'
+export * from './flow'
 export * from './handler'
 export * from './world'
 export * from './command'
@@ -15,4 +16,4 @@ export * from './context'
 export * from './meta'
 export * from './step'
 export * from './listener'
-export * from './exception'
+export * from './exception'

package/listener/index.ts

@@ -66,6 +66,8 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
         'Content-Type': 'application/json',
       },
     }
+    
+    let finales: Array<() => void | Promise<void>> = []
 
     try {
       const http = (await (async () => {
@@ -110,6 +112,7 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
           params: http.params.string,
           paramsType: 'string',
         })
+        finales = executed.finales;
 
         if (response.body === '' && executed.results.value !== undefined) {
           if (executed.emptyResult) {
@@ -124,6 +127,14 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
         }
 
         await runtime.emit('milkio:httpResponse', { executeId, logger, path: http.path.string as string, http, context: executed.context })
+        
+        for (const handler of finales) {
+          try {
+            await handler()
+          } catch (error) {
+            logger.error('An error occurred inside onFinally.', error)
+          }
+        }
 
         runtime.runtime.request.delete(executeId)
         return new Response(response.body, response)
@@ -134,6 +145,18 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
         if (routeSchema === undefined) throw reject('NOT_FOUND', { path: http.path.string as string })
         if (routeSchema.type !== 'stream') throw reject('UNACCEPTABLE', { expected: 'stream', message: `Not acceptable, the Accept in the request header should be "application/json". If you are using the "@milkio/stargate" package, please remove \`type: "stream"\` to the execute options.` })
 
+
+        const handleClose = async () => {
+          runtime.runtime.request.delete(executeId)
+          for (const handler of finales) {
+            try {
+              await handler()
+            } catch (error) {
+              logger.error('An error occurred inside onFinally.', error)
+            }
+          }
+        }
+
         const executed = await executer.__execute(routeSchema, {
           createdExecuteId: executeId,
           createdLogger: logger,
@@ -143,6 +166,7 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
           params: http.params.string,
           paramsType: 'string',
         })
+        finales = executed.finales;
         let stream: ReadableStream
         let control: ReadableStreamDirectController | ReadableStreamDefaultController
 
@@ -161,6 +185,7 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
                   }
                   else {
                     executed.results.value.return(undefined)
+                    await handleClose()
                     controller.close()
                   }
                 }
@@ -172,9 +197,11 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
                 controller.write(`data:${JSON.stringify([TSON.encode(result), null])}\n\n`)
               }
               await new Promise(resolve => setTimeout(resolve, 0))
+              await handleClose()
               controller.close()
             },
-            cancel() {
+            async cancel() {
+              await handleClose()
               control.close()
             },
           })
@@ -193,6 +220,7 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
                   }
                   else {
                     executed.results.value.return(undefined)
+                    await handleClose()
                     controller.close()
                   }
                 }
@@ -203,10 +231,12 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
                 result[exception.code] = exception.reject
                 controller.enqueue(`data:${JSON.stringify([TSON.encode(result), null])}\n\n`)
               }
+              await handleClose()
               await new Promise(resolve => setTimeout(resolve, 0))
               controller.close()
             },
-            cancel() {
+            async cancel() {
+              await handleClose()
               control.close()
             },
           })
@@ -218,7 +248,6 @@ export function __initListener(generated: GeneratedInit, runtime: any, executer:
 
         await runtime.emit('milkio:httpResponse', { executeId, logger, path: http.path.string as string, http, context: executed.context })
 
-        runtime.runtime.request.delete(executeId)
         return new Response(response.body, response)
       }
     }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "milkio",
   "type": "module",
-  "version": "1.0.0-alpha.63",
+  "version": "1.0.0-alpha.65",
   "module": "index.ts",
   "scripts": {},
   "peerDependencies": {
@@ -13,7 +13,7 @@
     "chalk": "^5.3.0",
     "date-fns": "^4.1.0",
     "milkid": "2.0.7",
-    "typia": "6.11.3"
+    "typia": "*"
   },
   "devDependencies": {
     "@types/bun": "latest",