homelab/traefik
1
0
Fork 0
Code Activity

Add GetBool function in KV provider, used by passHostHeader

Browse source
This commit is contained in:
emile 2015-11-22 21:24:43 +01:00
parent c8a0a83e2b
commit 6e62625ebf
3 changed files with 72 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

20
provider/kv.go
View file

@ -11,6 +11,7 @@ import (
"github.com/docker/libkv"
"github.com/docker/libkv/store"
"github.com/emilevauge/traefik/types"
"strconv"
)
// Kv holds common configurations of key-value providers.
@ -73,9 +74,10 @@ func (provider *Kv) loadConfig() *types.Configuration {
provider.Prefix,
}
var KvFuncMap = template.FuncMap{
"List": provider.list,
"Get": provider.get,
"Last": provider.last,
"List": provider.list,
"Get": provider.get,
"GetBool": provider.getBool,
"Last": provider.last,
}
configuration, err := provider.getConfiguration("templates/kv.tmpl", KvFuncMap, templateObjects)
@ -104,7 +106,7 @@ func (provider *Kv) get(keys ...string) string {
joinedKeys := strings.Join(keys, "")
keyPair, err := provider.kvclient.Get(joinedKeys)
if err != nil {
log.Debug("Error getting key: ", joinedKeys, err)
log.Error("Error getting key: ", joinedKeys, err)
return ""
} else if keyPair == nil {
return ""
@ -112,6 +114,16 @@ func (provider *Kv) get(keys ...string) string {
return string(keyPair.Value)
}
func (provider *Kv) getBool(keys ...string) bool {
value := provider.get(keys...)
b, err := strconv.ParseBool(string(value))
if err != nil {
log.Error("Error getting key: ", strings.Join(keys, ""), err)
return false
}
return b
}
func (provider *Kv) last(key string) string {
splittedKey := strings.Split(key, "/")
return splittedKey[len(splittedKey)-1]