任务三

internal/data/data.go

@@ -1,6 +1,7 @@
 package data
 
 import (
+	"fmt"
 	"kra/internal/conf"
 	"kra/internal/data/query"
 
@@ -20,8 +21,10 @@ type Data struct {
 }
 
 // NewDB 创建数据库连接
-func NewDB(c *conf.Data) (*gorm.DB, error) {
-	db, err := gorm.Open(mysql.Open(c.Database.Source), &gorm.Config{
+func NewDB(c *conf.Mysql) (*gorm.DB, error) {
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?%s",
+		c.Username, c.Password, c.Path, c.Port, c.DbName, c.Config)
+	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
 		Logger: logger.Default.LogMode(logger.Info),
 	})
 	if err != nil {