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

gabbyasuncion · web-flow · commit 6597bd7b8ea5 · 2021-11-02T10:17:29.000-04:00
diff --git a/mongo/gridfs/download_stream.go b/mongo/gridfs/download_stream.go
@@ -243,7 +243,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
 	}
 

Original file line number	Diff line number	Diff line change
`@@ -243,7 +243,14 @@ func (ds *DownloadStream) fillBuffer(ctx context.Context) error {`
`243`	`243`	`return err`
`244`	`244`	`}`
`245`	`245`
`246`		`- if chunkIndex.Int32() != ds.expectedChunk {`
	`246`	`+ var chunkIndexInt32 int32`
	`247`	`+ if chunkIndexInt64, ok := chunkIndex.Int64OK(); ok {`
	`248`	`+ chunkIndexInt32 = int32(chunkIndexInt64)`
	`249`	`+ } else {`
	`250`	`+ chunkIndexInt32 = chunkIndex.Int32()`
	`251`	`+ }`
	`252`	`+`
	`253`	`+ if chunkIndexInt32 != ds.expectedChunk {`
`247`	`254`	`return ErrWrongIndex`
`248`	`255`	`}`
`249`	`256`