mirror of
https://github.com/mulgadc/spinifex.git
synced 2026-10-21 11:46:37 +00:00
Julian Sommer
6737c42601
Move EBS, cluster, and node message types from config/ to types/. Remove duplicate NodeDiscoverResponse from gateway and daemon. Delete dead EC2StartInstances types and tests from config. Update all imports across source and test files.
75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
package daemon
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/mulgadc/hive/hive/types"
|
|
)
|
|
|
|
// errInsufficientCapacity is returned by allocateForLaunch when MinCount
|
|
// cannot be satisfied.
|
|
var errInsufficientCapacity = errors.New("insufficient capacity to satisfy MinCount")
|
|
|
|
// canAllocateCount returns how many instances of the given type can fit
|
|
// in the remaining capacity, capped at maxCount. Pure function — no locks
|
|
// or side effects.
|
|
func canAllocateCount(availVCPU, allocVCPU int, availMem, allocMem float64,
|
|
vCPUs int64, memMiB int64, maxCount int) int {
|
|
|
|
remainingVCPU := availVCPU - allocVCPU
|
|
remainingMem := availMem - allocMem
|
|
memoryGB := float64(memMiB) / 1024.0
|
|
|
|
countByCPU := maxCount
|
|
if vCPUs > 0 {
|
|
countByCPU = remainingVCPU / int(vCPUs)
|
|
}
|
|
|
|
countByMem := maxCount
|
|
if memoryGB > 0 {
|
|
countByMem = int(remainingMem / memoryGB)
|
|
}
|
|
|
|
result := min(countByMem, countByCPU)
|
|
result = min(result, maxCount)
|
|
return max(result, 0)
|
|
}
|
|
|
|
// resourceStatsForType computes the InstanceTypeCap for a single instance type
|
|
// given the remaining host resources. Pure function — no locks or side effects.
|
|
func resourceStatsForType(remainVCPU int, remainMem float64, it *ec2.InstanceTypeInfo) types.InstanceTypeCap {
|
|
vCPUs := instanceTypeVCPUs(it)
|
|
memGB := float64(instanceTypeMemoryMiB(it)) / 1024.0
|
|
|
|
count := 0
|
|
if vCPUs > 0 && memGB > 0 {
|
|
countVCPU := remainVCPU / int(vCPUs)
|
|
countMem := int(remainMem / memGB)
|
|
count = max(min(countMem, countVCPU), 0)
|
|
}
|
|
|
|
name := ""
|
|
if it.InstanceType != nil {
|
|
name = *it.InstanceType
|
|
}
|
|
|
|
return types.InstanceTypeCap{
|
|
Name: name,
|
|
VCPU: int(vCPUs),
|
|
MemoryGB: memGB,
|
|
Available: count,
|
|
}
|
|
}
|
|
|
|
// allocateForLaunch determines the number of instances to launch given
|
|
// available capacity and the MinCount/MaxCount constraints from a
|
|
// RunInstances request. Returns the launch count or an error if the
|
|
// minimum cannot be satisfied.
|
|
func allocateForLaunch(canAlloc, minCount, maxCount int) (int, error) {
|
|
if canAlloc < minCount {
|
|
return 0, errInsufficientCapacity
|
|
}
|
|
return min(canAlloc, maxCount), nil
|
|
}
|