es_index_data/createEsIndex/config/conf.go

package config

import (
	"fmt"
	"github.com/spf13/viper"
	"os"
	"time"
)

var (
	// Conf crocodile conf
	Conf *conf
)

// InitConf Config
func InitConf(conf string) {
	//_, err := toml.DecodeFile(conf, &Conf)
	//if err != nil {
	//	fmt.Printf("Err %v", err)
	//	os.Exit(1)
	//}

	viper.SetConfigFile("common.toml")
	viper.SetConfigName("common") // 配置文件名称(无扩展名)
	viper.SetConfigType("toml")   // 如果配置文件的名称中没有扩展名，则需要配置此项
	viper.AddConfigPath("./")
	err := viper.ReadInConfig() // 查找并读取配置文件
	if err != nil {             // 处理读取配置文件的错误
		fmt.Println("ReadInConfig err =>", err)
		return
	}
	err = viper.Unmarshal(&Conf)
	if err != nil {
		fmt.Println("Init Unmarshal err =>", err)
		os.Exit(1)
		return
	}

}

type conf struct {
	DB     db
	Udp    udp
	Mail   mail
	Log    log
	Env    env
	Others map[string]OthersData
	Pre    map[string]PreConf
}

type udp struct {
	LocPort string
	JyAddr  string
	JyPort  int
	NeAddr  string
	NePort  int
}

type udpNext struct {
	Addr  string
	Port  int
	Stype string
}

type mail struct {
	Send bool
	To   string
	Api  string
}

// Log Config
type log struct {
	LogPath    string
	MaxSize    int
	Compress   bool
	MaxAge     int
	MaxBackups int
	LogLevel   string
	Format     string
}

type db struct {
	MongoB mgo
	MongoP mgo
	MongoQ mgo
	MongoS mgo
	Es     es
	Oss    oss
	MysqlB mysql
}

type mysql struct {
	Addr     string
	Dbname   string
	Username string
	Password string
}

// oss oss 阿里云配置
type oss struct {
	Endpoint     string
	AccessKey    string
	AccessSecret string
	BucketName   string
	DetailBucket string
	Filesize     int
}
type mgo struct {
	Addr     string
	Dbname   string
	Coll     string
	Size     int
	User     string
	Password string
	Direct   bool
}

// env 全局的相关配置
type env struct {
	Stype    int  //默认0，正式环境；1是测试环境，不会执行定时任务更新采购单位、中标单位、数据检测
	OpenPre  bool //默认关闭，不开启预处理流程
	Alias    string
	SpecType string
	Dbfile   string //配置文件，读取bitmap
	Ai       bool   //是否Ai m模型测试
	Oss      bool   //是否针对详情，使用oss 获取内容
}

type es struct {
	Addr          string
	AddrP         string
	Size          int
	Username      string
	Password      string
	IndexB        string
	TypeB         string
	IndexP        string
	IndexPD       string
	BiddingDetail string
	DetailCount   int
	TypeP         string
	IndexWinner   string
	TypeWinner    string
	IndexBuyer    string
	TypeBuyer     string
	DetailFilter  []string
	IndexTmp      string
	//FieldEs              map[string]interface{}
	//FieldPurchasingist  map[string]interface{}
	//FieldProcurementList map[string]interface{}
	//FieldWinnerOrder     map[string]interface{}
	//Package              map[string]interface{}
	Addr2     string
	Username2 string
	Password2 string
	Indexb2   string
	Addr3     string // 华为云新集群，存储原来阿里云集群的标讯、项目数据
	Username3 string
	Password3 string
	Indexb3   string
}

// PreConf 预处理 配置
type PreConf struct {
	Addr     string
	Username string
	Password string
	Index    []string
}

type OthersData struct {
	MgoAddr     string
	MgoDB       string
	MgoColl     string
	MgoUsername string
	MgoPassword string
	EsAddr      string
	EsIndex     string
	EsUsername  string
	EsPassword  string
	NextAddr    string
	NextPort    int
	Data        map[string]interface{}
}

type duration struct {
	time.Duration
}

// UnmarshalText parse 10s to time.Time
func (d *duration) UnmarshalText(text []byte) error {
	var err error
	d.Duration, err = time.ParseDuration(string(text))
	return err
}