GODRIVER-2204: Add chunkIndex workaround to fillBuffer (#783)

mongodb · Nov 3, 2021 · 5089f6f · 5089f6f
1 parent 1715097
commit 5089f6f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/mongo/gridfs/download_stream.go b/mongo/gridfs/download_stream.go
@@ -242,7 +242,14 @@ func (ds *DownloadStream) fillBuffer(ctx context.Context) error {
 		return err
 	}
 
-	if chunkIndex.Int32() != ds.expectedChunk {
+	var chunkIndexInt32 int32
+	if chunkIndexInt64, ok := chunkIndex.Int64OK(); ok {
+		chunkIndexInt32 = int32(chunkIndexInt64)
+	} else {
+		chunkIndexInt32 = chunkIndex.Int32()
+	}
+
+	if chunkIndexInt32 != ds.expectedChunk {
 		return ErrWrongIndex
 	}