|
@@ -108,6 +108,10 @@ func fillCapacity(devicesList *DevicesList) {
|
|
DiskSizes := wmicGetinfo("diskdrive", "Size")
|
|
DiskSizes := wmicGetinfo("diskdrive", "Size")
|
|
for i, device := range devicesList.Devices {
|
|
for i, device := range devicesList.Devices {
|
|
for j := range DiskNames {
|
|
for j := range DiskNames {
|
|
|
|
+ //since Intel driver will alter drive name to "XXXX SCSI Disk Device"
|
|
|
|
+ //so remove the string to increase the match probability
|
|
|
|
+ DiskNames[j] = strings.ReplaceAll(DiskNames[j], " SCSI Disk Device", "")
|
|
|
|
+ //if the name match && capacity == 0
|
|
if device.Smart.ModelName == DiskNames[j] && devicesList.Devices[i].Smart.UserCapacity.Bytes == 0 {
|
|
if device.Smart.ModelName == DiskNames[j] && devicesList.Devices[i].Smart.UserCapacity.Bytes == 0 {
|
|
capacity, _ := strconv.ParseInt(DiskSizes[j], 10, 64)
|
|
capacity, _ := strconv.ParseInt(DiskSizes[j], 10, 64)
|
|
devicesList.Devices[i].Smart.UserCapacity.Bytes = capacity
|
|
devicesList.Devices[i].Smart.UserCapacity.Bytes = capacity
|