fix(bidengine): make GPU attributes as struct (#122)

Signed-off-by: Artur Troian <[email protected]>

bidengine/pricing.go

@@ -320,9 +320,18 @@ type storageElement struct {
 	Size  uint64 `json:"size"`
 }
 
+type gpuVendorAttributes struct {
+	Model string  `json:"model"`
+	RAM   *string `json:"ram,omitempty"`
+}
+
+type gpuAttributes struct {
+	Vendor map[string]gpuVendorAttributes `json:"vendor,omitempty"`
+}
+
 type gpuElement struct {
-	Units      uint64                       `json:"units"`
-	Attributes map[string]map[string]string `json:"attributes,omitempty"`
+	Units      uint64        `json:"units"`
+	Attributes gpuAttributes `json:"attributes"`
 }
 
 type dataForScriptElement struct {

bidengine/shellscript.go

@@ -60,16 +60,34 @@ func parseMemory(res *atypes.Memory) uint64 {
 
 func parseGPU(resource *atypes.GPU) gpuElement {
 	res := gpuElement{
-		Units:      resource.Units.Value(),
-		Attributes: make(map[string]map[string]string),
+		Units: resource.Units.Value(),
+		Attributes: gpuAttributes{
+			Vendor: make(map[string]gpuVendorAttributes),
+		},
 	}
 
 	for _, attr := range resource.Attributes {
 		tokens := strings.Split(attr.Key, "/")
-		if _, exists := res.Attributes[tokens[0]]; !exists {
-			res.Attributes[tokens[0]] = make(map[string]string)
+
+		// vendor/nvidia/model/a100
+		switch tokens[0] {
+		case "vendor":
+			vendor := tokens[1]
+			model := tokens[3]
+			var ram *string
+
+			// vendor/nvidia/model/a100/ram/80Gi
+			if len(tokens) == 6 && tokens[4] == "ram" {
+				ram = new(string)
+				*ram = tokens[5]
+			}
+
+			res.Attributes.Vendor[vendor] = gpuVendorAttributes{
+				Model: model,
+				RAM:   ram,
+			}
+		default:
 		}
-		res.Attributes[tokens[0]][tokens[1]] = tokens[2]
 	}
 
 	return res