@lark-apaas/client-toolkit 1.2.1-alpha.10 → 1.2.1-alpha.12

package/lib/utils/axiosConfig.js

@@ -131,22 +131,75 @@ async function logResponse(ok, responseOrError) {
 const requestStacktraceMap = new Map();
 function initAxiosConfig(axiosInstance) {
     if (!axiosInstance) axiosInstance = axios;
-    axiosInstance.interceptors.request.use((config)=>{
-        const method = (config.method || 'GET').toUpperCase();
-        const url = config.url || '';
-        const cleanedPath = url.split('?')[0].replace(/^\/spark\/p\/app_\w+/, '') || '/';
-        try {
-            const span = observable.startSpan(`${method} ${cleanedPath}`);
-            if (span) {
-                const traceInfo = `${span.spanContext().traceId}-${span.spanContext().spanId}`;
-                config.headers["X-Tt-TraceInfo"] = traceInfo;
-                config.__span = span;
-            }
-        } catch (error) {
-            console.error('Trace 配置失败:', error);
+    try {
+        if (axiosInstance && !axiosInstance._isTraced) {
+            axiosInstance._isTraced = true;
+            const methodsWithNoData = [
+                'get',
+                'delete',
+                'head',
+                'options'
+            ];
+            const methodsWithData = [
+                'post',
+                'put',
+                'patch',
+                "postForm",
+                "putForm",
+                "patchForm"
+            ];
+            const allMethods = [
+                'request',
+                ...methodsWithNoData,
+                ...methodsWithData
+            ];
+            allMethods.forEach((methodName)=>{
+                const originalMethod = axiosInstance[methodName];
+                if ('function' != typeof originalMethod) return;
+                axiosInstance[methodName] = function() {
+                    const args = Array.prototype.slice.call(arguments);
+                    try {
+                        let config;
+                        let url = '';
+                        let actualMethod = methodName.toUpperCase();
+                        if ('request' === methodName) if ('string' == typeof args[0]) {
+                            url = args[0];
+                            config = args[1] = args[1] || {};
+                        } else {
+                            config = args[0] = args[0] || {};
+                            url = config.url || '';
+                            actualMethod = (config.method || 'GET').toUpperCase();
+                        }
+                        else if (methodsWithNoData.includes(methodName)) {
+                            url = args[0];
+                            config = args[1] = args[1] || {};
+                        } else if (methodsWithData.includes(methodName)) {
+                            url = args[0];
+                            config = args[2] = args[2] || {};
+                        }
+                        if (config && !config.__span) {
+                            const cleanedPath = (url || '').split('?')[0].replace(/^\/spark\/p\/app_\w+/, '') || '/';
+                            const span = observable.startSpan(`${actualMethod} ${cleanedPath}`);
+                            if (span) {
+                                const spanContext = span.spanContext();
+                                if (spanContext && spanContext.traceId) {
+                                    const traceInfo = `${spanContext.traceId}-${spanContext.spanId}`;
+                                    if (!config.headers) config.headers = {};
+                                    config.headers['X-Tt-TraceInfo'] = traceInfo;
+                                    config.__span = span;
+                                }
+                            }
+                        }
+                    } catch (innerError) {
+                        console.error(`Trace wrapper inner error [${methodName}]:`, innerError);
+                    }
+                    return originalMethod.apply(this, args);
+                };
+            });
         }
-        return config;
-    });
+    } catch (e) {
+        console.error('Failed to init axios trace config wrapper:', e);
+    }
     axiosInstance.interceptors.response.use((response)=>{
         const { __span: span, _startTime: startTime, url = "" } = response.config || {};
         const method = (response.config.method || 'GET').toUpperCase();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lark-apaas/client-toolkit",
-  "version": "1.2.1-alpha.10",
+  "version": "1.2.1-alpha.12",
   "types": "./lib/index.d.ts",
   "main": "./lib/index.js",
   "files": [