adding big ddl command

backend/big_ddl.go

@@ -0,0 +1,77 @@
+package backend
+
+import (
+	"fmt"
+
+	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
+)
+
+const enablePerformanceSchemaSQL = `
+UPDATE performance_schema.setup_instruments
+SET ENABLED='YES', TIMED='YES'
+WHERE NAME LIKE 'stage/innodb/alter%';
+
+UPDATE performance_schema.setup_consumers
+SET ENABLED='YES'
+WHERE NAME LIKE '%stage%';
+`
+
+const createBigDDLSQL = `
+CREATE DATABASE IF NOT EXISTS TESTDB;
+USE TESTDB;
+
+CREATE TABLE IF NOT EXISTS  t1(x int primary key auto_increment);
+INSERT INTO t1() VALUES (),(),(),();
+INSERT INTO t1(x) SELECT x+(SELECT COUNT(*) FROM  t1) FROM t1;
+`
+
+// why using this COPY algorithm?
+// see https://dev.mysql.com/blog-archive/mysql-8-0-innodb-now-supports-instant-add-column/
+const triggerBigDDLSQL = `
+ALTER TABLE testdb.t1 ADD COLUMN col1 INT NOT NULL, ALGORITHM=COPY;
+`
+
+func CreateBigDDL(c pkg.Connect) error {
+	db, err := pkg.GetDB(c)
+	if err != nil {
+		return err
+	}
+
+	// Execute preparatory SQL to enable performance schema monitoring
+	err = pkg.BatchExec(db, enablePerformanceSchemaSQL)
+	if err != nil {
+		return fmt.Errorf("error enabling performance schema: %w", err)
+	}
+	logrus.Info("Performance schema monitoring enabled")
+
+	err = pkg.BatchExec(db, createBigDDLSQL)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		// Drop the TESTDB database
+		_, err = db.Exec("DROP DATABASE TESTDB")
+		if err != nil {
+			logrus.Errorf("error dropping database: %v", err)
+		}
+	}()
+
+	// Execute the doubling logic in a loop
+	for i := 0; i < 23; i++ {
+		logrus.Infof("inserting data, round %d/%d", i, 23)
+		err = pkg.Exec(db, "INSERT INTO t1(x) SELECT x + (SELECT COUNT(*) FROM t1) FROM t1")
+		if err != nil {
+			return fmt.Errorf("error doubling data: %w", err)
+		}
+	}
+
+	// Trigger the big DDL
+	fmt.Println("Triggering big DDL...")
+	_, err = db.Exec(triggerBigDDLSQL)
+	if err != nil {
+		return fmt.Errorf("error triggering big DDL: %w", err)
+	}
+
+	return nil
+}

backend/big_transaction.go

@@ -10,6 +10,7 @@ import (
 
 	"github.com/bo-er/corrupt-mysql/pkg"
 	"github.com/inhies/go-bytesize"
+	"github.com/sirupsen/logrus"
 )
 
 const prepareBigTransactionSQL = `
@@ -55,7 +56,7 @@ func getPower(size string) (int, error) {
 	if kb < 10 {
 		return 0, errors.New("please enter a number that's bigger than 10 kilobytes")
 	}
-	fmt.Println("bytes: ", kb)
+	logrus.Info("bytes: ", kb)
 	ratio := (kb / 10240) * 20
 	return int(math.Ceil(ratio)), nil
 }

backend/deadlock.go

@@ -1,9 +1,11 @@
 package backend
 
 import (
+	"fmt"
 	"sync"
 
 	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
 )
 
 const prepareDeadlockSQL = `
@@ -43,21 +45,49 @@ func CreateDeadlock(c pkg.Connect) error {
 	if err != nil {
 		return err
 	}
+
+	// Get the original innodb_lock_wait_timeout
+	var (
+		variableName    string
+		originalTimeout int
+	)
+	err = db.QueryRow("SHOW VARIABLES LIKE 'innodb_lock_wait_timeout'").Scan(&variableName, &originalTimeout)
+	if err != nil {
+		return fmt.Errorf("error getting original timeout: %w", err)
+	}
+	logrus.Infof("original timeout is %d", originalTimeout)
+	newTimeout := 60
+	err = pkg.Exec(db, "SET GLOBAL innodb_lock_wait_timeout = ?", newTimeout)
+	if err != nil {
+		return fmt.Errorf("error setting new timeout: %w", err)
+	}
+	defer func() {
+		// Restore the original innodb_lock_wait_timeout
+		err = pkg.Exec(db, "SET GLOBAL innodb_lock_wait_timeout = ?", originalTimeout)
+		if err != nil {
+			logrus.Errorf("error restoring original timeout: %v", err)
+		}
+	}()
+
 	err = pkg.BatchExec(db, prepareDeadlockSQL)
 	if err != nil {
 		return err
 	}
+
 	var wg sync.WaitGroup
 	wg.Add(2)
 	for _, sql := range []string{Thread1, Thread2} {
 		sql := sql
 		go func() {
+			defer wg.Done()
 			err = pkg.BatchExec(db, sql)
 			if err != nil {
-				panic(err.Error())
+				// Handle the deadlock error (e.g., log it)
+				logrus.Errorf("Error: %v", err)
 			}
 		}()
 	}
 	wg.Wait()
+
 	return nil
 }

cmd/big_ddl.go

@@ -0,0 +1,29 @@
+package cmd
+
+import (
+	"github.com/bo-er/corrupt-mysql/backend"
+	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+// bigDDLCmd creates a big DDL in a MySQL instance.
+var bigDDLCmd = &cobra.Command{
+	Use:   "bigddl",
+	Short: "creating a big DDL in a MySQL instance.",
+	Long:  `./corrupt-mysql bigddl -H10.186.62.63 -P25690 -uuniverse_udb -p123`,
+	Run: func(cmd *cobra.Command, args []string) {
+		connect := pkg.Connect{
+			User:     user,
+			Host:     host,
+			Password: password,
+			DBName:   "mysql",
+			Port:     port,
+		}
+		err := backend.CreateBigDDL(connect)
+		if err != nil {
+			logrus.Error(err)
+		}
+		logrus.Info("operation is done")
+	},
+}

cmd/big_transaction.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"github.com/bo-er/corrupt-mysql/backend"
 	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -25,7 +26,8 @@ var bigTransaction = &cobra.Command{
 
 		err := backend.CreatesBigTransactions(connect, args[0])
 		if err != nil {
-			panic(err.Error())
+			logrus.Error(err)
 		}
+		logrus.Info("operation is done")
 	},
 }

cmd/deadlock.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"github.com/bo-er/corrupt-mysql/backend"
 	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -21,7 +22,8 @@ var deadlockCmd = &cobra.Command{
 		}
 		err := backend.CreateDeadlock(connect)
 		if err != nil {
-			panic(err.Error())
+			logrus.Error(err)
 		}
+		logrus.Info("operation is done")
 	},
 }

cmd/high_cpu_usage.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"github.com/bo-er/corrupt-mysql/backend"
 	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -21,7 +22,8 @@ var highCPUCmd = &cobra.Command{
 		}
 		err := backend.MakeCPUUsageHigh(connect)
 		if err != nil {
-			panic(err.Error())
+			logrus.Error(err)
 		}
+		logrus.Info("operation is done")
 	},
 }

cmd/root.go

@@ -30,6 +30,7 @@ func init() {
 	rootCmd.AddCommand(deadlockCmd)
 	rootCmd.AddCommand(slowlogCmd)
 	rootCmd.AddCommand(bigTransaction)
+	rootCmd.AddCommand(bigDDLCmd)
 
 	rootCmd.PersistentFlags().StringVarP(&host, "host", "H", "", "host of the mysql server")
 	rootCmd.PersistentFlags().IntVarP(&port, "port", "P", 3306, "port of the mysql server")

cmd/slow_log.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"github.com/bo-er/corrupt-mysql/backend"
 	"github.com/bo-er/corrupt-mysql/pkg"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -21,7 +22,8 @@ var slowlogCmd = &cobra.Command{
 		}
 		err := backend.CreateSlowlog(connect)
 		if err != nil {
-			panic(err.Error())
+			logrus.Error(err)
 		}
+		logrus.Info("operation is done")
 	},
 }

main.go

@@ -1,7 +1,13 @@
 package main
 
-import "github.com/bo-er/corrupt-mysql/cmd"
+import (
+	"os"
+
+	"github.com/bo-er/corrupt-mysql/cmd"
+	"github.com/sirupsen/logrus"
+)
 
 func main() {
+	logrus.SetOutput(os.Stdout)
 	cmd.Execute()
 }

pkg/mysql.go

@@ -57,3 +57,12 @@ func BatchExec(db *sql.DB, sqls string) error {
 	}
 	return nil
 }
+
+func Exec(db *sql.DB, sql string, args ...any) error {
+	_, err := db.Exec(sql, args...)
+	if err != nil {
+		return fmt.Errorf("sql: %s,initializing databases: %s", sql, err.Error())
+	}
+	fmt.Println(sql)
+	return nil
+}