Skip to content

Commit baba861

Browse files
committed
feat(dal): add database driver logging
- Add logging for the selected database driver - Remove unnecessary empty lines in the code
1 parent a211bfa commit baba861

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

internal/dal/repository/repository.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import (
99
"github.com/duke-git/lancet/v2/cryptor"
1010
"github.com/glebarez/sqlite"
1111
"github.com/spf13/viper"
12+
"go.uber.org/zap"
1213
"gorm.io/driver/mysql"
1314
"gorm.io/driver/postgres"
1415
"gorm.io/gorm"
@@ -55,11 +56,11 @@ func NewDB(conf *viper.Viper, l *log.Logger) *gorm.DB {
5556
case "postgres":
5657
dialector = postgres.Open(dsn)
5758
case "sqlite":
58-
5959
dialector = sqlite.Open(dsn)
6060
default:
6161
panic("unknown db driver")
6262
}
63+
l.Info("db driver", zap.String("driver", dialector.Name()))
6364

6465
db, err = gorm.Open(dialector, &gorm.Config{
6566
QueryFields: true,
@@ -246,7 +247,6 @@ func autoMigrateAndInitialize(db *gorm.DB) {
246247
SiteKeyword: "网址导航",
247248
SiteDesc: "WebStack-Go - 基于 Golang 开源的网址导航网站",
248249
})
249-
250250
if err != nil {
251251
os.Exit(0)
252252
}

0 commit comments

Comments
 (0)