feat: 初始化 meshgo MQTT 服务
- 支持 TCP / WebSocket 监听，配置热重载，systemd 集成 - meshAuthHook 实现用户名/密码认证与 ACL - meshLogHook 打印所有 MQTT 事件（CONNECT/PUBLISH/SUBSCRIBE 等） - meshDBHook 将 msh/# 主题 payload 异步写入数据库 - 数据库支持 SQLite（默认）和 MySQL，自动初始化并补充缺失配置 - payload_log 表字段：ID、client、topic、qos、payload、created_at、sender_ip - 自动补充 config.yaml 缺失字段（文件存在时写回） - .gitignore 屏蔽 data/ 和 .workbuddy/
This commit is contained in:
@@ -0,0 +1,215 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"log"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
const DefaultConfigPath = "/etc/meshgo/config.yaml"
|
||||
|
||||
// Config 主配置结构
|
||||
type Config struct {
|
||||
Server ServerConfig `yaml:"server"`
|
||||
Auth AuthConfig `yaml:"auth"`
|
||||
Logging LoggingConfig `yaml:"logging"`
|
||||
Database DatabaseConfig `yaml:"database"`
|
||||
}
|
||||
|
||||
// ServerConfig MQTT 监听相关配置
|
||||
type ServerConfig struct {
|
||||
// TCP 监听地址，如 :1883
|
||||
TCPAddr string `yaml:"tcp_addr"`
|
||||
// WebSocket 监听地址，如 :8883（留空则不启动）
|
||||
WSAddr string `yaml:"ws_addr"`
|
||||
// 最大并发连接数，0 表示不限
|
||||
MaxConnections int `yaml:"max_connections"`
|
||||
// 客户端消息写入超时（秒）
|
||||
WriteTimeout int `yaml:"write_timeout"`
|
||||
}
|
||||
|
||||
// AuthConfig 认证配置
|
||||
type AuthConfig struct {
|
||||
// 是否开启用户名/密码认证
|
||||
Enabled bool `yaml:"enabled"`
|
||||
// 允许匿名连接
|
||||
AllowAnonymous bool `yaml:"allow_anonymous"`
|
||||
// 内置用户列表（简单场景使用）
|
||||
Users []UserEntry `yaml:"users"`
|
||||
}
|
||||
|
||||
// UserEntry 单条用户凭证
|
||||
type UserEntry struct {
|
||||
Username string `yaml:"username"`
|
||||
Password string `yaml:"password"`
|
||||
}
|
||||
|
||||
// LoggingConfig 日志配置
|
||||
type LoggingConfig struct {
|
||||
// debug / info / warn / error
|
||||
Level string `yaml:"level"`
|
||||
// 日志文件路径，留空则输出到 stdout
|
||||
File string `yaml:"file"`
|
||||
}
|
||||
|
||||
// DatabaseConfig 数据库配置
|
||||
type DatabaseConfig struct {
|
||||
// 是否启用数据库，默认 false
|
||||
Enabled bool `yaml:"enabled"`
|
||||
// 数据库类型：sqlite3（默认） / mysql
|
||||
Type string `yaml:"type"`
|
||||
// SQLite 数据库文件路径（相对于 dataDir）
|
||||
// MySQL 留空，DSN 填 DSN 字段
|
||||
File string `yaml:"file"`
|
||||
// MySQL DSN，如 user:password@tcp(127.0.0.1:3306)/meshgo
|
||||
// SQLite 模式下此字段被忽略
|
||||
DSN string `yaml:"dsn"`
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// 全局单例
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
var (
|
||||
current *Config
|
||||
mu sync.RWMutex
|
||||
)
|
||||
|
||||
// Get 返回当前配置的只读快照
|
||||
func Get() *Config {
|
||||
mu.RLock()
|
||||
defer mu.RUnlock()
|
||||
return current
|
||||
}
|
||||
|
||||
// Load 从磁盘加载配置，替换全局单例
|
||||
func Load(path string) error {
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cfg := defaultConfig()
|
||||
if err = yaml.Unmarshal(data, cfg); err != nil {
|
||||
return err
|
||||
}
|
||||
// 补充缺失字段的默认值
|
||||
applyDefaults(cfg)
|
||||
mu.Lock()
|
||||
current = cfg
|
||||
mu.Unlock()
|
||||
log.Printf("[config] 已加载配置文件: %s", path)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Reload 重新读取配置文件（SIGHUP 时调用）
|
||||
func Reload(path string) {
|
||||
log.Printf("[config] 收到重载信号，重新读取: %s", path)
|
||||
if err := Load(path); err != nil {
|
||||
log.Printf("[config] 重载失败，继续使用旧配置: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// EnsureConfigComplete 确保配置文件存在且字段完整
|
||||
// - 文件不存在：创建带注释的默认配置
|
||||
// - 文件存在但缺失字段：补充缺失字段并写回文件
|
||||
func EnsureConfigComplete(path string) error {
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
// 文件不存在，创建默认配置
|
||||
cfg := defaultConfig()
|
||||
return writeDefaultConfig(path, cfg)
|
||||
}
|
||||
|
||||
// 文件存在，解析并补充缺失字段
|
||||
cfg := defaultConfig()
|
||||
if err = yaml.Unmarshal(data, cfg); err != nil {
|
||||
return err
|
||||
}
|
||||
if changed := applyDefaults(cfg); changed {
|
||||
log.Printf("[config] 检测到配置文件缺失字段，已自动补充: %s", path)
|
||||
if err = writeDefaultConfig(path, cfg); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// writeDefaultConfig 写入带注释头的默认配置
|
||||
func writeDefaultConfig(path string, cfg *Config) error {
|
||||
data, err := yaml.Marshal(cfg)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
header := `# meshgo MQTT 服务配置文件
|
||||
# 修改后执行: systemctl reload meshgo 即可热重载（无需重启服务）
|
||||
#
|
||||
# 热重载支持的字段：auth / logging
|
||||
# 需要重启才能生效的字段：server.tcp_addr / server.ws_addr / database
|
||||
|
||||
`
|
||||
return os.WriteFile(path, append([]byte(header), data...), 0640)
|
||||
}
|
||||
|
||||
// applyDefaults 补充零值字段的默认值，返回是否发生了修改
|
||||
func applyDefaults(cfg *Config) bool {
|
||||
changed := false
|
||||
// Server
|
||||
if cfg.Server.TCPAddr == "" {
|
||||
cfg.Server.TCPAddr = ":1883"
|
||||
changed = true
|
||||
}
|
||||
if cfg.Server.WriteTimeout == 0 {
|
||||
cfg.Server.WriteTimeout = 3
|
||||
changed = true
|
||||
}
|
||||
// Auth
|
||||
if !cfg.Auth.Enabled {
|
||||
cfg.Auth.Enabled = false
|
||||
cfg.Auth.AllowAnonymous = true
|
||||
changed = true
|
||||
}
|
||||
// Logging
|
||||
if cfg.Logging.Level == "" {
|
||||
cfg.Logging.Level = "info"
|
||||
changed = true
|
||||
}
|
||||
// Database
|
||||
if cfg.Database.Type == "" {
|
||||
cfg.Database.Type = "sqlite3"
|
||||
changed = true
|
||||
}
|
||||
if cfg.Database.File == "" {
|
||||
cfg.Database.File = "meshgo.db"
|
||||
changed = true
|
||||
}
|
||||
return changed
|
||||
}
|
||||
|
||||
// defaultConfig 返回一份合理的默认配置
|
||||
func defaultConfig() *Config {
|
||||
return &Config{
|
||||
Server: ServerConfig{
|
||||
TCPAddr: ":1883",
|
||||
WSAddr: "",
|
||||
MaxConnections: 0,
|
||||
WriteTimeout: 3,
|
||||
},
|
||||
Auth: AuthConfig{
|
||||
Enabled: false,
|
||||
AllowAnonymous: true,
|
||||
Users: []UserEntry{},
|
||||
},
|
||||
Logging: LoggingConfig{
|
||||
Level: "info",
|
||||
File: "",
|
||||
},
|
||||
Database: DatabaseConfig{
|
||||
Enabled: false,
|
||||
Type: "sqlite3",
|
||||
File: "meshgo.db",
|
||||
DSN: "",
|
||||
},
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user