workload/schemachange: don't use CITEXT in mixed version cluster

Release note: None

Author: rafiss

pkg/workload/schemachange/BUILD.bazel

@@ -28,6 +28,7 @@ go_library(
         "//pkg/sql/catalog/catpb",
         "//pkg/sql/catalog/colinfo",
         "//pkg/sql/lexbase",
+        "//pkg/sql/oidext",
         "//pkg/sql/parser",
         "//pkg/sql/pgwire/pgcode",
         "//pkg/sql/pgwire/pgerror",

pkg/workload/schemachange/operation_generator.go

@@ -24,6 +24,7 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/sql/catalog/catpb"
 	"github.com/cockroachdb/cockroach/pkg/sql/catalog/colinfo"
 	"github.com/cockroachdb/cockroach/pkg/sql/lexbase"
+	"github.com/cockroachdb/cockroach/pkg/sql/oidext"
 	"github.com/cockroachdb/cockroach/pkg/sql/parser"
 	"github.com/cockroachdb/cockroach/pkg/sql/pgwire/pgcode"
 	"github.com/cockroachdb/cockroach/pkg/sql/randgen"
@@ -1283,6 +1284,15 @@ func (og *operationGenerator) createTable(ctx context.Context, tx pgx.Tx) (*opSt
 		}
 		return false
 	}()
+	hasCitextType := func() bool {
+		// Check if any of the columns have CITEXT types involved.
+		for _, def := range stmt.Defs {
+			if col, ok := def.(*tree.ColumnTableDef); ok && col.Type.SQLString() == "CITEXT" {
+				return true
+			}
+		}
+		return false
+	}()
 
 	// Randomly create as schema locked table.
 	versionBefore253, err := isClusterVersionLessThan(ctx, tx, clusterversion.V25_3.Version())
@@ -1314,6 +1324,8 @@ func (og *operationGenerator) createTable(ctx context.Context, tx pgx.Tx) (*opSt
 	opStmt.potentialExecErrors.addAll(codesWithConditions{
 		{code: pgcode.Syntax, condition: hasVectorType},
 		{code: pgcode.FeatureNotSupported, condition: hasVectorType},
+		{code: pgcode.Syntax, condition: hasCitextType},
+		{code: pgcode.FeatureNotSupported, condition: hasCitextType},
 	})
 	opStmt.sql = tree.Serialize(stmt)
 	return opStmt, nil
@@ -4090,8 +4102,18 @@ func (og *operationGenerator) randType(
 		return nil, nil, err
 	}
 
+	// Block CITEXT usage until v25.3 is finalized.
+	citextNotSupported, err := isClusterVersionLessThan(
+		ctx,
+		tx,
+		clusterversion.V25_3.Version())
+	if err != nil {
+		return nil, nil, err
+	}
+
 	typ := randgen.RandSortingType(og.params.rng)
-	for pgVectorNotSupported && typ.Family() == types.PGVectorFamily {
+	for (pgVectorNotSupported && typ.Family() == types.PGVectorFamily) ||
+		(citextNotSupported && typ.Oid() == oidext.T_citext) {
 		typ = randgen.RandSortingType(og.params.rng)
 	}
 
@@ -4274,6 +4296,11 @@ FROM
 		possibleParamReferences = append(possibleParamReferences, fmt.Sprintf(`enum_%d %s`, i, enum["name"]))
 	}
 
+	citextNotSupported, err := isClusterVersionLessThan(ctx, tx, clusterversion.V25_3.Version())
+	if err != nil {
+		return nil, err
+	}
+
 	// Generate random parameters / values for builtin types.
 	for i, typeVal := range randgen.SeedTypes {
 		// If we have types where invalid values can exist then skip over these,
@@ -4287,6 +4314,10 @@ FROM
 			continue
 		}
 
+		if citextNotSupported && typeVal.Oid() == oidext.T_citext {
+			continue
+		}
+
 		possibleReturnReferences = append(possibleReturnReferences, typeVal.SQLStandardName())
 		possibleParamReferences = append(possibleParamReferences, fmt.Sprintf("val_%d %s", i+len(enums), typeVal.SQLStandardName()))
 		optionalDefaultValue := randgen.RandDatum(og.params.rng, typeVal, true)