From 5a3208962abeb6cd40eb4af7e0099cca67f70058 Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Thu, 28 Oct 2021 20:30:28 +0800
Subject: [PATCH] auto reconnect to mysql when populate data

---
 examples/data.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/examples/data.go b/examples/data.go
index 5bf120f..b59dd40 100644
--- a/examples/data.go
+++ b/examples/data.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"io/ioutil"
 	"strings"
+	"time"
 
 	"github.com/yedf/dtm/common"
 	"github.com/yedf/dtm/dtmcli"
@@ -25,6 +26,10 @@ func RunSQLScript(conf map[string]string, script string, skipDrop bool) {
 			continue
 		}
 		_, err = dtmcli.DBExec(con, s)
+		for _, err = dtmcli.DBExec(con, s); err != nil; { // wait for mysql to start
+			time.Sleep(3 * time.Second)
+			_, err = dtmcli.DBExec(con, s)
+		}
 		dtmcli.FatalIfError(err)
 	}
 }