From 83611a11f9de3cb1c568d3451b249ace1731ea6d Mon Sep 17 00:00:00 2001
From: "landong.lk" <landong.lk@antfin.com>
Date: Fri, 22 Feb 2019 16:09:16 +0800
Subject: [PATCH] fix: useSetState bug

---
 src/useSetState.ts | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/useSetState.ts b/src/useSetState.ts
index 2ec750a8..3e44e0fd 100644
--- a/src/useSetState.ts
+++ b/src/useSetState.ts
@@ -3,8 +3,7 @@ import {useState} from 'react';
 const useSetState = <T extends object>(initialState: T = {} as T): [T, (patch: Partial<T> | Function) => void]=> {
   const [state, set] = useState<T>(initialState);
   const setState = patch => {
-    if (patch instanceof Function) set(prevState => Object.assign({}, prevState, patch(prevState)));
-    else set(Object.assign({}, state, patch));
+    set(prevState => Object.assign({}, prevState, patch instanceof Function ? patch(prevState) : patch));
   };
 
   return [state, setState];