archive-gh-me/zustand
1
0
Fork 0
mirror of https://github.com/pmndrs/zustand.git synced 2025-12-08 19:45:52 +00:00
Code Issues Projects Releases Wiki Activity

fix(persist): fix async migrate on persist middleware (#2877)

Browse Source 
* fix(perist): add support for async migrate on persist middleware

* feat(vitest): rename vitest.config.ts to vitest.config.mts

* fix: fix minor issues

* fix: fix minor issues

* feat: minor changes

* feat: revert vitest changes

---------

Co-authored-by: Daishi Kato <dai-shi@users.noreply.github.com>
This commit is contained in:
Danilo Britto 2024-12-04 08:06:36 -05:00 committed by GitHub
parent af7ff21190
commit 9bae06922a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/middleware/persist.ts
View File

@ -258,13 +258,14 @@ const persistImpl: PersistImpl = (config, baseOptions) => (set, get, api) => {
deserializedStorageValue.version !== options.version
) {
if (options.migrate) {
return [
true,
options.migrate(
deserializedStorageValue.state,
deserializedStorageValue.version,
),
] as const
const migration = options.migrate(
deserializedStorageValue.state,
deserializedStorageValue.version,
)
if (migration instanceof Promise) {
return migration.then((result) => [true, result] as const)
}
return [true, migration] as const
}
console.error(
`State loaded from storage couldn't be migrated since no migrate function was provided`,

4
tests/persistAsync.test.tsx
View File

@ -193,10 +193,10 @@ describe('persist middleware with async configuration', () => {
})
})
it('can migrate persisted state', async () => {
it('can async migrate persisted state', async () => {
const setItemSpy = vi.fn()
const onRehydrateStorageSpy = vi.fn()
const migrateSpy = vi.fn(() => ({ count: 99 }))
const migrateSpy = vi.fn(() => Promise.resolve({ count: 99 }))
const storage = {
getItem: async () =>

2
tests/persistSync.test.tsx
View File

@ -131,7 +131,7 @@ describe('persist middleware with sync configuration', () => {
expect(onRehydrateStorageSpy2).toBeCalledWith({ count: 42 }, undefined)
})
it('can migrate persisted state', () => {
it('can non-async migrate persisted state', () => {
const setItemSpy = vi.fn()
const onRehydrateStorageSpy = vi.fn()
const migrateSpy = vi.fn(() => ({ count: 99 }))