Add a comment

gallery-so · Oct 20, 2023 · 6d1d44e · 6d1d44e
1 parent a59b27c
commit 6d1d44e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmd/dataloaders/generator/dataloader.go b/cmd/dataloaders/generator/dataloader.go
@@ -253,6 +253,9 @@ func (d *Dataloader[TKey, TResult]) addKeyToBatch(key TKey, jsonKey string) (*ba
 		}
 
 		b := actual.(*batch[TKey, TResult])
+
+		// Prevent lock contention within a batch by allowing only the first maxBatchSize callers
+		// to obtain the lock.
 		numAssigned := atomic.AddInt32(&b.numAssigned, 1)
 		if numAssigned > int32(d.maxBatchSize) {
 			atomic.CompareAndSwapInt32(&d.currentBatchID, currentID, currentID+1)