240 lines
12 KiB
Go
240 lines
12 KiB
Go
package config
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"github.com/spf13/viper"
|
||
"joylink.club/iot/dto"
|
||
"log/slog"
|
||
)
|
||
|
||
const (
|
||
SystemName = "rtsts"
|
||
)
|
||
|
||
type AppConfig struct {
|
||
// Env string
|
||
Server server
|
||
Datasource datasource
|
||
Logging log
|
||
Messaging messaging
|
||
// Dynamics DynamicsConfig
|
||
// Vobc VobcConfig
|
||
// Interlock InterlockConfig
|
||
}
|
||
type server struct {
|
||
Port int
|
||
}
|
||
type datasource struct {
|
||
Dsn string
|
||
}
|
||
type log struct {
|
||
Level string // 日志打印级别 debug info warn error
|
||
Format string // 输出日志格式 logfmt, json
|
||
Path string // 输出日志文件路径
|
||
FileName string // 输出日志文件名称
|
||
FileMaxSize int // 【日志分割】单个日志文件最多存储量 单位(mb)
|
||
FileMaxBackups int // 【日志分割】日志备份文件最多数量
|
||
MaxAge int // 日志保留时间，单位: 天 (day)
|
||
Compress bool // 是否压缩日志
|
||
Stdout bool // 是否输出到控制台
|
||
}
|
||
|
||
type messaging struct {
|
||
Mqtt mqtt
|
||
}
|
||
|
||
// MQTT客户端配置
|
||
type mqtt struct {
|
||
Address string
|
||
Username string
|
||
Password string
|
||
}
|
||
|
||
// type centrifugo struct {
|
||
// TokenSecret string
|
||
// ApiKey string
|
||
// ApiEndpoint string
|
||
// Address string
|
||
// }
|
||
|
||
// 第三方配置结构
|
||
type ThirdPartyConfig struct {
|
||
Id int32 `json:"id"`
|
||
Dynamics DynamicsConfig `json:"dynamics" description:"动力学配置"`
|
||
Vobc VobcConfig `json:"vobc" description:"半实物配置"`
|
||
Interlocks []InterlockConfig `json:"interlock" description:"联锁配置"`
|
||
RsspAxleConfig RsspAxleConfig `json:"rsspAxleCfgs" description:"计轴通信配置"`
|
||
//ElectricMachinery ElectricMachineryConfig `json:"electricMachinery" description:"电机配置"`
|
||
ElectricMachinerys []ElectricMachineryConfig `json:"electricMachinerys" description:"电机配置"`
|
||
BtmCanet BtmCanetConfig `json:"btmCanet" description:"BTM关联的网关设备CANET配置"`
|
||
CidcModbus []CidcModbusConfig `json:"cidcModbus" description:"联锁驱采Modbus接口配置"`
|
||
Radar RadarConfig `json:"radar" description:"车载雷达相关配置"`
|
||
Acc AccConfig `json:"acc" description:"车载加速计"`
|
||
//PcSimConfig VehiclePCSimConfig `json:"pcSimConfig" description:"车载pc仿真平台通信"`
|
||
PcSimConfigs []VehiclePCSimConfig `json:"pcSimConfigs" description:"车载pc仿真平台通信"`
|
||
BtmVobc BtmVobcConfig `json:"btmVobc" description:"Btm Vobc 11号线相关配置"`
|
||
}
|
||
type RadarConfig struct {
|
||
Open bool `json:"open" description:"是否开启"`
|
||
RemoteIp string `json:"remoteIp" description:"远端接收列车信息ip"`
|
||
RemotePort uint32 `json:"remotePort" description:"远端接收列车信息端口"`
|
||
//LocalPort uint32 `json:"localPort" description:"本地端口"`
|
||
}
|
||
|
||
type AccConfig struct {
|
||
Open bool `json:"open" description:"是否开启"`
|
||
RemoteIp string `json:"remoteIp" description:"远端接收列车信息ip"`
|
||
RemotePort uint32 `json:"remotePort" description:"远端接收列车信息端口"`
|
||
}
|
||
type CidcModbusConfig struct {
|
||
Open bool `json:"open" description:"是否开启"`
|
||
Url string `json:"url" description:"接口URL【格式tcp://{ip}:{port}】" default:"tcp://127.0.0.1:502"` // 连接地址
|
||
UnitId uint32 `json:"unitId" description:"从机unitId"` // 从机unitId
|
||
Endianness dto.Modbus_Endianness `json:"endianness" description:"字节序【大端/小端】"` // 16位寄存器字节序
|
||
Interval uint32 `json:"interval" description:"定时请求间隔" default:"1000"` // 循环请求间隔(毫秒),0表示不主动请求,只当有变化时请求
|
||
Timeout uint32 `json:"timeout" description:"请求超时时间" default:"1000"` // 超时时间（毫秒）
|
||
Ecs string `json:"ecs" description:"联锁集中站"` // 所属集中站
|
||
Mapping []ModbusDcMapping `json:"mapping" description:"modbus数据与驱动/采集码表映射配置"`
|
||
}
|
||
type ModbusDcMapping struct {
|
||
Function dto.Modbus_Function `json:"function" description:"Modbus功能"` // 功能
|
||
Addr uint32 `json:"addr" description:"Modbus功能起始地址【位类型的功能为起始位地址,寄存器类型的功能为起始字(2个字节)地址】"` // 起始地址,当功能为位功能时,表示起始位地址,当功能为寄存器功能时,表示起始字(2个字节)地址
|
||
Quantity uint32 `json:"quantity" description:"Modbus读取数量【位类型的为位数,寄存器类型的为字(2个字节)数】"` // 数量,当功能为位功能时,表示位数,当功能为寄存器功能时,表示字(2个字节)数
|
||
// WriteStrategy proto.Modbus_WriteStrategy `json:"writeStrategy" description:"写入策略"` // 当功能为写入类功能时(不包含读写类功能),写策略
|
||
Type dto.DataType `json:"type" description:"映射的数据类型"` // 对应数据类型
|
||
Start uint32 `json:"start" description:"映射数据的起始地址【位类型的为起始位地址,寄存器类型的为起始字节地址】"` // 映射起始地址
|
||
}
|
||
type DynamicsConfig struct {
|
||
Ip string `json:"ip" description:"IP配置"`
|
||
UdpLocalPort int `json:"udpLocalPort" description:"本机监听接收端口"`
|
||
UdpRemotePort int `json:"udpRemotePort" description:"远端接收道岔信息端口"`
|
||
UdpRemoteTrainPort int `json:"udpRemoteTrainPort" description:"远端接收列车信息端口"`
|
||
HttpPort int `json:"httpPort" description:"http服务端口"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
}
|
||
type VobcConfig struct {
|
||
Ip string `json:"ip" description:"IP配置"`
|
||
LocalPort int `json:"localPort" description:"本机监听接收端口"`
|
||
RemotePort int `json:"remotePort" description:"远端接收列车信息端口"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
}
|
||
type InterlockConfig struct {
|
||
Open bool `json:"open" description:"是否开启"`
|
||
Ip string `json:"ip" description:"IP配置"`
|
||
LocalPort int `json:"localPort" description:"本机监听接收端口"`
|
||
RemotePort int `json:"remotePort" description:"远端接收采集信息端口"`
|
||
Code string `json:"code" description:"所属集中站"`
|
||
Line string `json:"line" description:"联锁通信协议所属线路"`
|
||
}
|
||
type ElectricMachineryConfig struct {
|
||
Ip string `json:"ip" description:"IP配置"`
|
||
RemotePort int `json:"remotePort" description:"远端接收信息端口"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
EndPointA bool `json:"endPointA" description:"一号电机"`
|
||
}
|
||
|
||
// BtmVobcConfig 11 号线 BTM vobc 网关设备配置
|
||
type BtmVobcConfig struct {
|
||
LocalUdpIp string `json:"LocalUdpIp" description:"本机监听接收UDP ip 配置"`
|
||
LocalUdpPort int `json:"localUdpPort" description:"本机监听接收UDP端口"`
|
||
RemoteIp string `json:"remoteIp" description:"btm串口设备IP配置"`
|
||
RemoteUdpPort int `json:"remoteUdpPort" description:"btm串口设备UDP端口"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
}
|
||
|
||
// BtmCanetConfig BTM CANET网关设备配置
|
||
type BtmCanetConfig struct {
|
||
LocalUdpPort int `json:"localUdpPort" description:"本机监听接收UDP端口"`
|
||
RemoteIp string `json:"remoteIp" description:"CANET设备IP配置"`
|
||
RemoteUdpPort int `json:"remoteUdpPort" description:"CANET设备UDP端口"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
}
|
||
|
||
type VehiclePCSimConfig2 struct {
|
||
TrainEnds bool `json:"trainEnds" description:"列车端点A"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
PcSimIp string `json:"pcSimIp" description:"pc仿真平台通信ip"`
|
||
PcSimPort uint32 `json:"pcSimPort" description:"pc仿真平台通信端口"`
|
||
LocalTestingPort uint32 `json:"localTestingPort" description:"本地测试端口"`
|
||
}
|
||
|
||
type VehiclePCSimConfig struct {
|
||
//TrainEnds bool `json:"trainEnds" description:"列车端点A?"`
|
||
ConfigName string `json:"configName" description:"连接名称"`
|
||
Open bool `json:"open" description:"是否开启"`
|
||
PcSimIp string `json:"pcSimIp" description:"pc仿真平台通信ip"`
|
||
PcSimPort uint32 `json:"pcSimPort" description:"pc仿真平台通信端口"`
|
||
//LocalTestingPort uint32 `json:"localTestingPort" description:"本地测试端口"`
|
||
}
|
||
|
||
// RsspAxleConfig Rssp计轴通信配置
|
||
type RsspAxleConfig struct {
|
||
Open bool `json:"open" description:"开启"`
|
||
StationCode string `json:"stationCode" description:"集中站编号"`
|
||
NetAConfig RsspNetConfig `json:"netAConfig" description:"A网配置"`
|
||
}
|
||
|
||
// RsspNetConfig 计轴通信配置
|
||
type RsspNetConfig struct {
|
||
RemoteIp string `json:"remoteIp" description:"远端IP"`
|
||
RemotePort int `json:"remotePort" description:"远端端口"`
|
||
LocalPort int `json:"localPort" description:"本地端口"`
|
||
|
||
RemoteAddr string `json:"sourceAddr" description:"联锁地址（16进制，2字节）"`
|
||
LocalAddr string `json:"targetAddr" description:"计轴地址（16进制，2字节）"`
|
||
|
||
RemoteSid1 string `json:"remoteSid1" description:"联锁SID_1（16进制，4字节）"`
|
||
RemoteSid2 string `json:"remoteSid2" description:"联锁SID_2（16进制，4字节）"`
|
||
LocalSid1 string `json:"localSid1" description:"计轴SID_1（16进制，4字节）"`
|
||
LocalSid2 string `json:"localSid2" description:"计轴SID_2（16进制，4字节）"`
|
||
|
||
RemoteSinit1 string `json:"remoteSinit1" description:"联锁SINIT_1（16进制，4字节）"`
|
||
RemoteSinit2 string `json:"remoteSinit2" description:"联锁SINIT_2（16进制，4字节）"`
|
||
LocalSinit1 string `json:"localSinit1" description:"计轴SINIT_1（16进制，4字节）"`
|
||
LocalSinit2 string `json:"localSinit2" description:"计轴SINIT_2（16进制，4字节）"`
|
||
|
||
RemoteDataVer1 string `json:"remoteDataVer1" description:"联锁DATAVER_1（16进制，4字节）"`
|
||
RemoteDataVer2 string `json:"remoteDataVer2" description:"联锁DATAVER_2（16进制，4字节）"`
|
||
LocalDataVer1 string `json:"localDataVer1" description:"计轴DATAVER_1（16进制，4字节）"`
|
||
LocalDataVer2 string `json:"localDataVer2" description:"计轴DATAVER_2（16进制，4字节）"`
|
||
|
||
MaxDeviation int `json:"maxDeviation" description:"可容忍的最大时序偏差"`
|
||
WaitSSRTimeout int `json:"waitSSRTimeout" description:"等待SSR回应的定时器超时值（ms）"`
|
||
Period int `json:"period" description:"RSD发送周期（ms）"`
|
||
}
|
||
|
||
///////////////////////////////////////////////////////////////////////////////////////
|
||
|
||
var Config AppConfig
|
||
|
||
// 获取配置文件名称,从运行flag参数config中获取，若未提供，使用默认'dev'
|
||
func getConfigName() string {
|
||
configName := ""
|
||
flag.StringVar(&configName, "config", "dev", "config name, eg: -config test")
|
||
flag.Parse()
|
||
if configName == "" {
|
||
configName = "dev"
|
||
}
|
||
slog.Info("读取配置文件", "配置文件名称", configName)
|
||
return configName
|
||
}
|
||
|
||
// 加载配置
|
||
func LoadConfig() {
|
||
cnf := viper.New()
|
||
cnf.SetConfigName(getConfigName())
|
||
cnf.SetConfigType("yml")
|
||
cnf.AddConfigPath("./config/")
|
||
cnf.AddConfigPath(".")
|
||
err := cnf.ReadInConfig()
|
||
if err != nil {
|
||
panic(fmt.Errorf("读取配置文件错误: %w", err))
|
||
}
|
||
err = cnf.Unmarshal(&Config)
|
||
if err != nil {
|
||
panic(fmt.Errorf("解析配置文件错误: %w", err))
|
||
}
|
||
slog.Info("成功加载配置", "config", Config)
|
||
}
|