89 lines
1.3 KiB
Go
89 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"authentication/data"
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
_ "github.com/jackc/pgconn"
|
|
_ "github.com/jackc/pgx/v4"
|
|
_ "github.com/jackc/pgx/v4/stdlib"
|
|
)
|
|
|
|
const webPort = "80"
|
|
|
|
var counts int64
|
|
|
|
type Config struct {
|
|
DB *sql.DB
|
|
Models data.Models
|
|
}
|
|
|
|
func main() {
|
|
// connect to DB
|
|
conn := connectToDB()
|
|
|
|
if conn == nil {
|
|
log.Panic("Can't connect to Postgres!")
|
|
}
|
|
|
|
// set up config
|
|
app := Config{
|
|
DB: conn,
|
|
Models: data.New(conn),
|
|
}
|
|
|
|
log.Printf("Starting authentication service on port %s\n", webPort)
|
|
|
|
// define http server
|
|
srv := &http.Server{
|
|
Addr: fmt.Sprintf(":%s", webPort),
|
|
Handler: app.routes(),
|
|
}
|
|
|
|
err := srv.ListenAndServe()
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
}
|
|
|
|
func openDB(dsn string) (*sql.DB, error) {
|
|
db, err := sql.Open("pgx", dsn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = db.Ping()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func connectToDB() *sql.DB {
|
|
dsn := os.Getenv("DSN")
|
|
|
|
for {
|
|
connection, err := openDB(dsn)
|
|
if err != nil {
|
|
log.Println("Postgres not yet ready ...: ", err)
|
|
counts++
|
|
} else {
|
|
log.Println("Connected to Postgres!")
|
|
return connection
|
|
}
|
|
|
|
if counts > 10 {
|
|
return nil
|
|
}
|
|
|
|
log.Println("Backing off for two seconds...")
|
|
time.Sleep(2 * time.Second)
|
|
}
|
|
}
|