From 9be44e4e73490c460283ad73d0303486113c5649 Mon Sep 17 00:00:00 2001
From: EnixCoda <enixcoda@gmail.com>
Date: Tue, 11 Sep 2018 21:42:25 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20handle=20updater=E2=80=99s=20return=20va?=
 =?UTF-8?q?lue?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/driver/connect.js | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/driver/connect.js b/src/driver/connect.js
index 19f1902..8451576 100644
--- a/src/driver/connect.js
+++ b/src/driver/connect.js
@@ -22,11 +22,9 @@ function link(instance, sources) {
       // luckily I don't need them :)
       let [updater, callback] = args
       if (typeof updater === 'function') {
-        callback = updater
-        callback(instance.state, instance.props)
-      } else {
-        instance.setState(updater, callback)
+        updater = updater(instance.state, instance.props)
       }
+      instance.setState(updater, callback)
     }
   }