d3m0k1d
71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
package collector
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/proto/proto"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
// ReportServices handles a unary service status update from an agent.
|
|
// Agents send their current services list, which is stored in the collector.
|
|
func (c *Collector) ReportServices(ctx context.Context, req *proto.ServicesUpdate) (*proto.ServicesUpdateResp, error) {
|
|
md, ok := metadata.FromIncomingContext(ctx)
|
|
if !ok {
|
|
return nil, fmt.Errorf("no metadata in context")
|
|
}
|
|
|
|
whoamiVals := md["whoami"]
|
|
if len(whoamiVals) == 0 {
|
|
return nil, fmt.Errorf("whoami metadata missing")
|
|
}
|
|
agentName := whoamiVals[0]
|
|
|
|
services := make([]Service, 0, len(req.Services))
|
|
for _, s := range req.Services {
|
|
services = append(services, Service{s.Name, s.Status})
|
|
}
|
|
|
|
if ok := c.tracker.UpdateServices(agentName, services); ok {
|
|
log.Printf("Updated services for agent %s: %v", agentName, services)
|
|
} else {
|
|
log.Printf("Warning: received services update for unknown agent %s", agentName)
|
|
}
|
|
|
|
return &proto.ServicesUpdateResp{}, nil
|
|
}
|
|
|
|
// ReportSystemMetrics handles system metrics update from an agent.
|
|
// Agents send their current system metrics (CPU, RAM, disk, network).
|
|
func (c *Collector) ReportSystemMetrics(ctx context.Context, req *proto.SystemMetrics) (*proto.SystemMetricsResp, error) {
|
|
md, ok := metadata.FromIncomingContext(ctx)
|
|
if !ok {
|
|
return nil, fmt.Errorf("no metadata in context")
|
|
}
|
|
|
|
whoamiVals := md["whoami"]
|
|
if len(whoamiVals) == 0 {
|
|
return nil, fmt.Errorf("whoami metadata missing")
|
|
}
|
|
agentName := whoamiVals[0]
|
|
|
|
metrics := SystemMetrics{
|
|
CPUPercent: req.CpuPercent,
|
|
MemoryPercent: req.MemoryPercent,
|
|
DiskPercent: req.DiskPercent,
|
|
NetworkRxBytes: req.NetworkRxBytes,
|
|
NetworkTxBytes: req.NetworkTxBytes,
|
|
}
|
|
|
|
if ok := c.tracker.UpdateSystemMetrics(agentName, metrics); ok {
|
|
log.Printf("Updated system metrics for agent %s: CPU=%.1f%%, RAM=%.1f%%, Disk=%.1f%%",
|
|
agentName, metrics.CPUPercent, metrics.MemoryPercent, metrics.DiskPercent)
|
|
} else {
|
|
log.Printf("Warning: received system metrics for unknown agent %s", agentName)
|
|
}
|
|
|
|
return &proto.SystemMetricsResp{}, nil
|
|
}
|