Aram Akhavan
c3b2eb2b4f
* Generate a UUIDv5 from a random namespace based on WWN, model name, and serial number * Migrate sqlite and influxdb data accordingly * Update frontend API routes and components * Fixes #923
100 lines
3.3 KiB
Go
100 lines
3.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/config"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/database"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/models/collector"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/notify"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gofrs/uuid/v5"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func UploadDeviceMetrics(c *gin.Context) {
|
|
//db := c.MustGet("DB").(*gorm.DB)
|
|
logger := c.MustGet("LOGGER").(*logrus.Entry)
|
|
appConfig := c.MustGet("CONFIG").(config.Interface)
|
|
//influxWriteDb := c.MustGet("INFLUXDB_WRITE").(*api.WriteAPIBlocking)
|
|
deviceRepo := c.MustGet("DEVICE_REPOSITORY").(database.DeviceRepo)
|
|
|
|
//appConfig := c.MustGet("CONFIG").(config.Interface)
|
|
|
|
scrutiny_uuid, err := uuid.FromString(c.Param("scrutiny_uuid"))
|
|
if err != nil {
|
|
logger.Errorln("Invalid scrutiny uuid", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
|
|
var collectorSmartData collector.SmartInfo
|
|
err = c.BindJSON(&collectorSmartData)
|
|
if err != nil {
|
|
logger.Errorln("Cannot parse SMART data", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
|
|
//update the device information if necessary
|
|
updatedDevice, err := deviceRepo.UpdateDevice(c, scrutiny_uuid, collectorSmartData)
|
|
if err != nil {
|
|
logger.Errorln("An error occurred while updating device data from smartctl metrics:", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
|
|
// insert smart info
|
|
smartData, err := deviceRepo.SaveSmartAttributes(c, scrutiny_uuid, collectorSmartData)
|
|
if err != nil {
|
|
logger.Errorln("An error occurred while saving smartctl metrics", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
|
|
if smartData.Status != pkg.DeviceStatusPassed {
|
|
//there is a failure detected by Scrutiny, update the device status on the homepage.
|
|
updatedDevice, err = deviceRepo.UpdateDeviceStatus(c, scrutiny_uuid, smartData.Status)
|
|
if err != nil {
|
|
logger.Errorln("An error occurred while updating device status", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
}
|
|
|
|
// save smart temperature data (ignore failures)
|
|
err = deviceRepo.SaveSmartTemperature(c, scrutiny_uuid, updatedDevice.DeviceProtocol, collectorSmartData, appConfig.GetBool(fmt.Sprintf("%s.collector.discard_sct_temp_history", config.DB_USER_SETTINGS_SUBKEY)))
|
|
if err != nil {
|
|
logger.Errorln("An error occurred while saving smartctl temp data", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
|
|
//check for error
|
|
if notify.ShouldNotify(
|
|
logger,
|
|
updatedDevice,
|
|
smartData,
|
|
scrutiny_uuid,
|
|
pkg.MetricsStatusThreshold(appConfig.GetInt(fmt.Sprintf("%s.metrics.status_threshold", config.DB_USER_SETTINGS_SUBKEY))),
|
|
pkg.MetricsStatusFilterAttributes(appConfig.GetInt(fmt.Sprintf("%s.metrics.status_filter_attributes", config.DB_USER_SETTINGS_SUBKEY))),
|
|
appConfig.GetBool(fmt.Sprintf("%s.metrics.repeat_notifications", config.DB_USER_SETTINGS_SUBKEY)),
|
|
c,
|
|
deviceRepo,
|
|
) {
|
|
//send notifications
|
|
|
|
liveNotify := notify.New(
|
|
logger,
|
|
appConfig,
|
|
updatedDevice,
|
|
false,
|
|
)
|
|
_ = liveNotify.Send() //we ignore error message when sending notifications.
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|