@midwayjs/piscina 4.0.0-beta.11 → 4.0.0-beta.12

package/dist/framework.js

@@ -44,11 +44,43 @@ let PiscinaWorkerFramework = class PiscinaWorkerFramework extends core_1.BaseFra
             throw new core_1.MidwayCommonError(`Task handler "${handler}" not found. Did you forget to use @PiscinaTask('${handler}') decorator?`);
         }
         const ctx = this.app.createAnonymousContext();
-        const taskInstance = await ctx.requestContext.getAsync(TaskClass);
-        if (!taskInstance || typeof taskInstance.execute !== 'function') {
-            throw new core_1.MidwayCommonError(`Task "${handler}" must implement execute method`);
-        }
-        return await taskInstance.execute(payload);
+        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: {
+                    handler,
+                },
+            })
+            : {};
+        return await traceService.runWithEntrySpan(`piscina ${handler}`, {
+            enable: traceEnabled,
+            carrier: entryCarrier,
+            attributes: {
+                'midway.protocol': 'piscina',
+                'midway.piscina.handler': handler,
+            },
+            meta: traceMetaResolver,
+            metaArgs: {
+                ctx,
+                carrier: entryCarrier,
+                request: payload,
+                custom: {
+                    handler,
+                },
+            },
+        }, async () => {
+            const taskInstance = await ctx.requestContext.getAsync(TaskClass);
+            if (!taskInstance || typeof taskInstance.execute !== 'function') {
+                throw new core_1.MidwayCommonError(`Task "${handler}" must implement execute method`);
+            }
+            return await taskInstance.execute(payload);
+        });
     }
     /**
      * 获取所有已注册的 handler

package/dist/worker-bootstrap.js

@@ -45,7 +45,6 @@ if (fullpath) {
             // 查找 worker 目录中的 tsconfig.json
             const workerTsConfig = path.join(baseDir, 'tsconfig.json');
             const hasWorkerTsConfig = fs.existsSync(workerTsConfig);
-            /* eslint-disable-next-line node/no-extraneous-require */
             require('ts-node').register(hasWorkerTsConfig
                 ? { project: workerTsConfig }
                 : {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@midwayjs/piscina",
-  "version": "4.0.0-beta.11",
+  "version": "4.0.0-beta.12",
   "description": "Midway Component for Piscina worker thread pool",
   "main": "dist/index.js",
   "typings": "index.d.ts",
@@ -28,12 +28,12 @@
   },
   "license": "MIT",
   "devDependencies": {
-    "@midwayjs/core": "^4.0.0-beta.11",
+    "@midwayjs/core": "^4.0.0-beta.12",
     "@midwayjs/logger": "^4.0.0",
-    "@midwayjs/mock": "^4.0.0-beta.11"
+    "@midwayjs/mock": "^4.0.0-beta.12"
   },
   "dependencies": {
     "piscina": "5.1.4"
   },
-  "gitHead": "6ef05719ca6e900f1ec34aff7a5c5a9614358c50"
+  "gitHead": "1c48179b7c827ba8ec9351e9b6c36ec1726d3e11"
 }