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 d681f27c0a
2 changed files with 22 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
database/db/new_test.go
View File

@@ -5,21 +5,21 @@ import (
)
func TestNew(t *testing.T) {
//mdb, err := New(WithMysql("127.0.0.1", 3306, "root", "MyPassw0rd", "mydb"))
//if err != nil {
// t.Fatal(err)
//}
//
//type User struct {
// Id uint64 `gorm:"primaryKey"`
// Username string `gorm:"unique"`
//}
//
//if err = mdb.Session(t.Context()).AutoMigrate(&User{}); err != nil {
// t.Fatal(err)
//}
//
//if err = mdb.Session(t.Context()).Create(&User{Username: "zyp"}).Error; err != nil {
// t.Fatal(err)
//}
mdb, err := New(WithMysql("127.0.0.1", 2881, "yizhi@test", "yizhi", "mie"))
if err != nil {
t.Fatal(err)
}
type User struct {
Id uint64 `gorm:"primaryKey"`
Username string `gorm:"unique"`
}
if err = mdb.Session(t.Context()).AutoMigrate(&User{}); err != nil {
t.Fatal(err)
}
if err = mdb.Session(t.Context()).Create(&User{Username: "zyp"}).Error; err != nil {
t.Fatal(err)
}
}

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
}
}