Complete with old blocks, then subscribe for new

Cerebellum-Network · Mar 15, 2024 · 792dd79 · 792dd79
1 parent 2ae98a8
commit 792dd79
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/blockchain/client.go b/blockchain/client.go
@@ -116,10 +116,6 @@ func (c *Client) RegisterEventsListener(begin types.BlockNumber, callback Events
 		}
 	}
 
-	c.mu.Lock()
-	c.eventsListeners[idx] = callback
-	c.mu.Unlock()
-
 	stopped := false
 
 	go func() {
@@ -189,6 +185,12 @@ func (c *Client) RegisterEventsListener(begin types.BlockNumber, callback Events
 				callback(events, header.Number, set.Block)
 			}
 		}
+
+		c.mu.Lock()
+		if !stopped {
+			c.eventsListeners[idx] = callback
+		}
+		c.mu.Unlock()
 	}()
 
 	once := sync.Once{}