Rename handler funcs

Signed-off-by: dusan <borovcanindusan1@gmail.com>

Author: dborovcanin

handler.go

@@ -7,13 +7,13 @@ type Handler interface {
 	// Name returns the protocol name.
 	Name() string
 
-	// HandleClientData processes data from client to server.
+	// ClientDataHandler processes data from client to server.
 	// Returns modified data and whether to continue proxying.
-	HandleClientData() HandleFunc
+	ClientDataHandler() HandleFunc
 
-	// HandleServerData processes data from server to client.
+	// ServerDataHandler processes data from server to client.
 	// Returns modified data and whether to continue proxying.
-	HandleServerData() HandleFunc
+	ServerDataHandler() HandleFunc
 
 	// OnConnect is called when a new connection is established.
 	OnConnect(ctx context.Context, conn ConnectionInfo) error

protocols/coap/protocol.go

@@ -30,7 +30,7 @@ func (h *CoAPHandler) Detect(data []byte) bool {
 	return version == 1
 }
 
-func (h *CoAPHandler) HandleClientData() mgate.HandleFunc {
+func (h *CoAPHandler) ClientDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		if h.logMessages && len(data) >= 4 {
 			msgType := (data[0] >> 4) & 0x03
@@ -41,7 +41,7 @@ func (h *CoAPHandler) HandleClientData() mgate.HandleFunc {
 	}
 }
 
-func (h *CoAPHandler) HandleServerData() mgate.HandleFunc {
+func (h *CoAPHandler) ServerDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		return data, true, nil
 	}

protocols/http/protocol.go

@@ -33,7 +33,7 @@ func (h *HTTPHandler) Detect(data []byte) bool {
 	return false
 }
 
-func (h *HTTPHandler) HandleClientData() mgate.HandleFunc {
+func (h *HTTPHandler) ClientDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		if h.logRequests {
 			lines := strings.Split(string(data), "\r\n")
@@ -45,7 +45,7 @@ func (h *HTTPHandler) HandleClientData() mgate.HandleFunc {
 	}
 }
 
-func (h *HTTPHandler) HandleServerData() mgate.HandleFunc {
+func (h *HTTPHandler) ServerDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		return data, true, nil
 	}

protocols/mqtt/protocol.go

@@ -31,7 +31,7 @@ func (h *MQTTHandler) Detect(data []byte) bool {
 	return packetType == 1
 }
 
-func (h *MQTTHandler) HandleClientData() mgate.HandleFunc {
+func (h *MQTTHandler) ClientDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		if h.logPackets && len(data) > 0 {
 			packetType := data[0] >> 4
@@ -49,7 +49,7 @@ func (h *MQTTHandler) HandleClientData() mgate.HandleFunc {
 	}
 }
 
-func (h *MQTTHandler) HandleServerData() mgate.HandleFunc {
+func (h *MQTTHandler) ServerDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		return data, true, nil
 	}

protocols/nats/protocol.go

@@ -35,7 +35,7 @@ func (h *NATSHandler) Detect(data []byte) bool {
 		bytes.HasPrefix(data, []byte("PING"))
 }
 
-func (h *NATSHandler) HandleClientData() mgate.HandleFunc {
+func (h *NATSHandler) ClientDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		if h.logOps {
 			h.logNATSOperation(data, conn.Client.Addr, "->")
@@ -44,7 +44,7 @@ func (h *NATSHandler) HandleClientData() mgate.HandleFunc {
 	}
 }
 
-func (h *NATSHandler) HandleServerData() mgate.HandleFunc {
+func (h *NATSHandler) ServerDataHandler() mgate.HandleFunc {
 	return func(ctx context.Context, data []byte, conn mgate.ConnectionInfo) ([]byte, bool, error) {
 		if h.logOps {
 			h.logNATSOperation(data, "Server", "<-")

tcp/proxy.go

@@ -108,13 +108,13 @@ func (p *TCPProxy) handleConnection(clientConn net.Conn) {
 	// Client to Server
 	go func() {
 		defer wg.Done()
-		p.pipe(clientConn, serverConn, p.handler.HandleClientData(), connInfo)
+		p.pipe(clientConn, serverConn, p.handler.ClientDataHandler(), connInfo)
 	}()
 
 	// Server to Client
 	go func() {
 		defer wg.Done()
-		p.pipe(serverConn, clientConn, p.handler.HandleServerData(), connInfo)
+		p.pipe(serverConn, clientConn, p.handler.ServerDataHandler(), connInfo)
 	}()
 
 	wg.Wait()
@@ -162,5 +162,3 @@ func (p *TCPProxy) Stop() error {
 	p.wg.Wait()
 	return nil
 }
-
-type handleFunc func(context.Context, []byte, mgate.ConnectionInfo) ([]byte, bool, error)