111 lines
4.0 KiB
Go
111 lines
4.0 KiB
Go
// Package config define config struct of model runtime service
|
|
package config
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// BaseConfig define config struct of base params config
|
|
type BaseConfig struct {
|
|
GridID int64 `mapstructure:"grid_id"`
|
|
ZoneID int64 `mapstructure:"zone_id"`
|
|
StationID int64 `mapstructure:"station_id"`
|
|
}
|
|
|
|
// ServiceConfig define config struct of service config
|
|
type ServiceConfig struct {
|
|
ServiceName string `mapstructure:"service_name"`
|
|
SecretKey string `mapstructure:"secret_key"`
|
|
}
|
|
|
|
// KafkaConfig define config struct of kafka config
|
|
type KafkaConfig struct {
|
|
Servers string `mapstructure:"Servers"`
|
|
GroupID string `mapstructure:"group_id"`
|
|
Topic string `mapstructure:"topic"`
|
|
AutoOffsetReset string `mapstructure:"auto_offset_reset"`
|
|
EnableAutoCommit string `mapstructure:"enable_auto_commit"`
|
|
ReadMessageTimeDuration string `mapstructure:"read_message_time_duration"`
|
|
}
|
|
|
|
// PostgresConfig define config struct of postgres config
|
|
type PostgresConfig struct {
|
|
Port int `mapstructure:"port"`
|
|
Host string `mapstructure:"host"`
|
|
DataBase string `mapstructure:"database"`
|
|
User string `mapstructure:"user"`
|
|
Password string `mapstructure:"password"`
|
|
}
|
|
|
|
// LoggerConfig define config struct of zap logger config
|
|
type LoggerConfig struct {
|
|
Mode string `mapstructure:"mode"`
|
|
Level string `mapstructure:"level"`
|
|
FilePath string `mapstructure:"filepath"`
|
|
MaxSize int `mapstructure:"maxsize"`
|
|
MaxBackups int `mapstructure:"maxbackups"`
|
|
MaxAge int `mapstructure:"maxage"`
|
|
Compress bool `mapstructure:"compress"`
|
|
}
|
|
|
|
// RedisConfig define config struct of redis config
|
|
type RedisConfig struct {
|
|
Addr string `mapstructure:"addr"`
|
|
Password string `mapstructure:"password"`
|
|
DB int `mapstructure:"db"`
|
|
PoolSize int `mapstructure:"poolsize"`
|
|
Timeout int `mapstructure:"timeout"`
|
|
}
|
|
|
|
// AntsConfig define config struct of ants pool config
|
|
type AntsConfig struct {
|
|
ParseConcurrentQuantity int `mapstructure:"parse_concurrent_quantity"` // parse comtrade file concurrent quantity
|
|
RTDReceiveConcurrentQuantity int `mapstructure:"rtd_receive_concurrent_quantity"` // polling real time data concurrent quantity
|
|
}
|
|
|
|
// DataRTConfig define config struct of data runtime server api config
|
|
type DataRTConfig struct {
|
|
Host string `mapstructure:"host"`
|
|
Port int64 `mapstructure:"port"`
|
|
PollingAPI string `mapstructure:"polling_api"`
|
|
Method string `mapstructure:"polling_api_method"`
|
|
}
|
|
|
|
// ModelRTConfig define config struct of model runtime server
|
|
type ModelRTConfig struct {
|
|
BaseConfig `mapstructure:"base"`
|
|
ServiceConfig `mapstructure:"service"`
|
|
PostgresConfig `mapstructure:"postgres"`
|
|
KafkaConfig `mapstructure:"kafka"`
|
|
LoggerConfig `mapstructure:"logger"`
|
|
AntsConfig `mapstructure:"ants"`
|
|
DataRTConfig `mapstructure:"dataRT"`
|
|
LockerRedisConfig RedisConfig `mapstructure:"locker_redis"`
|
|
StorageRedisConfig RedisConfig `mapstructure:"storage_redis"`
|
|
PostgresDBURI string `mapstructure:"-"`
|
|
}
|
|
|
|
// ReadAndInitConfig return modelRT project config struct
|
|
func ReadAndInitConfig(configDir, configName, configType string) (modelRTConfig ModelRTConfig) {
|
|
config := viper.New()
|
|
config.AddConfigPath(configDir)
|
|
config.SetConfigName(configName)
|
|
config.SetConfigType(configType)
|
|
if err := config.ReadInConfig(); err != nil {
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
|
|
panic(fmt.Sprintf("can not find conifg file:%s\n", err.Error()))
|
|
}
|
|
panic(err)
|
|
}
|
|
|
|
if err := config.Unmarshal(&modelRTConfig); err != nil {
|
|
panic(fmt.Sprintf("unmarshal modelRT config failed:%s\n", err.Error()))
|
|
}
|
|
|
|
// init postgres db uri
|
|
modelRTConfig.PostgresDBURI = fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s", modelRTConfig.PostgresConfig.Host, modelRTConfig.PostgresConfig.Port, modelRTConfig.PostgresConfig.User, modelRTConfig.PostgresConfig.Password, modelRTConfig.PostgresConfig.DataBase)
|
|
return modelRTConfig
|
|
}
|