2022-01-23 02:19:32 +01:00
|
|
|
package database
|
|
|
|
|
|
|
|
import (
|
2022-02-22 00:08:19 +01:00
|
|
|
"context"
|
2022-01-23 02:19:32 +01:00
|
|
|
"fmt"
|
2022-01-25 23:58:41 +01:00
|
|
|
"log"
|
|
|
|
"time"
|
2022-01-23 02:19:32 +01:00
|
|
|
|
2022-02-22 00:08:19 +01:00
|
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
2022-01-23 02:19:32 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
var LogName string
|
|
|
|
|
2022-02-22 00:08:19 +01:00
|
|
|
// ParseConfig parses the database config string and sets LogName.
|
|
|
|
func ParseConfig(config string) (*pgxpool.Config, error) {
|
|
|
|
dbConfig, err := pgxpool.ParseConfig(config)
|
|
|
|
LogName = fmt.Sprintf("%s@%s:%s/%s", dbConfig.ConnConfig.User, dbConfig.ConnConfig.Host, dbConfig.ConnConfig.Port, dbConfig.ConnConfig.Database)
|
|
|
|
return dbConfig, err
|
|
|
|
}
|
2022-01-25 23:58:41 +01:00
|
|
|
|
2022-02-22 00:08:19 +01:00
|
|
|
func Open(dbConfig *pgxpool.Config) (*pgxpool.Pool, error) {
|
2022-01-25 23:58:41 +01:00
|
|
|
log.Printf("database: connecting...")
|
2022-02-22 00:08:19 +01:00
|
|
|
db, err := pgxpool.ConnectConfig(context.Background(), dbConfig)
|
2022-01-25 23:58:41 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2022-02-22 00:08:19 +01:00
|
|
|
err = db.Ping(context.Background())
|
2022-01-25 23:58:41 +01:00
|
|
|
for i := 1; err != nil && i < 10; i++ {
|
|
|
|
log.Printf("database: connection failed, retrying in %d seconds...", i)
|
|
|
|
time.Sleep(time.Duration(i) * time.Second)
|
2022-02-22 00:08:19 +01:00
|
|
|
err = db.Ping(context.Background())
|
2022-01-25 23:58:41 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if err == nil {
|
|
|
|
log.Println("database: connection successful")
|
|
|
|
}
|
|
|
|
|
|
|
|
return db, err
|
2022-01-23 02:19:32 +01:00
|
|
|
}
|