From 0329c12173bec73c8b85f5bc7876f24418d614a8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Mon, 22 Jul 2024 13:04:32 +0200
Subject: [PATCH] Fix relay close message handling

---
 relay/messages/message.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/relay/messages/message.go b/relay/messages/message.go
index 564c82e03..0211e1900 100644
--- a/relay/messages/message.go
+++ b/relay/messages/message.go
@@ -43,6 +43,8 @@ func (m MsgType) String() string {
 		return "transport"
 	case MsgTypeClose:
 		return "close"
+	case MsgTypeHealthCheck:
+		return "health check"
 	default:
 		return "unknown"
 	}
@@ -150,7 +152,7 @@ func UnmarshalHelloResponse(msg []byte) (string, error) {
 func MarshalCloseMsg() []byte {
 	msg := make([]byte, 1)
 	msg[0] = byte(MsgTypeClose)
-	return healthCheckMsg
+	return msg
 }
 
 // Transport message