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

gabbyasuncion · gabbyasuncion · commit 5089f6f4f0b3 · 2021-11-03T11:07:15.000-04:00
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
 	}
 

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