fixup to cmd plugin

Author: tehut

plugins/device/cmd/example/device.go

@@ -299,9 +299,8 @@ func (d *NvidiaDevice) fingerprint(ctx context.Context, devices chan *device.Fin
 			devices <- device.NewFingerprintError(err)
 			return
 		}
-		deviceGroups := make([]*device.DeviceGroup, 0)
 		diffDevices := d.diffFiles(files)
-		deviceGroups = d.getDeviceGroup(diffDevices)
+		deviceGroups := d.getDeviceGroup(diffDevices)
 		devices <- device.NewFingerprint(deviceGroups...)
 
 	}
@@ -376,8 +375,16 @@ func (d *NvidiaDevice) getDeviceGroup(devices []*device.Device) []*device.Device
 	groupByTuple := make(map[string][]*device.Device, len(dSlice)/2)
 	inactive := make(map[string][]*device.Device, 2)
 
-	for _, v := range devices {
-		index := rand.Intn(len(dSlice))
+	for n, v := range devices {
+		// this bit sets the device model based on the strings in dSlice
+		dSliceRange := len(dSlice) - 1
+		var index int
+		if n <= dSliceRange {
+			index = n
+		} else {
+			index = rand.Intn(dSliceRange)
+		}
+
 		tuple := strings.Join([]string{vendor, deviceType, dSlice[index]}, "/")
 		if !v.Healthy {
 			if _, ok := inactive[tuple]; !ok {
@@ -402,7 +409,7 @@ func (d *NvidiaDevice) getDeviceGroup(devices []*device.Device) []*device.Device
 				d.logger.Error("unexpected tuple split", "part0", parts[0], "part1", parts[1], "part2", parts[2])
 				continue
 			}
-			mem := int64(rand.Intn(20))
+			mem := int64(rand.Intn(50))
 			vendor := "nvidia"
 			shared := v[0].Shared.String()
 			devGroups = append(devGroups, &device.DeviceGroup{