add indexMode to test index operator

Signed-off-by: Weizhen Wang <[email protected]>

Author: hawkingrei

ddl/ddl.go

@@ -249,12 +249,18 @@ ParallelExecuteOperations executes process:
 func ParallelExecuteOperations(c *testCase, ops []ddlTestOpExecutor, postOp func() error) error {
 	perm := rand.Perm(len(ops))
 	taskCh := make(chan *ddlJobTask, len(ops))
+	var probability map[int]float64
+	if isIndexMode {
+		probability = mapOfDDLKindProbabilityInIndexMode
+	} else {
+		probability = mapOfDDLKindProbability
+	}
 	for _, idx := range perm {
 		if c.isStop() {
 			return nil
 		}
 		op := ops[idx]
-		if rand.Float64() > mapOfDDLKindProbability[op.ddlKind] {
+		if rand.Float64() > probability[op.ddlKind] {
 			continue
 		}
 		op.executeFunc(op.config, taskCh)

ddl/ddl_ops.go

@@ -206,6 +206,33 @@ var mapOfDDLKindProbability = map[DDLKind]float64{
 	ddlSetTiflashReplica:            0.30,
 }
 
+// mapOfDDLKindProbabilityInIndexMode use to control every kind of ddl request execute probability in index mode.
+var mapOfDDLKindProbabilityInIndexMode = map[DDLKind]float64{
+	ddlAddTable:  0.05,
+	ddlDropTable: 0.05,
+
+	ddlAddIndex:  0.85,
+	ddlDropIndex: 0.5,
+
+	ddlAddColumn:     0.2,
+	ddlModifyColumn:  0.1,
+	ddlModifyColumn2: 0.1,
+	ddlDropColumn:    0.1,
+
+	ddlCreateView: 0.07,
+
+	ddlCreateSchema:                 0.02,
+	ddlDropSchema:                   0.02,
+	ddlRenameTable:                  0.12,
+	ddlRenameIndex:                  0.12,
+	ddlTruncateTable:                0.12,
+	ddlShardRowID:                   0.07,
+	ddlRebaseAutoID:                 0.03,
+	ddlSetDefaultValue:              0.07,
+	ddlModifyTableComment:           0.07,
+	ddlModifyTableCharsetAndCollate: 0.07,
+}
+
 type ddlJob struct {
 	id         int
 	schemaName string

ddl/run.go

@@ -17,6 +17,7 @@ import (
 var defaultPushMetricsInterval = 15 * time.Second
 var enableTransactionTestFlag = "0"
 var enableTransactionTest = false
+var isIndexMode = false
 
 func init() {
 	if enableTransactionTestFlag == "1" {
@@ -35,7 +36,8 @@ func OpenDB(dsn string, maxIdleConns int) (*sql.DB, error) {
 	return db, nil
 }
 
-func Run(dbAddr string, dbName string, concurrency int, tablesToCreate int, mysqlCompatible bool, testTp DDLTestType, testTime time.Duration) {
+func Run(dbAddr string, dbName string, concurrency int, tablesToCreate int, mysqlCompatible bool, testTp DDLTestType, testTime time.Duration, indexMode bool) {
+	isIndexMode = indexMode
 	wrapCtx := context.WithCancel
 	if testTime > 0 {
 		wrapCtx = func(ctx context.Context) (context.Context, context.CancelFunc) {

main.go

@@ -34,6 +34,7 @@ var (
 	mysqlCompatible = flag.Bool("mysql-compatible", false, "disable TiDB-only features")
 	testTime        = flag.Duration("time", 2*time.Hour, "test time")
 	output          = flag.String("output", "", "output file")
+	indexMode       = flag.Bool("index-mode", false, "test more index operator")
 )
 
 func prepareEnv() {
@@ -74,5 +75,5 @@ func main() {
 		log.Fatalf("unknown test mode: %s", *mode)
 	}
 	prepareEnv()
-	Run(*dbAddr, *dbName, *concurrency, *tablesToCreate, *mysqlCompatible, testType, *testTime)
+	Run(*dbAddr, *dbName, *concurrency, *tablesToCreate, *mysqlCompatible, testType, *testTime, *indexMode)
 }