mirror of
https://github.com/astaxie/beego.git
synced 2024-11-25 14:00:55 +00:00
beego config module json support get data like key:🔑:key
This commit is contained in:
parent
34ba7a8e6c
commit
3a0b2e3b95
@ -5,6 +5,7 @@ import (
|
||||
"errors"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
@ -33,42 +34,73 @@ func (js *JsonConfig) Parse(filename string) (ConfigContainer, error) {
|
||||
|
||||
type JsonConfigContainer struct {
|
||||
data map[string]interface{}
|
||||
sync.Mutex
|
||||
sync.RWMutex
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) Bool(key string) (bool, error) {
|
||||
if v, ok := c.data[key].(bool); ok {
|
||||
val := c.getdata(key)
|
||||
if val != nil {
|
||||
if v, ok := val.(bool); ok {
|
||||
return v, nil
|
||||
}
|
||||
} else {
|
||||
return false, errors.New("not bool value")
|
||||
}
|
||||
} else {
|
||||
return false, errors.New("not exist key:" + key)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) Int(key string) (int, error) {
|
||||
if v, ok := c.data[key].(float64); ok {
|
||||
val := c.getdata(key)
|
||||
if val != nil {
|
||||
if v, ok := val.(float64); ok {
|
||||
return int(v), nil
|
||||
}
|
||||
} else {
|
||||
return 0, errors.New("not int value")
|
||||
}
|
||||
} else {
|
||||
return 0, errors.New("not exist key:" + key)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) Int64(key string) (int64, error) {
|
||||
if v, ok := c.data[key].(float64); ok {
|
||||
val := c.getdata(key)
|
||||
if val != nil {
|
||||
if v, ok := val.(float64); ok {
|
||||
return int64(v), nil
|
||||
}
|
||||
} else {
|
||||
return 0, errors.New("not int64 value")
|
||||
}
|
||||
} else {
|
||||
return 0, errors.New("not exist key:" + key)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) Float(key string) (float64, error) {
|
||||
if v, ok := c.data[key].(float64); ok {
|
||||
val := c.getdata(key)
|
||||
if val != nil {
|
||||
if v, ok := val.(float64); ok {
|
||||
return v, nil
|
||||
}
|
||||
} else {
|
||||
return 0.0, errors.New("not float64 value")
|
||||
}
|
||||
} else {
|
||||
return 0.0, errors.New("not exist key:" + key)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) String(key string) string {
|
||||
if v, ok := c.data[key].(string); ok {
|
||||
val := c.getdata(key)
|
||||
if val != nil {
|
||||
if v, ok := val.(string); ok {
|
||||
return v
|
||||
}
|
||||
} else {
|
||||
return ""
|
||||
}
|
||||
} else {
|
||||
return ""
|
||||
}
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) Set(key, val string) error {
|
||||
@ -79,10 +111,41 @@ func (c *JsonConfigContainer) Set(key, val string) error {
|
||||
}
|
||||
|
||||
func (c *JsonConfigContainer) DIY(key string) (v interface{}, err error) {
|
||||
if v, ok := c.data[key]; ok {
|
||||
return v, nil
|
||||
}
|
||||
val := c.getdata(key)
|
||||
if val != nil {
|
||||
return val, nil
|
||||
} else {
|
||||
return nil, errors.New("not exist key")
|
||||
}
|
||||
}
|
||||
|
||||
//section.key or key
|
||||
func (c *JsonConfigContainer) getdata(key string) interface{} {
|
||||
c.RLock()
|
||||
defer c.RUnlock()
|
||||
if len(key) == 0 {
|
||||
return nil
|
||||
}
|
||||
sectionkey := strings.Split(key, "::")
|
||||
if len(sectionkey) >= 2 {
|
||||
cruval, ok := c.data[sectionkey[0]]
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
for _, key := range sectionkey[1:] {
|
||||
if v, ok := cruval.(map[string]interface{}); !ok {
|
||||
return nil
|
||||
} else if cruval, ok = v[key]; !ok {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return cruval
|
||||
} else {
|
||||
if v, ok := c.data[key]; ok {
|
||||
return v
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
@ -18,7 +18,12 @@ var jsoncontext = `{
|
||||
"port": "port",
|
||||
"database": "database",
|
||||
"username": "username",
|
||||
"password": "password"
|
||||
"password": "password",
|
||||
"conns":{
|
||||
"maxconnection":12,
|
||||
"autoconnect":true,
|
||||
"connectioninfo":"info"
|
||||
}
|
||||
}
|
||||
}`
|
||||
|
||||
@ -70,9 +75,19 @@ func TestJson(t *testing.T) {
|
||||
if jsonconf.String("name") != "astaxie" {
|
||||
t.Fatal("get name error")
|
||||
}
|
||||
if jsonconf.String("database::host") != "host" {
|
||||
t.Fatal("get database::host error")
|
||||
}
|
||||
if jsonconf.String("database::conns::connectioninfo") != "info" {
|
||||
t.Fatal("get database::conns::connectioninfo error")
|
||||
}
|
||||
if maxconnection, err := jsonconf.Int("database::conns::maxconnection"); err != nil || maxconnection != 12 {
|
||||
t.Fatal("get database::conns::maxconnection error")
|
||||
}
|
||||
if db, err := jsonconf.DIY("database"); err != nil {
|
||||
t.Fatal(err)
|
||||
} else if m, ok := db.(map[string]interface{}); !ok {
|
||||
t.Log(db)
|
||||
t.Fatal("db not map[string]interface{}")
|
||||
} else {
|
||||
if m["host"].(string) != "host" {
|
||||
|
Loading…
Reference in New Issue
Block a user