mirror of https://github.com/status-im/fathom.git
81 lines
1.4 KiB
Go
81 lines
1.4 KiB
Go
package datastore
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
_ "github.com/go-sql-driver/mysql" // mysql driver
|
|
_ "github.com/lib/pq" // postgresql driver
|
|
"github.com/rubenv/sql-migrate"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
// DB ...
|
|
var DB *sql.DB
|
|
|
|
// Init creates a database connection pool
|
|
func Init() *sql.DB {
|
|
driver := os.Getenv("ANA_DATABASE_DRIVER")
|
|
if driver == "" {
|
|
driver = "mysql"
|
|
}
|
|
|
|
DB = New(driver, getDSN(driver))
|
|
|
|
// run migrations
|
|
runMigrations(driver)
|
|
|
|
return DB
|
|
}
|
|
|
|
// New creates a new database pool
|
|
func New(driver string, config string) *sql.DB {
|
|
db, err := sql.Open(driver, config)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if err = db.Ping(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
func getDSN(driver string) string {
|
|
var dsn = fmt.Sprintf(
|
|
"%s:%s@%s/%s",
|
|
os.Getenv("ANA_DATABASE_USER"),
|
|
os.Getenv("ANA_DATABASE_PASSWORD"),
|
|
os.Getenv("ANA_DATABASE_HOST"),
|
|
os.Getenv("ANA_DATABASE_NAME"),
|
|
)
|
|
|
|
switch driver {
|
|
case "postgres":
|
|
dsn = "postgres://" + dsn
|
|
case "mysql":
|
|
dsn = dsn + "?parseTime=true"
|
|
}
|
|
|
|
return dsn
|
|
}
|
|
|
|
func runMigrations(driver string) {
|
|
migrations := migrate.FileMigrationSource{
|
|
Dir: "pkg/datastore/migrations",
|
|
}
|
|
|
|
migrate.SetTable("migrations")
|
|
|
|
n, err := migrate.Exec(DB, driver, migrations, migrate.Up)
|
|
|
|
if err != nil {
|
|
log.Fatal("Database migrations failed: ", err)
|
|
}
|
|
|
|
if n > 0 {
|
|
log.Printf("Applied %d database migrations!\n", n)
|
|
}
|
|
}
|