2024-01-20 01:27:10 +02:00
|
|
|
package svc
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"time"
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
"gorm.io/driver/sqlite"
|
|
|
|
"gorm.io/gorm"
|
2024-01-20 01:27:10 +02:00
|
|
|
"tea.chunkbyte.com/kato/drive-health/lib/config"
|
|
|
|
"tea.chunkbyte.com/kato/drive-health/lib/hardware"
|
|
|
|
)
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
var db *gorm.DB
|
2024-01-20 01:27:10 +02:00
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
func InitDB() {
|
|
|
|
var err error
|
|
|
|
dbPath := config.GetConfiguration().DatabaseFilePath
|
|
|
|
if dbPath == "" {
|
|
|
|
dbPath = "./data.db"
|
2024-01-20 01:27:10 +02:00
|
|
|
}
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
|
2024-01-20 01:27:10 +02:00
|
|
|
if err != nil {
|
2024-01-20 18:35:50 +02:00
|
|
|
panic("failed to connect database")
|
2024-01-20 01:27:10 +02:00
|
|
|
}
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
// Migrate the schema
|
|
|
|
db.AutoMigrate(&hardware.HardDrive{}, &hardware.HardDriveTemperature{})
|
2024-01-20 01:27:10 +02:00
|
|
|
}
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
func GetDatabaseRef() *gorm.DB {
|
|
|
|
return db
|
2024-01-20 01:27:10 +02:00
|
|
|
}
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
func LogDriveTemps() error {
|
|
|
|
drives, err := hardware.GetSystemHardDrives(db, nil, nil)
|
2024-01-20 01:27:10 +02:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
for _, hdd := range drives {
|
|
|
|
temp := hdd.GetTemperature()
|
|
|
|
db.Create(&hardware.HardDriveTemperature{
|
|
|
|
HardDriveID: hdd.ID,
|
|
|
|
TimeStamp: time.Now(),
|
|
|
|
Temperature: temp,
|
|
|
|
})
|
2024-01-20 01:27:10 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
func RunLoggerService() {
|
|
|
|
fmt.Println("Initializing Temperature Logging Service...")
|
2024-01-20 01:27:10 +02:00
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
tickTime := time.Duration(config.GetConfiguration().DiskFetchFrequency) * time.Second
|
2024-01-20 01:27:10 +02:00
|
|
|
|
2024-01-20 18:35:50 +02:00
|
|
|
// Snapshot taking routine
|
2024-01-20 01:27:10 +02:00
|
|
|
go func() {
|
|
|
|
for {
|
2024-01-20 18:35:50 +02:00
|
|
|
time.Sleep(tickTime)
|
|
|
|
err := LogDriveTemps()
|
2024-01-20 01:27:10 +02:00
|
|
|
if err != nil {
|
2024-01-20 18:35:50 +02:00
|
|
|
fmt.Printf("🛑 Temperature logging failed: %s\n", err)
|
2024-01-20 01:27:10 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
}
|