check ctx.Done in database.IndexDatabaseWithIterator

whosonfirst · Oct 5, 2021 · 6015543 · 6015543
1 parent 5e5e326
commit 6015543
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/database/iterator.go b/database/iterator.go
@@ -15,6 +15,13 @@ func IndexDatabaseWithIterator(ctx context.Context, db SpatialDatabase, iterator
 
 	iter_cb := func(ctx context.Context, path string, fh io.ReadSeeker, args ...interface{}) error {
 
+		select {
+		case <-ctx.Done():
+			return nil
+		default:
+			// pass
+		}
+
 		body, err := io.ReadAll(fh)
 
 		if err != nil {