data_field_dispose/data_fusion/config/conf.go

package config

import (
	"fmt"
	"os"
	"time"

	"github.com/BurntSushi/toml"
)

var (
	// Conf crocodile conf
	Conf *conf
)

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

type conf struct {
	Serve serve
	DB    db
	Mail  mail
	Log   log
}

type serve struct {
	Thread int
	Fields []string
	Weight map[string]int
}

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 {
	Mongo  mgo
	MongoS mgo
	Redis  redis
}

type mgo struct {
	Addr     string
	Dbname   string
	Coll     string
	Record   string
	Size     int
	User     string
	Password string
}

type redis struct {
	Addr string
	Db   int
}

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
}