From 8c70b7d7ffed473ec79d90a00838d624da40f611 Mon Sep 17 00:00:00 2001
From: Zoltan Papp <zoltan.pmail@gmail.com>
Date: Sun, 9 Jun 2024 12:41:52 +0200
Subject: [PATCH] Replace ws lib on client side

---
 relay/client/client.go                      | 2 +-
 relay/client/dialer/wsnhooyr/client_conn.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/relay/client/client.go b/relay/client/client.go
index aa4fff5d7..0a1066145 100644
--- a/relay/client/client.go
+++ b/relay/client/client.go
@@ -3,6 +3,7 @@ package client
 import (
 	"context"
 	"fmt"
+	ws "github.com/netbirdio/netbird/relay/client/dialer/wsnhooyr"
 	"io"
 	"net"
 	"sync"
@@ -10,7 +11,6 @@ import (
 
 	log "github.com/sirupsen/logrus"
 
-	"github.com/netbirdio/netbird/relay/client/dialer/ws"
 	"github.com/netbirdio/netbird/relay/messages"
 )
 
diff --git a/relay/client/dialer/wsnhooyr/client_conn.go b/relay/client/dialer/wsnhooyr/client_conn.go
index bd3db74eb..8f02b6701 100644
--- a/relay/client/dialer/wsnhooyr/client_conn.go
+++ b/relay/client/dialer/wsnhooyr/client_conn.go
@@ -75,5 +75,5 @@ func (c *Conn) SetDeadline(t time.Time) error {
 }
 
 func (c *Conn) Close() error {
-	return c.Conn.Close(websocket.StatusNormalClosure, "")
+	return c.Conn.CloseNow()
 }