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
Show Stats 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 log = logger('system');
const config = persistConfigs[key as keyof typeof persistConfigs];
if (!config) {
throw new Error(`No unserialize config for slice "${key}"`);
const persistConfig = persistConfigs[key as keyof typeof persistConfigs];
if (!persistConfig) {
throw new Error(`No persist config for slice "${key}"`);
}
try {
const { initialState, migrate } = config;
const { initialState, migrate } = persistConfig;
const parsed = JSON.parse(data);
// strip out old keys
const stripped = pick(parsed, keys(initialState));
@ -134,12 +134,16 @@ const unserialize: UnserializeFunction = (data, key) => {
return transformed;
} catch (err) {
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) => {
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);
};