From 4ec0dff05e517c6c79b7606f9dc9c54b2015df59 Mon Sep 17 00:00:00 2001
From: Diego Bernardes <diego.bernardes@outlook.com>
Date: Sat, 23 Dec 2023 22:00:56 +0000
Subject: [PATCH] Check if context is canceled during event sent

---
 client.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/client.go b/client.go
index ab56179..187f7d0 100644
--- a/client.go
+++ b/client.go
@@ -161,11 +161,14 @@ func (c *Client) sendRequest(ctx context.Context, req proto.Message) (proto.Mess
 	}
 }
 
-func (c *Client) sendEvent(_ context.Context, e proto.Message) error {
+func (c *Client) sendEvent(ctx context.Context, e proto.Message) error {
 	payload, err := proto.Marshal(e)
 	if err != nil {
 		return fmt.Errorf("failed to marshal event: %w", err)
 	}
+	if err := ctx.Err(); err != nil {
+		return fmt.Errorf("context error: %w", err)
+	}
 	if errSend := c.transport.send(payload); errSend != nil {
 		return fmt.Errorf("failed to send the message: %w", errSend)
 	}