models

type BasicNodeAttributes struct {
	NVLink string `yaml:"nvlink,omitempty"`
}

type CapacityBlock struct {
	Nodes      []string            `yaml:"nodes"`
	Attributes BasicNodeAttributes `yaml:"attributes"`
}

type GPU struct {
	Index     int    `yaml:"index"`
	PCIBusID  string `yaml:"pci_bus_id"`
	UUID      string `yaml:"uuid"`
	Model     string `yaml:"model"`
	MemoryMiB int    `yaml:"memory_mib"`
}

type Model struct {
	Switches       map[string]*Switch       `yaml:"switches"`
	Nodes          map[string]*Node         `yaml:"nodes"`
	CapacityBlocks map[string]CapacityBlock `yaml:"capacity_blocks"`

	// derived
	Instances []topology.ComputeInstances `yaml:"-"`
}

type Node struct {
	Name          string         `yaml:"name,omitempty"`
	Attributes    NodeAttributes `yaml:"attributes"`
	CapacityBlock string         `yaml:"capacity_block_id"`

	Metadata  map[string]string `yaml:"-"`
	NetLayers []string          `yaml:"-"`
}

type NodeAttributes struct {
	BasicNodeAttributes `yaml:",inline"`
	Status              string `yaml:"status,omitempty"`
	Timestamp           string `yaml:"timestamp,omitempty"`
	GPUs                []GPU  `yaml:"gpus,omitempty"`
}

type Switch struct {
	Name     string            `yaml:"name,omitempty"`
	Metadata map[string]string `yaml:"metadata"`
	Switches []string          `yaml:"switches"`
	Nodes    []string          `yaml:"nodes"`
}