mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-23 00:27:55 +08:00
5ce6bbf076
glide has its own requirements. My previous workaround caused me some code checkin errors. Need to fix this.
131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package util
|
|
|
|
// Copyright 2011 Numerotron Inc.
|
|
// Use of this source code is governed by an MIT-style license
|
|
// that can be found in the LICENSE file.
|
|
//
|
|
// Developed at www.stathat.com by Patrick Crosby
|
|
// Contact us on twitter with any questions: twitter.com/stat_hat
|
|
|
|
// The jconfig package provides a simple, basic configuration file parser using JSON.
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"os"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
)
|
|
|
|
type Config struct {
|
|
data map[string]interface{}
|
|
filename string
|
|
}
|
|
|
|
func newConfig() *Config {
|
|
result := new(Config)
|
|
result.data = make(map[string]interface{})
|
|
return result
|
|
}
|
|
|
|
// Loads config information from a JSON file
|
|
func LoadConfig(filename string) *Config {
|
|
result := newConfig()
|
|
result.filename = filename
|
|
err := result.parse()
|
|
if err != nil {
|
|
glog.Fatalf("error loading config file %s: %s", filename, err)
|
|
}
|
|
return result
|
|
}
|
|
|
|
// Loads config information from a JSON string
|
|
func LoadConfigString(s string) *Config {
|
|
result := newConfig()
|
|
err := json.Unmarshal([]byte(s), &result.data)
|
|
if err != nil {
|
|
glog.Fatalf("error parsing config string %s: %s", s, err)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (c *Config) StringMerge(s string) {
|
|
next := LoadConfigString(s)
|
|
c.merge(next.data)
|
|
}
|
|
|
|
func (c *Config) LoadMerge(filename string) {
|
|
next := LoadConfig(filename)
|
|
c.merge(next.data)
|
|
}
|
|
|
|
func (c *Config) merge(ndata map[string]interface{}) {
|
|
for k, v := range ndata {
|
|
c.data[k] = v
|
|
}
|
|
}
|
|
|
|
func (c *Config) parse() error {
|
|
f, err := os.Open(c.filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
b := new(bytes.Buffer)
|
|
_, err = b.ReadFrom(f)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = json.Unmarshal(b.Bytes(), &c.data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Returns a string for the config variable key
|
|
func (c *Config) GetString(key string) string {
|
|
result, present := c.data[key]
|
|
if !present {
|
|
return ""
|
|
}
|
|
return result.(string)
|
|
}
|
|
|
|
// Returns an int for the config variable key
|
|
func (c *Config) GetInt(key string) int {
|
|
x, ok := c.data[key]
|
|
if !ok {
|
|
return -1
|
|
}
|
|
return int(x.(float64))
|
|
}
|
|
|
|
// Returns a float for the config variable key
|
|
func (c *Config) GetFloat(key string) float64 {
|
|
x, ok := c.data[key]
|
|
if !ok {
|
|
return -1
|
|
}
|
|
return x.(float64)
|
|
}
|
|
|
|
// Returns a bool for the config variable key
|
|
func (c *Config) GetBool(key string) bool {
|
|
x, ok := c.data[key]
|
|
if !ok {
|
|
return false
|
|
}
|
|
return x.(bool)
|
|
}
|
|
|
|
// Returns an array for the config variable key
|
|
func (c *Config) GetArray(key string) []interface{} {
|
|
result, present := c.data[key]
|
|
if !present {
|
|
return []interface{}(nil)
|
|
}
|
|
return result.([]interface{})
|
|
}
|