added gorm logger
parent
e538b9644f
commit
bcbbfdf6be
|
|
@ -8,10 +8,11 @@ import (
|
|||
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
// ConnectDB - returns a pointer to a new database connection
|
||||
func ConnectDB() error {
|
||||
func ConnectDB(gormLogger logger.Interface) error {
|
||||
// var erre error
|
||||
|
||||
// log.Println("Getting environment values")
|
||||
|
|
@ -32,7 +33,9 @@ func ConnectDB() error {
|
|||
|
||||
log.Println("Opening connection to database")
|
||||
|
||||
db, err := gorm.Open(postgres.Open(DBURL), &gorm.Config{})
|
||||
db, err := gorm.Open(postgres.Open(DBURL), &gorm.Config{
|
||||
Logger: gormLogger,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
|
|
|
|||
13
main.go
13
main.go
|
|
@ -18,6 +18,7 @@ import (
|
|||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/gofiber/fiber/v2/middleware/cors"
|
||||
"github.com/gofiber/fiber/v2/middleware/logger"
|
||||
gormlogger "gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
|
@ -34,6 +35,16 @@ func main() {
|
|||
mw := io.MultiWriter(os.Stdout, logFile)
|
||||
log.SetOutput(mw)
|
||||
|
||||
gormLogger := gormlogger.New(
|
||||
log.New(logFile, "\r\n", log.LstdFlags),
|
||||
gormlogger.Config{
|
||||
SlowThreshold: time.Second,
|
||||
LogLevel: gormlogger.Info,
|
||||
IgnoreRecordNotFoundError: true,
|
||||
Colorful: false,
|
||||
},
|
||||
)
|
||||
|
||||
log.Println("==================================================================")
|
||||
|
||||
// Get release type from command line
|
||||
|
|
@ -81,7 +92,7 @@ func main() {
|
|||
}))
|
||||
|
||||
// Connects to database
|
||||
if err = database.ConnectDB(); err != nil {
|
||||
if err = database.ConnectDB(gormLogger); err != nil {
|
||||
panic("Could not connect to database")
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue