mirror of https://github.com/status-im/consul.git
778 lines
23 KiB
Go
778 lines
23 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package proxystateconverter
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/hashicorp/consul/proto-public/pbmesh/v2beta1/pbproxystate"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hashicorp/consul/agent/xds/response"
|
|
|
|
"github.com/hashicorp/consul/agent/proxycfg"
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"google.golang.org/protobuf/types/known/durationpb"
|
|
)
|
|
|
|
// routesFromSnapshot returns the xDS API representation of the "routes" in the
|
|
// snapshot.
|
|
func (s *Converter) routesFromSnapshot(cfgSnap *proxycfg.ConfigSnapshot) error {
|
|
if cfgSnap == nil {
|
|
return errors.New("nil config given")
|
|
}
|
|
|
|
switch cfgSnap.Kind {
|
|
case structs.ServiceKindConnectProxy:
|
|
return s.routesForConnectProxy(cfgSnap)
|
|
// TODO(proxystate): Ingress Gateways will be added in the future.
|
|
//case structs.ServiceKindIngressGateway:
|
|
// return s.routesForIngressGateway(cfgSnap)
|
|
// TODO(proxystate): API Gateways will be added in the future.
|
|
//case structs.ServiceKindAPIGateway:
|
|
// return s.routesForAPIGateway(cfgSnap)
|
|
// TODO(proxystate): Terminating Gateways will be added in the future.
|
|
//case structs.ServiceKindTerminatingGateway:
|
|
// return s.routesForTerminatingGateway(cfgSnap)
|
|
// TODO(proxystate): Mesh Gateways will be added in the future.
|
|
//case structs.ServiceKindMeshGateway:
|
|
// return s.routesForMeshGateway(cfgSnap)
|
|
default:
|
|
return fmt.Errorf("Invalid service kind: %v", cfgSnap.Kind)
|
|
}
|
|
}
|
|
|
|
// routesFromSnapshotConnectProxy returns the xDS API representation of the
|
|
// "routes" in the snapshot.
|
|
func (s *Converter) routesForConnectProxy(cfgSnap *proxycfg.ConfigSnapshot) error {
|
|
for uid, chain := range cfgSnap.ConnectProxy.DiscoveryChain {
|
|
if chain.Default {
|
|
continue
|
|
}
|
|
|
|
virtualHost, err := s.makeUpstreamHostForDiscoveryChain(cfgSnap, uid, chain, []string{"*"}, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if virtualHost == nil {
|
|
continue
|
|
}
|
|
|
|
route := &pbproxystate.Route{
|
|
VirtualHosts: []*pbproxystate.VirtualHost{virtualHost},
|
|
}
|
|
s.proxyState.Routes[uid.EnvoyID()] = route
|
|
}
|
|
addressesMap := make(map[string]map[string]string)
|
|
err := cfgSnap.ConnectProxy.DestinationsUpstream.ForEachKeyE(func(uid proxycfg.UpstreamID) error {
|
|
svcConfig, ok := cfgSnap.ConnectProxy.DestinationsUpstream.Get(uid)
|
|
if !ok || svcConfig == nil {
|
|
return nil
|
|
}
|
|
if !structs.IsProtocolHTTPLike(svcConfig.Protocol) {
|
|
// Routes can only be defined for HTTP services
|
|
return nil
|
|
}
|
|
|
|
for _, address := range svcConfig.Destination.Addresses {
|
|
|
|
routeName := clusterNameForDestination(cfgSnap, "~http", fmt.Sprintf("%d", svcConfig.Destination.Port), svcConfig.NamespaceOrDefault(), svcConfig.PartitionOrDefault())
|
|
if _, ok := addressesMap[routeName]; !ok {
|
|
addressesMap[routeName] = make(map[string]string)
|
|
}
|
|
// cluster name is unique per address/port so we should not be doing any override here
|
|
clusterName := clusterNameForDestination(cfgSnap, svcConfig.Name, address, svcConfig.NamespaceOrDefault(), svcConfig.PartitionOrDefault())
|
|
addressesMap[routeName][clusterName] = address
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for routeName, clusters := range addressesMap {
|
|
route, err := s.makeRouteForAddresses(clusters)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if route != nil {
|
|
s.proxyState.Routes[routeName] = route
|
|
}
|
|
}
|
|
|
|
// TODO(rb): make sure we don't generate an empty result
|
|
return nil
|
|
}
|
|
|
|
func (s *Converter) makeRouteForAddresses(addresses map[string]string) (*pbproxystate.Route, error) {
|
|
route, err := makeAddressesRoute(addresses)
|
|
if err != nil {
|
|
s.Logger.Error("failed to make route", "cluster", "error", err)
|
|
return nil, err
|
|
}
|
|
|
|
return route, nil
|
|
}
|
|
|
|
// TODO(proxystate): Terminating Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func routesForTerminatingGateway
|
|
|
|
// TODO(proxystate): Mesh Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func routesForMeshGateway
|
|
|
|
func makeAddressesRoute(addresses map[string]string) (*pbproxystate.Route, error) {
|
|
route := &pbproxystate.Route{}
|
|
for clusterName, address := range addresses {
|
|
destination := makeRouteDestinationFromName(clusterName)
|
|
virtualHost := &pbproxystate.VirtualHost{
|
|
Name: clusterName,
|
|
Domains: []string{address},
|
|
RouteRules: []*pbproxystate.RouteRule{
|
|
{
|
|
Match: makeDefaultRouteMatch(),
|
|
Destination: destination,
|
|
},
|
|
},
|
|
}
|
|
route.VirtualHosts = append(route.VirtualHosts, virtualHost)
|
|
}
|
|
|
|
// sort virtual hosts to have a stable order
|
|
sort.SliceStable(route.VirtualHosts, func(i, j int) bool {
|
|
return route.VirtualHosts[i].Name > route.VirtualHosts[j].Name
|
|
})
|
|
return route, nil
|
|
}
|
|
|
|
// makeRouteDestinationFromName (fka makeRouteActionFromName)
|
|
func makeRouteDestinationFromName(clusterName string) *pbproxystate.RouteDestination {
|
|
return &pbproxystate.RouteDestination{
|
|
Destination: &pbproxystate.RouteDestination_Cluster{
|
|
Cluster: &pbproxystate.DestinationCluster{
|
|
Name: clusterName,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// TODO(proxystate): Ingress Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func routesForIngressGateway
|
|
|
|
// TODO(proxystate): API Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func routesForAPIGateway
|
|
// func buildHTTPRouteUpstream
|
|
|
|
// TODO(proxystate): Ingress Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func findIngressServiceMatchingUpstream
|
|
|
|
// TODO(proxystate): Ingress Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func generateUpstreamIngressDomains
|
|
|
|
// TODO(proxystate): Ingress Gateways will be added in the future.
|
|
// Functions to add from agent/xds/clusters.go:
|
|
// func generateUpstreamAPIsDomains
|
|
|
|
func (s *Converter) makeUpstreamHostForDiscoveryChain(
|
|
cfgSnap *proxycfg.ConfigSnapshot,
|
|
uid proxycfg.UpstreamID,
|
|
chain *structs.CompiledDiscoveryChain,
|
|
serviceDomains []string,
|
|
forMeshGateway bool,
|
|
) (*pbproxystate.VirtualHost, error) {
|
|
var routeRules []*pbproxystate.RouteRule
|
|
|
|
startNode := chain.Nodes[chain.StartNode]
|
|
if startNode == nil {
|
|
return nil, fmt.Errorf("missing first node in compiled discovery chain for: %s", chain.ServiceName)
|
|
}
|
|
|
|
upstreamsSnapshot, err := cfgSnap.ToConfigSnapshotUpstreams()
|
|
if err != nil && !forMeshGateway {
|
|
return nil, err
|
|
}
|
|
|
|
switch startNode.Type {
|
|
case structs.DiscoveryGraphNodeTypeRouter:
|
|
routeRules = make([]*pbproxystate.RouteRule, 0, len(startNode.Routes))
|
|
|
|
for _, discoveryRoute := range startNode.Routes {
|
|
routeMatch := makeRouteMatchForDiscoveryRoute(discoveryRoute)
|
|
|
|
var (
|
|
routeDestination *pbproxystate.RouteDestination
|
|
err error
|
|
)
|
|
|
|
nextNode := chain.Nodes[discoveryRoute.NextNode]
|
|
|
|
var lb *structs.LoadBalancer
|
|
if nextNode.LoadBalancer != nil {
|
|
lb = nextNode.LoadBalancer
|
|
}
|
|
|
|
switch nextNode.Type {
|
|
case structs.DiscoveryGraphNodeTypeSplitter:
|
|
routeDestination, err = s.makeRouteDestinationForSplitter(upstreamsSnapshot, nextNode.Splits, chain, forMeshGateway)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
case structs.DiscoveryGraphNodeTypeResolver:
|
|
rd, ok := s.makeRouteDestinationForChainCluster(upstreamsSnapshot, nextNode.Resolver.Target, chain, forMeshGateway)
|
|
if !ok {
|
|
continue
|
|
}
|
|
routeDestination = rd
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unexpected graph node after route %q", nextNode.Type)
|
|
}
|
|
|
|
routeDestination.DestinationConfiguration = &pbproxystate.DestinationConfiguration{}
|
|
if err := injectLBToDestinationConfiguration(lb, routeDestination.DestinationConfiguration); err != nil {
|
|
return nil, fmt.Errorf("failed to apply load balancer configuration to route action: %v", err)
|
|
}
|
|
|
|
// TODO(rb): Better help handle the envoy case where you need (prefix=/foo/,rewrite=/) and (exact=/foo,rewrite=/) to do a full rewrite
|
|
|
|
destination := discoveryRoute.Definition.Destination
|
|
|
|
routeRule := &pbproxystate.RouteRule{}
|
|
|
|
if destination != nil {
|
|
configHandle := routeDestination.DestinationConfiguration
|
|
if destination.PrefixRewrite != "" {
|
|
configHandle.PrefixRewrite = destination.PrefixRewrite
|
|
}
|
|
|
|
if destination.RequestTimeout > 0 || destination.IdleTimeout > 0 {
|
|
configHandle.TimeoutConfig = &pbproxystate.TimeoutConfig{}
|
|
}
|
|
if destination.RequestTimeout > 0 {
|
|
configHandle.TimeoutConfig.Timeout = durationpb.New(destination.RequestTimeout)
|
|
}
|
|
if destination.IdleTimeout > 0 {
|
|
configHandle.TimeoutConfig.IdleTimeout = durationpb.New(destination.IdleTimeout)
|
|
}
|
|
|
|
if destination.HasRetryFeatures() {
|
|
configHandle.RetryPolicy = getRetryPolicyForDestination(destination)
|
|
}
|
|
|
|
if err := injectHeaderManipToRoute(destination, routeRule); err != nil {
|
|
return nil, fmt.Errorf("failed to apply header manipulation configuration to route: %v", err)
|
|
}
|
|
}
|
|
|
|
routeRule.Match = routeMatch
|
|
routeRule.Destination = routeDestination
|
|
|
|
routeRules = append(routeRules, routeRule)
|
|
}
|
|
|
|
case structs.DiscoveryGraphNodeTypeSplitter:
|
|
routeDestination, err := s.makeRouteDestinationForSplitter(upstreamsSnapshot, startNode.Splits, chain, forMeshGateway)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var lb *structs.LoadBalancer
|
|
if startNode.LoadBalancer != nil {
|
|
lb = startNode.LoadBalancer
|
|
}
|
|
routeDestination.DestinationConfiguration = &pbproxystate.DestinationConfiguration{}
|
|
if err := injectLBToDestinationConfiguration(lb, routeDestination.DestinationConfiguration); err != nil {
|
|
return nil, fmt.Errorf("failed to apply load balancer configuration to route action: %v", err)
|
|
}
|
|
|
|
defaultRoute := &pbproxystate.RouteRule{
|
|
Match: makeDefaultRouteMatch(),
|
|
Destination: routeDestination,
|
|
}
|
|
|
|
routeRules = []*pbproxystate.RouteRule{defaultRoute}
|
|
|
|
case structs.DiscoveryGraphNodeTypeResolver:
|
|
routeDestination, ok := s.makeRouteDestinationForChainCluster(upstreamsSnapshot, startNode.Resolver.Target, chain, forMeshGateway)
|
|
if !ok {
|
|
break
|
|
}
|
|
var lb *structs.LoadBalancer
|
|
if startNode.LoadBalancer != nil {
|
|
lb = startNode.LoadBalancer
|
|
}
|
|
routeDestination.DestinationConfiguration = &pbproxystate.DestinationConfiguration{}
|
|
if err := injectLBToDestinationConfiguration(lb, routeDestination.DestinationConfiguration); err != nil {
|
|
return nil, fmt.Errorf("failed to apply load balancer configuration to route action: %v", err)
|
|
}
|
|
|
|
if startNode.Resolver.RequestTimeout > 0 {
|
|
to := &pbproxystate.TimeoutConfig{
|
|
Timeout: durationpb.New(startNode.Resolver.RequestTimeout),
|
|
}
|
|
routeDestination.DestinationConfiguration.TimeoutConfig = to
|
|
}
|
|
defaultRoute := &pbproxystate.RouteRule{
|
|
Match: makeDefaultRouteMatch(),
|
|
Destination: routeDestination,
|
|
}
|
|
|
|
routeRules = []*pbproxystate.RouteRule{defaultRoute}
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unknown first node in discovery chain of type: %s", startNode.Type)
|
|
}
|
|
|
|
host := &pbproxystate.VirtualHost{
|
|
Name: uid.EnvoyID(),
|
|
Domains: serviceDomains,
|
|
RouteRules: routeRules,
|
|
}
|
|
|
|
return host, nil
|
|
}
|
|
|
|
func getRetryPolicyForDestination(destination *structs.ServiceRouteDestination) *pbproxystate.RetryPolicy {
|
|
retryPolicy := &pbproxystate.RetryPolicy{}
|
|
if destination.NumRetries > 0 {
|
|
retryPolicy.NumRetries = response.MakeUint32Value(int(destination.NumRetries))
|
|
}
|
|
|
|
// The RetryOn magic values come from: https://www.envoyproxy.io/docs/envoy/v1.10.0/configuration/http_filters/router_filter#config-http-filters-router-x-envoy-retry-on
|
|
var retryStrings []string
|
|
|
|
if len(destination.RetryOn) > 0 {
|
|
retryStrings = append(retryStrings, destination.RetryOn...)
|
|
}
|
|
|
|
if destination.RetryOnConnectFailure {
|
|
// connect-failure can be enabled by either adding connect-failure to the RetryOn list or by using the legacy RetryOnConnectFailure option
|
|
// Check that it's not already in the RetryOn list, so we don't set it twice
|
|
connectFailureExists := false
|
|
for _, r := range retryStrings {
|
|
if r == "connect-failure" {
|
|
connectFailureExists = true
|
|
}
|
|
}
|
|
if !connectFailureExists {
|
|
retryStrings = append(retryStrings, "connect-failure")
|
|
}
|
|
}
|
|
|
|
if len(destination.RetryOnStatusCodes) > 0 {
|
|
retryStrings = append(retryStrings, "retriable-status-codes")
|
|
retryPolicy.RetriableStatusCodes = destination.RetryOnStatusCodes
|
|
}
|
|
|
|
retryPolicy.RetryOn = strings.Join(retryStrings, ",")
|
|
|
|
return retryPolicy
|
|
}
|
|
|
|
func makeRouteMatchForDiscoveryRoute(discoveryRoute *structs.DiscoveryRoute) *pbproxystate.RouteMatch {
|
|
match := discoveryRoute.Definition.Match
|
|
if match == nil || match.IsEmpty() {
|
|
return makeDefaultRouteMatch()
|
|
}
|
|
|
|
routeMatch := &pbproxystate.RouteMatch{}
|
|
|
|
switch {
|
|
case match.HTTP.PathExact != "":
|
|
routeMatch.PathMatch = &pbproxystate.PathMatch{
|
|
PathMatch: &pbproxystate.PathMatch_Exact{
|
|
Exact: match.HTTP.PathExact,
|
|
},
|
|
}
|
|
case match.HTTP.PathPrefix != "":
|
|
routeMatch.PathMatch = &pbproxystate.PathMatch{
|
|
PathMatch: &pbproxystate.PathMatch_Prefix{
|
|
Prefix: match.HTTP.PathPrefix,
|
|
},
|
|
}
|
|
case match.HTTP.PathRegex != "":
|
|
routeMatch.PathMatch = &pbproxystate.PathMatch{
|
|
PathMatch: &pbproxystate.PathMatch_Regex{
|
|
Regex: match.HTTP.PathRegex,
|
|
},
|
|
}
|
|
default:
|
|
routeMatch.PathMatch = &pbproxystate.PathMatch{
|
|
PathMatch: &pbproxystate.PathMatch_Prefix{
|
|
Prefix: "/",
|
|
},
|
|
}
|
|
}
|
|
|
|
if len(match.HTTP.Header) > 0 {
|
|
routeMatch.HeaderMatches = make([]*pbproxystate.HeaderMatch, 0, len(match.HTTP.Header))
|
|
for _, hdr := range match.HTTP.Header {
|
|
headerMatch := &pbproxystate.HeaderMatch{
|
|
Name: hdr.Name,
|
|
}
|
|
|
|
switch {
|
|
case hdr.Exact != "":
|
|
headerMatch.Match = &pbproxystate.HeaderMatch_Exact{
|
|
Exact: hdr.Exact,
|
|
}
|
|
case hdr.Regex != "":
|
|
headerMatch.Match = &pbproxystate.HeaderMatch_Regex{
|
|
Regex: hdr.Regex,
|
|
}
|
|
case hdr.Prefix != "":
|
|
headerMatch.Match = &pbproxystate.HeaderMatch_Prefix{
|
|
Prefix: hdr.Prefix,
|
|
}
|
|
case hdr.Suffix != "":
|
|
headerMatch.Match = &pbproxystate.HeaderMatch_Suffix{
|
|
Suffix: hdr.Suffix,
|
|
}
|
|
case hdr.Present:
|
|
headerMatch.Match = &pbproxystate.HeaderMatch_Present{
|
|
Present: true,
|
|
}
|
|
default:
|
|
continue // skip this impossible situation
|
|
}
|
|
|
|
if hdr.Invert {
|
|
headerMatch.InvertMatch = true
|
|
}
|
|
|
|
routeMatch.HeaderMatches = append(routeMatch.HeaderMatches, headerMatch)
|
|
}
|
|
}
|
|
|
|
if len(match.HTTP.Methods) > 0 {
|
|
routeMatch.MethodMatches = append(routeMatch.MethodMatches, match.HTTP.Methods...)
|
|
}
|
|
|
|
if len(match.HTTP.QueryParam) > 0 {
|
|
routeMatch.QueryParameterMatches = make([]*pbproxystate.QueryParameterMatch, 0, len(match.HTTP.QueryParam))
|
|
for _, qm := range match.HTTP.QueryParam {
|
|
|
|
queryMatcher := &pbproxystate.QueryParameterMatch{
|
|
Name: qm.Name,
|
|
}
|
|
|
|
switch {
|
|
case qm.Exact != "":
|
|
queryMatcher.Match = &pbproxystate.QueryParameterMatch_Exact{
|
|
Exact: qm.Exact,
|
|
}
|
|
case qm.Regex != "":
|
|
queryMatcher.Match = &pbproxystate.QueryParameterMatch_Regex{
|
|
Regex: qm.Regex,
|
|
}
|
|
case qm.Present:
|
|
queryMatcher.Match = &pbproxystate.QueryParameterMatch_Present{
|
|
Present: true,
|
|
}
|
|
default:
|
|
continue // skip this impossible situation
|
|
}
|
|
|
|
routeMatch.QueryParameterMatches = append(routeMatch.QueryParameterMatches, queryMatcher)
|
|
}
|
|
}
|
|
|
|
return routeMatch
|
|
}
|
|
|
|
func makeDefaultRouteMatch() *pbproxystate.RouteMatch {
|
|
return &pbproxystate.RouteMatch{
|
|
PathMatch: &pbproxystate.PathMatch{
|
|
PathMatch: &pbproxystate.PathMatch_Prefix{
|
|
Prefix: "/",
|
|
},
|
|
},
|
|
// TODO(banks) Envoy supports matching only valid GRPC
|
|
// requests which might be nice to add here for gRPC services
|
|
// but it's not supported in our current envoy SDK version
|
|
// although docs say it was supported by 1.8.0. Going to defer
|
|
// that until we've updated the deps.
|
|
}
|
|
}
|
|
|
|
func (s *Converter) makeRouteDestinationForChainCluster(
|
|
upstreamsSnapshot *proxycfg.ConfigSnapshotUpstreams,
|
|
targetID string,
|
|
chain *structs.CompiledDiscoveryChain,
|
|
forMeshGateway bool,
|
|
) (*pbproxystate.RouteDestination, bool) {
|
|
clusterName := s.getTargetClusterName(upstreamsSnapshot, chain, targetID, forMeshGateway)
|
|
if clusterName == "" {
|
|
return nil, false
|
|
}
|
|
return makeRouteDestinationFromName(clusterName), true
|
|
}
|
|
|
|
func (s *Converter) makeRouteDestinationForSplitter(
|
|
upstreamsSnapshot *proxycfg.ConfigSnapshotUpstreams,
|
|
splits []*structs.DiscoverySplit,
|
|
chain *structs.CompiledDiscoveryChain,
|
|
forMeshGateway bool,
|
|
) (*pbproxystate.RouteDestination, error) {
|
|
clusters := make([]*pbproxystate.L7WeightedDestinationCluster, 0, len(splits))
|
|
for _, split := range splits {
|
|
nextNode := chain.Nodes[split.NextNode]
|
|
|
|
if nextNode.Type != structs.DiscoveryGraphNodeTypeResolver {
|
|
return nil, fmt.Errorf("unexpected splitter destination node type: %s", nextNode.Type)
|
|
}
|
|
targetID := nextNode.Resolver.Target
|
|
|
|
clusterName := s.getTargetClusterName(upstreamsSnapshot, chain, targetID, forMeshGateway)
|
|
if clusterName == "" {
|
|
continue
|
|
}
|
|
|
|
// The smallest representable weight is 1/10000 or .01% but envoy
|
|
// deals with integers so scale everything up by 100x.
|
|
weight := int(split.Weight * 100)
|
|
|
|
clusterWeight := &pbproxystate.L7WeightedDestinationCluster{
|
|
Name: clusterName,
|
|
Weight: response.MakeUint32Value(weight),
|
|
}
|
|
if err := injectHeaderManipToWeightedCluster(split.Definition, clusterWeight); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
clusters = append(clusters, clusterWeight)
|
|
}
|
|
|
|
if len(clusters) <= 0 {
|
|
return nil, fmt.Errorf("number of clusters in splitter must be > 0; got %d", len(clusters))
|
|
}
|
|
|
|
return &pbproxystate.RouteDestination{
|
|
Destination: &pbproxystate.RouteDestination_WeightedClusters{
|
|
WeightedClusters: &pbproxystate.L7WeightedClusterGroup{
|
|
Clusters: clusters,
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func injectLBToDestinationConfiguration(lb *structs.LoadBalancer, destinationConfig *pbproxystate.DestinationConfiguration) error {
|
|
if lb == nil || !lb.IsHashBased() {
|
|
return nil
|
|
}
|
|
|
|
result := make([]*pbproxystate.LoadBalancerHashPolicy, 0, len(lb.HashPolicies))
|
|
for _, policy := range lb.HashPolicies {
|
|
if policy.SourceIP {
|
|
p := &pbproxystate.LoadBalancerHashPolicy{
|
|
Policy: &pbproxystate.LoadBalancerHashPolicy_ConnectionProperties{
|
|
ConnectionProperties: &pbproxystate.ConnectionPropertiesPolicy{
|
|
SourceIp: true,
|
|
Terminal: policy.Terminal,
|
|
},
|
|
},
|
|
}
|
|
result = append(result, p)
|
|
continue
|
|
}
|
|
|
|
switch policy.Field {
|
|
case structs.HashPolicyHeader:
|
|
p := &pbproxystate.LoadBalancerHashPolicy{
|
|
Policy: &pbproxystate.LoadBalancerHashPolicy_Header{
|
|
Header: &pbproxystate.HeaderPolicy{
|
|
Name: policy.FieldValue,
|
|
Terminal: policy.Terminal,
|
|
},
|
|
},
|
|
}
|
|
result = append(result, p)
|
|
|
|
case structs.HashPolicyCookie:
|
|
|
|
cookie := &pbproxystate.CookiePolicy{
|
|
Name: policy.FieldValue,
|
|
Terminal: policy.Terminal,
|
|
}
|
|
if policy.CookieConfig != nil {
|
|
cookie.Path = policy.CookieConfig.Path
|
|
|
|
if policy.CookieConfig.TTL != 0*time.Second {
|
|
cookie.Ttl = durationpb.New(policy.CookieConfig.TTL)
|
|
}
|
|
|
|
// Envoy will generate a session cookie if the ttl is present and zero.
|
|
if policy.CookieConfig.Session {
|
|
cookie.Ttl = durationpb.New(0 * time.Second)
|
|
}
|
|
}
|
|
p := &pbproxystate.LoadBalancerHashPolicy{
|
|
Policy: &pbproxystate.LoadBalancerHashPolicy_Cookie{
|
|
Cookie: cookie,
|
|
},
|
|
}
|
|
result = append(result, p)
|
|
|
|
case structs.HashPolicyQueryParam:
|
|
p := &pbproxystate.LoadBalancerHashPolicy{
|
|
Policy: &pbproxystate.LoadBalancerHashPolicy_QueryParameter{
|
|
QueryParameter: &pbproxystate.QueryParameterPolicy{
|
|
Name: policy.FieldValue,
|
|
Terminal: policy.Terminal,
|
|
},
|
|
},
|
|
}
|
|
result = append(result, p)
|
|
|
|
default:
|
|
return fmt.Errorf("unsupported load balancer hash policy field: %v", policy.Field)
|
|
}
|
|
}
|
|
|
|
destinationConfig.HashPolicies = result
|
|
return nil
|
|
}
|
|
|
|
func injectHeaderManipToRoute(dest *structs.ServiceRouteDestination, r *pbproxystate.RouteRule) error {
|
|
if !dest.RequestHeaders.IsZero() {
|
|
r.HeaderMutations = append(
|
|
r.HeaderMutations,
|
|
makeRequestHeaderAdd(dest.RequestHeaders.Add, true)...,
|
|
)
|
|
r.HeaderMutations = append(
|
|
r.HeaderMutations,
|
|
makeRequestHeaderAdd(dest.RequestHeaders.Set, false)...,
|
|
)
|
|
r.HeaderMutations = append(
|
|
r.HeaderMutations,
|
|
makeRequestHeaderRemove(dest.RequestHeaders.Remove),
|
|
)
|
|
}
|
|
if !dest.ResponseHeaders.IsZero() {
|
|
r.HeaderMutations = append(
|
|
r.HeaderMutations,
|
|
makeResponseHeaderAdd(dest.ResponseHeaders.Add, true)...,
|
|
)
|
|
r.HeaderMutations = append(
|
|
r.HeaderMutations,
|
|
makeResponseHeaderAdd(dest.ResponseHeaders.Set, false)...,
|
|
)
|
|
r.HeaderMutations = append(
|
|
r.HeaderMutations,
|
|
makeResponseHeaderRemove(dest.ResponseHeaders.Remove),
|
|
)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func injectHeaderManipToWeightedCluster(split *structs.ServiceSplit, c *pbproxystate.L7WeightedDestinationCluster) error {
|
|
if !split.RequestHeaders.IsZero() {
|
|
c.HeaderMutations = append(
|
|
c.HeaderMutations,
|
|
makeRequestHeaderAdd(split.RequestHeaders.Add, true)...,
|
|
)
|
|
c.HeaderMutations = append(
|
|
c.HeaderMutations,
|
|
makeRequestHeaderAdd(split.RequestHeaders.Set, false)...,
|
|
)
|
|
c.HeaderMutations = append(
|
|
c.HeaderMutations,
|
|
makeRequestHeaderRemove(split.RequestHeaders.Remove),
|
|
)
|
|
}
|
|
if !split.ResponseHeaders.IsZero() {
|
|
c.HeaderMutations = append(
|
|
c.HeaderMutations,
|
|
makeResponseHeaderAdd(split.ResponseHeaders.Add, true)...,
|
|
)
|
|
c.HeaderMutations = append(
|
|
c.HeaderMutations,
|
|
makeResponseHeaderAdd(split.ResponseHeaders.Set, false)...,
|
|
)
|
|
c.HeaderMutations = append(
|
|
c.HeaderMutations,
|
|
makeResponseHeaderRemove(split.ResponseHeaders.Remove),
|
|
)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func makeRequestHeaderAdd(vals map[string]string, add bool) []*pbproxystate.HeaderMutation {
|
|
mutations := make([]*pbproxystate.HeaderMutation, 0, len(vals))
|
|
|
|
appendAction := pbproxystate.AppendAction_APPEND_ACTION_OVERWRITE_IF_EXISTS_OR_ADD
|
|
if add {
|
|
appendAction = pbproxystate.AppendAction_APPEND_ACTION_APPEND_IF_EXISTS_OR_ADD
|
|
}
|
|
|
|
for k, v := range vals {
|
|
m := &pbproxystate.HeaderMutation{
|
|
Action: &pbproxystate.HeaderMutation_RequestHeaderAdd{
|
|
RequestHeaderAdd: &pbproxystate.RequestHeaderAdd{
|
|
Header: &pbproxystate.Header{
|
|
Key: k,
|
|
Value: v,
|
|
},
|
|
AppendAction: appendAction,
|
|
},
|
|
},
|
|
}
|
|
mutations = append(mutations, m)
|
|
}
|
|
return mutations
|
|
}
|
|
|
|
func makeRequestHeaderRemove(values []string) *pbproxystate.HeaderMutation {
|
|
return &pbproxystate.HeaderMutation{
|
|
Action: &pbproxystate.HeaderMutation_RequestHeaderRemove{
|
|
RequestHeaderRemove: &pbproxystate.RequestHeaderRemove{
|
|
HeaderKeys: values,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func makeResponseHeaderAdd(vals map[string]string, add bool) []*pbproxystate.HeaderMutation {
|
|
mutations := make([]*pbproxystate.HeaderMutation, 0, len(vals))
|
|
|
|
appendAction := pbproxystate.AppendAction_APPEND_ACTION_OVERWRITE_IF_EXISTS_OR_ADD
|
|
if add {
|
|
appendAction = pbproxystate.AppendAction_APPEND_ACTION_APPEND_IF_EXISTS_OR_ADD
|
|
}
|
|
|
|
for k, v := range vals {
|
|
m := &pbproxystate.HeaderMutation{
|
|
Action: &pbproxystate.HeaderMutation_ResponseHeaderAdd{
|
|
ResponseHeaderAdd: &pbproxystate.ResponseHeaderAdd{
|
|
Header: &pbproxystate.Header{
|
|
Key: k,
|
|
Value: v,
|
|
},
|
|
AppendAction: appendAction,
|
|
},
|
|
},
|
|
}
|
|
mutations = append(mutations, m)
|
|
}
|
|
return mutations
|
|
}
|
|
|
|
func makeResponseHeaderRemove(values []string) *pbproxystate.HeaderMutation {
|
|
return &pbproxystate.HeaderMutation{
|
|
Action: &pbproxystate.HeaderMutation_ResponseHeaderRemove{
|
|
ResponseHeaderRemove: &pbproxystate.ResponseHeaderRemove{
|
|
HeaderKeys: values,
|
|
},
|
|
},
|
|
}
|
|
}
|