Black-Hole's Blog

In love

对Promise增加统一操作扩展

Posted at — Jan 6, 2018

前言

在ES6里，增加了Promise方法。而Promise的回调函数只有then和catch两种方法。

而后，Promise又添加了两种附加方法，当然需要自己去添加进去。

可以去看一下上文的链接去了解一下，或者去下面的链接，看下官方源码是如何实现的: done 和 finally

正文

但是却没有一个针对then和catch的统一操作。

如果在最后处理的时候，then和catch的代码处理逻辑差不多的情况下，可能就需要写两份差不多的代码量。

当然可能会有些人会把差不多的代码处理逻辑提取到一个函数里，但是这样也不太美观，这个时候如果有一个回调函数可以同时处理resolve和reject就好了。

我们可以对Promise方法添加一个原型函数。这个函数来捕获resolve和reject，然后进行处理返回，就好了。代码很简单，如下：

Promise.prototype.unified = function (callback) {
  this.then(
    data => callback(true, data),
    data => callback(false, data)
  )
}

使用的方法也很简单，我们先写一个没有统一操作的Promise代码：

let promise = new Promise(function(resolve, reject) {
  if (false){
    setTimeout(() => resolve('success'), 1000)
  } else {
    setTimeout(() => reject('error'), 1000)
  }
})

promise
  .then((data) => {
    console.log(
      state: true,
      data: data,
      msg: 'operation successful'
    )
  })
  .catch((data) => {
    console.log(
      state: false,
      data: data,
      msg: 'operation failed'
    )
  })

然后我们使用unified方法重写一个：

let promise = new Promise(function(resolve, reject) {
  if (false){
    setTimeout(() => resolve('success'), 1000)
  } else {
    setTimeout(() => reject('error'), 1000)
  }
})

promise.unified((state, data) => {
  const msg = state ? 'operation successful' : 'operation failed'
  console.log(
    state,
    data,
    msg
  )
})

是不是方法了很多，当然此方法属于代码耦合了。也请谨慎使用，不然后期维护会很麻烦。

comments powered by Disqus