Prevent sending data to closed lease channel

Azure · Sep 5, 2019 · 7dd2941 · 7dd2941
1 parent e1a121a
commit 7dd2941
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/changelog.md b/changelog.md
@@ -1,6 +1,7 @@
 # Change Log
 
 ## `head`
+- fix send on closed channel for GetLeases
 
 ## `v2.0.2`
 - enable partitionKey for sendBatch to fix [#128](https://github.com/Azure/azure-event-hubs-go/issues/128)

diff --git a/storage/storage.go b/storage/storage.go
@@ -192,6 +192,8 @@ func (sl *LeaserCheckpointer) GetLeases(ctx context.Context) ([]eph.LeaseMarker,
 	partitionIDs := sl.processor.GetPartitionIDs()
 	leaseCh := make(chan leaseGetResult)
 	defer close(leaseCh)
+	ctx, cancel := context.WithCancel(ctx)
+	defer cancel()
 
 	for _, partitionID := range partitionIDs {
 		go func(pID string) {