Skip to content

Commit 12083ad

Browse files
committed
Code review explicitly set the context
1 parent bce9f74 commit 12083ad

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

driver-reactive-streams/src/main/com/mongodb/reactivestreams/client/internal/gridfs/GridFSUploadPublisherImpl.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -252,8 +252,10 @@ private Mono<Long> createSaveChunksMono(final AtomicBoolean terminated) {
252252
.append("n", chunkIndex.getAndIncrement())
253253
.append("data", data);
254254

255-
return clientSession == null ? chunksCollection.insertOne(chunkDocument)
256-
: chunksCollection.insertOne(clientSession, chunkDocument);
255+
Mono<InsertOneResult> insertMono = clientSession == null
256+
? Mono.from(chunksCollection.insertOne(chunkDocument))
257+
: Mono.from(chunksCollection.insertOne(clientSession, chunkDocument));
258+
return insertMono.contextWrite(sink.contextView());
257259
})
258260
.subscribe(null, sink::error, () -> sink.success(lengthInBytes.get()));
259261
});

0 commit comments

Comments
 (0)