cnlink/conf/conf.go

package conf

import (
	"bytes"
	"io/ioutil"
	"log"
	"os"
	"runtime"
	"github.com/BurntSushi/toml"
)

type sequenceDB struct {
	DSN          string `toml:"dsn"`
	MaxIdleConns int    `toml:"max_idle_conns"`
	MaxOpenConns int    `toml:"max_open_conns"`
}

type http struct {
	Port string `toml:"port"`
}

type shortDB struct {
	ReadDSN      string `toml:"read_dsn"`
	WriteDSN     string `toml:"write_dsn"`
	MaxIdleConns int    `toml:"max_idle_conns"`
	MaxOpenConns int    `toml:"max_open_conns"`
}

type common struct {
	BlackShortURLs    []string `toml:"black_short_urls"`
	BlackShortURLsMap map[string]bool
	BaseString        string `toml:"base_string"`
	BaseStringLength  uint64
	DomainName        string `toml:"domain_name"`
	Schema            string `toml:"schema"`
}

type config struct {
	Http       http       `toml:"http"`
	SequenceDB sequenceDB `toml:"sequence_db"`
	ShortDB    shortDB    `toml:"short_db"`
	Common     common     `toml:"common"`
}

var Conf config
var Version string
func MustParseConfig(configFile string) {
	if fileInfo, err := os.Stat(configFile); err != nil {
		if os.IsNotExist(err) {
			log.Panicf("configuration file %v does not exist.", configFile)
		} else {
			log.Panicf("configuration file %v can not be stated. %v", configFile, err)
		}
	} else {
		if fileInfo.IsDir() {
			log.Panicf("%v is a directory name", configFile)
		}
	}

	content, err := ioutil.ReadFile(configFile)
	if err != nil {
		log.Panicf("read configuration file error. %v", err)
	}
	content = bytes.TrimSpace(content)

	err = toml.Unmarshal(content, &Conf)
	if err != nil {
		log.Panicf("unmarshal toml object error. %v", err)
	}

	// short url black list
	Conf.Common.BlackShortURLsMap = make(map[string]bool)
	for _, blackShortURL := range Conf.Common.BlackShortURLs {
		Conf.Common.BlackShortURLsMap[blackShortURL] = true
	}

	// base string
	Conf.Common.BaseStringLength = uint64(len(Conf.Common.BaseString))
}
func init() {
	runtime.GOMAXPROCS(runtime.NumCPU())
	log.SetFlags(log.LstdFlags | log.Lshortfile)
}