亲宝软件园·资讯

展开

Proxy更优雅处理异常

zxg_神说要有光 人气:0

代码不会全部按照我们的预期运行,可能会有意料之外的情况,为了保证程序的健壮性,要进行异常处理。

比如一个对象的所有方法,都应该做异常处理,但是,如果每个方法都加 try catch 又太麻烦:

const obj = {
    aaa() {
        try {
            // aaa
        } catch(e) {
            // xxxx
        } 
    },
    bbb() {
        try {
            // bbb
        } catch(e) {
            // xxxx
        } 
    },
    ccc() {
        try {
            // ccc
        } catch(e) {
            // xxxx
        } 
    }
}

有没有一种方式既能对所有方法做异常处理,又不用重复写这么多次呢?

有,就是代理模式。

代理模式是通过对目标对象做一层包装,提供和目标对象同名的方法,最终的功能实现还是调用目标对象的方法,但可以额外添加一些职责,比如日志、权限等,透明地对目标对象做一些扩充。

比如 React 里的高阶组件就是代理模式的实现,可以透明的扩展被包装的组件的功能。

很明显,这里的异常处理,也可以用代理的方式来做。但不用完全自己实现,ES6 提供了 Proxy,可以基于它来实现。

定义 createProxy 方法来实现代理,创建一个 Proxy 对象,对目标对象 target 做一层包装,定义 get、set 时的处理:

function createProxy(target) {
    const proxy = createExceptionProxy();
    return new Proxy(target, {
        get: proxy,
        set: proxy
    });
}

function createExceptionProxy() {
    return (target, prop) => {
        if (!(prop in target)) {
            return;
        }

        if (typeof target[prop] === 'function') {
            return createExceptionZone(target, prop);
        }

        return target[prop];
    }
}

如果 target 不包含 prop,就返回空,否则返回对应的属性值 target[prop]。

如果属性值是函数,则做一层包装:

function createExceptionZone(target, prop) {
    return (...args) => {
        let result;
        ExceptionsZone.run(() => {
          result = target[prop](...args);
        });
        return result;
    };
}

最终的功能实现还是调用 target,传入参数,把调用结果作为代理方法的结果返回。

包装这一层的目的是为了做异常处理,也就是 ExceptionsZone.run 做的事情:

class ExceptionsZone {
    static exceptionHandler = new ExceptionHandler();

    static run(callback) {
      try {
        callback();
      } catch (e) {
        this.exceptionHandler.handle(e);
      }
    }
}

调用目标方法,并做 try catch,当出现异常的时候,用 ExceptionHandler 来处理。

这里的异常处理我们就简单打印下日志:

class ExceptionHandler {
    handle(exception) {
        console.log('记录错误:',exception.message, exception.stack);
    }
}

这样就实现了给目标对象的所有方法添加异常处理的目的。

测试下:

const obj = {
    name: 'guang',
    say() {
        console.log('Hi, I\'m ' + this.name);
    },
    coding() {
        //xxx
        throw new Error('bug');
    }
    coding2() {
        //xxx
        throw new Error('bug2');
    }
}

const proxy = createProxy(obj);

proxy.say();
proxy.coding();

这里的 coding、coding2 方法都会抛出异常,但并没有做异常处理,我们用代理给它加上:

我们成功地通过代理模式给对象方法添加了异常处理!

但是现在这样还是有问题的,比如我把 coding 方法改为 async 的就不行了:

那怎么办呢?能不能统一对异步和同步方法做代理呢?

确实没办法,因为没法区分方法是同步还是异步,而且这两种方法的调用方式也不同,但我们可以单独提供一个 runner 方法来运行这些异步逻辑:

class ExceptionsZone {
    static exceptionHandler = new ExceptionHandler();

    static async asyncRun(callback) {
      try {
        await callback();
      } catch (e) {
        this.exceptionHandler.handle(e);
      }
    }
}

然后这样运行:

(async function() {
    await ExceptionsZone.asyncRun(proxy.coding2);
})();

这样就能处理异步逻辑中的异常了:

我们通过代理的方式给对象的所有同步方法添加了异常处理,然后又提供了运行异步方法的 runner 函数,对异步的异常做了处理,结合这两种方式,优雅地给目标对象的所有方法加上了异常处理。

可能你会说,代理就代理,你定义这么多 class 干啥?

因为这段逻辑是我从 Nest.js 源码里摘出来的,它源码里就是这样来给对象添加异常处理的:

异步逻辑也是单独提供了个方法来运行:

我觉得这个透明给对象添加异常处理的方式很优雅,就把它从 Nest.js 源码里抽了出来。

完整代码如下:

function createProxy(target) {
    const proxy = createExceptionProxy();
    return new Proxy(target, {
        get: proxy,
        set: proxy
    });
}

function createExceptionProxy() {
    return (receiver, prop) => {
        if (!(prop in receiver)) {
            return;
        }

        if (typeof receiver[prop] === 'function') {
            return createExceptionZone(receiver, prop);
        }

        return receiver[prop];
    }
}

function createExceptionZone(receiver, prop) {
    return (...args) => {
        let result;
        ExceptionsZone.run(() => {
          result = receiver[prop](...args);
        });
        return result;
    };
}

class ExceptionHandler {
    handle(exception) {
        console.log('记录错误:',exception.message, exception.stack);
    }
}

class ExceptionsZone {
    static exceptionHandler = new ExceptionHandler();
  
    static run(callback) {
      try {
        callback();
      } catch (e) {
        this.exceptionHandler.handle(e);
      }
    }
  
    static async asyncRun(callback) {
      try {
        await callback();
      } catch (e) {
        this.exceptionHandler.handle(e);
      }
    }
}

const obj = {
    name: 'guang',
    say() {
        console.log('Hi, I\'m ' + this.name);
    },
    coding() {
        //xxx
        throw new Error('bug');
    },
    async coding2() {
        //xxx
        throw new Error('bug2');
    }
}

const proxy = createProxy(obj);

proxy.say();
proxy.coding();

(async function() {
    await ExceptionsZone.asyncRun(proxy.coding2);
})();

总结

为了保证健壮性,我们要对所有可能报错的代码添加异常处理,但是每个方法都添加 try catch 又太麻烦,所以我们利用 Proxy 实现了代理,透明的给对象的所有方法都添加上了异常处理。

但是,代理添加的只是同步的异常处理,并没有捕获异步逻辑的异常,我们可以单独定义一个函数来运行异步方法。

结合代理 + 提供运行异步方法的 runner 这两种方式,就能给一个没有做任何异常处理的对象加上异常处理。是不是很优雅~

加载全部内容

相关教程
猜你喜欢
用户评论