suport redux enhancer with extraEnhancers

Author: sorrycc

src/createDva.js

@@ -191,6 +191,13 @@ export default function createDva(createOpts) {
         'app.start: extraReducers is conflict with other reducers',
       );
 
+      // extra enhancers
+      const extraEnhancers = plugin.get('extraEnhancers');
+      invariant(
+        Array.isArray(extraEnhancers),
+        'app.start: extraEnhancers should be array',
+      );
+
       // create store
       const extraMiddlewares = plugin.get('onAction');
       const reducerEnhancer = plugin.get('onReducer');
@@ -206,6 +213,7 @@ export default function createDva(createOpts) {
       const enhancers = [
         applyMiddleware(...middlewares),
         devtools(),
+        ...extraEnhancers,
       ];
       const store = this._store = createStore(
         createReducer(),

src/plugin.js

@@ -12,6 +12,7 @@ class Plugin {
       onReducer: [],
       onEffect: [],
       extraReducers: [],
+      extraEnhancers: [],
     };
   }
 
@@ -21,7 +22,11 @@ class Plugin {
     for (const key in plugin) {
       if (Object.prototype.hasOwnProperty.call(plugin, key)) {
         invariant(hooks[key], `plugin.use: unknown plugin property: ${key}`);
-        hooks[key].push(plugin[key]);
+        if (key === 'extraEnhancers') {
+          hooks[key] = plugin[key];
+        } else {
+          hooks[key].push(plugin[key]);
+        }
       }
     }
   }

test/dva-test.js

@@ -114,4 +114,25 @@ describe('dva', () => {
     app._store.dispatch({ type: 'test' });
     expect(count).toEqual(3);
   });
+
+  it('opts.extraEnhancers', () => {
+    let count = 0;
+    const countEnhancer = storeCreator => (reducer, preloadedState, enhancer) => {
+      const store = storeCreator(reducer, preloadedState, enhancer);
+      const oldDispatch = store.dispatch;
+      store.dispatch = (action) => {
+        count += 1;
+        oldDispatch(action);
+      };
+      return store;
+    };
+    const app = dva({
+      extraEnhancers: [countEnhancer],
+    });
+    app.router(() => 1);
+    app.start();
+
+    // @@router/LOCATION_CHANGE
+    expect(count).toEqual(1);
+  });
 });