Moved to useLayoutEffect for subscription (#50)

Copied from #47 but for Preact module
storeon · May 10, 2019 · 64360cb · 64360cb
1 parent 2cc8382
commit 64360cb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/preact/index.js b/preact/index.js
@@ -2,13 +2,16 @@ var hooks = require('preact/hooks')
 
 var StoreContext = require('./context')
 
+var useIsomorphicLayoutEffect =
+  typeof window !== 'undefined' ? hooks.useLayoutEffect : hooks.useEffect
+
 module.exports = function () {
   var keys = [].slice.call(arguments)
 
   var store = hooks.useContext(StoreContext)
   var rerender = hooks.useState({ })
 
-  hooks.useEffect(function () {
+  useIsomorphicLayoutEffect(function () {
     return store.on('@changed', function (_, changed) {
       var changesInKeys = keys.some(function (key) {
         return key in changed