update tidb

db/tidb/bank.go

@@ -45,6 +45,7 @@ func (c *bankClient) SetUp(ctx context.Context, nodes []string, node string) err
 		return nil
 	}
 
+	log.Printf("begin to create table accounts on node %s", node)
 	sql := `create table if not exists accounts
 			(id     int not null primary key,
 			balance bigint not null)`

db/tidb/db.go

@@ -34,7 +34,8 @@ var (
 
 // db is the TiDB database.
 type db struct {
-	nodes []string
+	nodes          []string
+	installBlocker util.BlockRunner
 }
 
 // SetUp initializes the database.
@@ -46,15 +47,30 @@ func (db *db) SetUp(ctx context.Context, nodes []string, node string) error {
 
 	db.nodes = nodes
 
+	db.installBlocker.Init(len(nodes))
+
 	log.Printf("install archieve on node %s", node)
-	if err := util.InstallArchive(ctx, node, archiveURL, deployDir); err != nil {
+
+	var err error
+	db.installBlocker.Run(func() {
+		err = util.InstallArchive(ctx, node, archiveURL, deployDir)
+	})
+	if err != nil {
 		return err
 	}
 
 	util.Mkdir(ctx, node, path.Join(deployDir, "conf"))
 	util.Mkdir(ctx, node, path.Join(deployDir, "log"))
 
-	if err := util.WriteFile(ctx, node, pdConfig, strconv.Quote("[replication]\nmax-replicas=5")); err != nil {
+	pdCfs := []string{
+		"tick-interval=\"100ms\"",
+		"election-interval=\"500ms\"",
+		"tso-save-interval=\"500ms\"",
+		"[replication]",
+		"max-replicas=5",
+	}
+
+	if err := util.WriteFile(ctx, node, pdConfig, strconv.Quote(strings.Join(pdCfs, "\n"))); err != nil {
 		return err
 	}
 
@@ -161,8 +177,19 @@ func (db *db) start(ctx context.Context, node string, inSetUp bool) error {
 		return err
 	}
 
+	var err error
 	if inSetUp {
-		time.Sleep(30 * time.Second)
+		for i := 0; i < 12; i++ {
+			if err = ssh.Exec(ctx, node, "curl", fmt.Sprintf("http://%s:10080/status", node)); err == nil {
+				break
+			}
+			log.Printf("try to wait tidb run on %s", node)
+			time.Sleep(10 * time.Second)
+		}
+	}
+
+	if err != nil {
+		return err
 	}
 
 	if !util.IsDaemonRunning(ctx, node, tidbBinary, tidbPID) {

pkg/control/control.go

@@ -107,6 +107,8 @@ func (c *Controller) Run() {
 	c.tearDownDB()
 
 	c.recorder.Close()
+
+	log.Printf("finish test")
 }
 
 func (c *Controller) syncExec(f func(i int)) {
@@ -158,7 +160,7 @@ func (c *Controller) setUpClient() {
 }
 
 func (c *Controller) tearDownClient() {
-	log.Printf("begin to set up client")
+	log.Printf("begin to tear down client")
 	c.syncExec(func(i int) {
 		client := c.clients[i]
 		node := c.nodes[i]
@@ -173,6 +175,8 @@ func (c *Controller) onClientLoop(i int) {
 	client := c.clients[i]
 	node := c.nodes[i]
 
+	log.Printf("begin to run command on node %s", node)
+
 	ctx, cancel := context.WithTimeout(c.ctx, c.cfg.RunTime)
 	defer cancel()