Added Integer size check to the download publisher

rozza · rozza · commit 5640ad5aad67 · 2020-03-10T12:11:44.000Z
JAVA-3650
diff --git a/driver-reactive-streams/src/main/com/mongodb/reactivestreams/client/internal/GridFSDownloadPublisherImpl.java b/driver-reactive-streams/src/main/com/mongodb/reactivestreams/client/internal/GridFSDownloadPublisherImpl.java
@@ -223,7 +223,9 @@ private void tryProcess() {
                     }
 
                     int byteBufferSize = Math.max(chunkSize, bufferSizeBytes);
-                    byteBufferSize =  Math.min(Long.valueOf(remaining).intValue(), byteBufferSize);
+                    if (remaining < Integer.MAX_VALUE) {
+                        byteBufferSize = Math.min(Long.valueOf(remaining).intValue(), byteBufferSize);
+                    }
                     ByteBuffer byteBuffer = ByteBuffer.allocate(byteBufferSize);
 
                     if (currentBatchSize == 0) {

Original file line number	Diff line number	Diff line change
`@@ -223,7 +223,9 @@ private void tryProcess() {`
`223`	`223`	`}`
`224`	`224`
`225`	`225`	`int byteBufferSize = Math.max(chunkSize, bufferSizeBytes);`
`226`		`- byteBufferSize = Math.min(Long.valueOf(remaining).intValue(), byteBufferSize);`
	`226`	`+ if (remaining < Integer.MAX_VALUE) {`
	`227`	`+ byteBufferSize = Math.min(Long.valueOf(remaining).intValue(), byteBufferSize);`
	`228`	`+ }`
`227`	`229`	`ByteBuffer byteBuffer = ByteBuffer.allocate(byteBufferSize);`
`228`	`230`
`229`	`231`	`if (currentBatchSize == 0) {`