DEVPROD-26636 Replace the parquet-go library with parquet-go/parquet-go (evergreen-ci#10013)

Author: malikchaya2

agent/command/results_utils.go

@@ -16,8 +16,7 @@ import (
 	"github.com/evergreen-ci/evergreen/model/testresult"
 	"github.com/evergreen-ci/pail"
 	"github.com/evergreen-ci/utility"
-	goparquet "github.com/fraugster/parquet-go"
-	"github.com/fraugster/parquet-go/floor"
+	"github.com/parquet-go/parquet-go"
 	"github.com/pkg/errors"
 )
 
@@ -145,10 +144,7 @@ func uploadParquet(ctx context.Context, credentials evergreen.S3Credentials, out
 	}
 	defer w.Close()
 
-	pw := floor.NewWriter(goparquet.NewFileWriter(w, goparquet.WithSchemaDefinition(task.ParquetTestResultsSchemaDef)))
-	defer pw.Close()
-
-	return errors.Wrap(pw.Write(results), "writing Parquet test results")
+	return errors.Wrap(parquet.Write(w, []testresult.ParquetTestResults{*results}), "writing Parquet test results")
 }
 
 func makeTestResultsInfo(t task.Task, displayTaskInfo *apimodels.DisplayTaskInfo) testresult.TestResultsInfo {

agent/command/results_utils_test.go

@@ -18,9 +18,8 @@ import (
 	"github.com/evergreen-ci/evergreen/testutil"
 	"github.com/evergreen-ci/pail"
 	"github.com/evergreen-ci/utility"
-	goparquet "github.com/fraugster/parquet-go"
-	"github.com/fraugster/parquet-go/floor"
 	"github.com/mongodb/grip/sometimes"
+	"github.com/parquet-go/parquet-go"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
@@ -194,9 +193,7 @@ func saveTestResults(t *testing.T, ctx context.Context, testBucket pail.Bucket,
 	require.NoError(t, err)
 	defer func() { assert.NoError(t, w.Close()) }()
 
-	pw := floor.NewWriter(goparquet.NewFileWriter(w, goparquet.WithSchemaDefinition(task.ParquetTestResultsSchemaDef)))
-	require.NoError(t, pw.Write(savedParquet))
-	require.NoError(t, pw.Close())
+	require.NoError(t, parquet.Write(w, []testresult.ParquetTestResults{savedParquet}))
 	require.NoError(t, db.Insert(ctx, testresult.Collection, tr))
 	require.NoError(t, svc.AppendTestResultMetadata(resultTestutil.MakeAppendTestResultMetadataReq(ctx, savedResults, tr.ID)))
 	return savedResults

cmd/load-smoke-data/load-smoke-data.go

@@ -19,11 +19,10 @@ import (
 	"github.com/evergreen-ci/evergreen/model/testresult"
 	"github.com/evergreen-ci/pail"
 	"github.com/evergreen-ci/utility"
-	goparquet "github.com/fraugster/parquet-go"
-	"github.com/fraugster/parquet-go/floor"
 	"github.com/mongodb/amboy"
 	"github.com/mongodb/amboy/queue"
 	"github.com/mongodb/grip"
+	"github.com/parquet-go/parquet-go"
 	"github.com/pkg/errors"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
@@ -144,9 +143,9 @@ func writeDummyTestResultToLocalBucket(ctx context.Context) error {
 		CreatedAt: tr.CreatedAt,
 		Results:   make([]testresult.ParquetTestResult, 1),
 	}
-	pw := floor.NewWriter(goparquet.NewFileWriter(w, goparquet.WithSchemaDefinition(task.ParquetTestResultsSchemaDef)))
-	pw.Write(savedParquet)
-	pw.Close()
+	if err := parquet.Write(w, []testresult.ParquetTestResults{savedParquet}); err != nil {
+		return errors.Wrap(err, "writing dummy Parquet test results")
+	}
 	return nil
 }
 

go.mod

@@ -1,6 +1,6 @@
 module github.com/evergreen-ci/evergreen
 
-go 1.24.0
+go 1.24.9
 
 require (
 	github.com/99designs/gqlgen v0.17.86
@@ -171,12 +171,12 @@ require (
 	github.com/evergreen-ci/evg-lint v0.0.0-20251215145242-23eaa365e48f
 	github.com/evergreen-ci/plank v0.0.0-20251203163536-53406252f581
 	github.com/evergreen-ci/test-selection-client v0.0.0-20251016163227-83399b69e34c
-	github.com/fraugster/parquet-go v0.11.0
 	github.com/google/go-github/v70 v70.0.0
 	github.com/gorilla/csrf v1.7.3
 	github.com/gorilla/handlers v1.5.2
 	github.com/kanopy-platform/kanopy-oidc-lib v0.1.3
 	github.com/mongodb/jasper v0.0.0-20260115181313-e094cd64f89f
+	github.com/parquet-go/parquet-go v0.29.0
 	github.com/shirou/gopsutil/v3 v3.24.5
 	github.com/sirupsen/logrus v1.9.4
 	github.com/vikstrous/dataloadgen v0.0.10
@@ -186,8 +186,6 @@ require (
 )
 
 require (
-	github.com/apache/thrift v0.16.0 // indirect
-	github.com/araddon/dateparse v0.0.0-20210429162001-6b43995a97de // indirect
 	github.com/aws/aws-sdk-go-v2/service/signin v1.0.7 // indirect
 	github.com/aws/aws-sdk-go-v2/service/sns v1.39.8 // indirect
 	github.com/cenkalti/backoff/v5 v5.0.3 // indirect
@@ -209,7 +207,10 @@ require (
 	github.com/moby/sys/userns v0.1.0 // indirect
 	github.com/montanaflynn/stats v0.7.1 // indirect
 	github.com/okta/okta-jwt-verifier-golang/v2 v2.1.1 // indirect
+	github.com/parquet-go/bitpack v1.0.0 // indirect
+	github.com/parquet-go/jsonlite v1.0.0 // indirect
 	github.com/segmentio/asm v1.2.1 // indirect
+	github.com/twpayne/go-geom v1.6.1 // indirect
 	github.com/urfave/cli/v3 v3.6.1 // indirect
 	go.opentelemetry.io/auto/sdk v1.2.1 // indirect
 	go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.39.0 // indirect
@@ -236,6 +237,4 @@ require (
 	gotest.tools/v3 v3.5.1 // indirect
 )
 
-replace github.com/fraugster/parquet-go => github.com/evergreen-ci/parquet-go v0.0.0-20260116211725-cd13d4127a88
-
 tool github.com/99designs/gqlgen