Make use persistent in-memory state start with correct state & check …

…equality before setState
YSaxon · Apr 11, 2023 · c83e6c8 · c83e6c8
1 parent df510e1
commit c83e6c8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/web/hooks/use-persistent-in-memory-state.ts b/web/hooks/use-persistent-in-memory-state.ts
@@ -1,10 +1,13 @@
 import { safeJsonParse } from 'common/util/json'
-import { useEffect, useState } from 'react'
+import { useEffect } from 'react'
+import { useStateCheckEquality } from './use-state-check-equality'
 
 const store: { [key: string]: any } = {}
 
 export const usePersistentInMemoryState = <T>(initialValue: T, key: string) => {
-  const [state, setState] = useState<T>(initialValue)
+  const [state, setState] = useStateCheckEquality<T>(
+    safeJsonParse(store[key]) ?? initialValue
+  )
 
   useEffect(() => {
     const storedValue = safeJsonParse(store[key]) ?? initialValue