enable websockets (#4)

Signed-off-by: Peter Jausovec <peter.jausovec@solo.io>
Co-authored-by: Peter Jausovec <peter.jausovec@solo.io>

Author: EItanya

cmd/atenet/internal/app/router/xds.go

@@ -62,6 +62,7 @@ const (
 	IngressHTTPSListener = "ingress_https_listener"
 	RouteName            = "substrate_routes"
 	ClusterName          = "ate-cluster"
+	websocketUpgradeType = "websocket"
 )
 
 // XdsServer implements an aggregated discovery service server for dynamic Envoy router nodes.
@@ -287,6 +288,9 @@ func (x *XdsServer) buildRoutes() *routev3.RouteConfiguration {
 									Cluster: "dynamic_forward_proxy_cluster",
 								},
 								Timeout: durationpb.New(10 * time.Second),
+								UpgradeConfigs: []*routev3.RouteAction_UpgradeConfig{
+									{UpgradeType: websocketUpgradeType},
+								},
 							},
 						},
 					},
@@ -334,6 +338,10 @@ func (x *XdsServer) buildHcm(statPrefix string) *anypb.Any {
 	hcm, _ := anypb.New(&hcmv3.HttpConnectionManager{
 		StatPrefix:        statPrefix,
 		GenerateRequestId: &wrapperspb.BoolValue{Value: true},
+		UpgradeConfigs: []*hcmv3.HttpConnectionManager_UpgradeConfig{
+			{UpgradeType: websocketUpgradeType},
+		},
+		StreamIdleTimeout: durationpb.New(0),
 		AccessLog: []*accesslogv3.AccessLog{
 			{
 				Name: "envoy.access_loggers.stdout",

cmd/atenet/internal/app/router/xds_test.go

@@ -24,6 +24,7 @@ import (
 	clusterv3 "github.com/envoyproxy/go-control-plane/envoy/config/cluster/v3"
 	listenerv3 "github.com/envoyproxy/go-control-plane/envoy/config/listener/v3"
 	routev3 "github.com/envoyproxy/go-control-plane/envoy/config/route/v3"
+	hcmv3 "github.com/envoyproxy/go-control-plane/envoy/extensions/filters/network/http_connection_manager/v3"
 	cachev3 "github.com/envoyproxy/go-control-plane/pkg/cache/v3"
 	resourcev3 "github.com/envoyproxy/go-control-plane/pkg/resource/v3"
 )
@@ -116,6 +117,10 @@ func TestXdsServer_UpdateSnapshot(t *testing.T) {
 		if fallbackRoute.GetMatch().GetPrefix() != "/" {
 			t.Errorf("Expected path mapping prefix '/', got '%s'", fallbackRoute.GetMatch().GetPrefix())
 		}
+		routeAction := fallbackRoute.GetRoute()
+		if len(routeAction.GetUpgradeConfigs()) != 1 || routeAction.GetUpgradeConfigs()[0].GetUpgradeType() != websocketUpgradeType {
+			t.Errorf("Expected route websocket upgrade config, got %+v", routeAction.GetUpgradeConfigs())
+		}
 	}
 
 	// Verify listeners generated
@@ -135,6 +140,15 @@ func TestXdsServer_UpdateSnapshot(t *testing.T) {
 		if sa.GetAddress() != "0.0.0.0" {
 			t.Errorf("Expected address '0.0.0.0', got %s", sa.GetAddress())
 		}
+
+		hcmAny := l.GetFilterChains()[0].GetFilters()[0].GetTypedConfig()
+		hcm := &hcmv3.HttpConnectionManager{}
+		if err := hcmAny.UnmarshalTo(hcm); err != nil {
+			t.Fatalf("Failed to unmarshal HCM: %v", err)
+		}
+		if len(hcm.GetUpgradeConfigs()) != 1 || hcm.GetUpgradeConfigs()[0].GetUpgradeType() != websocketUpgradeType {
+			t.Errorf("Expected HCM websocket upgrade config, got %+v", hcm.GetUpgradeConfigs())
+		}
 	}
 }