From 42331cde577ff7c65721c3392c9e801d30b94333 Mon Sep 17 00:00:00 2001
From: zhaoyupeng <zhaoyupeng@yizhisec.com>
Date: Fri, 31 Oct 2025 22:23:35 +0800
Subject: [PATCH] chore: add init db debug log

---
 database/db/new_test.go | 34 +++++++++++++++++-----------------
 database/db/option.go   |  5 +++++
 2 files changed, 22 insertions(+), 17 deletions(-)

diff --git a/database/db/new_test.go b/database/db/new_test.go
index d7ba95b..d549c3f 100644
--- a/database/db/new_test.go
+++ b/database/db/new_test.go
@@ -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)
+	}
 }
diff --git a/database/db/option.go b/database/db/option.go
index 4faf60b..a1e857f 100644
--- a/database/db/option.go
+++ b/database/db/option.go
@@ -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
 		}
 	}