withSuspense() HOC for React.lazy() + React.Suspense

withSuspense.js

import React from 'react';
import { CircularProgress, LinearProgress } from '@material-ui/core/';

/**
 * Wraps the React Component with React.Suspense and FallbackComponent while loading.
 * @param {React.Component} WrappedComponent - lazy loading component to wrap.
 * @param {React.Component} FallbackComponent - component to show while the WrappedComponent is loading.
 */
export const withSuspense = (WrappedComponent, FallbackComponent = null) => {
	return class extends React.Component {
		render() {
			if (!FallbackComponent) FallbackComponent = <LinearProgress />; // by default
			return (
				<React.Suspense fallback={FallbackComponent}>
					<WrappedComponent {...this.props} />
				</React.Suspense>
			);
		}
	};
};

...

// Usage
const lazySomeComponent = React.lazy(() => import('./xxx/SomeComponent'));

export const SomeComponent = withSuspense(lazySomeComponent);
export const SomeComponentWithCircularProgress = withSuspense(lazySomeComponent, <CircularProgress />);
export const SomeComponentWithDiv = withSuspense(lazySomeComponent, <div>Loading...</div>);

or for functional component :

---

import React from 'react';
import { CircularProgress, LinearProgress } from '@material-ui/core/';

/**

• Wraps the React Component with React.Suspense and FallbackComponent while loading.
• @param {React.Component} WrappedComponent - lazy loading component to wrap.
• @param {React.Component} FallbackComponent - component to show while the WrappedComponent is loading.
  */
  const withSuspense = (WrappedComponent, FallbackComponent = ) => {
  return (props) => (
  <React.Suspense fallback={FallbackComponent}>
  <WrappedComponent {...props} />
  </React.Suspense>
  );
  };

// Usage
const lazySomeComponent = React.lazy(() => import('./xxx/SomeComponent'));

export const SomeComponent = withSuspense(lazySomeComponent);
export const SomeComponentWithCircularProgress = withSuspense(lazySomeComponent, );
export const SomeComponentWithDiv = withSuspense(lazySomeComponent,

Loading...

);