public/photoview
1
0
Fork 0
mirror of https://git.vectorsigma.ru/public/photoview.git synced 2026-08-03 21:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve mysql connection retry on ping fail

Browse Source
This commit is contained in:
viktorstrate
2020-04-30 15:00:18 +02:00
parent ff0ea48aca
commit deb2fd9f4f
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
api/database/mysql.go
View File

@@ -41,31 +41,32 @@ func SetupDatabase() (*sql.DB, error) {
log.Printf("Connecting to database: %s", address)
var db *sql.DB
db, err = sql.Open("mysql", address.String())
if err != nil {
return nil, errors.New("Could not connect to database, exiting")
}
tryCount := 0
for {
db, err = sql.Open("mysql", address.String())
if err != nil {
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := db.PingContext(ctx); err != nil {
if tryCount < 4 {
tryCount++
log.Printf("WARN: Could not connect to database: %s. Will retry after 1 second", err.Error())
log.Printf("WARN: Could not ping database: %s, Will retry after 1 second", err)
time.Sleep(time.Second)
continue
} else {
return nil, errors.New("Could not connect to database, exiting")
return nil, errors.Wrap(err, "Could not ping database, exiting")
}
}
break
}
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := db.PingContext(ctx); err != nil {
return nil, errors.Wrap(err, "Could not ping database, exiting")
}
db.SetMaxOpenConns(80)
return db, nil