Merge #37684

37684: roachtest: revert not disabling merge queue r=jeffrey-xiao a=jeffrey-xiao

Roachtests need to remain compatible with 19.1 and having the merge
queue enabled when performing a split will throw an error.

Release note: None

Co-authored-by: Jeffrey Xiao <[email protected]>

pkg/cmd/roachtest/election.go

@@ -42,6 +42,8 @@ func registerElectionAfterRestart(r *registry) {
 			c.Run(ctx, c.Node(1), `./cockroach sql --insecure -e "
         CREATE DATABASE IF NOT EXISTS test;
         CREATE TABLE test.kv (k INT PRIMARY KEY, v INT);
+        -- Prevent the merge queue from immediately discarding our splits.
+        SET CLUSTER SETTING kv.range_merge.queue_enabled = false;
         ALTER TABLE test.kv SPLIT AT SELECT generate_series(0, 10000, 100)"`)
 
 			start := timeutil.Now()

pkg/workload/indexes/indexes.go

@@ -96,6 +96,10 @@ func (w *indexes) Hooks() workload.Hooks {
 			return nil
 		},
 		PostLoad: func(sqlDB *gosql.DB) error {
+			// Prevent the merge queue from immediately discarding our splits.
+			if err := maybeDisableMergeQueue(sqlDB); err != nil {
+				return err
+			}
 			// Split at the beginning of each index so that as long as the
 			// table has a single index, all writes will be multi-range.
 			for i := 0; i < w.idxs; i++ {
@@ -109,6 +113,17 @@ func (w *indexes) Hooks() workload.Hooks {
 	}
 }
 
+func maybeDisableMergeQueue(sqlDB *gosql.DB) error {
+	var ok bool
+	if err := sqlDB.QueryRow(
+		`SELECT count(*) > 0 FROM [ SHOW ALL CLUSTER SETTINGS ] AS _ (v) WHERE v = 'kv.range_merge.queue_enabled'`,
+	).Scan(&ok); err != nil || !ok {
+		return err
+	}
+	_, err := sqlDB.Exec("SET CLUSTER SETTING kv.range_merge.queue_enabled = false")
+	return err
+}
+
 // Tables implements the Generator interface.
 func (w *indexes) Tables() []workload.Table {
 	// Construct the schema with all indexes.

pkg/workload/workload.go

@@ -541,8 +541,24 @@ func Setup(
 	return size, nil
 }
 
+func maybeDisableMergeQueue(db *gosql.DB) error {
+	var ok bool
+	if err := db.QueryRow(
+		`SELECT count(*) > 0 FROM [ SHOW ALL CLUSTER SETTINGS ] AS _ (v) WHERE v = 'kv.range_merge.queue_enabled'`,
+	).Scan(&ok); err != nil || !ok {
+		return err
+	}
+	_, err := db.Exec("SET CLUSTER SETTING kv.range_merge.queue_enabled = false")
+	return err
+}
+
 // Split creates the range splits defined by the given table.
 func Split(ctx context.Context, db *gosql.DB, table Table, concurrency int) error {
+	// Prevent the merge queue from immediately discarding our splits.
+	if err := maybeDisableMergeQueue(db); err != nil {
+		return err
+	}
+
 	if table.Splits.NumBatches <= 0 {
 		return nil
 	}