60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"embed"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
var Pool *pgxpool.Pool
|
|
|
|
func Connect(url string) error {
|
|
// Create context timeout, if connection takes more than 5 seconds, it will be canceled automatically
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
// Always cancel timeout to clean up resources
|
|
defer cancel()
|
|
|
|
// Open new connection with database url
|
|
var err error
|
|
Pool, err = pgxpool.New(ctx, url)
|
|
if err != nil {
|
|
log.Printf("[ERROR] Failed to connect to DB: %v\n", err)
|
|
return err
|
|
}
|
|
|
|
// Ping database to check if its online
|
|
err = Pool.Ping(ctx)
|
|
if err != nil {
|
|
log.Printf("[ERROR] Failed to ping DB: %v", err)
|
|
return err
|
|
}
|
|
|
|
// Connection successful
|
|
log.Printf("[INFO] Connected to DB: %v\n", url)
|
|
return nil
|
|
}
|
|
|
|
//go:embed migrations.sql
|
|
var migrationFS embed.FS
|
|
|
|
func Migrate() error {
|
|
// Read file from embed
|
|
sql, err := migrationFS.ReadFile("migrations.sql")
|
|
if err != nil {
|
|
log.Printf("[ERROR] Failed to read migrations.sql: %v\n", err)
|
|
return err
|
|
}
|
|
|
|
// Execute sql file into database
|
|
_, err = Pool.Exec(context.Background(), string(sql))
|
|
if err != nil {
|
|
log.Printf("[ERROR] Failed to migrate DB: %v\n", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|