Use healthcheck for outgoing websocket connections

Signed-off-by: Vasyl Gello <[email protected]>

src/core/link_ws.go

@@ -32,6 +32,20 @@ type wsServer struct {
 	ctx context.Context
 }
 
+func (c *linkWSConn) CheckHealth(healthUrl string) {
+	for {
+		if c != nil {
+			resp, err := http.Get(healthUrl)
+				if err == nil {
+				resp.Body.Close()
+			}
+			time.Sleep(30 * time.Second)
+		} else {
+			return
+		}
+	}
+}
+
 func (l *linkWSListener) Accept() (net.Conn, error) {
 	qs := <-l.ch
 	if qs == nil {
@@ -94,10 +108,15 @@ func (l *linkWS) dial(ctx context.Context, url *url.URL, info linkInfo, options
 	if err != nil {
 		return nil, err
 	}
+	healthUrl := *url
+	healthUrl.Scheme = "http"
+	healthUrlStr := healthUrl.JoinPath("/h").String()
 	netconn := websocket.NetConn(ctx, wsconn, websocket.MessageBinary)
-	return &linkWSConn{
+	conn := &linkWSConn{
 		Conn: netconn,
-	}, nil
+	}
+	go conn.CheckHealth(healthUrlStr)
+	return conn, nil
 }
 
 func (l *linkWS) listen(ctx context.Context, url *url.URL, _ string) (net.Listener, error) {

src/core/link_wss.go

@@ -4,7 +4,9 @@ import (
 	"context"
 	"fmt"
 	"net"
+	"net/http"
 	"net/url"
+	"time"
 
 	"github.com/Arceliar/phony"
 	"nhooyr.io/websocket"
@@ -19,6 +21,20 @@ type linkWSSConn struct {
 	net.Conn
 }
 
+func (c *linkWSSConn) CheckHealth(healthUrl string) {
+	for {
+		if c != nil {
+			resp, err := http.Get(healthUrl)
+				if err == nil {
+				resp.Body.Close()
+			}
+			time.Sleep(30 * time.Second)
+		} else {
+			return
+		}
+	}
+}
+
 func (l *links) newLinkWSS() *linkWSS {
 	lwss := &linkWSS{
 		links:     l,
@@ -34,10 +50,15 @@ func (l *linkWSS) dial(ctx context.Context, url *url.URL, info linkInfo, options
 	if err != nil {
 		return nil, err
 	}
+	healthUrl := *url
+	healthUrl.Scheme = "https"
+	healthUrlStr := healthUrl.JoinPath("/h").String()
 	netconn := websocket.NetConn(ctx, wsconn, websocket.MessageBinary)
-	return &linkWSSConn{
+	conn := &linkWSSConn{
 		Conn: netconn,
-	}, nil
+	}
+	go conn.CheckHealth(healthUrlStr)
+	return conn, nil
 }
 
 func (l *linkWSS) listen(ctx context.Context, url *url.URL, _ string) (net.Listener, error) {