@@ -33,6 +33,7 @@ Example, if you have a non-standard port, set it with -P, default is 3306:
3333 server , _ := cmd .Flags ().GetString ("server" )
3434 port , _ := cmd .Flags ().GetString ("port" )
3535 database , _ := cmd .Flags ().GetString ("database" )
36+ retries , _ := cmd .Flags ().GetInt ("retries" )
3637
3738 // check if required flags are set
3839 if user == "" || password == "" || server == "" || database == "" {
@@ -43,7 +44,7 @@ Example, if you have a non-standard port, set it with -P, default is 3306:
4344 dsn := user + ":" + password + "@tcp(" + server + ":" + port + ")/" + database
4445
4546 // Try to connect with db
46- db , err := waitForDB (dsn , 10 , time .Second * time .Duration (timer ))
47+ db , err := waitForDB (dsn , retries , time .Second * time .Duration (timer ))
4748 if err != nil {
4849 log .Fatalf ("DB connection failed: %v" , err )
4950 }
@@ -60,6 +61,7 @@ func init() {
6061 sqlCmd .Flags ().StringP ("server" , "s" , "" , "Database server" )
6162 sqlCmd .Flags ().StringP ("port" , "P" , "3306" , "Database port" )
6263 sqlCmd .Flags ().StringP ("database" , "d" , "" , "Database name" )
64+ sqlCmd .Flags ().IntP ("retries" , "r" , 10 , "Number of retries" )
6365}
6466
6567func waitForDB (dsn string , retries int , delay time.Duration ) (* sql.DB , error ) {
0 commit comments