eventRT/config/config.go

// Package config define config struct of model runtime service
package config

import (
	"fmt"
	"net/url"

	"github.com/spf13/viper"
)

// 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"`
}

// MongoDBConfig define config struct of mongoDB config
type MongoDBConfig struct {
	Host     string `mapstructure:"host"`
	User     string `mapstructure:"user"`
	Password string `mapstructure:"password"`
	Database string `mapstructure:"database"`
	AuthDB   string `mapstructure:"auth_db"`
	Port     int    `mapstructure:"port"`
	Timeout  int    `mapstructure:"timeout"`
}

type RabbitMQConfig struct {
	CACertPath         string `mapstructure:"ca_cert_path"`
	ClientKeyPath      string `mapstructure:"client_key_path"`
	ClientKeyPassword  string `mapstructure:"client_key_password"`
	ClientCertPath     string `mapstructure:"client_cert_path"`
	InsecureSkipVerify bool   `mapstructure:"insecure_skip_verify"`
	ServerName         string `mapstructure:"server_name"`
	User               string `mapstructure:"user"`
	Password           string `mapstructure:"password"`
	Host               string `mapstructure:"host"`
	Port               int    `mapstructure:"port"`
}

// ServiceConfig define config struct of service config
type ServiceConfig struct {
	ServiceAddr string `mapstructure:"service_addr"`
	ServiceName string `mapstructure:"service_name"`
	SecretKey   string `mapstructure:"secret_key"`
	DeployEnv   string `mapstructure:"deploy_env"`
}

// EventRTConfig define config struct of model runtime server
type EventRTConfig struct {
	ServiceConfig  `mapstructure:"service"`
	LoggerConfig   `mapstructure:"logger"`
	MongoDBConfig  `mapstructure:"mongodb"`
	RabbitMQConfig `mapstructure:"rabbitMQ"`
	MongoDBURI     string
}

// ReadAndInitConfig return eventRT project config struct
func ReadAndInitConfig(configDir, configName, configType string) (eventRTConfig EventRTConfig) {
	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(&eventRTConfig); err != nil {
		panic(fmt.Sprintf("unmarshal eventRT config failed:%s\n", err.Error()))
	}

	if eventRTConfig.MongoDBConfig.Timeout <= 0 {
		eventRTConfig.MongoDBConfig.Timeout = 10
	}
	if eventRTConfig.MongoDBConfig.AuthDB == "" {
		eventRTConfig.MongoDBConfig.AuthDB = "admin"
	}

	// init mongoDB uri
	user := url.QueryEscape(eventRTConfig.MongoDBConfig.User)
	password := url.QueryEscape(eventRTConfig.MongoDBConfig.Password)
	eventRTConfig.MongoDBURI = fmt.Sprintf("mongodb://%s:%s@%s:%d/?authSource=%s", user, password, eventRTConfig.MongoDBConfig.Host, eventRTConfig.MongoDBConfig.Port, eventRTConfig.MongoDBConfig.AuthDB)
	return eventRTConfig
}