From dd0821c49194a83bd04d41e06e55bfec14e99337 Mon Sep 17 00:00:00 2001
From: Tom Coleman <tom@thesnail.org>
Date: Tue, 14 Sep 2021 20:54:00 +1000
Subject: [PATCH] Ensure we don't clobber multiple source container state
 updates

---
 addons/docs/src/blocks/SourceContainer.tsx | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/addons/docs/src/blocks/SourceContainer.tsx b/addons/docs/src/blocks/SourceContainer.tsx
index b11e20135a4..4cae3680f96 100644
--- 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 };
+        setSources((current) => {
+          const newSources = { ...current, [id]: newItem };
 
-        if (!deepEqual(sources, newSources)) {
-          setSources(newSources);
-        }
+          if (!deepEqual(current, newSources)) {
+            return newSources;
+          }
+          return current;
+        });
       }
     };