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