yizhisec/packages
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: add init db debug log

Browse Source
This commit is contained in:
zhaoyupeng
2025-10-31 22:23:35 +08:00
parent 36fd04cac9
commit 42331cde57
2 changed files with 22 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
database/db/option.go
View File

@@ -3,6 +3,8 @@ package db
import (
"context"
"fmt"
"gitea.loveuer.com/yizhisec/packages/logger"
)
type config struct {
@@ -25,6 +27,7 @@ func WithCtx(ctx context.Context) OptionFn {
func WithMysql(host string, port int, user string, password string, database string) OptionFn {
return func(c *config) {
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", user, password, host, port, database)
logger.Debug("db init with mysql, dsn: %s", dsn)
c.mysql = &dsn
}
}
@@ -32,6 +35,7 @@ func WithMysql(host string, port int, user string, password string, database str
func WithPg(host string, port int, user string, password string, database string) OptionFn {
return func(c *config) {
dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai", host, user, password, database, port)
logger.Debug("db init with pg, dsn: %s", dsn)
c.pg = &dsn
}
}
@@ -39,6 +43,7 @@ func WithPg(host string, port int, user string, password string, database string
func WithSqlite(path string) OptionFn {
return func(c *config) {
if path != "" {
logger.Debug("db init with sqlite, path: %s", path)
c.sqlite = &path
}
}