// State currently does not return tasks or lrp rootfs, because the
// auctioneer currently does not need them.
func (a *AuctionCellRep) State() (rep.CellState, error) {
logger := a.logger.Session("auction-state")
logger.Info("providing")
totalResources, err := a.fetchResourcesVia(a.client.TotalResources)
if err != nil {
logger.Error("failed-to-get-total-resources", err)
return rep.CellState{}, err
}
availableResources, err := a.fetchResourcesVia(a.client.RemainingResources)
if err != nil {
logger.Error("failed-to-get-remaining-resource", err)
return rep.CellState{}, err
}
containers, err := a.client.ListContainers(nil)
if err != nil {
logger.Error("failed-to-fetch-containers", err)
return rep.CellState{}, err
}
var taskGuid, taskDomain string
var key *models.ActualLRPKey
var keyErr error
tasks := []rep.Task{}
lrps := []rep.LRP{}
for i := range containers {
container := &containers[i]
resource := rep.Resource{MemoryMB: int32(container.MemoryMB), DiskMB: int32(container.DiskMB)}
if container.Tags[rep.LifecycleTag] == rep.TaskLifecycle {
taskGuid, taskDomain, keyErr = taskKeyFromTags(container.Tags)
if keyErr != nil {
logger.Error("failed-to-extract-task key", keyErr)
continue
}
tasks = append(tasks, rep.NewTask(taskGuid, taskDomain, resource))
} else {
key, keyErr = rep.ActualLRPKeyFromTags(container.Tags)
if keyErr != nil {
logger.Error("failed-to-extract-key", keyErr)
continue
}
lrps = append(lrps, rep.NewLRP(*key, resource))
}
}
state := rep.NewCellState(a.rootFSProviders, availableResources, totalResources, lrps, tasks, a.zone, a.evacuationReporter.Evacuating())
a.logger.Info("provided", lager.Data{
"available-resources": state.AvailableResources,
"total-resources": state.TotalResources,
"num-lrps": len(state.LRPs),
"zone": state.Zone,
"evacuating": state.Evacuating,
})
return state, nil
}
func BuildCellState(
zone string,
memoryMB int32,
diskMB int32,
containers int,
evacuating bool,
rootFSProviders rep.RootFSProviders,
lrps []rep.LRP,
) rep.CellState {
totalResources := rep.NewResources(memoryMB, diskMB, containers)
availableResources := totalResources.Copy()
for i := range lrps {
availableResources.Subtract(&lrps[i].Resource)
}
Expect(availableResources.MemoryMB).To(BeNumerically(">=", 0), "Check your math!")
Expect(availableResources.DiskMB).To(BeNumerically(">=", 0), "Check your math!")
Expect(availableResources.Containers).To(BeNumerically(">=", 0), "Check your math!")
return rep.NewCellState(rootFSProviders, availableResources, totalResources, lrps, nil, zone, evacuating)
}
// State currently does not return tasks or lrp rootfs, because the
// auctioneer currently does not need them.
func (a *AuctionCellRep) State() (rep.CellState, error) {
logger := a.logger.Session("auction-state")
logger.Info("providing")
containers, err := a.client.ListContainers(executor.Tags{})
if err != nil {
logger.Error("failed-to-fetch-containers", err)
return rep.CellState{}, err
}
totalResources, err := a.client.TotalResources()
if err != nil {
logger.Error("failed-to-get-total-resources", err)
return rep.CellState{}, err
}
availableResources, err := a.client.RemainingResourcesFrom(containers)
if err != nil {
logger.Error("failed-to-get-remaining-resource", err)
return rep.CellState{}, err
}
var key *models.ActualLRPKey
var keyErr error
lrps := []rep.LRP{}
tasks := []rep.Task{}
for i := range containers {
container := &containers[i]
resource := rep.Resource{MemoryMB: int32(container.MemoryMB), DiskMB: int32(container.DiskMB)}
if container.Tags == nil {
logger.Error("failed-to-extract-container-tags", nil)
continue
}
switch container.Tags[rep.LifecycleTag] {
case rep.LRPLifecycle:
key, keyErr = rep.ActualLRPKeyFromTags(container.Tags)
if keyErr != nil {
logger.Error("failed-to-extract-key", keyErr)
continue
}
lrps = append(lrps, rep.NewLRP(*key, resource))
case rep.TaskLifecycle:
domain := container.Tags[rep.DomainTag]
tasks = append(tasks, rep.NewTask(container.Guid, domain, resource))
}
}
state := rep.NewCellState(
a.rootFSProviders,
a.convertResources(availableResources),
a.convertResources(totalResources),
lrps,
tasks,
a.zone,
a.evacuationReporter.Evacuating(),
)
a.logger.Info("provided", lager.Data{
"available-resources": state.AvailableResources,
"total-resources": state.TotalResources,
"num-lrps": len(state.LRPs),
"zone": state.Zone,
"evacuating": state.Evacuating,
})
return state, nil
}