render children from props instead of state Cc: Sebastian Cruz @Sebas…

…tiansc
krishnaglick · Apr 2, 2019 · 35442b3 · 35442b3
1 parent 7978843
commit 35442b3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -11,18 +11,18 @@ export function withContext (contextTypes = {}) {
     }
     static contextTypes = contextTypes
     state = {
-      elementToRender: this.props.children
+      hasError: false
     }
 
     componentDidCatch () {
       this.setState({
-        elementToRender: this.props.fallBack()
+        hasError: true
       })
     }
 
     render () {
       if (server) return server._render(this, ProvideContext)
-      return <div>{this.state.elementToRender}</div>
+      return <div>{this.state.hasError ? this.props.fallBack() : this.props.children}</div>
     }
   }