state: abort list operation on context cancellation

ipfs-cluster · Jan 26, 2022 · cb57579 · cb57579
1 parent dca8923
commit cb57579
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/state/dsstate/datastore.go b/state/dsstate/datastore.go
@@ -142,6 +142,12 @@ func (st *State) List(ctx context.Context) ([]*api.Pin, error) {
 
 	total := 0
 	for r := range results.Next() {
+		// Abort if we shutdown.
+		select {
+		case <-ctx.Done():
+			return nil, ctx.Err()
+		default:
+		}
 		if r.Error != nil {
 			logger.Errorf("error in query result: %s", r.Error)
 			return pins, r.Error