Skip to content

Commit 29b80e3

Browse files
committed
Fix debug mode enabled by default in db
1 parent a16a763 commit 29b80e3

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

db.go

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"gorm.io/driver/postgres"
77
"gorm.io/driver/sqlite"
88
"gorm.io/gorm"
9+
"gorm.io/gorm/logger"
910
)
1011

1112
const dbVersion = "1"
@@ -50,23 +51,31 @@ func (h *Headscale) initDB() error {
5051
func (h *Headscale) openDB() (*gorm.DB, error) {
5152
var db *gorm.DB
5253
var err error
54+
55+
var log logger.Interface
56+
if h.dbDebug {
57+
log = logger.Default
58+
} else {
59+
log = logger.Default.LogMode(logger.Silent)
60+
}
61+
5362
switch h.dbType {
5463
case "sqlite3":
5564
db, err = gorm.Open(sqlite.Open(h.dbString), &gorm.Config{
5665
DisableForeignKeyConstraintWhenMigrating: true,
66+
Logger: log,
5767
})
5868
case "postgres":
5969
db, err = gorm.Open(postgres.Open(h.dbString), &gorm.Config{
6070
DisableForeignKeyConstraintWhenMigrating: true,
71+
Logger: log,
6172
})
6273
}
6374

6475
if err != nil {
6576
return nil, err
6677
}
67-
if h.dbDebug {
68-
db.Debug()
69-
}
78+
7079
return db, nil
7180
}
7281

0 commit comments

Comments
 (0)