marvin/marvinblum
1
0
Fork 0
mirror of https://github.com/Kugelschieber/marvinblum.git synced 2026-01-18 06:40:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added automatic database migration.

Browse Source
This commit is contained in:
Marvin Blum
2020-09-10 22:13:07 +02:00
committed by Marvin Blum
parent 5b800991c8
commit 47e6ecfadd
7 changed files with 804 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
db/db.go Normal file
View File

@@ -0,0 +1,29 @@
package db
import (
"fmt"
"github.com/emvi/logbuch"
"os"
"strconv"
"time"
)
const (
connectionString = `host=%s port=%s user=%s password=%s dbname=%s sslmode=%s sslcert=%s sslkey=%s sslrootcert=%s connectTimeout=%s timezone=%s`
)
func GetConnectionString() string {
host := os.Getenv("MB_DB_HOST")
port := os.Getenv("MB_DB_PORT")
user := os.Getenv("MB_DB_USER")
password := os.Getenv("MB_DB_PASSWORD")
schema := os.Getenv("MB_DB_SCHEMA")
sslMode := os.Getenv("MB_DB_SSLMODE")
sslCert := os.Getenv("MB_DB_SSLCERT")
sslKey := os.Getenv("MB_DB_SSLKEY")
sslRootCert := os.Getenv("MB_DB_SSLROOTCERT")
zone, offset := time.Now().Zone()
timezone := zone + strconv.Itoa(-offset/3600)
logbuch.Info("Setting time zone", logbuch.Fields{"timezone": timezone})
return fmt.Sprintf(connectionString, host, port, user, password, schema, sslMode, sslCert, sslKey, sslRootCert, "30", timezone)
}

48
db/migrate.go Normal file
View File

@@ -0,0 +1,48 @@
package db
import (
"fmt"
"github.com/emvi/logbuch"
"github.com/golang-migrate/migrate/v4"
_ "github.com/golang-migrate/migrate/v4/database/postgres"
_ "github.com/golang-migrate/migrate/v4/source/file"
"os"
)
const (
migrateConnectionString = `postgres://%s:%s/%s?user=%s&password=%s&sslmode=%s&sslcert=%s&sslkey=%s&sslrootcert=%s&connect_timeout=60`
)
func Migrate() {
logbuch.Info("Migrating database schema (if required)")
m, err := migrate.New("file://schema", getMigrationConnectionString())
if err != nil {
logbuch.Fatal("Error migrating database schema", logbuch.Fields{"err": err})
return
}
if err := m.Up(); err != nil && err != migrate.ErrNoChange {
logbuch.Fatal("Error migrating database schema", logbuch.Fields{"err": err})
return
}
if sourceErr, dbErr := m.Close(); sourceErr != nil || dbErr != nil {
logbuch.Fatal("Error migrating database schema", logbuch.Fields{"source_err": sourceErr, "db_err": dbErr})
}
logbuch.Info("Done migrating database schema")
}
func getMigrationConnectionString() string {
host := os.Getenv("MB_DB_HOST")
port := os.Getenv("MB_DB_PORT")
user := os.Getenv("MB_DB_USER")
password := os.Getenv("MB_DB_PASSWORD")
schema := os.Getenv("MB_DB_SCHEMA")
sslMode := os.Getenv("MB_DB_SSLMODE")
sslCert := os.Getenv("MB_DB_SSLCERT")
sslKey := os.Getenv("MB_DB_SSLKEY")
sslRootCert := os.Getenv("MB_DB_SSLROOTCERT")
return fmt.Sprintf(migrateConnectionString, host, port, schema, user, password, sslMode, sslCert, sslKey, sslRootCert)
}