Fix async api no opt

dist/index.esm.js

@@ -100,6 +100,9 @@ function groupByType(recordData) {
 function sortByCount(groupData) {
   return groupData.count;
 }
+function isPromise(value) {
+  return value instanceof Promise;
+}
 
 let proxyed$1 = false;
 const env$2 = getEnv();
@@ -311,14 +314,7 @@ function proxyAPI() {
         checkWarningRules(type, "post");
         return result;
       } else {
-        updateMeta(type, (meta) => {
-          meta.parallelism++;
-        });
-        checkWarningRules(type);
-        const opt = args[0] || {};
-        const successRaw = opt.success;
-        const failRaw = opt.fail;
-        opt.success = function(...args2) {
+        let asyncSuccessCallbackProxy = function(args2) {
           recordData.endTime = +/* @__PURE__ */ new Date();
           recordData.duration = recordData.endTime - recordData.startTime;
           const postDataGens = getDataGenerators(type, "post");
@@ -332,9 +328,7 @@ function proxyAPI() {
           });
           checkWarningRules(type, "post");
           recordData = null;
-          successRaw && successRaw.apply(this, args2);
-        };
-        opt.fail = function(...args2) {
+        }, asyncFailCallbackProxy = function(args2) {
           const res = args2[0];
           if (env$1 === "ali") {
             recordData.errno = res.error;
@@ -350,10 +344,37 @@ function proxyAPI() {
           });
           checkWarningRules(type, "post");
           recordData = null;
-          failRaw && failRaw.apply(this, args2);
         };
-        args[0] = opt;
-        return original.apply(this, args);
+        updateMeta(type, (meta) => {
+          meta.parallelism++;
+        });
+        checkWarningRules(type);
+        const opt = args[0];
+        if (opt) {
+          const successRaw = opt.success;
+          const failRaw = opt.fail;
+          opt.success = function(...args2) {
+            asyncSuccessCallbackProxy(args2);
+            successRaw && successRaw.apply(this, args2);
+          };
+          opt.fail = function(...args2) {
+            asyncFailCallbackProxy(args2);
+            failRaw && failRaw.apply(this, args2);
+          };
+          args[0] = opt;
+          return original.apply(this, args);
+        }
+        const originalResult = original.apply(this, args);
+        if (isPromise(originalResult)) {
+          return originalResult.then((result) => {
+            asyncSuccessCallbackProxy([result]);
+            return Promise.resolve(result);
+          }).catch((err) => {
+            asyncFailCallbackProxy([err]);
+            return Promise.reject(err);
+          });
+        }
+        return originalResult;
       }
     };
     Object.defineProperty(envObj, type, {

src/proxyAPI.ts

@@ -1,4 +1,4 @@
-import { getEnvObj, getEnv } from './utils'
+import { getEnvObj, getEnv, isPromise } from './utils'
 import { addRecordData, updateMeta, checkWarningRules, getDataGenerators } from './monitor'
 import type { IAnyObject, StackConfig, RecordData } from './types'
 
@@ -148,10 +148,9 @@ export function proxyAPI() {
           meta.parallelism++
         })
         checkWarningRules(type)
-        const opt = args[0] || {}
-        const successRaw = opt.success
-        const failRaw = opt.fail
-        opt.success = function (...args: any[]) {
+        const opt = args[0]
+
+        function asyncSuccessCallbackProxy(args: any[]) {
           recordData.endTime = +new Date()
           recordData.duration = recordData.endTime - recordData.startTime
           const postDataGens = getDataGenerators(type, 'post')
@@ -166,10 +165,9 @@ export function proxyAPI() {
           checkWarningRules(type, 'post')
           // for gc
           recordData = null as unknown as RecordData
-          successRaw && successRaw.apply(this, args)
         }
 
-        opt.fail = function (...args: any[]) {
+        function asyncFailCallbackProxy(args: any[]) {
           const res = args[0]
           if (env === 'ali') {
             recordData.errno = res.error
@@ -186,10 +184,35 @@ export function proxyAPI() {
           checkWarningRules(type, 'post')
           // for gc
           recordData = null as unknown as RecordData
-          failRaw && failRaw.apply(this, args)
         }
-        args[0] = opt
-        return original.apply(this, args)
+
+        if (opt) {
+          const successRaw = opt.success
+          const failRaw = opt.fail
+          opt.success = function (...args: any[]) {
+            asyncSuccessCallbackProxy(args)
+            successRaw && successRaw.apply(this, args)
+          }
+          opt.fail = function (...args: any[]) {
+            asyncFailCallbackProxy(args)
+            failRaw && failRaw.apply(this, args)
+          }
+
+          args[0] = opt
+
+          return original.apply(this, args)
+        }
+        const originalResult = original.apply(this, args)
+        if (isPromise(originalResult)) {
+          return originalResult.then((result: any) => {
+            asyncSuccessCallbackProxy([result])
+            return Promise.resolve(result)
+          }).catch((err: any) => {
+            asyncFailCallbackProxy([err])
+            return Promise.reject(err)
+          })
+        }
+        return originalResult
       }
     }
 

src/utils/index.ts

@@ -64,5 +64,8 @@ export function sortByCount (groupData: GroupData) {
   return groupData.count
 }
 
+export function isPromise (value: any) {
+  return value instanceof Promise
+}