func HandleSlotAllocations(zkHelper *utils.ZkHelper, slotCount int, nodeId int) *SlotAllocation {
allocationsPath := "/yundis/allocations"
allocations := GetSlotAllocationsFromZk(zkHelper)
if allocations != nil { //init from zk
log.Info("Init slot allocation from zk.")
//return allocations
} else { //init the cluster, and save to zk
allocations = CreateSlotAlloction(slotCount, nodeId)
//self.SetAllocations(slotAllocations)
bytes, err := allocations.ToNodeData()
if err != nil {
log.Errorf("Can convert slotallocation to json str. err : %s", err)
}
//create this path, and initial the data into zk
_, err = zkHelper.GetZkConn().Create(allocationsPath, bytes, 0, zk.WorldACL(zk.PermAll))
if err != nil {
log.Errorf("Can not write allocations to zk. err : %s", err)
}
}
return allocations
/*
log.Info("Watching the change of /yundis/allocations.")
_, _, ch, err := zkHelper.GetZkConn().GetW("/yundis/allocations")
if err != nil {
log.Errorf("Can not watch path /yundis/allocations, err:%s", err)
return err
}
go func() {
for {
event := <-ch
log.Infof("The value of /yundis/allocations changed. %+v", event)
values, _, ch1, err1 := zkHelper.GetZkConn().GetW("/yundis/allocations")
if err1 == nil {
ch = ch1
log.Infof("The value of /yundis/allocations:%s", string(values))
//handle the new value of /yundis/allocations
newAllocations, err := InitSlotAlloctionWithData(string(values))
if err != nil {
log.Errorf("Can not init Allocation by data from zk. err: %s", err)
} else {
HandleAllocationChange(self.allocations, newAllocations, self.slotinfoMaps, self.zkHelper)
log.Info("Update the allocations.")
self.SetAllocations(newAllocations)
}
} else {
log.Errorf("Can not watching the value of /yundis/allocations, err:%s", err1)
break
}
time.Sleep(time.Second)
}
}()*/
}
/**
* init slot allocation from zk.
*/
func GetSlotAllocationsFromZk(zkHelper *utils.ZkHelper) *SlotAllocation {
allocationsPath := "/yundis/allocations"
if b := zkHelper.PathExist(allocationsPath); b {
bytes, _, err := zkHelper.GetZkConn().Get(allocationsPath)
if err != nil {
log.Errorf("Read the data of '/yundis/allocations' error. err : %s", err)
}
allocations, err := InitSlotAlloctionWithJson(string(bytes))
if err != nil {
log.Errorf("Parse the json data error. err:%s", err)
} else {
return allocations
}
}
return nil
}
