From c0243580c0203056955053ea12e36289867e0f96 Mon Sep 17 00:00:00 2001 From: Deluan Date: Tue, 11 May 2021 19:08:06 -0400 Subject: [PATCH] Integrate goose log with our own log system --- db/db.go | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/db/db.go b/db/db.go index 79c8fa322..640861b89 100644 --- a/db/db.go +++ b/db/db.go @@ -2,6 +2,7 @@ package db import ( "database/sql" + "fmt" "os" "sync" @@ -41,6 +42,7 @@ func Db() *sql.DB { func EnsureLatestVersion() { db := Db() + goose.SetLogger(&logAdapter{}) // Disable foreign_keys to allow re-creating tables in migrations _, err := db.Exec("PRAGMA foreign_keys=off") @@ -65,3 +67,27 @@ func EnsureLatestVersion() { os.Exit(1) } } + +type logAdapter struct{} + +func (l *logAdapter) Fatal(v ...interface{}) { + log.Error(fmt.Sprint(v...)) + os.Exit(-1) +} + +func (l *logAdapter) Fatalf(format string, v ...interface{}) { + log.Error(fmt.Sprintf(format, v...)) + os.Exit(-1) +} + +func (l *logAdapter) Print(v ...interface{}) { + log.Info(fmt.Sprint(v...)) +} + +func (l *logAdapter) Println(v ...interface{}) { + log.Info(fmt.Sprintln(v...)) +} + +func (l *logAdapter) Printf(format string, v ...interface{}) { + log.Info(fmt.Sprintf(format, v...)) +}