-
Notifications
You must be signed in to change notification settings - Fork 683
/
transforms.go
204 lines (193 loc) · 7.3 KB
/
transforms.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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
package ambex
import (
"context"
"fmt"
api "github.com/datawire/ambassador/pkg/api/envoy/api/v2"
v2 "github.com/datawire/ambassador/pkg/api/envoy/api/v2"
core "github.com/datawire/ambassador/pkg/api/envoy/api/v2/core"
v2endpoint "github.com/datawire/ambassador/pkg/api/envoy/api/v2/endpoint"
listener "github.com/datawire/ambassador/pkg/api/envoy/api/v2/listener"
http "github.com/datawire/ambassador/pkg/api/envoy/config/filter/network/http_connection_manager/v2"
ctypes "github.com/datawire/ambassador/pkg/envoy-control-plane/cache/types"
"github.com/datawire/ambassador/pkg/envoy-control-plane/resource/v2"
"github.com/datawire/ambassador/pkg/envoy-control-plane/wellknown"
"github.com/datawire/dlib/dlog"
"github.com/golang/protobuf/ptypes"
"google.golang.org/protobuf/proto"
)
// ListenerToRdsListener will take a listener definition and extract any inline RouteConfigurations
// replacing them with a reference to an RDS supplied route configuration. It does not modify the
// supplied listener, any configuration included in the result is copied from the input.
//
// If the input listener does not match the expected form it is simply copied, i.e. it is the
// identity transform for any inputs not matching the expected form.
//
// Example Input (that will get transformed in a non-identity way):
// - a listener configured with an http connection manager
// - that specifies an http router
// - that supplies its RouteConfiguration inline via the route_config field
//
// {
// "name": "...",
// ...,
// "filter_chains": [
// {
// "filter_chain_match": {...},
// "filters": [
// {
// "name": "envoy.filters.network.http_connection_manager",
// "typed_config": {
// "@type": "type.googleapis.com/envoy.config.filter.network.http_connection_manager.v2.HttpConnectionManager",
// "http_filters": [...],
// "route_config": {
// "virtual_hosts": [
// {
// "name": "ambassador-listener-8443-*",
// "domains": ["*"],
// "routes": [...],
// }
// ]
// }
// }
// }
// ]
// }
// ]
// }
//
// Example Output:
// - a duplicate listener that defines the "rds" field instead of the "route_config" field
// - and a list of route configurations
// - with route_config_name supplied in such a way as to correlate the two together
//
// lnr, routes, err := ListenerToRdsListener(...)
//
// lnr = {
// "name": "...",
// ...,
// "filter_chains": [
// {
// "filter_chain_match": {...},
// "filters": [
// {
// "name": "envoy.filters.network.http_connection_manager",
// "typed_config": {
// "@type": "type.googleapis.com/envoy.config.filter.network.http_connection_manager.v2.HttpConnectionManager",
// "http_filters": [...],
// "rds": {
// "config_source": {
// "ads": {}
// },
// "route_config_name": "ambassador-listener-8443-routeconfig-0"
// }
// }
// }
// ]
// }
// ]
// }
//
// routes = [
// {
// "name": "ambassador-listener-8443-routeconfig-0",
// "virtual_hosts": [
// {
// "name": "ambassador-listener-8443-*",
// "domains": ["*"],
// "routes": [...],
// }
// ]
// }
// ]
func ListenerToRdsListener(lnr *api.Listener) (*api.Listener, []*api.RouteConfiguration, error) {
l := proto.Clone(lnr).(*api.Listener)
var routes []*api.RouteConfiguration
for _, fc := range l.FilterChains {
for _, f := range fc.Filters {
if f.Name != wellknown.HTTPConnectionManager {
// We only know how to create an rds listener for HttpConnectionManager
// listeners. We must ignore all other listeners.
continue
}
// Note that the hcm configuration is stored in a protobuf any, so √the
// GetHTTPConnectionManager is actually returning an unmarshalled copy.
hcm := resource.GetHTTPConnectionManager(f)
if hcm != nil {
// RouteSpecifier is a protobuf oneof that corresponds to the rds, route_config, and
// scoped_routes fields. Only one of those may be set at a time.
rs, ok := hcm.RouteSpecifier.(*http.HttpConnectionManager_RouteConfig)
if ok {
rc := rs.RouteConfig
if rc.Name == "" {
// Generate a unique name for the RouteConfiguration that we can use to
// correlate the listener to the RDS record. We use the listener name plus
// an index because there can be more than one route configuration
// associated with a given listener.
rc.Name = fmt.Sprintf("%s-routeconfig-%d", l.Name, len(routes))
}
routes = append(routes, rc)
// Now that we have extracted and named the RouteConfiguration, we change the
// RouteSpecifier from the inline RouteConfig variation to RDS via ADS. This
// will cause it to use whatever ADS source is defined in the bootstrap
// configuration.
hcm.RouteSpecifier = &http.HttpConnectionManager_Rds{
Rds: &http.Rds{
ConfigSource: &core.ConfigSource{
ConfigSourceSpecifier: &core.ConfigSource_Ads{
Ads: &core.AggregatedConfigSource{},
},
},
RouteConfigName: rc.Name,
},
}
}
// Because the hcm is a protobuf any, we need to remarshal it, we can't simply
// expect the above modifications to take effect on our clone of the input. There is
// also a protobuf oneof that includes the deprecated config and typed_config
// fields.
any, err := ptypes.MarshalAny(hcm)
if err != nil {
return nil, nil, err
}
f.ConfigType = &listener.Filter_TypedConfig{TypedConfig: any}
}
}
}
return l, routes, nil
}
// JoinEdsClusters will perform an outer join operation between the eds clusters in the supplied
// clusterlist and the eds endpoint data in the supplied map. It will return a slice of
// ClusterLoadAssignments (cast to []ctypes.Resource) with endpoint data for all the eds clusters in
// the supplied list. If there is no map entry for a given cluster, an empty ClusterLoadAssignment
// will be synthesized. The result is a set of endpoints that are consistent (by the
// go-control-plane's definition of consistent) with the input clusters.
func JoinEdsClusters(ctx context.Context, clusters []ctypes.Resource, edsEndpoints map[string]*v2.ClusterLoadAssignment) (endpoints []ctypes.Resource) {
for _, clu := range clusters {
c := clu.(*v2.Cluster)
// Don't mess with non EDS clusters.
if c.EdsClusterConfig == nil {
continue
}
// By default envoy will use the cluster name to lookup ClusterLoadAssignments unless the
// ServiceName is supplied in the EdsClusterConfig.
ref := c.EdsClusterConfig.ServiceName
if ref == "" {
ref = c.Name
}
var source string
ep, ok := edsEndpoints[ref]
if ok {
source = "found"
} else {
ep = &v2.ClusterLoadAssignment{
ClusterName: ref,
Endpoints: []*v2endpoint.LocalityLbEndpoints{},
}
source = "synthesized"
dlog.Warnf(ctx, "no endpoints for cluster %s: %s", c.Name, ref)
}
dlog.Debugf(ctx, "%s ClusterLoadAssignment for cluster %s: %v", source, c.Name, ep)
endpoints = append(endpoints, ep)
}
return
}