fix errors

Signed-off-by: Nikita Skrynnik <nikita.skrynnik@xored.com>
NikitaSkrynnik · Sep 4, 2023 · 5df3ea9 · 5df3ea9
1 parent f96fdf6
commit 5df3ea9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pkg/networkservice/common/heal/client.go b/pkg/networkservice/common/heal/client.go
@@ -60,6 +60,7 @@ func (h *healClient) Request(ctx context.Context, request *networkservice.Networ
 	// Cancel any existing eventLoop
 	loopHandle, loaded := loadAndDelete(ctx)
 	if loaded {
+		log.FromContext(ctx).Info("Stopping Event Loop. All pings should be stopped now")
 		loopHandle.cancel()
 		if started, ok := <-loopHandle.healingStartedCh; ok {
 			loopHandle.healingStarted = started
@@ -79,6 +80,7 @@ func (h *healClient) Request(ctx context.Context, request *networkservice.Networ
 		}
 		return nil, err
 	}
+	log.FromContext(ctx).Info("Starting Event Loop Again. All pings should be resumed now")
 	eventLoopErr := h.startEventLoop(ctx, conn)
 	if eventLoopErr != nil {
 		closeCtx, closeCancel := closeCtxFunc()

diff --git a/pkg/networkservice/common/heal/eventloop.go b/pkg/networkservice/common/heal/eventloop.go
@@ -216,9 +216,11 @@ func (cev *eventLoop) monitorDataPlane() <-chan struct{} {
 		for {
 			select {
 			case <-ticker.C:
-				deadlineCtx, deadlineCancel := context.WithDeadline(cev.chainCtx, time.Now().Add(cev.heal.livenessCheckTimeout))
+				cev.logger.Info("Starting Liveness Check")
+				deadlineCtx, deadlineCancel := context.WithDeadline(cev.eventLoopCtx, time.Now().Add(cev.heal.livenessCheckTimeout))
 				alive := cev.heal.livenessCheck(deadlineCtx, cev.conn)
 				deadlineCancel()
+				cev.logger.Info("Finishing Liveness Check")
 				if !alive {
 					// Start healing
 					return