1 files changed, 8 insertions, 4 deletions
diff --git a/internal/clients/handlers/handler.go b/internal/clients/handlers/handler.go
index 2013be0..c53ca34 100644
--- a/internal/clients/handlers/handler.go
+++ b/internal/clients/handlers/handler.go
@@ -1,12 +1,16 @@
 package handlers
 
-import "io"
+import (
+	"context"
+	"io"
+)
 
 // Handler provides all methods which can be run on any client handler.
 type Handler interface {
 	io.ReadWriter
-	Ping() error
-	Stop()
-	SendCommand(command string) error
+	SendMessage(command string) error
 	Server() string
+	Status() int
+	WithCancel(ctx context.Context) (context.Context, context.CancelFunc)
+	Done() <-chan struct{}
 }