62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
|
package crush
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/unistack-org/crushkv/crushmap"
|
||
|
|
)
|
||
|
|
|
||
|
|
func makeNode(m *crushmap.Map, bucket *crushmap.Bucket, parent *TestingNode) *TestingNode {
|
||
|
|
var child *TestingNode
|
||
|
|
node := new(TestingNode)
|
||
|
|
node.ID = bucket.TypeName + ":" + bucket.Name
|
||
|
|
node.Type = uint16(bucket.TypeID)
|
||
|
|
node.Weight = bucket.Weight
|
||
|
|
node.Alg = bucket.Alg
|
||
|
|
//node.Children = make([]Node, len(bucket.Items))
|
||
|
|
node.Parent = parent
|
||
|
|
if parent != nil {
|
||
|
|
node.ID = parent.ID + "->" + node.ID
|
||
|
|
// parent.Weight += node.Weight
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, item := range bucket.Items {
|
||
|
|
childBucket := m.GetBucketByName(item.Name)
|
||
|
|
if childBucket != nil {
|
||
|
|
child = makeNode(m, childBucket, node)
|
||
|
|
} else {
|
||
|
|
idx := strings.Index(item.Name, ".")
|
||
|
|
child = &TestingNode{
|
||
|
|
ID: item.Name,
|
||
|
|
Type: m.GetTypeIDByName(item.Name[:idx]),
|
||
|
|
Weight: item.Weight,
|
||
|
|
Parent: node,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
child.ID = node.ID + "->" + child.ID
|
||
|
|
if parent != nil {
|
||
|
|
parent.Weight += child.Weight
|
||
|
|
}
|
||
|
|
|
||
|
|
switch child.Alg {
|
||
|
|
case "straw2":
|
||
|
|
child.Selector = NewStraw2Selector(child)
|
||
|
|
}
|
||
|
|
|
||
|
|
node.Children = append(node.Children, child)
|
||
|
|
}
|
||
|
|
|
||
|
|
if node.Weight == 0 {
|
||
|
|
for _, child := range node.Children {
|
||
|
|
node.Weight += child.GetWeight()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
switch bucket.Alg {
|
||
|
|
case "straw2":
|
||
|
|
node.Selector = NewStraw2Selector(node)
|
||
|
|
}
|
||
|
|
|
||
|
|
return node
|
||
|
|
}
|