mirror of https://github.com/rocboss/paopao-ce
Michael Li
2 years ago
parent
6738df8913
commit
72136a2266
@ -1,41 +0,0 @@
|
||||
package cfg
|
||||
|
||||
var (
|
||||
_features = newEmptyFeatures()
|
||||
|
||||
// Use alias of Features.Use func
|
||||
Use = _features.Use
|
||||
|
||||
// UseDeafult alias of Features.UseDefault func
|
||||
UseDefault = _features.UseDefault
|
||||
|
||||
// As alias of Features.Cfg func
|
||||
As = _features.Cfg
|
||||
|
||||
// If alias of Features.CfgIf func
|
||||
If = _features.CfgIf
|
||||
|
||||
// In alias of Features.CfgIn func
|
||||
In = _features.CfgIn
|
||||
|
||||
// Be alias of Feaures.CfgBe func
|
||||
Be = _features.CfgBe
|
||||
|
||||
// Not alias of Features.CfgNot func
|
||||
Not = _features.CfgNot
|
||||
)
|
||||
|
||||
// Initialize initialize features in cfg pkg
|
||||
func Initialize(suites map[string][]string, kv map[string]string) {
|
||||
_features = NewFeatures(suites, kv)
|
||||
{
|
||||
// must re-assign variable below
|
||||
Use = _features.Use
|
||||
UseDefault = _features.UseDefault
|
||||
As = _features.Cfg
|
||||
If = _features.CfgIf
|
||||
In = _features.CfgIn
|
||||
Be = _features.CfgBe
|
||||
Not = _features.CfgNot
|
||||
}
|
||||
}
|
||||
@ -1,78 +0,0 @@
|
||||
package cfg
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestCfg(t *testing.T) {
|
||||
suites := map[string][]string{
|
||||
"default": {"Sms", "Alipay", "Zinc", "MySQL", "Redis", "AliOSS", "LogZinc"},
|
||||
"develop": {"Zinc", "MySQL", "AliOSS", "LogFile"},
|
||||
"slim": {"Zinc", "MySQL", "Redis", "AliOSS", "LogFile"},
|
||||
}
|
||||
kv := map[string]string{
|
||||
"sms": "SmsJuhe",
|
||||
}
|
||||
|
||||
Initialize(suites, kv)
|
||||
UseDefault()
|
||||
|
||||
if !If("Sms") {
|
||||
t.Error(`want If("Sms") == true but not`)
|
||||
}
|
||||
|
||||
if v, exist := As("Sms"); exist && v != "SmsJuhe" {
|
||||
t.Errorf(`want As("Sms") == "SmsJuhe", true but got: "%s", "%t"`, v, exist)
|
||||
}
|
||||
|
||||
matched := false
|
||||
Be("Alipay", func() {
|
||||
matched = true
|
||||
})
|
||||
if !matched {
|
||||
t.Error(`want Be("Alipay", ...) matched but not`)
|
||||
}
|
||||
|
||||
matched = false
|
||||
Not("LogFile", func() {
|
||||
matched = true
|
||||
})
|
||||
if !matched {
|
||||
t.Error(`want Not("LogFile", ...) matched but not`)
|
||||
}
|
||||
|
||||
var m1, m2, m3, m4 bool
|
||||
In(Actions{
|
||||
"Sms": func() {
|
||||
m1 = true
|
||||
},
|
||||
"Alipay": func() {
|
||||
m2 = true
|
||||
},
|
||||
"Meili": func() {
|
||||
m4 = true
|
||||
},
|
||||
}, func() {
|
||||
m3 = true
|
||||
})
|
||||
if !m1 || !m2 || m3 || m4 {
|
||||
t.Errorf(`In("Sms", "Alipay", "Meili", ...) not correct -> m1: %t m2:%t m3:%t m4:%t`, m1, m2, m3, m4)
|
||||
}
|
||||
|
||||
m1 = false
|
||||
m2 = false
|
||||
m3 = false
|
||||
In(Actions{
|
||||
"LogFile": func() {
|
||||
m1 = true
|
||||
},
|
||||
"Meili": func() {
|
||||
m2 = true
|
||||
},
|
||||
}, func() {
|
||||
m3 = true
|
||||
})
|
||||
if m1 || m2 || !m3 {
|
||||
t.Errorf(`In("Zinc", "MySQL", ...) not correct -> m1: %t m2:%t m3:%t`, m1, m2, m3)
|
||||
}
|
||||
}
|
||||
@ -1,149 +0,0 @@
|
||||
package cfg
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/rocboss/paopao-ce/pkg/types"
|
||||
)
|
||||
|
||||
// Features fetures info struct
|
||||
type Features struct {
|
||||
kv map[string]string
|
||||
suites map[string][]string
|
||||
features map[string]string
|
||||
}
|
||||
|
||||
// Actions feature-func map alias type
|
||||
type Actions map[string]types.Fn
|
||||
|
||||
// NewFeatures create new Features instance
|
||||
func NewFeatures(suites map[string][]string, kv map[string]string) *Features {
|
||||
f := newEmptyFeatures()
|
||||
for k, v := range suites {
|
||||
if len(k) > 0 {
|
||||
for i := 0; i < len(v); i++ {
|
||||
// ignore empty string
|
||||
if len(v[i]) == 0 {
|
||||
lastIdx := len(v) - 1
|
||||
v[i] = v[lastIdx]
|
||||
v = v[:lastIdx]
|
||||
i--
|
||||
}
|
||||
}
|
||||
if len(v) > 0 {
|
||||
f.suites[k] = v
|
||||
}
|
||||
}
|
||||
}
|
||||
for k, v := range kv {
|
||||
if len(k) > 0 && len(v) > 0 {
|
||||
f.kv[k] = v
|
||||
}
|
||||
}
|
||||
f.UseDefault()
|
||||
return f
|
||||
}
|
||||
|
||||
func newEmptyFeatures() *Features {
|
||||
return &Features{
|
||||
suites: make(map[string][]string),
|
||||
kv: make(map[string]string),
|
||||
features: make(map[string]string),
|
||||
}
|
||||
}
|
||||
|
||||
// UseDefault use default suite for features
|
||||
func (f *Features) UseDefault() {
|
||||
f.Use([]string{"default"}, true)
|
||||
}
|
||||
|
||||
// Use use custom suite for features
|
||||
func (f *Features) Use(suite []string, noDefault bool) {
|
||||
if noDefault && len(f.features) != 0 {
|
||||
f.features = make(map[string]string)
|
||||
}
|
||||
features := f.flatFeatures(suite)
|
||||
for _, feature := range features {
|
||||
if len(feature) == 0 {
|
||||
continue
|
||||
}
|
||||
f.features[feature] = f.kv[feature]
|
||||
}
|
||||
}
|
||||
|
||||
func (f *Features) flatFeatures(suite []string) []string {
|
||||
features := make([]string, 0, len(suite)+10)
|
||||
for s := suite[:]; len(s) > 0; s = s[:len(s)-1] {
|
||||
item := strings.TrimSpace(strings.ToLower(s[0]))
|
||||
if len(item) > 0 {
|
||||
if items, exist := f.suites[item]; exist {
|
||||
s = append(s, items...)
|
||||
}
|
||||
features = append(features, item)
|
||||
}
|
||||
s[0] = s[len(s)-1]
|
||||
}
|
||||
return features
|
||||
}
|
||||
|
||||
// Cfg get value by key if exist
|
||||
func (f *Features) Cfg(key string) (string, bool) {
|
||||
key = strings.ToLower(key)
|
||||
value, exist := f.features[key]
|
||||
return value, exist
|
||||
}
|
||||
|
||||
// CfgIf check expression is true. if expression just have a string like
|
||||
// `Sms` is mean `Sms` whether define in suite feature settings. expression like
|
||||
// `Sms = SmsJuhe` is mean whether `Sms` define in suite feature settings and value
|
||||
// is `SmsJuhe`
|
||||
func (f *Features) CfgIf(expression string) bool {
|
||||
kv := strings.Split(expression, "=")
|
||||
key := strings.Trim(strings.ToLower(kv[0]), " ")
|
||||
v, ok := f.features[key]
|
||||
if len(kv) == 2 && ok && strings.Trim(kv[1], " ") == v {
|
||||
return true
|
||||
} else if len(kv) == 1 && ok {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// CfgIn range actions to check item's expression is true then do the handle, defFn will handle
|
||||
// if all items are not matched,
|
||||
func (f *Features) CfgIn(actions Actions, defAct ...types.Fn) {
|
||||
itemMatched := false
|
||||
for expression, handle := range actions {
|
||||
if f.CfgIf(expression) && handle != nil {
|
||||
handle()
|
||||
itemMatched = true
|
||||
}
|
||||
}
|
||||
if !itemMatched && len(defAct) > 0 {
|
||||
for _, handle := range defAct {
|
||||
if handle != nil {
|
||||
handle()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// CfgBe check expression is true then do the handle. if expression just have a string like
|
||||
// `Sms` is mean `Sms` whether defined in suite feature settings. expression like
|
||||
// `Sms = SmsJuhe` is mean whether `Sms` define in suite feature settings and value
|
||||
// is `SmsJuhe`
|
||||
func (f *Features) CfgBe(expression string, handle types.Fn) {
|
||||
if f.CfgIf(expression) && handle != nil {
|
||||
handle()
|
||||
}
|
||||
}
|
||||
|
||||
// CfgNot check expression is not true then do the handle. if expression just have a string like
|
||||
// `Sms` is mean `Sms` whether defined in suite feature settings. expression like
|
||||
// `Sms = SmsJuhe` is mean whether `Sms` define in suite feature settings and value
|
||||
// is `SmsJuhe`
|
||||
func (f *Features) CfgNot(expression string, handle func()) {
|
||||
if !f.CfgIf(expression) {
|
||||
handle()
|
||||
}
|
||||
}
|
||||
@ -1,114 +0,0 @@
|
||||
package cfg
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestUseDefault(t *testing.T) {
|
||||
suites := map[string][]string{
|
||||
"default": {"Sms", "Alipay", "Zinc", "MySQL", "Redis", "AliOSS", "LogZinc"},
|
||||
"develop": {"Zinc", "MySQL", "AliOSS", "LogFile"},
|
||||
"slim": {"Zinc", "MySQL", "Redis", "AliOSS", "LogFile"},
|
||||
}
|
||||
kv := map[string]string{
|
||||
"sms": "SmsJuhe",
|
||||
}
|
||||
f := NewFeatures(suites, kv)
|
||||
for _, data := range []struct {
|
||||
key string
|
||||
expect string
|
||||
exist bool
|
||||
}{
|
||||
{"Sms", "SmsJuhe", true},
|
||||
{"Alipay", "", true},
|
||||
{"Zinc", "", true},
|
||||
{"Redis", "", true},
|
||||
{"Database", "", false},
|
||||
} {
|
||||
if v, ok := f.Cfg(data.key); ok != data.exist || v != data.expect {
|
||||
t.Errorf("key: %s expect: %s exist: %t got v: %s ok: %t", data.key, data.expect, data.exist, v, ok)
|
||||
}
|
||||
}
|
||||
for exp, res := range map[string]bool{
|
||||
"Sms": true,
|
||||
"Sms = SmsJuhe": true,
|
||||
"SmsJuhe": false,
|
||||
"default": true,
|
||||
} {
|
||||
if ok := f.CfgIf(exp); res != ok {
|
||||
t.Errorf("CfgIf(%s) want %t got %t", exp, res, ok)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestUse(t *testing.T) {
|
||||
suites := map[string][]string{
|
||||
"default": {"Sms", "Alipay", "Zinc", "MySQL", "Redis", "AliOSS", "LogZinc"},
|
||||
"develop": {"Zinc", "MySQL", "AliOSS", "LogFile"},
|
||||
"slim": {"Zinc", "MySQL", "Redis", "AliOSS", "LogFile"},
|
||||
}
|
||||
kv := map[string]string{
|
||||
"sms": "SmsJuhe",
|
||||
}
|
||||
f := NewFeatures(suites, kv)
|
||||
|
||||
f.Use([]string{"develop"}, true)
|
||||
for _, data := range []struct {
|
||||
key string
|
||||
expect string
|
||||
exist bool
|
||||
}{
|
||||
{"Sms", "", false},
|
||||
{"Alipay", "", false},
|
||||
{"Zinc", "", true},
|
||||
{"Redis", "", false},
|
||||
{"Database", "", false},
|
||||
} {
|
||||
if v, ok := f.Cfg(data.key); ok != data.exist || v != data.expect {
|
||||
t.Errorf("key: %s expect: %s exist: %t got v: %s ok: %t", data.key, data.expect, data.exist, v, ok)
|
||||
}
|
||||
}
|
||||
for exp, res := range map[string]bool{
|
||||
"Sms": false,
|
||||
"Sms = SmsJuhe": false,
|
||||
"SmsJuhe": false,
|
||||
"default": false,
|
||||
"develop": true,
|
||||
} {
|
||||
if ok := f.CfgIf(exp); res != ok {
|
||||
t.Errorf("CfgIf(%s) want %t got %t", exp, res, ok)
|
||||
}
|
||||
}
|
||||
|
||||
f.UseDefault()
|
||||
f.Use([]string{"slim", "", "demo"}, false)
|
||||
for _, data := range []struct {
|
||||
key string
|
||||
expect string
|
||||
exist bool
|
||||
}{
|
||||
{"Sms", "SmsJuhe", true},
|
||||
{"Alipay", "", true},
|
||||
{"Zinc", "", true},
|
||||
{"Redis", "", true},
|
||||
{"Database", "", false},
|
||||
{"demo", "", true},
|
||||
} {
|
||||
if v, ok := f.Cfg(data.key); ok != data.exist || v != data.expect {
|
||||
t.Errorf("key: %s expect: %s exist: %t got v: %s ok: %t", data.key, data.expect, data.exist, v, ok)
|
||||
}
|
||||
}
|
||||
for exp, res := range map[string]bool{
|
||||
"Sms": true,
|
||||
"Sms = SmsJuhe": true,
|
||||
"SmsJuhe": false,
|
||||
"default": true,
|
||||
"develop": false,
|
||||
"slim": true,
|
||||
"demo": true,
|
||||
} {
|
||||
if ok := f.CfgIf(exp); res != ok {
|
||||
t.Errorf("CfgIf(%s) want %t got %t", exp, res, ok)
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue