// Package config 负责读程序的配置文件(application.properties)。 // // 大白话理解:程序启动时先看看配置文件里写了啥,没写就用默认值。 // 配置文件就在程序旁边,叫 application.properties,用记事本就能改。 package config import ( "runtime" "strconv" "strings" "github.com/magiconair/properties" ) // Config 程序配置: // - Port:网页服务用哪个端口(默认 8080) // - Dsn:数据库文件叫什么(默认 media-info.db,放在程序目录下) // - AutoOpen:启动后要不要自动打开浏览器 // - FFmpegPath:ffmpeg 程序路径(用于把带 EXIF 旋转标记的图片转正);留空自动在 PATH 里找 // - RotateImages:是否处理图片 EXIF 旋转(默认 true);false 则完全不处理,按存储方向/尺寸归类 // - Workers:探测/MIME/移动阶段的并发 worker 数(默认 4;设 0 表示用满全部 CPU 核数) // // 注意:NSFW 检测相关配置已改为从数据库读取(通过网页「系统配置」设置) type Config struct { Port int Dsn string AutoOpen bool FFmpegPath string RotateImages bool Workers int } // appConfig 当前生效的配置(程序运行期间一直不变)。 var appConfig Config // GetAppConfig 返回当前生效的配置。 func GetAppConfig() Config { return appConfig } // LoadAppConfig 从 application.properties 读配置;文件不存在就用默认值。 func LoadAppConfig() { // 先给一套默认值 appConfig = Config{Port: 8080, Dsn: "media-info.db", AutoOpen: true, RotateImages: true, Workers: 4} p, err := properties.LoadFile("application.properties", properties.UTF8) if err != nil { return // 没有配置文件,就用默认值 } // 配置文件里写了就覆盖默认值 if port := p.GetInt("server.port", 0); port > 0 { appConfig.Port = port } if v, ok := p.Get("database.dsn"); ok && v != "" { appConfig.Dsn = v } appConfig.AutoOpen = p.GetBool("server.autoOpen", true) // ffmpeg 路径:留空表示自动在 PATH 里找 if v, ok := p.Get("ffmpeg.path"); ok && v != "" { appConfig.FFmpegPath = v } appConfig.RotateImages = p.GetBool("organize.rotateImages", true) // 并发 worker 数:只认非负整数,填了别的就保持默认 if v, ok := p.Get("organize.workers"); ok && strings.TrimSpace(v) != "" { if n, err := strconv.Atoi(strings.TrimSpace(v)); err == nil && n >= 0 { appConfig.Workers = n } } } // WorkerCount 返回探测/MIME/移动阶段使用的并发 worker 数: // - 配置里 organize.workers > 0 时用配置值(默认 4); // - 配置为 0 时用满全部 CPU 核数。 func WorkerCount() int { if appConfig.Workers > 0 { return appConfig.Workers } n := runtime.NumCPU() if n < 1 { return 1 } return n }