Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): better error handling for persist serialize function

Browse Source
This commit is contained in:
psychedelicious
2024-02-03 20:53:33 +11:00
committed by Kent Keirsey
parent d713620d9e
commit 3205371654
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
invokeai/frontend/web/src/app/store/store.ts
View File

@ -109,12 +109,12 @@ const persistConfigs: { [key in keyof typeof allReducers]?: PersistConfig } = {
const unserialize: UnserializeFunction = (data, key) => { const unserialize: UnserializeFunction = (data, key) => {
const log = logger('system'); const log = logger('system');
const config = persistConfigs[key as keyof typeof persistConfigs]; const persistConfig = persistConfigs[key as keyof typeof persistConfigs];
if (!config) { if (!persistConfig) {
throw new Error(`No unserialize config for slice "${key}"`); throw new Error(`No persist config for slice "${key}"`);
} }
try { try {
const { initialState, migrate } = config; const { initialState, migrate } = persistConfig;
const parsed = JSON.parse(data); const parsed = JSON.parse(data);
// strip out old keys // strip out old keys
const stripped = pick(parsed, keys(initialState)); const stripped = pick(parsed, keys(initialState));
@ -134,12 +134,16 @@ const unserialize: UnserializeFunction = (data, key) => {
return transformed; return transformed;
} catch (err) { } catch (err) {
log.warn({ error: serializeError(err) }, `Error rehydrating slice "${key}", falling back to default initial state`); log.warn({ error: serializeError(err) }, `Error rehydrating slice "${key}", falling back to default initial state`);
return config.initialState; return persistConfig.initialState;
} }
}; };
export const serialize: SerializeFunction = (data, key) => { export const serialize: SerializeFunction = (data, key) => {
const result = omit(data, persistConfigs[key as keyof typeof persistConfigs]?.persistDenylist ?? []); const persistConfig = persistConfigs[key as keyof typeof persistConfigs];
if (!persistConfig) {
throw new Error(`No persist config for slice "${key}"`);
}
const result = omit(data, persistConfig.persistDenylist);
return JSON.stringify(result); return JSON.stringify(result);
}; };