update mechanism

Weaverse · Sep 26, 2019 · 6c2ad06 · 6c2ad06
1 parent 9d673ca
commit 6c2ad06
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "global-state-hook",
-	"version": "1.1.1",
+	"version": "1.1.2",
 	"description": "~200kb for exactly what you need to manage global state",
 	"source": "src/index.ts",
 	"main": "dist/index.js",

diff --git a/src/index.ts b/src/index.ts
@@ -24,11 +24,11 @@ export interface IStateUpdater {
 }
 
 export function useSubscription(subscriber: ISubscription, pick: string[] = []): IStateUpdater {
-	const { state } = subscriber
+	let { state } = subscriber
 	const [, setUpdate] = React.useState()
 
 	const setState = React.useCallback((newState: State) => {
-		Object.assign(state, newState)
+		typeof newState === "object" ? Object.assign(state, newState) : (state = newState)
 		subscriber.listener.forEach(fn => fn(newState))
 	}, [])