set state on onRehydrateStorage

[kuus]

Hi, looking at https://docs.pmnd.rs/zustand/integrations/persisting-store-data#how-can-i-check-if-my-store-has-been-hydrated
it seems it is possible to do this:

const useBoundStore = create(
  persist(
    (set, get) => ({
      // ...
      _hasHydrated: false,
      setHasHydrated: (state) => {
        set({
          _hasHydrated: state
        });
      }
    }),
    {
      // ...
      onRehydrateStorage: () => (state) => {
        state.setHasHydrated(true)
      }
    }
  )
);

but when the callback fires setHasHydrated is actually undefined, am I missing something?

[kuus]

Maybe related to #1527?

[kuus]

This is what I am simply trying to do

const _useUser = create<UserStore>()(
  persist<UserStore>(
    (set, _get) => ({
      _(partialState) {
        set(partialState);
      },
    }),
    {
      name: "user",
      onRehydrateStorage: (state) => {
        const { _ } = state;

        const asyncHydrate = async () => {
          _({ loading: true });

          const res = await api.get("User");
          if (res.ok) {
            // console.log("useUser:asyncHidrate with data", res.data);
            _({ user: res.data, loading: false });
          } else {
            _({ loading: false });
          }
        };
        asyncHydrate();

        return (_state, _error) => {
          _({ isReady: true });
        };
      },
    }
  )
);

but my private _ set state method is undefined

[dai-shi]

Maybe related to #1527?

Does it fail with only some specific versions?

[kuus]

I was on 4.2.0, now I updated to 4.3.2 and it happens the same

[kuus]

Compared to the issue 1527 the partialize is not involved here

[dai-shi]

it seems it is possible to do this:

If the code is identical from the docs, it can be a bug somewhere (in docs or in code).

[kuus]

my bad @dai-shi I forgot the first function
I was doing

onRehydrateStorage: (state) => {

instead of

onRehydrateStorage: () => (state) => {

[Karthik-B-06]

This worked for me. Thanks :) 🎉 🎉