MTP20_IF/config/config.go

package config

import (
	"errors"
	"fmt"
	"strconv"

	"github.com/beevik/etree"
)

var SerCfg *ServiceConfig

type WebConfig struct {
	ListenAddr string
	DebugMode  bool
}

type LogConfig struct {
	LogPath  string
	LogLevel int
}

type DbConfig struct {
	DbDriver  string
	DbAddress string
	DbName    string
	DbPort    string
	DbUser    string
	DbPwd     string
}

type RedisConfig struct {
	Address string
	Port    int
	TimeOut int
	ConnNum int
	DbNum   int
	Pwd     string
}

type MqConfig struct {
	Url      string
	Exchange string
}

type ServiceConfig struct {
	WebCfg   WebConfig
	LogCfg   LogConfig
	DbCfg    DbConfig
	RedisCfg RedisConfig
	MqCfg    MqConfig
}

func (c *ServiceConfig) Init(path string) error {
	SerCfg = &ServiceConfig{}

	cfgfile := etree.NewDocument()
	err := cfgfile.ReadFromFile(path)
	if err != nil {
		fmt.Println("Read config file failed")
		return err
	}

	root := cfgfile.SelectElement("Configuration")
	if root == nil {
		return errors.New("read config file failed")
	}

	// web配置
	websettings := root.SelectElements("WebSetting")
	for _, setting := range websettings {
		web_listenaddr := setting.SelectElement("ListenAddress")
		if web_listenaddr == nil {
			return errors.New("read web listen address failed")
		}
		SerCfg.WebCfg.ListenAddr = web_listenaddr.SelectAttrValue("value", "0.0.0.0:8080")

		// 调试模式
		debug_mode := setting.SelectElement("DebugMode")
		if debug_mode == nil {
			return errors.New("read debug mode failed")
		}
		ret, err := strconv.ParseUint(debug_mode.SelectAttrValue("value", "0"), 10, 32)
		if err != nil {
			return errors.New("read debug mode is invalid:")
		}
		SerCfg.WebCfg.DebugMode = (ret == 1)
	}

	// 日志配置
	logsettings := root.SelectElements("LogSetting")
	for _, setting := range logsettings {
		// 日志路径
		log_path := setting.SelectElement("LogPath")
		if log_path == nil {
			return errors.New("read log path failed")
		}
		SerCfg.LogCfg.LogPath = log_path.SelectAttrValue("value", "./")

		// 日志级别
		log_level := setting.SelectElement("LogLevel")
		if log_level == nil {
			return errors.New("read log level failed")
		}
		ret, err := strconv.ParseUint(log_level.SelectAttrValue("value", "5"), 10, 32)
		if err != nil {
			return errors.New("read log level is invalid:")
		}
		SerCfg.LogCfg.LogLevel = int(ret)
	}

	// 数据库配置
	dbsettings := root.SelectElements("DbSetting")
	for _, setting := range dbsettings {
		// 数据库驱动
		db_driver := setting.SelectElement("DbDriver")
		if db_driver == nil {
			return errors.New("read db driver failed")
		}
		SerCfg.DbCfg.DbDriver = db_driver.SelectAttrValue("value", "oci8")

		// 数据库地址
		db_address := setting.SelectElement("DbAddress")
		if db_address == nil {
			return errors.New("read db address failed")
		}
		SerCfg.DbCfg.DbAddress = db_address.SelectAttrValue("value", "0.0.0.0")

		// 数据库名
		db_name := setting.SelectElement("DbName")
		if db_name == nil {
			return errors.New("read db name failed")
		}
		SerCfg.DbCfg.DbName = db_name.SelectAttrValue("value", "orcl")

		// 数据库端口
		db_port := setting.SelectElement("DbPort")
		if db_port == nil {
			return errors.New("read db port failed")
		}
		SerCfg.DbCfg.DbPort = db_port.SelectAttrValue("value", "1521")

		// 数据库名
		db_user := setting.SelectElement("DbUser")
		if db_user == nil {
			return errors.New("read db user failed")
		}
		SerCfg.DbCfg.DbUser = db_user.SelectAttrValue("value", "mtp2_test01")

		// 数据库密码
		db_pwd := setting.SelectElement("DbPwd")
		if db_pwd == nil {
			return errors.New("read db password failed")
		}
		SerCfg.DbCfg.DbPwd = db_pwd.SelectAttrValue("value", "muchinfo")
	}

	// Redis配置
	redissettings := root.SelectElements("RedisSetting")
	for _, setting := range redissettings {
		// 地址
		redis_address := setting.SelectElement("Address")
		if redis_address == nil {
			return errors.New("read redis address failed")
		}
		SerCfg.RedisCfg.Address = redis_address.SelectAttrValue("value", "127.0.0.1")

		// 端口
		redis_port := setting.SelectElement("Port")
		if redis_port == nil {
			return errors.New("read redis port failed")
		}
		ret, err := strconv.ParseUint(redis_port.SelectAttrValue("value", "5007"), 10, 32)
		if err != nil {
			return errors.New("read redis port is invalid:")
		}
		SerCfg.RedisCfg.Port = int(ret)

		// 超时时长
		redis_timeout := setting.SelectElement("Timeout")
		if redis_timeout == nil {
			return errors.New("read redis timeout failed")
		}
		ret, err = strconv.ParseUint(redis_timeout.SelectAttrValue("value", "3"), 10, 32)
		if err != nil {
			return errors.New("read redis timeout is invalid:")
		}
		SerCfg.RedisCfg.TimeOut = int(ret)

		// 连接数
		redis_connnum := setting.SelectElement("ConnNum")
		if redis_connnum == nil {
			return errors.New("read redis connnet num failed")
		}
		ret, err = strconv.ParseUint(redis_connnum.SelectAttrValue("value", "1"), 10, 32)
		if err != nil {
			return errors.New("read redis connnet num is invalid:")
		}
		SerCfg.RedisCfg.ConnNum = int(ret)

		// 库编号
		redis_dbnum := setting.SelectElement("DbNum")
		if redis_dbnum == nil {
			return errors.New("read redis db num failed")
		}
		ret, err = strconv.ParseUint(redis_dbnum.SelectAttrValue("value", "0"), 10, 32)
		if err != nil {
			return errors.New("read redis db num is invalid:")
		}
		SerCfg.RedisCfg.DbNum = int(ret)

		// 密码
		redis_pwd := setting.SelectElement("Pwd")
		if redis_pwd == nil {
			return errors.New("read redis password failed")
		}
		SerCfg.RedisCfg.Pwd = redis_pwd.SelectAttrValue("value", "")
	}

	// RabbitMQ配置
	mqsettings := root.SelectElements("MqSetting")
	for _, setting := range mqsettings {
		// url
		url := setting.SelectElement("Url")
		if url == nil {
			return errors.New("read rabbitmq url failed")
		}
		SerCfg.MqCfg.Url = url.SelectAttrValue("value", "")

		// exchange
		exchange := setting.SelectElement("Exchange")
		if exchange == nil {
			return errors.New("read rabbitmq exchange failed")
		}
		SerCfg.MqCfg.Exchange = exchange.SelectAttrValue("value", "entry")
	}

	return nil
}

func (c *ServiceConfig) GetWebListenAddr() string {
	return SerCfg.WebCfg.ListenAddr
}

func (c *ServiceConfig) GetDebugMode() bool {
	return SerCfg.WebCfg.DebugMode
}

func (c *ServiceConfig) GetLogSetting() (string, int) {
	return SerCfg.LogCfg.LogPath, SerCfg.LogCfg.LogLevel
}

func (c *ServiceConfig) GetDbDriver() string {
	return SerCfg.DbCfg.DbDriver
}

func (c *ServiceConfig) GetDbSource() string {
	dbsource := fmt.Sprintf("%s/%s@%s:%s/%s",
		SerCfg.DbCfg.DbUser,
		SerCfg.DbCfg.DbPwd,
		SerCfg.DbCfg.DbAddress,
		SerCfg.DbCfg.DbPort,
		SerCfg.DbCfg.DbName)

	return dbsource
}

func (c *ServiceConfig) GetRedisAddress() string {
	redisaccress := fmt.Sprintf("%s:%d",
		SerCfg.RedisCfg.Address,
		SerCfg.RedisCfg.Port)

	return redisaccress
}

func (c *ServiceConfig) GetRedisPwd() string {
	return SerCfg.RedisCfg.Pwd
}

func (c *ServiceConfig) GetRedisDbNum() int {
	return SerCfg.RedisCfg.DbNum
}

func (c *ServiceConfig) GetRedisConnNum() int {
	return SerCfg.RedisCfg.ConnNum
}

func (c *ServiceConfig) GetRedisTimeout() int {
	return SerCfg.RedisCfg.TimeOut
}