main.go

package psmconfig

import (
	"flag"
	"io/ioutil"
	"log"

	yaml "gopkg.in/yaml.v2"
)

// Config Global Variable
// Store Configuration Options
var Config *Conf

// Conf Struct
type Conf struct {
	Name   string `yaml:"name,omitempty"`
	Env    string `yaml:"env,omitempty"`
	Host   string `yaml:"host,omitempty"`
	Port   int    `yaml:"port,omitempty"`
	Static string `yaml:"static,omitempty"`
	Mongo  struct {
		Host string `yaml:"host,omitempty"`
		Port int    `yaml:"port,omitempty"`
		User string `yaml:"user,omitempty"`
		Pass string `yaml:"pass,omitempty"`
		Path string `yaml:"path,omitempty"`
	} `yaml:"mongo,omitempty"`
	CouchBase struct {
		Host string `yaml:"host,omitempty"`
		Port int    `yaml:"port,omitempty"`
		User string `yaml:"user,omitempty"`
		Pass string `yaml:"pass,omitempty"`
		Path string `yaml:"path,omitempty"`
	} `yaml:"couchbase,omitempty"`
	ES struct {
		Host string `yaml:"host,omitempty"`
		Port int    `yaml:"port,omitempty"`
		User string `yaml:"user,omitempty"`
		Pass string `yaml:"pass,omitempty"`
		Path string `yaml:"path,omitempty"`
	} `yaml:"es,omitempty"`
	Redis struct {
		Host string `yaml:"host,omitempty"`
		Port int    `yaml:"port,omitempty"`
		User string `yaml:"user,omitempty"`
		Pass string `yaml:"pass,omitempty"`
		Path string `yaml:"path,omitempty"`
	} `yaml:"redis,omitempty"`
	Twitter struct {
		ConsumerKey       string `yaml:"consumer-key,omitempty"`
		ConsumerSecret    string `yaml:"consumer-secret,omitempty"`
		AccessToken       string `yaml:"access-token,omitempty"`
		AccessTokenSecret string `yaml:"access-token-secret,omitempty"`
	} `yaml:"twitter,omitempty"`
	Log   struct{}          `yaml:"log,omitempty"`
	Proto struct{}          `yaml:"proto,omitempty"`
	API   map[string]string `yaml:"api,omitempty"`
}

// NewConfig Init
func NewConfig() {
	var path string
	flag.StringVar(&path, "config", "conf/default.yaml", "Parse Configuration File")
	flag.Parse()

	data, err := ioutil.ReadFile(path)
	if err != nil {
		log.Printf("[PSM-CONFIG] Configuration parsing error: %v", err)
		panic(err)
	}

	yaml.Unmarshal(data, &Config)
	log.Printf("[PSM-CONFIG] Configuration loaded: %v", Config)
}