@midwayjs/cos 4.0.0-beta.8 → 4.0.0

package/dist/manager.d.ts

@@ -1,10 +1,15 @@
-import { ServiceFactory } from '@midwayjs/core';
+import { ServiceFactory, MidwayTraceService } from '@midwayjs/core';
 import * as COS from 'cos-nodejs-sdk-v5';
 export declare class COSServiceFactory extends ServiceFactory<COS> {
     cosConfig: any;
     init(): Promise<void>;
     logger: any;
+    protected traceService: MidwayTraceService;
+    protected traceMetaResolver: any;
+    protected traceEnabled: any;
+    protected traceInjector: any;
     createClient(config: COS.COSOptions): Promise<COS>;
+    protected bindTraceContext(client: COS): void;
     getName(): string;
 }
 export declare class COSService implements COS {

package/dist/manager.js

@@ -21,10 +21,59 @@ let COSServiceFactory = class COSServiceFactory extends core_1.ServiceFactory {
         });
     }
     logger;
+    traceService;
+    traceMetaResolver;
+    traceEnabled;
+    traceInjector;
     async createClient(config) {
         assert.ok(config.SecretKey && config.SecretId, '[@midwayjs/cos] secretId secretKey is required on config');
         this.logger.info('[midway:cos] init %s', config.SecretKey);
-        return new COS(config);
+        const client = new COS(config);
+        this.bindTraceContext(client);
+        return client;
+    }
+    bindTraceContext(client) {
+        if (!client || !this.traceService) {
+            return;
+        }
+        const rawRequest = client.request?.bind(client);
+        if (!rawRequest) {
+            return;
+        }
+        client.request = async (...args) => {
+            if (typeof args[1] === 'function') {
+                return rawRequest(...args);
+            }
+            const params = args?.[0] || {};
+            const apiName = params.Action || params.action || 'request';
+            const rawCarrier = typeof this.traceInjector === 'function'
+                ? this.traceInjector({
+                    request: params,
+                    custom: {
+                        action: String(apiName),
+                    },
+                })
+                : {};
+            const carrier = rawCarrier && typeof rawCarrier === 'object' ? rawCarrier : {};
+            return await this.traceService.runWithExitSpan(`cos.${String(apiName).toLowerCase()}`, {
+                enable: this.traceEnabled !== false,
+                carrier,
+                attributes: {
+                    'midway.protocol': 'cos',
+                    'midway.cos.action': String(apiName),
+                },
+                meta: this.traceMetaResolver,
+                metaArgs: {
+                    carrier,
+                    request: params,
+                    custom: {
+                        action: String(apiName),
+                    },
+                },
+            }, async () => {
+                return await rawRequest(...args);
+            });
+        };
     }
     getName() {
         return 'cos';
@@ -45,6 +94,22 @@ __decorate([
     (0, core_1.Logger)('coreLogger'),
     __metadata("design:type", Object)
 ], COSServiceFactory.prototype, "logger", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", core_1.MidwayTraceService)
+], COSServiceFactory.prototype, "traceService", void 0);
+__decorate([
+    (0, core_1.Config)('cos.tracing.meta'),
+    __metadata("design:type", Object)
+], COSServiceFactory.prototype, "traceMetaResolver", void 0);
+__decorate([
+    (0, core_1.Config)('cos.tracing.enable'),
+    __metadata("design:type", Object)
+], COSServiceFactory.prototype, "traceEnabled", void 0);
+__decorate([
+    (0, core_1.Config)('cos.tracing.injector'),
+    __metadata("design:type", Object)
+], COSServiceFactory.prototype, "traceInjector", void 0);
 exports.COSServiceFactory = COSServiceFactory = __decorate([
     (0, core_1.Provide)(),
     (0, core_1.Scope)(core_1.ScopeEnum.Singleton)

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@midwayjs/cos",
   "description": "midway cos component",
-  "version": "4.0.0-beta.8",
+  "version": "4.0.0",
   "main": "dist/index.js",
   "typings": "index.d.ts",
   "files": [
@@ -10,8 +10,8 @@
     "index.d.ts"
   ],
   "devDependencies": {
-    "@midwayjs/core": "^4.0.0-beta.8",
-    "@midwayjs/mock": "^4.0.0-beta.8",
+    "@midwayjs/core": "^4.0.0",
+    "@midwayjs/mock": "^4.0.0",
     "dotenv": "16.4.7"
   },
   "dependencies": {
@@ -37,5 +37,5 @@
     "type": "git",
     "url": "https://github.com/midwayjs/midway.git"
   },
-  "gitHead": "355e55949fdd132b0bdcb4830222a0a027e92ded"
+  "gitHead": "014f32c23ebc1d5ac21777c76be2fd373ce992d8"
 }