first commit

params.go

@@ -0,0 +1,81 @@
+package main
+
+import (
+	"io/ioutil"
+	"github.com/influxdb/influxdb/client"
+
+	goyaml "gopkg.in/yaml.v2"
+)
+
+const PARAM_FILE string = "params.yml"
+
+type Params struct {
+	Port                 int                  `yaml:"port,omitempty"`
+	Host                 string               `yaml:"host,omitempty"`
+	InfluxDBClientConfig *client.ClientConfig `yaml:"influxdb,omitempty"`
+	InfluxDBClient       *client.Client       `yaml:"-"`
+}
+
+func (this *Params) LoadFromYamlFile(fileName string) error {
+	data, err := ioutil.ReadFile(fileName)
+	if err != nil {
+		return err
+	}
+	return this.LoadFromYamlData(data)
+}
+
+func (this *Params) LoadFromYamlData(data []byte) error {
+	var err error
+	p := new(Params)
+	if err = goyaml.Unmarshal(data, p); err != nil {
+		return err
+	}
+
+	if p.Host != "" {
+		this.Host = p.Host
+	}
+	if p.Port > 0 {
+		this.Port = p.Port
+	}
+	if p.InfluxDBClientConfig == nil {
+		return nil
+	}
+	if p.InfluxDBClientConfig.Host != "" {
+		this.InfluxDBClientConfig.Host = p.InfluxDBClientConfig.Host
+	}
+	if p.InfluxDBClientConfig.Database != "" {
+		this.InfluxDBClientConfig.Database = p.InfluxDBClientConfig.Database
+	}
+	if p.InfluxDBClientConfig.Password != "" {
+		this.InfluxDBClientConfig.Password = p.InfluxDBClientConfig.Password
+	}
+	if p.InfluxDBClientConfig.Username != "" {
+		this.InfluxDBClientConfig.Username = p.InfluxDBClientConfig.Username
+	}
+	this.InfluxDBClientConfig.IsSecure = p.InfluxDBClientConfig.IsSecure
+
+	return nil
+}
+
+func (this *Params) initInfluxDB() error {
+	var err error
+	this.InfluxDBClient, err = client.NewClient(this.InfluxDBClientConfig)
+
+	return err
+}
+
+func GetDefaultParams() *Params {
+	influxdbClientConfig := &client.ClientConfig{}
+	return &Params{7777, "localhost", influxdbClientConfig, nil}
+}
+
+var params = GetDefaultParams()
+
+func init() {
+	if err := params.LoadFromYamlFile(PARAM_FILE); err != nil {
+		panic(err)
+	}
+	if err := params.initInfluxDB(); err != nil {
+		panic(err)
+	}
+}