From 43139e348a6672499e7491317f73bb74a3cad345 Mon Sep 17 00:00:00 2001
From: Darren Shepherd <darren@rancher.com>
Date: Mon, 10 May 2021 15:06:46 -0700
Subject: [PATCH] Make handshake timeout consistent with rancher agent

---
 pkg/aggregation/server.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pkg/aggregation/server.go b/pkg/aggregation/server.go
index 3f68598..9d39c3d 100644
--- a/pkg/aggregation/server.go
+++ b/pkg/aggregation/server.go
@@ -15,10 +15,14 @@ import (
 	"github.com/sirupsen/logrus"
 )
 
+const (
+	HandshakeTimeOut = 10 * time.Second
+)
+
 func ListenAndServe(ctx context.Context, url string, caCert []byte, token string, handler http.Handler) {
 	dialer := websocket.Dialer{
 		Proxy:            http.ProxyFromEnvironment,
-		HandshakeTimeout: 45 * time.Second,
+		HandshakeTimeout: HandshakeTimeOut,
 	}
 
 	if caCert != nil && len(caCert) == 0 {