@midwayjs/one-shot 4.0.0-beta.10 → 4.0.0-beta.12

package/dist/framework.js

@@ -31,14 +31,46 @@ let MidwayOneShotFramework = class MidwayOneShotFramework extends core_1.BaseFra
             ...ctxData,
             payload,
         });
-        const fn = await this.applyMiddleware(async (ctx) => {
-            const instance = (await ctx.requestContext.getAsync(Runner));
-            if (!instance?.run) {
-                throw new core_1.MidwayCommonError('One-shot runner must implement run().');
-            }
-            return await instance.run(payload, ctx);
-        });
-        return (await fn(ctx));
+        const traceService = this.applicationContext.get(core_1.MidwayTraceService);
+        const traceMetaResolver = this.configurationOptions?.tracing?.meta;
+        const traceEnabled = this.configurationOptions?.tracing?.enable !== false;
+        const traceExtractor = this.configurationOptions?.tracing
+            ?.extractor;
+        const entryCarrier = typeof traceExtractor === 'function'
+            ? traceExtractor({
+                ctx,
+                request: payload,
+                custom: {
+                    runner: Runner.name || 'runner',
+                },
+            })
+            : {};
+        return (await traceService.runWithEntrySpan(`oneshot ${Runner.name || 'runner'}`, {
+            enable: traceEnabled,
+            carrier: entryCarrier,
+            attributes: {
+                'midway.protocol': 'one-shot',
+                'midway.oneshot.runner': Runner.name || 'runner',
+            },
+            meta: traceMetaResolver,
+            metaArgs: {
+                ctx,
+                carrier: entryCarrier,
+                request: payload,
+                custom: {
+                    runner: Runner.name || 'runner',
+                },
+            },
+        }, async () => {
+            const fn = await this.applyMiddleware(async (ctx) => {
+                const instance = (await ctx.requestContext.getAsync(Runner));
+                if (!instance?.run) {
+                    throw new core_1.MidwayCommonError('One-shot runner must implement run().');
+                }
+                return await instance.run(payload, ctx);
+            });
+            return await fn(ctx);
+        }));
     }
 };
 exports.MidwayOneShotFramework = MidwayOneShotFramework;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@midwayjs/one-shot",
-  "version": "4.0.0-beta.10",
+  "version": "4.0.0-beta.12",
   "description": "Midway One-shot Framework",
   "main": "dist/index.js",
   "typings": "index.d.ts",
@@ -23,14 +23,14 @@
     "lint": "mwts check"
   },
   "dependencies": {
-    "@midwayjs/core": "^4.0.0-beta.10"
+    "@midwayjs/core": "^4.0.0-beta.12"
   },
   "devDependencies": {
-    "@midwayjs/mock": "^4.0.0-beta.10"
+    "@midwayjs/mock": "^4.0.0-beta.12"
   },
   "engines": {
     "node": ">=20"
   },
   "license": "MIT",
-  "gitHead": "1b1856629913703f67304155aaf611ec936a81ac"
+  "gitHead": "1c48179b7c827ba8ec9351e9b6c36ec1726d3e11"
 }