Skip to content

Commit f42d250

Browse files
EricWaiEricWai
committed
playbook(memcache): support instances and instances_sequence
Signed-off-by: EricWai <[email protected]>
1 parent 9d83a9c commit f42d250

File tree

12 files changed

+576
-63
lines changed

12 files changed

+576
-63
lines changed

‎internal/configure/hosts/hc_get.go

+19-24
Original file line numberDiff line numberDiff line change
@@ -25,66 +25,61 @@
2525
package hosts
2626

2727
import (
28-
comm "github.com/opencurve/curveadm/internal/configure/common"
2928
"github.com/opencurve/curveadm/internal/configure/curveadm"
3029
"github.com/opencurve/curveadm/internal/utils"
3130
"github.com/opencurve/curveadm/pkg/module"
31+
"github.com/opencurve/curveadm/pkg/variable"
3232
)
3333

34-
func (hc *HostConfig) get(i *comm.Item) interface{} {
34+
func (hc *HostConfig) get(i *item) interface{} {
3535
if v, ok := hc.config[i.Key()]; ok {
3636
return v
3737
}
3838

39-
defaultValue := i.DefaultValue()
39+
defaultValue := i.defaultValue
4040
if defaultValue != nil && utils.IsFunc(defaultValue) {
4141
return defaultValue.(func(*HostConfig) interface{})(hc)
4242
}
4343
return defaultValue
4444
}
4545

46-
func (hc *HostConfig) getString(i *comm.Item) string {
46+
func (hc *HostConfig) getString(i *item) string {
4747
v := hc.get(i)
4848
if v == nil {
4949
return ""
5050
}
5151
return v.(string)
5252
}
5353

54-
func (hc *HostConfig) getInt(i *comm.Item) int {
54+
func (hc *HostConfig) getInt(i *item) int {
5555
v := hc.get(i)
5656
if v == nil {
5757
return 0
5858
}
5959
return v.(int)
6060
}
6161

62-
func (hc *HostConfig) getBool(i *comm.Item) bool {
62+
func (hc *HostConfig) getBool(i *item) bool {
6363
v := hc.get(i)
6464
if v == nil {
6565
return false
6666
}
6767
return v.(bool)
6868
}
6969

70-
func (hc *HostConfig) GetHost() string { return hc.getString(CONFIG_HOST) }
71-
func (hc *HostConfig) GetHostname() string { return hc.getString(CONFIG_HOSTNAME) }
72-
func (hc *HostConfig) GetSSHHostname() string { return hc.getString(CONFIG_SSH_HOSTNAME) }
73-
func (hc *HostConfig) GetSSHPort() int { return hc.getInt(CONFIG_SSH_PORT) }
74-
func (hc *HostConfig) GetPrivateKeyFile() string { return hc.getString(CONFIG_PRIVATE_CONFIG_FILE) }
75-
func (hc *HostConfig) GetForwardAgent() bool { return hc.getBool(CONFIG_FORWARD_AGENT) }
76-
func (hc *HostConfig) GetBecomeUser() string { return hc.getString(CONFIG_BECOME_USER) }
77-
func (hc *HostConfig) GetLabels() []string { return hc.labels }
78-
func (hc *HostConfig) GetEnvs() []string { return hc.envs }
79-
80-
func (hc *HostConfig) GetUser() string {
81-
user := hc.getString(CONFIG_USER)
82-
if user == "${user}" {
83-
return utils.GetCurrentUser()
84-
}
85-
return user
86-
}
87-
70+
func (hc *HostConfig) GetHost() string { return hc.getString(CONFIG_HOST) }
71+
func (hc *HostConfig) GetHostname() string { return hc.getString(CONFIG_HOSTNAME) }
72+
func (hc *HostConfig) GetSSHHostname() string { return hc.getString(CONFIG_SSH_HOSTNAME) }
73+
func (hc *HostConfig) GetUser() string { return hc.getString(CONFIG_USER) }
74+
func (hc *HostConfig) GetSSHPort() int { return hc.getInt(CONFIG_SSH_PORT) }
75+
func (hc *HostConfig) GetPrivateKeyFile() string { return hc.getString(CONFIG_PRIVATE_CONFIG_FILE) }
76+
func (hc *HostConfig) GetForwardAgent() bool { return hc.getBool(CONFIG_FORWARD_AGENT) }
77+
func (hc *HostConfig) GetBecomeUser() string { return hc.getString(CONFIG_BECOME_USER) }
78+
func (hc *HostConfig) GetLabels() []string { return hc.labels }
79+
func (hc *HostConfig) GetEnvs() []string { return hc.envs }
80+
func (hc *HostConfig) GetInstances() int { return hc.instances }
81+
func (hc *HostConfig) GetInstancesSequence() int { return hc.instancesSequence }
82+
func (hc *HostConfig) GetVariables() *variable.Variables { return hc.variables }
8883
func (hc *HostConfig) GetSSHConfig() *module.SSHConfig {
8984
hostname := hc.GetSSHHostname()
9085
if len(hostname) == 0 {

‎internal/configure/hosts/hc_item.go

+144-18
Original file line numberDiff line numberDiff line change
@@ -26,75 +26,201 @@ package hosts
2626

2727
import (
2828
"fmt"
29+
"github.com/opencurve/curveadm/internal/errno"
2930

30-
comm "github.com/opencurve/curveadm/internal/configure/common"
3131
"github.com/opencurve/curveadm/internal/utils"
3232
)
3333

3434
const (
35+
REQUIRE_ANY = iota
36+
REQUIRE_INT
37+
REQUIRE_STRING
38+
REQUIRE_BOOL
39+
REQUIRE_POSITIVE_INTEGER
40+
REQUIRE_STRING_SLICE
41+
3542
DEFAULT_SSH_PORT = 22
3643
)
3744

45+
type (
46+
// config item
47+
item struct {
48+
key string
49+
require int
50+
exclude bool // exclude for service config
51+
defaultValue interface{} // nil means no default value
52+
}
53+
54+
itemSet struct {
55+
items []*item
56+
key2item map[string]*item
57+
}
58+
)
59+
3860
var (
39-
itemset = comm.NewItemSet()
61+
itemset = &itemSet{
62+
items: []*item{},
63+
key2item: map[string]*item{},
64+
}
4065

41-
CONFIG_HOST = itemset.Insert(
66+
CONFIG_HOST = itemset.insert(
4267
"host",
43-
comm.REQUIRE_STRING,
68+
REQUIRE_STRING,
4469
false,
4570
nil,
4671
)
4772

48-
CONFIG_HOSTNAME = itemset.Insert(
73+
CONFIG_HOSTNAME = itemset.insert(
4974
"hostname",
50-
comm.REQUIRE_STRING,
75+
REQUIRE_STRING,
5176
false,
5277
nil,
5378
)
5479

55-
CONFIG_SSH_HOSTNAME = itemset.Insert(
80+
CONFIG_SSH_HOSTNAME = itemset.insert(
5681
"ssh_hostname",
57-
comm.REQUIRE_STRING,
82+
REQUIRE_STRING,
5883
false,
5984
nil,
6085
)
6186

62-
CONFIG_USER = itemset.Insert(
87+
CONFIG_USER = itemset.insert(
6388
"user",
64-
comm.REQUIRE_STRING,
89+
REQUIRE_STRING,
6590
false,
6691
func(hc *HostConfig) interface{} {
6792
return utils.GetCurrentUser()
6893
},
6994
)
7095

71-
CONFIG_SSH_PORT = itemset.Insert(
96+
CONFIG_SSH_PORT = itemset.insert(
7297
"ssh_port",
73-
comm.REQUIRE_POSITIVE_INTEGER,
98+
REQUIRE_POSITIVE_INTEGER,
7499
false,
75100
DEFAULT_SSH_PORT,
76101
)
77102

78-
CONFIG_PRIVATE_CONFIG_FILE = itemset.Insert(
103+
CONFIG_PRIVATE_CONFIG_FILE = itemset.insert(
79104
"private_key_file",
80-
comm.REQUIRE_STRING,
105+
REQUIRE_STRING,
81106
false,
82107
func(hc *HostConfig) interface{} {
83108
return fmt.Sprintf("%s/.ssh/id_rsa", utils.GetCurrentHomeDir())
84109
},
85110
)
86111

87-
CONFIG_FORWARD_AGENT = itemset.Insert(
112+
CONFIG_FORWARD_AGENT = itemset.insert(
88113
"forward_agent",
89-
comm.REQUIRE_BOOL,
114+
REQUIRE_BOOL,
90115
false,
91116
false,
92117
)
93118

94-
CONFIG_BECOME_USER = itemset.Insert(
119+
CONFIG_BECOME_USER = itemset.insert(
95120
"become_user",
96-
comm.REQUIRE_STRING,
121+
REQUIRE_STRING,
97122
false,
98123
nil,
99124
)
100125
)
126+
127+
func convertSlice[T int | string | any](key, value any) ([]T, error) {
128+
var slice []T
129+
if !utils.IsAnySlice(value) || len(value.([]any)) == 0 {
130+
return slice, errno.ERR_CONFIGURE_VALUE_REQUIRES_NONEMPTY_SLICE
131+
}
132+
anySlice := value.([]any)
133+
switch anySlice[0].(type) {
134+
case T:
135+
for _, str := range anySlice {
136+
slice = append(slice, str.(T))
137+
}
138+
default:
139+
return slice, errno.ERR_UNSUPPORT_CONFIGURE_VALUE_TYPE.
140+
F("%s: %v", key, value)
141+
}
142+
143+
return slice, nil
144+
}
145+
146+
func (i *item) Key() string {
147+
return i.key
148+
}
149+
150+
func (itemset *itemSet) insert(key string, require int, exclude bool, defaultValue interface{}) *item {
151+
i := &item{key, require, exclude, defaultValue}
152+
itemset.key2item[key] = i
153+
itemset.items = append(itemset.items, i)
154+
return i
155+
}
156+
157+
func (itemset *itemSet) get(key string) *item {
158+
return itemset.key2item[key]
159+
}
160+
161+
func (itemset *itemSet) getAll() []*item {
162+
return itemset.items
163+
}
164+
165+
func (itemset *itemSet) Build(key string, value interface{}) (interface{}, error) {
166+
item := itemset.get(key)
167+
if item == nil {
168+
return value, nil
169+
}
170+
171+
v, ok := utils.All2Str(value)
172+
if !ok {
173+
if !utils.IsAnySlice(value) {
174+
return nil, errno.ERR_UNSUPPORT_CONFIGURE_VALUE_TYPE.
175+
F("%s: %v", key, value)
176+
}
177+
}
178+
179+
switch item.require {
180+
case REQUIRE_ANY:
181+
// do nothing
182+
183+
case REQUIRE_STRING:
184+
if len(v) == 0 {
185+
return nil, errno.ERR_CONFIGURE_VALUE_REQUIRES_NON_EMPTY_STRING.
186+
F("%s: %v", key, value)
187+
} else {
188+
return v, nil
189+
}
190+
191+
case REQUIRE_INT:
192+
if v, ok := utils.Str2Int(v); !ok {
193+
return nil, errno.ERR_CONFIGURE_VALUE_REQUIRES_INTEGER.
194+
F("%s: %v", key, value)
195+
} else {
196+
return v, nil
197+
}
198+
199+
case REQUIRE_POSITIVE_INTEGER:
200+
if v, ok := utils.Str2Int(v); !ok {
201+
return nil, errno.ERR_CONFIGURE_VALUE_REQUIRES_INTEGER.
202+
F("%s: %v", key, value)
203+
} else if v <= 0 {
204+
return nil, errno.ERR_CONFIGURE_VALUE_REQUIRES_POSITIVE_INTEGER.
205+
F("%s: %v", key, value)
206+
} else {
207+
return v, nil
208+
}
209+
210+
case REQUIRE_BOOL:
211+
if v, ok := utils.Str2Bool(v); !ok {
212+
return nil, errno.ERR_CONFIGURE_VALUE_REQUIRES_BOOL.
213+
F("%s: %v", key, value)
214+
} else {
215+
return v, nil
216+
}
217+
218+
case REQUIRE_STRING_SLICE:
219+
return convertSlice[string](key, value)
220+
221+
default:
222+
// do nothing
223+
}
224+
225+
return value, nil
226+
}

0 commit comments

Comments
 (0)