app.module.ts

export class AppModule {
    public constructor(
      private readonly transferState: TransferState,
      private readonly store: Store<fromRoot.State>,
  ) {
      const isBrowser = this.transferState.hasKey<any>(NGRX_STATE);

      if (isBrowser) {
          this.onBrowser();
      } else {
          this.onServer();
      }
  }
  onServer() {

      this.transferState.onSerialize(NGRX_STATE, () => {
          let state;
          this.store.subscribe( ( saveState: any ) => {
              console.log('Set for browser', JSON.stringify(saveState));
              state = saveState;
          }).unsubscribe();

          return state;
      });
  }

  onBrowser() {
      const state = this.transferState.get<any>(NGRX_STATE, null);
      this.transferState.remove(NGRX_STATE);
      this.store.dispatch( { type: 'SET_ROOT_STATE', payload: state } );
      console.log('Got state from server', JSON.stringify(state));
  }
 }