Last active
June 11, 2020 15:53
-
-
Save alexandr-g/4eae32d32fe1e8112297682df194451c to your computer and use it in GitHub Desktop.
Setting up Apollo Client with `withApollo()` HOC for NextJS application
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// apollo/client.js | |
import React from 'react' | |
import Head from 'next/head' | |
import { ApolloProvider } from '@apollo/react-hooks' | |
import { ApolloClient } from 'apollo-client' | |
import { InMemoryCache } from 'apollo-cache-inmemory' | |
let globalApolloClient = null | |
/** | |
* Creates and provides the apolloContext | |
* to a next.js PageTree. Use it by wrapping | |
* your PageComponent via HOC pattern. | |
* @param {Function|Class} PageComponent | |
* @param {Object} [config] | |
* @param {Boolean} [config.ssr=true] | |
*/ | |
export function withApollo(PageComponent, { ssr = true } = {}) { | |
const WithApollo = ({ apolloClient, apolloState, ...pageProps }) => { | |
const client = apolloClient || initApolloClient(apolloState) | |
return ( | |
<ApolloProvider client={client}> | |
<PageComponent {...pageProps} /> | |
</ApolloProvider> | |
) | |
} | |
// Set the correct displayName in development | |
if (process.env.NODE_ENV !== 'production') { | |
const displayName = | |
PageComponent.displayName || PageComponent.name || 'Component' | |
if (displayName === 'App') { | |
console.warn('This withApollo HOC only works with PageComponents.') | |
} | |
WithApollo.displayName = `withApollo(${displayName})` | |
} | |
if (ssr || PageComponent.getInitialProps) { | |
WithApollo.getInitialProps = async (ctx) => { | |
const { AppTree } = ctx | |
// Initialize ApolloClient, add it to the ctx object so | |
// we can use it in `PageComponent.getInitialProp`. | |
const apolloClient = (ctx.apolloClient = initApolloClient()) | |
// Run wrapped getInitialProps methods | |
let pageProps = {} | |
if (PageComponent.getInitialProps) { | |
pageProps = await PageComponent.getInitialProps(ctx) | |
} | |
// Only on the server: | |
if (typeof window === 'undefined') { | |
// When redirecting, the response is finished. | |
// No point in continuing to render | |
if (ctx.res && ctx.res.finished) { | |
return pageProps | |
} | |
// Only if ssr is enabled | |
if (ssr) { | |
try { | |
// Run all GraphQL queries | |
const { getDataFromTree } = await import('@apollo/react-ssr') | |
await getDataFromTree( | |
<AppTree | |
pageProps={{ | |
...pageProps, | |
apolloClient, | |
}} | |
/> | |
) | |
} catch (error) { | |
// Prevent Apollo Client GraphQL errors from crashing SSR. | |
// Handle them in components via the data.error prop: | |
// https://www.apollographql.com/docs/react/api/react-apollo.html#graphql-query-data-error | |
console.error('Error while running `getDataFromTree`', error) | |
} | |
// getDataFromTree does not call componentWillUnmount | |
// head side effect therefore need to be cleared manually | |
Head.rewind() | |
} | |
} | |
// Extract query data from the Apollo store | |
const apolloState = apolloClient.cache.extract() | |
return { | |
...pageProps, | |
apolloState, | |
} | |
} | |
} | |
return WithApollo | |
} | |
/** | |
* Always creates a new apollo client on the server | |
* Creates or reuses apollo client in the browser. | |
* @param {Object} initialState | |
*/ | |
function initApolloClient(initialState) { | |
// Make sure to create a new client for every server-side request so that data | |
// isn't shared between connections (which would be bad) | |
if (typeof window === 'undefined') { | |
return createApolloClient(initialState) | |
} | |
// Reuse client on the client-side | |
if (!globalApolloClient) { | |
globalApolloClient = createApolloClient(initialState) | |
} | |
return globalApolloClient | |
} | |
/** | |
* Creates and configures the ApolloClient | |
* @param {Object} [initialState={}] | |
*/ | |
function createApolloClient(initialState = {}) { | |
const ssrMode = typeof window === 'undefined' | |
const cache = new InMemoryCache().restore(initialState) | |
// Check out https://github.com/vercel/next.js/pull/4611 if you want to use the AWSAppSyncClient | |
return new ApolloClient({ | |
ssrMode, | |
link: createIsomorphLink(), | |
cache, | |
}) | |
} | |
function createIsomorphLink() { | |
if (typeof window === 'undefined') { | |
const { SchemaLink } = require('apollo-link-schema') | |
const { schema } = require('./schema') | |
return new SchemaLink({ schema }) | |
} else { | |
const { HttpLink } = require('apollo-link-http') | |
return new HttpLink({ | |
uri: '/api/graphql', | |
credentials: 'same-origin', | |
}) | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment