Method sendHeartbeatLoop

internal/worker_node/worker_node.go:233–267

(cfg *config.WorkerNodeConfig)

}

func (w *WorkerNode) sendHeartbeatLoop(cfg *config.WorkerNodeConfig) {
	pollContext, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	pollErr := wait.PollUntilContextCancel(pollContext, 2500*time.Millisecond, true,
		func(ctx context.Context) (done bool, err error) {
			workerStatistics, err := w.getWorkerStatistics()
			if err != nil {
				return false, err
			}

			resp, err := w.cpApi.NodeHeartbeat(ctx, workerStatistics)

			// In case we don't manage to connect, we give up
			if err != nil || resp == nil {
				return false, err
			}

			return resp.Success, nil
		},
	)
	if pollErr != nil {
		logrus.Warnf("Failed to send a heartbeat to the control plane : %s", pollErr)
		logrus.Warnf("Trying to establish connection with some other control plane replica.")
		cpApi, err := grpc_helpers.NewControlPlaneConnection(cfg.ControlPlaneAddress)
		if err != nil {
			logrus.Fatalf("Cannot establish connection with any of the specified control plane(s) (error : %s)", err.Error())
		} else {
			w.cpApi = cpApi
			logrus.Infof("Control plance changed successfully.")
		}
	} else {
		logrus.Debug("Sent heartbeat to the control plane")
	}
}