更新

pkg/config/config.go

@@ -24,7 +24,8 @@ var AppConfig *Config
 func InitConfig() {
 	viper.SetConfigName("config")
 	viper.SetConfigType("yaml")
-	viper.AddConfigPath(".") // 在当前目录查找 config.yaml
+	viper.AddConfigPath(".")        // 在当前目录查找 config.yaml
+	viper.AddConfigPath("./config") // 兼容多一层目录的情况
 
 	if err := viper.ReadInConfig(); err != nil {
 		panic(fmt.Errorf("无法读取配置文件: %w", err))
@@ -34,3 +35,15 @@ func InitConfig() {
 		panic(fmt.Errorf("无法解析配置结构体: %w", err))
 	}
 }
+
+// GetDSN 生成 MySQL 连接字符串
+func GetDSN() string {
+	db := AppConfig.Database
+	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local",
+		db.User, db.Password, db.Host, db.Port, db.Dbname, db.Charset)
+}
+
+// GetServerPort 获取服务器端口字符串
+func GetServerPort() string {
+	return fmt.Sprintf(":%d", AppConfig.Server.Port)
+}