func ParseConfigData(data []byte) (*Config, error) {
var cfg Config
if err := yaml.Unmarshal([]byte(data), &cfg); err != nil {
return nil, err
}
return &cfg, nil
}
func TestParseRule(t *testing.T) {
var s = `
schemas:
-
db: kingshard
nodes: [node1, node2, node3]
rules:
default: node1
shard:
-
table: test_shard_hash
key: id
nodes: [node2, node3]
locations: [16,16]
type: hash
-
table: test_shard_range
key: id
type: range
nodes: [node2, node3]
locations: [16,16]
table_row_limit: 10000
`
var cfg config.Config
if err := yaml.Unmarshal([]byte(s), &cfg); err != nil {
t.Fatal(err)
}
rt, err := NewRouter(&cfg.Schemas[0])
if err != nil {
t.Fatal(err)
}
if rt.DefaultRule.Nodes[0] != "node1" {
t.Fatal("default rule parse not correct.")
}
hashRule := rt.GetRule("test_shard_hash")
if hashRule.Type != HashRuleType {
t.Fatal(hashRule.Type)
}
if len(hashRule.Nodes) != 2 || hashRule.Nodes[0] != "node2" || hashRule.Nodes[1] != "node3" {
t.Fatal("parse nodes not correct.")
}
if n := hashRule.FindNode(uint64(11)); n != "node2" {
t.Fatal(n)
}
rangeRule := rt.GetRule("test_shard_range")
if rangeRule.Type != RangeRuleType {
t.Fatal(rangeRule.Type)
}
if n := rangeRule.FindNode(10000 - 1); n != "node2" {
t.Fatal(n)
}
defaultRule := rt.GetRule("defaultRule_table")
if defaultRule == nil {
t.Fatal("must not nil")
}
if defaultRule.Type != DefaultRuleType {
t.Fatal(defaultRule.Type)
}
if defaultRule.Shard == nil {
t.Fatal("nil error")
}
if n := defaultRule.FindNode(11); n != "node1" {
t.Fatal(n)
}
}