65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
|
package hardware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/anatol/smart.go"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
type HardDrive struct {
|
||
|
|
ID uint `gorm:"primarykey"`
|
||
|
|
CreatedAt time.Time
|
||
|
|
UpdatedAt time.Time
|
||
|
|
DeletedAt gorm.DeletedAt `gorm:"index"`
|
||
|
|
Name string
|
||
|
|
Transport string
|
||
|
|
Size string
|
||
|
|
Model string
|
||
|
|
Serial string
|
||
|
|
Type string
|
||
|
|
Temperatures []HardDriveTemperature `gorm:"foreignKey:HardDriveID"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type HardDriveTemperature struct {
|
||
|
|
gorm.Model
|
||
|
|
HardDriveID uint
|
||
|
|
TimeStamp time.Time
|
||
|
|
Temperature int
|
||
|
|
}
|
||
|
|
|
||
|
|
// A snapshot in time of the current state of the harddrives
|
||
|
|
type HardwareSnapshot struct {
|
||
|
|
TimeStamp time.Time
|
||
|
|
HDD []*HardDrive
|
||
|
|
}
|
||
|
|
|
||
|
|
type Snapshots struct {
|
||
|
|
List []*HardwareSnapshot
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fetch the temperature of the device, optinally update the reference object
|
||
|
|
func (h *HardDrive) GetTemperature() int {
|
||
|
|
// Fetch the device by name
|
||
|
|
disk, err := smart.Open("/dev/" + h.Name)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Failed to open device %s: %s\n", h.Name, err)
|
||
|
|
return -1
|
||
|
|
}
|
||
|
|
defer disk.Close()
|
||
|
|
|
||
|
|
// Fetch SMART data
|
||
|
|
smartInfo, err := disk.ReadGenericAttributes()
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Failed to get SMART data for %s: %s\n", h.Name, err)
|
||
|
|
return -1
|
||
|
|
}
|
||
|
|
|
||
|
|
// Parse the temperature
|
||
|
|
temperature := int(smartInfo.Temperature)
|
||
|
|
|
||
|
|
// Return the found value
|
||
|
|
return temperature
|
||
|
|
}
|