Black-Hole's Blog

In love

对Promise增加统一操作扩展

Posted at — Jan 6, 2018

前言

在ES6里,增加了Promise方法。而Promise的回调函数只有thencatch两种方法。

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

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

正文

但是却没有一个针对thencatch的统一操作。

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

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

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

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
  )
})

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