avoid null issue

districtr · Nov 11, 2024 · 9cff4fb · 9cff4fb
1 parent 11ad326
commit 9cff4fb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/app/src/app/store/mapStore.ts b/app/src/app/store/mapStore.ts
@@ -382,18 +382,20 @@ export const useMapStore = create(
     ),
     {
       diff: (pastState: Partial<MapStore>, currentState: Partial<MapStore>) => {
-        currentState.zoneAssignments.keys().forEach((geoid) => {
+        if (!currentState.zoneAssignments || !pastState.zoneAssignments) return pastState;
+        for (const geoid of currentState.zoneAssignments.keys()) {
           if (!pastState.zoneAssignments.has(geoid)) {
             pastState.zoneAssignments.set(geoid, null);
           }
-        });
+        }
         return pastState as Partial<MapStore>;
       },
       equality: (pastState, currentState) => {
         return (pastState.zoneAssignments === currentState.zoneAssignments) &&
         (pastState.zoneAssignments.size === currentState.zoneAssignments.size);
       },
       limit: 7,
+      // @ts-ignore: save only partial store
       partialize: (state: Partial<MapStore>) => {
         const { zoneAssignments } = state;
         return { zoneAssignments } as Partial<MapStore>;