()
| 221 | } |
| 222 | |
| 223 | func (w *WorkerNode) getWorkerStatistics() (*proto.NodeHeartbeatMessage, error) { |
| 224 | hardwareUsage := hardware.GetHardwareUsage() |
| 225 | |
| 226 | return &proto.NodeHeartbeatMessage{ |
| 227 | NodeID: w.Name, |
| 228 | CpuUsage: hardwareUsage.CpuUsage, |
| 229 | MemoryUsage: hardwareUsage.MemoryUsage, |
| 230 | }, nil |
| 231 | } |
| 232 | |
| 233 | func (w *WorkerNode) sendHeartbeatLoop(cfg *config.WorkerNodeConfig) { |
| 234 | pollContext, cancel := context.WithTimeout(context.Background(), 5*time.Second) |
no test coverage detected