func NewTestCluster(nodescount int) *TestCluster {
tc := new(TestCluster)
firstport := 20000
tc.nodes = make([]*process.Process, nodescount)
nodes := make([]gostore.ConfigNode, nodescount)
for i := 0; i < nodescount; i++ {
nodes[i].NodeId = uint16(i)
nodes[i].NodeIP = "127.0.0.1"
nodes[i].TCPPort = uint16(firstport + i*10)
nodes[i].UDPPort = uint16(firstport + i*10 + 1)
}
rings := make([]gostore.ConfigRing, 1)
rings[0].Id = 0
rings[0].ReplicationFactor = 3
for i := 0; i < nodescount; i++ {
datadir := fmt.Sprintf("data/%d", i)
conf := new(gostore.Config)
conf.CurrentNode = uint16(i)
conf.Nodes = nodes
conf.Rings = rings
conf.GlobalRing = 0
conf.Services = make([]gostore.ConfigService, 1)
// add FS
conf.Services[0].Id = 2
conf.Services[0].Type = "fs"
conf.Services[0].CustomConfig = make(map[string]interface{})
conf.Services[0].CustomConfig["DataDir"] = datadir
conf.Services[0].CustomConfig["ApiAddress"] = fmt.Sprintf("127.0.0.1:%d", (firstport + i*10 + 2))
// Clear and create data dir
os.RemoveAll(datadir)
os.Mkdir(datadir, 0777)
tc.nodes[i] = process.NewProcess(*conf)
}
return tc
}
func StartNode(id int) {
var nodes []gostore.ConfigNode
if id != 0 {
nodes = make([]gostore.ConfigNode, 2)
} else {
nodes = make([]gostore.ConfigNode, 1)
}
// make the master
nodes[0].NodeId = uint16(0)
nodes[0].NodeIP = "127.0.0.1"
nodes[0].TCPPort = uint16(firstport + 0*10)
nodes[0].UDPPort = uint16(firstport + 0*10 + 1)
nodes[0].Rings = make([]gostore.ConfigNodeRing, 1)
nodes[0].Rings[0].RingId = 1
nodes[0].Rings[0].Token = "0000000000000000000000"
// make myself
if id != 0 {
nodes[1].NodeId = uint16(id)
nodes[1].NodeIP = "127.0.0.1"
nodes[1].TCPPort = uint16(firstport + id*10)
nodes[1].UDPPort = uint16(firstport + id*10 + 1)
}
rings := make([]gostore.ConfigRing, 2)
rings[0].Id = 0
rings[0].ReplicationFactor = 3
rings[1].Id = 1
rings[1].ReplicationFactor = 3
datadir := fmt.Sprintf("data/%d", id)
conf := new(gostore.Config)
conf.CurrentNode = uint16(id)
conf.Nodes = nodes
conf.Rings = rings
conf.Services = make([]gostore.ConfigService, 2)
// add cluster service
conf.Services[0].Id = 1
conf.Services[0].Type = "cls"
conf.Services[0].CustomConfig = make(map[string]interface{})
conf.Services[0].CustomConfig["DataDir"] = datadir
conf.Services[0].CustomConfig["MasterRing"] = 1.0
// add FS
conf.Services[1].Id = 2
conf.Services[1].Type = "fs"
conf.Services[1].CustomConfig = make(map[string]interface{})
conf.Services[1].CustomConfig["DataDir"] = datadir
conf.Services[1].CustomConfig["ApiAddress"] = fmt.Sprintf("127.0.0.1:%d", (firstport + id*10 + 2))
// Clear and create data dir
os.RemoveAll(datadir)
os.Mkdir(datadir, 0777)
tc.nodes[id] = process.NewProcess(*conf)
}