import { createStore, applyMiddleware } from 'redux'
import rootReducer from 'your/path/reducers'
import { composeRootReducer, resetMiddleware, resetReduxState } from 'redux-reset-state'
 
// Some code is omitted here
 
const store = createStore(composeRootReducer(rootReducer), initialState, applyMiddleware(
  resetMiddleware
))
 
// If the shape of your application state is like this: { state: { key: { "state-key": 'value' } } }, 
// you can exec the following line of code to reset the state: 
resetReduxState([['state', 'key', 'state-key'], ['state', 'key']])
 

import React, { Component } from 'react'
import { resetReduxState } from 'redux-reset-state'
 
export default function (stateKeys) {
 
  return function wrapWithRedsetStateWhenUnmount(WrappedComponent) {
    class RedsetStateWhenUnmount extends Component {
      static displayName = `Reset(${JSON.stringify(stateKeys)})(${WrappedComponent.displayName || WrappedComponent.name || 'Component'})`
      
      componentWillUnmount() {
        resetReduxState(stateKeys)
      }
      
      render() {
        return <WrappedComponent resetReduxState={resetReduxState} {...this.props} />;
      }
    }
 
    return RedsetStateWhenUnmount;
  }
}