718370f4c9
Allow plugin to take an etc_type keyword argument for using the TYPE_ETC ubid type, and remove the separate definitions in every model that uses the TYPE_ETC ubid type. This was the cleanest way to DRY things up. You cannot extend the models with a module to do this before including ResourceMethods, because then ResourceMethods::ClassMethods will override it, and you cannot extend the models with a module to do this after including ResourceMethods, because the inclusion will not work correctly due to the eager definition of @ubid_format. Best reviewed without whitespace differences.
51 lines
1.3 KiB
Ruby
51 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../model"
|
|
|
|
class PciDevice < Sequel::Model
|
|
many_to_one :vm_host
|
|
many_to_one :vm
|
|
|
|
def self.device_name(device_id)
|
|
# https://download.nvidia.com/XFree86/Linux-x86_64/535.98/README/supportedchips.html
|
|
case device_id
|
|
when "20b5"
|
|
"NVIDIA A100 80GB PCIe"
|
|
when "27b0"
|
|
"NVIDIA RTX 4000 SFF Ada Generation"
|
|
else
|
|
"PCI device"
|
|
end
|
|
end
|
|
|
|
plugin ResourceMethods, etc_type: true
|
|
|
|
def is_gpu
|
|
["0300", "0302"].include? device_class
|
|
end
|
|
|
|
def name
|
|
PciDevice.device_name(device)
|
|
end
|
|
end
|
|
|
|
# Table: pci_device
|
|
# Columns:
|
|
# id | uuid | PRIMARY KEY
|
|
# slot | text | NOT NULL
|
|
# device_class | text | NOT NULL
|
|
# vendor | text | NOT NULL
|
|
# device | text | NOT NULL
|
|
# numa_node | integer |
|
|
# iommu_group | integer | NOT NULL
|
|
# enabled | boolean | NOT NULL DEFAULT true
|
|
# vm_host_id | uuid | NOT NULL
|
|
# vm_id | uuid |
|
|
# Indexes:
|
|
# pci_device_pkey | PRIMARY KEY btree (id)
|
|
# pci_device_vm_host_id_slot_key | UNIQUE btree (vm_host_id, slot)
|
|
# pci_device_vm_id_index | btree (vm_id)
|
|
# Foreign key constraints:
|
|
# pci_device_vm_host_id_fkey | (vm_host_id) REFERENCES vm_host(id)
|
|
# pci_device_vm_id_fkey | (vm_id) REFERENCES vm(id)
|