redux中间件探秘

发表于 更新于 分类于
对redux中间件的实现进行探索

从redux-thunk引出思考

在使用redux-thunk进行异步action书写的时候，我经常好奇redux到底如何运作，让asyncAction成为可能

为了探究，我们必须看一下redux-thunk的源码了。幸运的是redux-thunk的源码很少。。。至于为什么，下面立马讲解。

redux-thunk的源码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
// redux-thunk source code
function createThunkMiddleware(extraArgument) {
  return ({ dispatch, getState }) => next => action => {
    if (typeof action === 'function') {
      return action(dispatch, getState, extraArgument);
    }

    return next(action);
  };
}

const thunk = createThunkMiddleware();
thunk.withExtraArgument = createThunkMiddleware;

export default thunk;

从源码可以看出该中间件仅仅只是一个工厂函数，输出了一个嵌套工厂函数的工厂函数，那个最终参数带着next的返回函数，就是redux所需要适应的中间件。

以es6箭头语法看来可能比较麻烦，我们可以试着把这个代码直接转成es5的形式看看。

1
2
3
4
5
6
7
8
9
10
11
12
function createThunkMiddleware(extraArgument) {
  return function (storeOrFakeStore) {
    var dispatch = storeOrFakeStore.dispatch;
    var getState = storeOrFakeStore.getState;
    return function (next) {
      return function (action) {
          return action(dispatch, getState, extraArgument);
      }
      return next(action);
    }
  };
}

从这个源码可以看出，本身中间件是会接受当前store或者一个fakeStore（这个fakeStore可能仅仅只承载了store的两个api，dispatch和getState），并将dispatch和getState这两个store的方法传进可能执行异步操作的action函数里。这样，action完成异步操作以后，同样被赋予了dispatch的权利，就能够将状态通过action流转到下一个场景了。

那同学们又会问了，这个next是个啥？恩，其实这个next其实就是下一个要处理action的中间件，毕竟中间件是一个接着一个的对吧。如果大家写过koa或者express应该对这个next会熟悉很多。

接下来我们来看看react源码中的createStore模块是怎么应用中间件的。

我们使用createStore api创建store的时候发生了什么

applyMiddleware同样也是一个工厂，如果读者您用过redux中间件的话，你应该知道redux创建store是怎样创建的

1
2
3
4
const store = createStore(
  reducer,
  applyMiddleware(...middleware)
)

以下是createStore的源码，我们只看相关的一部分

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
export default function createStore(reducer, preloadedState, enhancer) {
  if (typeof preloadedState === 'function' && typeof enhancer === 'undefined') {
    enhancer = preloadedState
    preloadedState = undefined
  }

  if (typeof enhancer !== 'undefined') {
    if (typeof enhancer !== 'function') {
      throw new Error('Expected the enhancer to be a function.')
    }

    return enhancer(createStore)(reducer, preloadedState)
  }
  ...
}

结合store的声明和使用，我们可以知道redux的第三个参数可以接受一个叫做增强器的东西，如果存在增强器，则直接调用增强器方法，返回新的store并增强redux的功能。而使用中间件的时候，redux将存在的applyMiddleware工厂方法作为增强器应用在了redux上。

applyMiddleware api如何实现中间件操作的。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
export default function applyMiddleware(...middlewares) {
  return (createStore) => (reducer, preloadedState, enhancer) => {
    const store = createStore(reducer, preloadedState, enhancer)
    let dispatch = store.dispatch
    let chain = []

    const middlewareAPI = {
      getState: store.getState,
      dispatch: (action) => dispatch(action)
    }
    chain = middlewares.map(middleware => middleware(middlewareAPI))
    dispatch = compose(...chain)(store.dispatch)

    return {
      ...store,
      dispatch
    }
  }
}

applyMiddleware工厂函数对传入的中间件进行了compose操作，使中间件互相之间呈嵌套的形式，这样在中间件里的next函数就可以next()执行下去了。。。新的dispatch会从第一个中间件开始触发，这样，在我们调用store.dispatch的时候，就会将中间件走一遍了。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
// compose函数
export default function compose(...funcs) {
  if (funcs.length === 0) {
    return arg => arg
  }

  if (funcs.length === 1) {
    return funcs[0]
  }
  // 如果存在多个中间件，直接使用reduce方法将各个中间件嵌套起来。
  // 于是我们在使用中间件的时候就要注意了，中间件本质是一个拦截操作
  // 如果我们有两个中间件对某一个类型的action先后做了拦截，我们必须注意
  // 在createStore的时候插入中间件的顺序，中间件方法的执行是有序的。
  return funcs.reduce((a, b) => (...args) => a(b(...args)))
}

你也可以制作一个中间件

怎么样，redux的源码简单吗？简单到同学们也能自己开发中间件，现在大家自己也可以动手写自己的react中间件了。

“我，鸡哥，打钱!”