Ensure we don't clobber multiple source container state updates

storybookjs · Sep 14, 2021 · dd0821c · dd0821c
1 parent a308946
commit dd0821c
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/addons/docs/src/blocks/SourceContainer.tsx b/addons/docs/src/blocks/SourceContainer.tsx
@@ -21,11 +21,14 @@ export const SourceContainer: FC<{}> = ({ children }) => {
   useEffect(() => {
     const handleSnippetRendered = (id: StoryId, newItem: SourceItem) => {
       if (newItem !== sources[id]) {
-        const newSources = { ...sources, [id]: newItem };
-
-        if (!deepEqual(sources, newSources)) {
-          setSources(newSources);
-        }
+        setSources((current) => {
+          const newSources = { ...current, [id]: newItem };
+
+          if (!deepEqual(current, newSources)) {
+            return newSources;
+          }
+          return current;
+        });
       }
     };