chore: add cockroachdb 26.1 to the matrix

Author: tstirrat15

.github/workflows/build-test.yaml

@@ -262,7 +262,11 @@ jobs:
       fail-fast: false
       matrix:
         datastore: ["crdb"]
-        crdbversion: ["25.2.0", "25.3.0"]
+        crdbversion: 
+          # the version many folks are on
+          - "25.2.0"
+          # the most recent version
+          - "26.1.0"
     steps:
       - uses: "actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8" # v4.2.2
         if: |
@@ -297,7 +301,11 @@ jobs:
       fail-fast: false
       matrix:
         datastore: ["crdb"]
-        crdbversion: ["25.2.0", "25.3.0"]
+        crdbversion:
+          # the version many folks are on
+          - "25.2.0"
+          # the most recent version
+          - "26.1.0"
     steps:
       - uses: "actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8" # v4.2.2
         if: |

internal/datastore/crdb/crdb_test.go

@@ -17,6 +17,7 @@ import (
 	"net"
 	"os"
 	"path/filepath"
+	"strings"
 	"testing"
 	"time"
 
@@ -112,6 +113,15 @@ func TestCRDBDatastoreWithoutIntegrity(t *testing.T) {
 		GCWindow(veryLargeGCWindow),
 		WithAcquireTimeout(5*time.Second),
 	))
+
+	t.Run("TestVersionReading", createDatastoreTest(
+		b,
+		// TODO
+		VersionReadingTest,
+		RevisionQuantization(0),
+		GCWindow(veryLargeGCWindow),
+		WithAcquireTimeout(5*time.Second),
+	))
 }
 
 type datastoreTestFunc func(t *testing.T, ds datastore.Datastore)
@@ -960,3 +970,32 @@ func TestRegisterPrometheusCollectors(t *testing.T) {
 	require.NotNil(t, poolReadMetric)
 	require.Equal(t, float64(readMaxConns), poolReadMetric.GetGauge().GetValue()) //nolint:testifylint // we expect exact values
 }
+
+func VersionReadingTest(t *testing.T, rawDS datastore.Datastore) {
+	require := require.New(t)
+
+	expectedVersionList := strings.Split(crdbTestVersion(), ".")
+	expectedMajor := expectedVersionList[0]
+	expectedMinor := expectedVersionList[1]
+	expectedPatch := expectedVersionList[2]
+
+	const (
+		readMaxConns = 10
+	)
+	// Set up a raw connection to the DB
+	initPoolConfig, err := pgxpool.ParseConfig(fmt.Sprintf("postgres://db:password@pg.example.com:5432/mydb?pool_max_conns=%d", readMaxConns))
+	require.NoError(err)
+	initPool, err := pool.NewRetryPool(t.Context(), "read", initPoolConfig, nil, 18, 20)
+	require.NoError(err)
+	t.Cleanup(func() {
+		initPool.Close()
+	})
+
+	var version crdbVersion
+	err = queryServerVersion(t.Context(), initPool, &version)
+	require.NoError(err)
+
+	require.Equal(expectedMajor, version.Major)
+	require.Equal(expectedMinor, version.Minor)
+	require.Equal(expectedPatch, version.Patch)
+}

internal/datastore/crdb/version.go

@@ -2,48 +2,33 @@ package crdb
 
 import (
 	"context"
-	"errors"
 	"fmt"
 	"regexp"
 	"strconv"
 
 	"github.com/jackc/pgx/v5"
-	"github.com/jackc/pgx/v5/pgconn"
 	"github.com/rs/zerolog"
 
 	pgxcommon "github.com/authzed/spicedb/internal/datastore/postgres/common"
 )
 
 const (
-	queryVersionJSON = "SELECT crdb_internal.active_version()::jsonb;"
-	queryVersion     = "SELECT version();"
-
-	errFunctionDoesNotExist = "42883"
+	queryVersion = "SELECT version();"
 )
 
 var versionRegex = regexp.MustCompile(`v([0-9]+)\.([0-9]+)\.([0-9]+)(?:-([0-9A-Za-z-]+(?:\.[0-9A-Za-z-]+)*))?(?:\+[0-9A-Za-z-]+)?`)
 
+// queryServerVersion reads the version() string out of CRDB
+// and then parses it using a regex to determine the semver.
 func queryServerVersion(ctx context.Context, db pgxcommon.DBFuncQuerier, version *crdbVersion) error {
+	var versionStr string
 	if err := db.QueryRowFunc(ctx, func(ctx context.Context, row pgx.Row) error {
-		return row.Scan(version)
-	}, queryVersionJSON); err != nil {
-		var pgerr *pgconn.PgError
-		if !errors.As(err, &pgerr) || pgerr.Code != errFunctionDoesNotExist {
-			return err
-		}
-
-		// The crdb_internal.active_version() wasn't added until v22.1.X, try to parse the version
-		var versionStr string
-		if err := db.QueryRowFunc(ctx, func(ctx context.Context, row pgx.Row) error {
-			return row.Scan(&versionStr)
-		}, queryVersion); err != nil {
-			return err
-		}
-
-		return parseVersionStringInto(versionStr, version)
+		return row.Scan(&versionStr)
+	}, queryVersion); err != nil {
+		return err
 	}
 
-	return nil
+	return parseVersionStringInto(versionStr, version)
 }
 
 func parseVersionStringInto(versionStr string, version *crdbVersion) error {

internal/datastore/crdb/version/version.go

@@ -8,4 +8,4 @@ const MinimumSupportedCockroachDBVersion = "23.1.30"
 // LatestTestedCockroachDBVersion is the latest version of CockroachDB that has been tested with this driver.
 //
 // NOTE: must match a tag on DockerHub for the `cockroachdb/cockroach` image, without the `v` prefix.
-const LatestTestedCockroachDBVersion = "25.3.2"
+const LatestTestedCockroachDBVersion = "26.1.0"