bc0243a47a
Now that rhizome has switched to use CloudHypervisor 46.0 by default on Ubuntu 24.04 hosts, this code is no longer needed. This keeps the os_filter support in the allocator, as I expect that could be generally useful in future cases. Revert "Include Vm ch_version in GithubRunner#log_duration if available" This reverts commit511832a26d. Revert "Don't disable hugepages when enabling CloudHypervisor 46.0" This reverts commit0b20a316f2. Revert "Use CloudHypervisor 46.0 for GHA in E2E tests" This reverts commit144ad31f33. Revert "Use CloudHypervisor 46.0 for a percentage of GitHub Action VMs added to VmPool" This reverts commit9476744edb. Revert "Use CloudHypervisor 46.0 for a percentage of GitHub Action VMs" This reverts commit57d9aac730. Revert "Require Ubuntu 24.04 host when allocating VM with CloudHypervisor 46.0" This reverts commit3a9ccfb434.
88 lines
2.4 KiB
Ruby
88 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "net/ssh"
|
|
|
|
class Prog::Vm::VmPool < Prog::Base
|
|
subject_is :vm_pool
|
|
|
|
def self.assemble(size:, vm_size:, boot_image:, location_id:, storage_size_gib:,
|
|
storage_encrypted:, storage_skip_sync:, arch:)
|
|
DB.transaction do
|
|
vm_pool = VmPool.create_with_id(
|
|
size:,
|
|
vm_size:,
|
|
boot_image:,
|
|
location_id:,
|
|
storage_size_gib:,
|
|
storage_encrypted:,
|
|
storage_skip_sync:,
|
|
arch:
|
|
)
|
|
Strand.create(prog: "Vm::VmPool", label: "create_new_vm") { it.id = vm_pool.id }
|
|
end
|
|
end
|
|
|
|
def before_run
|
|
when_destroy_set? do
|
|
unless ["destroy", "wait_vms_destroy"].include?(strand.label)
|
|
hop_destroy
|
|
end
|
|
end
|
|
end
|
|
|
|
label def create_new_vm
|
|
storage_params = {
|
|
size_gib: vm_pool.storage_size_gib,
|
|
encrypted: vm_pool.storage_encrypted,
|
|
skip_sync: vm_pool.storage_skip_sync
|
|
}
|
|
ps = Prog::Vnet::SubnetNexus.assemble(
|
|
Config.vm_pool_project_id,
|
|
location_id: vm_pool.location_id,
|
|
allow_only_ssh: true
|
|
).subject
|
|
|
|
Prog::Vm::Nexus.assemble_with_sshable(
|
|
Config.vm_pool_project_id,
|
|
unix_user: "runneradmin",
|
|
sshable_unix_user: "runneradmin",
|
|
size: vm_pool.vm_size,
|
|
location_id: vm_pool.location_id,
|
|
boot_image: vm_pool.boot_image,
|
|
storage_volumes: [storage_params],
|
|
enable_ip4: true,
|
|
pool_id: vm_pool.id,
|
|
arch: vm_pool.arch,
|
|
swap_size_bytes: 4294963200,
|
|
private_subnet_id: ps.id
|
|
)
|
|
|
|
hop_wait
|
|
end
|
|
|
|
label def wait
|
|
if vm_pool.size - vm_pool.vms.count > 0
|
|
idle_cpus = VmHost.where(allocation_state: "accepting", arch: vm_pool.arch, location_id: [Location::GITHUB_RUNNERS_ID, Location::HETZNER_HEL1_ID, Location::HETZNER_FSN1_ID]).select_map { sum((total_cores - used_cores) * total_cpus / total_cores) }.first.to_i
|
|
waiting_cpus = Vm.where(Sequel.like(:boot_image, "github%")).where(allocated_at: nil, arch: vm_pool.arch).sum(:vcpus).to_i
|
|
pool_vm_cpus = Validation.validate_vm_size(vm_pool.vm_size, vm_pool.arch).vcpus
|
|
hop_create_new_vm if idle_cpus - waiting_cpus - pool_vm_cpus >= 0
|
|
end
|
|
nap 30
|
|
end
|
|
|
|
label def destroy
|
|
vm_pool.vms.each do |vm|
|
|
vm.private_subnets.each { it.incr_destroy }
|
|
vm.incr_destroy
|
|
end
|
|
hop_wait_vms_destroy
|
|
end
|
|
|
|
label def wait_vms_destroy
|
|
nap 10 if vm_pool.vms.count > 0
|
|
|
|
vm_pool.destroy
|
|
pop "pool destroyed"
|
|
end
|
|
end
|