Tweak API, add (de)hydrate

Author: gaearon

src/Dispatcher.js

@@ -3,17 +3,22 @@ function dispatch(store, atom, action) {
 }
 
 export default class Dispatcher {
-  constructor(store, atom) {
+  constructor(store) {
     this.store = store;
+    this.hydrate();
+  }
+
+  hydrate({ atom, subscriptions = [] } = {}) {
     this.atom = atom;
-    this.subscriptions = [];
+    this.subscriptions = subscriptions;
     this.dispatch({});
   }
 
-  receive(dispatcher) {
-    this.atom = dispatcher.atom;
-    this.subscriptions = dispatcher.subscriptions;
-    this.dispatch({});
+  dehydrate() {
+    const { atom, subscriptions } = this;
+    delete this.atom;
+    this.subscriptions = [];
+    return { atom, subscriptions };
   }
 
   dispatch(action) {
@@ -40,9 +45,4 @@ export default class Dispatcher {
     const { atom, subscriptions } = this;
     subscriptions.forEach(listener => listener(atom));
   }
-
-  dispose() {
-    this.atom = undefined;
-    this.subscriptions = [];
-  }
 }

src/components/Provider.js

@@ -20,12 +20,9 @@ export default class Provider {
   }
 
   componentWillReceiveProps(nextProps) {
-    nextProps.dispatcher.receive(this.props.dispatcher);
-    this.props.dispatcher.dispose();
-  }
-
-  componentWillUnmount() {
-    this.props.dispatcher.dispose();
+    nextProps.dispatcher.hydrate(
+      this.props.dispatcher.dehydrate()
+    );
   }
 
   subscribe(listener) {