-
Notifications
You must be signed in to change notification settings - Fork 2.8k
/
annotations.go
74 lines (56 loc) · 2.71 KB
/
annotations.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
// SPDX-License-Identifier: Apache-2.0
// Copyright Authors of Cilium
package ciliumenvoyconfig
import (
"strings"
envoy_config_cluster_v3 "github.com/cilium/proxy/go/envoy/config/cluster/v3"
envoy_config_listener "github.com/cilium/proxy/go/envoy/config/listener/v3"
envoy_config_route_v3 "github.com/cilium/proxy/go/envoy/config/route/v3"
http_connection_manager_v3 "github.com/cilium/proxy/go/envoy/extensions/filters/network/http_connection_manager/v3"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"github.com/cilium/cilium/pkg/annotation"
)
const (
lbEnabledAnnotation = annotation.ServicePrefix + "/lb-l7"
lbModeAnnotation = annotation.ServicePrefix + "/lb-l7-algorithm"
)
type clusterMutator func(*envoy_config_cluster_v3.Cluster) *envoy_config_cluster_v3.Cluster
type httpConnectionManagerMutator func(*http_connection_manager_v3.HttpConnectionManager) *http_connection_manager_v3.HttpConnectionManager
type listenerMutator func(*envoy_config_listener.Listener) *envoy_config_listener.Listener
type routeMutator func(route *envoy_config_route_v3.Route) *envoy_config_route_v3.Route
type routeConfigMutator func(*envoy_config_route_v3.RouteConfiguration) *envoy_config_route_v3.RouteConfiguration
type virtualHostMutator func(*envoy_config_route_v3.VirtualHost) *envoy_config_route_v3.VirtualHost
// isLBProtocolAnnotationEnabled returns true if the load balancer protocol is enabled
func isLBProtocolAnnotationEnabled(obj metav1.Object) bool {
return obj.GetAnnotations()[lbEnabledAnnotation] == "enabled"
}
// getLBProtocolModelAnnotation returns the load balancer mode
func getLBProtocolModelAnnotation(obj metav1.Object) string {
return obj.GetAnnotations()[lbModeAnnotation]
}
//
// ClusterMutator functions
//
// grpcHttpConnectionManagerMutator returns a function that mutates the upgradeConfigs for grpc protocol
func lbModeClusterMutator(obj metav1.Object) clusterMutator {
return func(cluster *envoy_config_cluster_v3.Cluster) *envoy_config_cluster_v3.Cluster {
lbMode := getLBProtocolModelAnnotation(obj)
if lbMode == "" {
return cluster
}
cluster.LbPolicy = envoy_config_cluster_v3.Cluster_LbPolicy(envoy_config_cluster_v3.Cluster_LbPolicy_value[strings.ToUpper(lbMode)])
return cluster
}
}
//
// HTTPConnectionManagerMutator functions
//
// grpcHttpConnectionManagerMutator returns a function that mutates the upgradeConfigs for grpc protocol
func grpcHttpConnectionManagerMutator(_ metav1.Object) httpConnectionManagerMutator {
return func(manager *http_connection_manager_v3.HttpConnectionManager) *http_connection_manager_v3.HttpConnectionManager {
manager.UpgradeConfigs = []*http_connection_manager_v3.HttpConnectionManager_UpgradeConfig{
{UpgradeType: "websocket"},
}
return manager
}
}