hldrCenter/server/config/config.go

package config

import (
	"os"

	"gopkg.in/yaml.v3"
)

// Config 应用配置结构体
type Config struct {
	Server     ServerConfig     `yaml:"server"`
	OSS        OSSConfig        `yaml:"oss"`
	MySQL      MySQLConfig      `yaml:"mysql"`
	Upload     UploadConfig     `yaml:"upload"`
	FileUpload FileUploadConfig `yaml:"fileupload"`
}

// ServerConfig 服务器配置
type ServerConfig struct {
	Port           string   `yaml:"port"`
	AllowedOrigins []string `yaml:"allowed_origins"`
}

// OSSConfig 阿里云OSS配置
type OSSConfig struct {
	Endpoint        string `yaml:"endpoint"`
	BucketName      string `yaml:"bucket_name"`
	AccessKeyID     string `yaml:"access_key_id"`
	AccessKeySecret string `yaml:"access_key_secret"`
	ObjectPrefix    string `yaml:"object_prefix"`
}

// MySQLConfig MySQL数据库配置
type MySQLConfig struct {
	Host     string `yaml:"host"`
	Port     int    `yaml:"port"`
	Username string `yaml:"username"`
	Password string `yaml:"password"`
	Database string `yaml:"database"`
	Charset  string `yaml:"charset"`
}

// UploadConfig 上传配置
type UploadConfig struct {
	AllowImageTypes string `yaml:"allow_image_types"`
	MaxFileSize     int64  `yaml:"max_file_size"`
}

type FileUploadConfig struct {
	MaxFileSize    int64  `yaml:"max_file_size"`    // 最大文件大小（字节）
	AllowFileTypes string `yaml:"allow_file_types"` // 允许的文件类型，用逗号分隔
	ChunkSize      int64  `yaml:"chunk_size"`       // 分片大小（5MB）
	TempDir        string `yaml:"temp_dir"`         // 分片临时存储目录
}

// LoadConfig 从文件加载配置
func LoadConfig(filename string) (*Config, error) {
	data, err := os.ReadFile(filename)
	if err != nil {
		return nil, err
	}

	var config Config
	if err := yaml.Unmarshal(data, &config); err != nil {
		return nil, err
	}

	return &config, nil
}