修改文件上传逻辑配置

This commit is contained in:
2025-10-31 17:35:44 +08:00
parent d634ce1993
commit 49f948b062
2 changed files with 18 additions and 4 deletions

View File

@@ -12,6 +12,7 @@ type Config struct {
OSS OSSConfig `yaml:"oss"` OSS OSSConfig `yaml:"oss"`
MySQL MySQLConfig `yaml:"mysql"` MySQL MySQLConfig `yaml:"mysql"`
Upload UploadConfig `yaml:"upload"` Upload UploadConfig `yaml:"upload"`
FileUpload FileUploadConfig `yaml:"fileupload"`
} }
// ServerConfig 服务器配置 // ServerConfig 服务器配置
@@ -45,6 +46,13 @@ type UploadConfig struct {
MaxFileSize int64 `yaml:"max_file_size"` 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 从文件加载配置 // LoadConfig 从文件加载配置
func LoadConfig(filename string) (*Config, error) { func LoadConfig(filename string) (*Config, error) {
data, err := os.ReadFile(filename) data, err := os.ReadFile(filename)

View File

@@ -20,3 +20,9 @@ mysql:
upload: upload:
allow_image_types: "image/jpeg,image/png,image/gif,image/webp" allow_image_types: "image/jpeg,image/png,image/gif,image/webp"
max_file_size: 5242880 # 5MB (5 * 1024 * 1024) max_file_size: 5242880 # 5MB (5 * 1024 * 1024)
fileupload:
max_file_size: 1073741824 # 1GB
allow_file_types: "video/mp4,video/avi,application/pdf,image/jpeg,image/png,application/vnd.openxmlformats-officedocument.wordprocessingml.document,text/plain,application/json"
chunk_size: 5242880 # 分片容量
temp_dir: "./temp/chunks" # 临时分片目录(需提前创建,确保权限)