ensure cursor is closed

rwynn · Nov 14, 2019 · 32a9d23 · 32a9d23
1 parent 6d019d8
commit 32a9d23
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gtm.go b/gtm.go
@@ -1193,8 +1193,9 @@ func GetCollectionInfo(ctx *OpCtx, client *mongo.Client, ns string) (info *Colle
 	}
 	var cursor *mongo.Cursor
 	cursor, err = client.Database(n.database).ListCollections(context.Background(), bson.M{"name": n.collection})
-	if err == nil {
-		if cursor.Next(context.Background()) {
+	if cursor != nil {
+		defer cursor.Close(context.Background())
+		if err == nil && cursor.Next(context.Background()) {
 			err = cursor.Decode(info)
 		}
 	}