Skip to content

Commit f07f94c

Browse files
committed
workload: enable rangefeed with --changefeed
This commit enables rangefeed when the --changefeed flag is specified with the workload command.
1 parent 8aa3669 commit f07f94c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pkg/workload/changefeeds/changefeeds.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,18 @@ func AddChangefeedToQueryLoad(
5252
if err != nil {
5353
return err
5454
}
55+
var rangefeedEnabled bool
56+
if err := conn.QueryRow(ctx, "SHOW CLUSTER SETTING kv.rangefeed.enabled").Scan(&rangefeedEnabled); err != nil {
57+
return err
58+
}
59+
if !rangefeedEnabled {
60+
// This will fail if the workload is running against a secondary tenant,
61+
// which cannot modify cluster settings but generally have rangefeeds
62+
// enabled by default.
63+
if _, err := conn.Exec(ctx, "SET CLUSTER SETTING kv.rangefeed.enabled = true"); err != nil {
64+
return err
65+
}
66+
}
5567
if _, err := conn.Exec(ctx, fmt.Sprintf("USE %q", dbName)); err != nil {
5668
return err
5769
}

0 commit comments

Comments
 (0)