Skip to content

Commit 27ef9a0

Browse files
author
marcel
committed
added sql retries flag
1 parent f96ff58 commit 27ef9a0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

cmd/sql.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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

6567
func waitForDB(dsn string, retries int, delay time.Duration) (*sql.DB, error) {

0 commit comments

Comments
 (0)