PCSM-261: Add truncate tag to handle floats in BSON

Author: chupe

topo/topo.go

@@ -127,11 +127,11 @@ type DBStats struct {
 // CollStats represents the result of the [GetCollStats].
 type CollStats struct {
 	// Count is the number of documents in the collection.
-	Count int64 `bson:"count"`
+	Count int64 `bson:"count,truncate"`
 	// Size is the total size of the collection.
-	Size int64 `bson:"size"`
+	Size int64 `bson:"size,truncate"`
 	// AvgObjSize is the average size of documents in the collection.
-	AvgObjSize int64 `bson:"avgObjSize"`
+	AvgObjSize int64 `bson:"avgObjSize,truncate"`
 }
 
 // SayHello runs the db.hello() command and returns the [Hello].

topo/topo_test.go

@@ -6,9 +6,33 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"go.mongodb.org/mongo-driver/v2/bson"
 	"go.mongodb.org/mongo-driver/v2/mongo"
 )
 
+func TestCollStats_DecodeFromFloat64(t *testing.T) {
+	t.Parallel()
+
+	input := bson.M{
+		"count":      0.0,
+		"size":       73179136.0,
+		"avgObjSize": 0.0,
+	}
+
+	data, err := bson.Marshal(input)
+	require.NoError(t, err)
+
+	var stats CollStats
+	err = bson.Unmarshal(data, &stats)
+	require.NoError(t, err)
+
+	assert.Equal(t, int64(0), stats.Count)
+	assert.Equal(t, int64(73179136), stats.Size)
+	assert.Equal(t, int64(0), stats.AvgObjSize)
+}
+
 func TestRunWithRetry_NonTransientError(t *testing.T) {
 	t.Parallel()