feat: init pc-monitor project
- Client: Go-based Windows hardware monitoring (CPU, GPU, memory, disk, network, power) - Server: Go + Gin + SQLite backend with REST API - Frontend: Vue 3 + Element Plus dashboard - Docker deployment support - Windows service installation script
This commit is contained in:
89
server/api/handler/alert.go
Normal file
89
server/api/handler/alert.go
Normal file
@@ -0,0 +1,89 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"pc-monitor-server/model"
|
||||
"pc-monitor-server/service"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type AlertHandler struct {
|
||||
service *service.AlertService
|
||||
}
|
||||
|
||||
func NewAlertHandler(service *service.AlertService) *AlertHandler {
|
||||
return &AlertHandler{service: service}
|
||||
}
|
||||
|
||||
type CreateRuleRequest struct {
|
||||
DeviceID string `json:"device_id" binding:"required"`
|
||||
Metric string `json:"metric" binding:"required"`
|
||||
Operator string `json:"operator" binding:"required"`
|
||||
Threshold float64 `json:"threshold" binding:"required"`
|
||||
Duration int `json:"duration"`
|
||||
}
|
||||
|
||||
func (h *AlertHandler) CreateRule(c *gin.Context) {
|
||||
var req CreateRuleRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
rule := &model.AlertRule{
|
||||
ID: uuid.New().String(),
|
||||
DeviceID: req.DeviceID,
|
||||
Metric: req.Metric,
|
||||
Operator: req.Operator,
|
||||
Threshold: req.Threshold,
|
||||
Duration: req.Duration,
|
||||
CreatedAt: time.Now(),
|
||||
}
|
||||
|
||||
if err := h.service.CreateRule(rule); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"rule": rule})
|
||||
}
|
||||
|
||||
func (h *AlertHandler) GetRulesByDevice(c *gin.Context) {
|
||||
deviceID := c.Param("id")
|
||||
rules, err := h.service.GetRulesByDevice(deviceID)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"rules": rules})
|
||||
}
|
||||
|
||||
func (h *AlertHandler) DeleteRule(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
if err := h.service.DeleteRule(id); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"message": "Rule deleted"})
|
||||
}
|
||||
|
||||
func (h *AlertHandler) GetActiveAlerts(c *gin.Context) {
|
||||
alerts, err := h.service.GetActiveAlerts()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"alerts": alerts})
|
||||
}
|
||||
|
||||
func (h *AlertHandler) ResolveAlert(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
if err := h.service.ResolveAlert(id); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"message": "Alert resolved"})
|
||||
}
|
||||
75
server/api/handler/device.go
Normal file
75
server/api/handler/device.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"pc-monitor-server/service"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type DeviceHandler struct {
|
||||
service *service.DeviceService
|
||||
}
|
||||
|
||||
func NewDeviceHandler(service *service.DeviceService) *DeviceHandler {
|
||||
return &DeviceHandler{service: service}
|
||||
}
|
||||
|
||||
type RegisterRequest struct {
|
||||
Hostname string `json:"hostname" binding:"required"`
|
||||
OS string `json:"os" binding:"required"`
|
||||
IP string `json:"ip" binding:"required"`
|
||||
}
|
||||
|
||||
func (h *DeviceHandler) Register(c *gin.Context) {
|
||||
var req RegisterRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
device, err := h.service.Register(req.Hostname, req.OS, req.IP)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"device": device})
|
||||
}
|
||||
|
||||
func (h *DeviceHandler) GetAll(c *gin.Context) {
|
||||
devices, err := h.service.GetAll()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"devices": devices})
|
||||
}
|
||||
|
||||
func (h *DeviceHandler) GetByID(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
device, err := h.service.GetByID(id)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "Device not found"})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"device": device})
|
||||
}
|
||||
|
||||
func (h *DeviceHandler) Delete(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
if err := h.service.Delete(id); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"message": "Device deleted"})
|
||||
}
|
||||
|
||||
func (h *DeviceHandler) Heartbeat(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
if err := h.service.Heartbeat(id); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"message": "OK"})
|
||||
}
|
||||
80
server/api/handler/metrics.go
Normal file
80
server/api/handler/metrics.go
Normal file
@@ -0,0 +1,80 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"pc-monitor-server/model"
|
||||
"pc-monitor-server/service"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type MetricsHandler struct {
|
||||
service *service.MetricsService
|
||||
}
|
||||
|
||||
func NewMetricsHandler(service *service.MetricsService) *MetricsHandler {
|
||||
return &MetricsHandler{service: service}
|
||||
}
|
||||
|
||||
type ReportRequest struct {
|
||||
DeviceID string `json:"device_id" binding:"required"`
|
||||
model.Metrics
|
||||
}
|
||||
|
||||
func (h *MetricsHandler) Report(c *gin.Context) {
|
||||
var req ReportRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
metrics := req.Metrics
|
||||
metrics.DeviceID = req.DeviceID
|
||||
metrics.Timestamp = time.Now()
|
||||
|
||||
if err := h.service.Save(&metrics); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"message": "Metrics saved"})
|
||||
}
|
||||
|
||||
func (h *MetricsHandler) GetLatest(c *gin.Context) {
|
||||
deviceID := c.Param("id")
|
||||
metrics, err := h.service.GetLatest(deviceID)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "No metrics found"})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"metrics": metrics})
|
||||
}
|
||||
|
||||
func (h *MetricsHandler) GetHistory(c *gin.Context) {
|
||||
deviceID := c.Param("id")
|
||||
|
||||
startStr := c.Query("start")
|
||||
endStr := c.Query("end")
|
||||
|
||||
start := time.Now().Add(-1 * time.Hour)
|
||||
end := time.Now()
|
||||
|
||||
if startStr != "" {
|
||||
if t, err := time.Parse(time.RFC3339, startStr); err == nil {
|
||||
start = t
|
||||
}
|
||||
}
|
||||
if endStr != "" {
|
||||
if t, err := time.Parse(time.RFC3339, endStr); err == nil {
|
||||
end = t
|
||||
}
|
||||
}
|
||||
|
||||
metrics, err := h.service.GetHistory(deviceID, start, end, 1000)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"metrics": metrics})
|
||||
}
|
||||
Reference in New Issue
Block a user