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
55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/database"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/models"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/samber/lo"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// register devices that are detected by various collectors.
|
|
// This function is run everytime a collector is about to start a run. It can be used to update device metadata.
|
|
func RegisterDevices(c *gin.Context) {
|
|
deviceRepo := c.MustGet("DEVICE_REPOSITORY").(database.DeviceRepo)
|
|
logger := c.MustGet("LOGGER").(*logrus.Entry)
|
|
|
|
var collectorDeviceWrapper models.DeviceWrapper
|
|
err := c.BindJSON(&collectorDeviceWrapper)
|
|
if err != nil {
|
|
logger.Errorln("Cannot parse detected devices", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"success": false})
|
|
return
|
|
}
|
|
|
|
// Filter any device without a scrutiny UUID. This should never happen...
|
|
detectedStorageDevices := lo.Filter[models.Device](collectorDeviceWrapper.Data, func(dev models.Device, _ int) bool {
|
|
return !dev.ScrutinyUUID.IsNil()
|
|
})
|
|
|
|
errs := []error{}
|
|
for _, dev := range detectedStorageDevices {
|
|
//insert devices into DB (and update specified columns if device is already registered)
|
|
// update device fields that may change: (DeviceType, HostID)
|
|
if err := deviceRepo.RegisterDevice(c, dev); err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
}
|
|
|
|
if len(errs) > 0 {
|
|
logger.Errorln("An error occurred while registering devices", errs)
|
|
c.JSON(http.StatusInternalServerError, gin.H{
|
|
"success": false,
|
|
})
|
|
return
|
|
} else {
|
|
c.JSON(http.StatusOK, models.DeviceWrapper{
|
|
Success: true,
|
|
Data: detectedStorageDevices,
|
|
})
|
|
return
|
|
}
|
|
}
|