File tree Expand file tree Collapse file tree 1 file changed +12
-3
lines changed
Expand file tree Collapse file tree 1 file changed +12
-3
lines changed Original file line number Diff line number Diff line change 66 "gorm.io/driver/postgres"
77 "gorm.io/driver/sqlite"
88 "gorm.io/gorm"
9+ "gorm.io/gorm/logger"
910)
1011
1112const dbVersion = "1"
@@ -50,23 +51,31 @@ func (h *Headscale) initDB() error {
5051func (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
You can’t perform that action at this time.
0 commit comments