fix: fixed a bug with outgoing messages (#40)

Co-authored-by: alexandertsoy <atsoy@b2broker.com>
b2broker · Oct 12, 2023 · c7276d4 · c7276d4
1 parent 4e86aae
commit c7276d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/initiator.go b/initiator.go
@@ -18,6 +18,7 @@ type InitiatorHandler interface {
 	StopWithError(err error)
 	Send(message SendingMessage) error
 	Context() context.Context
+	Stop()
 }
 
 // Initiator provides the client-side service functionality.
@@ -92,6 +93,8 @@ func (c *Initiator) Serve() error {
 
 				err := c.conn.Write(msg)
 				if err != nil {
+					c.handler.Stop()
+
 					return ErrConnClosed
 				}
 			}