Files
ubicloud/serializers/vm.rb
Benjamin Satzger 6fbe28de36 Add GPU configuration to VM serializer
Add a new field `gpu` to the detailed section of VM. It would contain
`2x NVIDIA A100 80GB PCIe" in case 2 GPUs of device type 20b5
are attached to the VM.
2025-05-26 16:48:14 +02:00

42 lines
1.1 KiB
Ruby

# frozen_string_literal: true
class Serializers::Vm < Serializers::Base
def self.serialize_internal(vm, options = {})
base = {
id: vm.ubid,
name: vm.name,
state: vm.display_state,
location: vm.display_location,
size: vm.display_size,
unix_user: vm.unix_user,
storage_size_gib: vm.storage_size_gib,
ip6: vm.ip6,
ip4_enabled: vm.ip4_enabled,
ip4: vm.ephemeral_net4
}
if options[:include_path]
base[:path] = vm.path
end
if options[:detailed]
gpu_devices = vm.pci_devices.select { |d| ["0300", "0302"].include?(d.device_class) }
gpu = "#{gpu_devices.count}x #{PciDevice.device_name(gpu_devices.first.device)}" if gpu_devices.any?
base.merge!(
firewalls: Serializers::Firewall.serialize(vm.firewalls, {include_path: true}),
private_ipv4: vm.private_ipv4,
private_ipv6: vm.private_ipv6,
subnet: vm.nics.first.private_subnet.name,
gpu:
)
end
if options[:load_balancer]
base[:load_balancer_state] = vm.load_balancer_vm_ports.first&.state
end
base
end
end