From 77753da0d384225cbd4661e95a23ccebf321c5d4 Mon Sep 17 00:00:00 2001
From: nquidox <nquidox@gmail.com>
Date: Thu, 2 Apr 2026 18:01:12 +0300
Subject: [PATCH] method fix

---
 internal/taskAgent/service.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/internal/taskAgent/service.go b/internal/taskAgent/service.go
index fa6ba7f..c430492 100644
--- a/internal/taskAgent/service.go
+++ b/internal/taskAgent/service.go
@@ -32,8 +32,8 @@ func (s *service) FetchTasks(ctx context.Context) (received []structs.Task, err
 		return nil, err
 	}
 
+	log.Infof("%v Receiving tasks", pkgLogHeader)
 	for {
-		log.Infof("%v Receiving tasks", pkgLogHeader)
 		task, err := stream.Recv()
 		if err == io.EOF {
 			break
@@ -47,8 +47,8 @@ func (s *service) FetchTasks(ctx context.Context) (received []structs.Task, err
 			MerchUuid: task.MerchUuid,
 			Origins:   task.Origins,
 		})
-		log.WithField("Count", len(received)).Infof("%v End receiving tasks", pkgLogHeader)
 	}
+	log.WithField("Count", len(received)).Infof("%v End receiving tasks", pkgLogHeader)
 
 	return received, nil
 }
@@ -81,5 +81,11 @@ func (s *service) SendResults(ctx context.Context, results []structs.Result) err
 		return err
 	}
 
+	if _, err = stream.CloseAndRecv(); err != nil {
+		log.WithError(err).Errorf("%v Failed to receive server response", pkgLogHeader)
+		return err
+	}
+
+	log.Debugf("%v End send results", pkgLogHeader)
 	return nil
 }