File tree Expand file tree Collapse file tree 2 files changed +16
-7
lines changed Expand file tree Collapse file tree 2 files changed +16
-7
lines changed Original file line number Diff line number Diff line change @@ -621,6 +621,8 @@ async def create_collection(
621
621
and name in await self ._list_collection_names (filter = {"name" : name }, session = s )
622
622
):
623
623
raise CollectionInvalid ("collection %s already exists" % name )
624
+ if s :
625
+ s ._leave_alive = False
624
626
coll = AsyncCollection (
625
627
self ,
626
628
name ,
@@ -1206,10 +1208,12 @@ async def _list_collection_names(
1206
1208
if not filter or (len (filter ) == 1 and "name" in filter ):
1207
1209
kwargs ["nameOnly" ] = True
1208
1210
1209
- return [
1210
- result ["name" ]
1211
- async for result in await self ._list_collections_helper (session = session , ** kwargs )
1212
- ]
1211
+ cursor = await self ._list_collections_helper (session = session , ** kwargs )
1212
+ results = [result ["name" ] async for result in cursor ]
1213
+
1214
+ await cursor .close ()
1215
+
1216
+ return results
1213
1217
1214
1218
async def list_collection_names (
1215
1219
self ,
Original file line number Diff line number Diff line change @@ -621,6 +621,8 @@ def create_collection(
621
621
and name in self ._list_collection_names (filter = {"name" : name }, session = s )
622
622
):
623
623
raise CollectionInvalid ("collection %s already exists" % name )
624
+ if s :
625
+ s ._leave_alive = False
624
626
coll = Collection (
625
627
self ,
626
628
name ,
@@ -1200,9 +1202,12 @@ def _list_collection_names(
1200
1202
if not filter or (len (filter ) == 1 and "name" in filter ):
1201
1203
kwargs ["nameOnly" ] = True
1202
1204
1203
- return [
1204
- result ["name" ] for result in self ._list_collections_helper (session = session , ** kwargs )
1205
- ]
1205
+ cursor = self ._list_collections_helper (session = session , ** kwargs )
1206
+ results = [result ["name" ] for result in cursor ]
1207
+
1208
+ cursor .close ()
1209
+
1210
+ return results
1206
1211
1207
1212
def list_collection_names (
1208
1213
self ,
You can’t perform that action at this time.
0 commit comments