From 516ac073a0ef6af74b2bd8cb1f04285300c0bfc9 Mon Sep 17 00:00:00 2001
From: Yashash H L <yashashhl25@gmail.com>
Date: Thu, 26 Sep 2024 10:09:06 +0530
Subject: [PATCH] pass eot request

Signed-off-by: Yashash H L <yashashhl25@gmail.com>
---
 pkg/sdkclient/sinker/client.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pkg/sdkclient/sinker/client.go b/pkg/sdkclient/sinker/client.go
index 9e9cb31386..51c1273568 100644
--- a/pkg/sdkclient/sinker/client.go
+++ b/pkg/sdkclient/sinker/client.go
@@ -154,6 +154,17 @@ func (c *client) SinkFn(ctx context.Context, requests []*sinkpb.SinkRequest) ([]
 		}
 	}
 
+	// send eot request
+	eotRequest := &sinkpb.SinkRequest{
+		Status: &sinkpb.SinkRequest_Status{
+			Eot: true,
+		},
+	}
+
+	if err := c.sinkStream.Send(eotRequest); err != nil {
+		return nil, fmt.Errorf("failed to send eot request: %v", err)
+	}
+
 	// Wait for the corresponding responses
 	var responses []*sinkpb.SinkResponse
 	for i := 0; i < len(requests); i++ {