Esempio n. 1
0
func (tl *TeamMaster) findServers(dc *resource.DataCenter, req *resource.AllocationRequest) (ret []resource.Allocation) {
	// sort racks by unallocated resources
	var racks []*resource.Rack
	for _, rack := range dc.GetRacks() {
		racks = append(racks, rack)
	}
	sort.Sort(ByAvailableResources(racks))

	requests := make([]*resource.ComputeRequest, 0, len(req.Requests))
	for i := range req.Requests {
		requests = append(requests, &req.Requests[i])
	}
	sort.Sort(ByRequestedResources(requests))

	for _, rack := range racks {
		allocated, requests := tl.allocateServersOnRack(dc, rack, requests)
		ret = append(ret, allocated...)
		if len(requests) == 0 {
			break
		}
	}
	return
}
Esempio n. 2
0
func (tl *TeamMaster) allocateServersOnRack(dc *resource.DataCenter, rack *resource.Rack, requests []*resource.ComputeRequest) (
	allocated []resource.Allocation, remainingRequests []*resource.ComputeRequest) {
	var j = -1
	for _, agent := range rack.Agents {
		if j >= len(requests) {
			break
		}
		available := agent.Resource.Minus(agent.Allocated)
		hasAllocation := true
		for available.GreaterThanZero() && hasAllocation && j < len(requests) {
			hasAllocation = false

			j++
			if j >= len(requests) {
				break
			}
			request := requests[j]

			// fmt.Printf("available %v, requested %v\n", available, request.ComputeResource)
			if available.Covers(request.ComputeResource) {
				allocated = append(allocated, resource.Allocation{
					Location:  agent.Location,
					Allocated: request.ComputeResource,
				})
				agent.Allocated = agent.Allocated.Plus(request.ComputeResource)
				rack.Allocated = rack.Allocated.Plus(request.ComputeResource)
				dc.Allocated = dc.Allocated.Plus(request.ComputeResource)
				tl.MasterResource.Topology.Allocated = tl.MasterResource.Topology.Allocated.Plus(request.ComputeResource)
				available = available.Minus(request.ComputeResource)
				hasAllocation = true
			} else {
				remainingRequests = append(remainingRequests, request)
			}
		}
	}
	return
}