mirror of
https://github.com/traefik/traefik
synced 2026-02-03 11:10:33 +00:00
114 lines
3.8 KiB
Go
114 lines
3.8 KiB
Go
package dynamic
|
|
|
|
import (
|
|
"reflect"
|
|
)
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPConfiguration contains all the UDP configuration parameters.
|
|
type UDPConfiguration struct {
|
|
Routers map[string]*UDPRouter `json:"routers,omitempty" toml:"routers,omitempty" yaml:"routers,omitempty" export:"true"`
|
|
Services map[string]*UDPService `json:"services,omitempty" toml:"services,omitempty" yaml:"services,omitempty" export:"true"`
|
|
}
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPService defines the configuration for a UDP service. All fields are mutually exclusive.
|
|
type UDPService struct {
|
|
LoadBalancer *UDPServersLoadBalancer `json:"loadBalancer,omitempty" toml:"loadBalancer,omitempty" yaml:"loadBalancer,omitempty" export:"true"`
|
|
Weighted *UDPWeightedRoundRobin `json:"weighted,omitempty" toml:"weighted,omitempty" yaml:"weighted,omitempty" label:"-" export:"true"`
|
|
}
|
|
|
|
// Merge merges another UDPService into this one.
|
|
// Returns true if the merge succeeds, false if configurations conflict.
|
|
func (s *UDPService) Merge(other *UDPService) bool {
|
|
if s.LoadBalancer == nil || other.LoadBalancer == nil {
|
|
return reflect.DeepEqual(s, other)
|
|
}
|
|
|
|
return s.LoadBalancer.Merge(other.LoadBalancer)
|
|
}
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPWeightedRoundRobin is a weighted round robin UDP load-balancer of services.
|
|
type UDPWeightedRoundRobin struct {
|
|
Services []UDPWRRService `json:"services,omitempty" toml:"services,omitempty" yaml:"services,omitempty" export:"true"`
|
|
}
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPWRRService is a reference to a UDP service load-balanced with weighted round robin.
|
|
type UDPWRRService struct {
|
|
Name string `json:"name,omitempty" toml:"name,omitempty" yaml:"name,omitempty" export:"true"`
|
|
Weight *int `json:"weight,omitempty" toml:"weight,omitempty" yaml:"weight,omitempty" export:"true"`
|
|
}
|
|
|
|
// SetDefaults sets the default values for a UDPWRRService.
|
|
func (w *UDPWRRService) SetDefaults() {
|
|
defaultWeight := 1
|
|
w.Weight = &defaultWeight
|
|
}
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPRouter defines the configuration for an UDP router.
|
|
type UDPRouter struct {
|
|
EntryPoints []string `json:"entryPoints,omitempty" toml:"entryPoints,omitempty" yaml:"entryPoints,omitempty" export:"true"`
|
|
Service string `json:"service,omitempty" toml:"service,omitempty" yaml:"service,omitempty" export:"true"`
|
|
}
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPServersLoadBalancer defines the configuration for a load-balancer of UDP servers.
|
|
type UDPServersLoadBalancer struct {
|
|
Servers []UDPServer `json:"servers,omitempty" toml:"servers,omitempty" yaml:"servers,omitempty" label-slice-as-struct:"server" export:"true"`
|
|
}
|
|
|
|
// Merge merges the other load balancer into this one.
|
|
// Returns true if merge succeeded, false if configurations conflict.
|
|
func (l *UDPServersLoadBalancer) Merge(other *UDPServersLoadBalancer) bool {
|
|
if !l.mergeable(other) {
|
|
return false
|
|
}
|
|
|
|
// Deduplicate and append servers.
|
|
uniq := make(map[string]struct{}, len(l.Servers))
|
|
for _, server := range l.Servers {
|
|
uniq[server.Address] = struct{}{}
|
|
}
|
|
for _, server := range other.Servers {
|
|
if _, ok := uniq[server.Address]; !ok {
|
|
l.Servers = append(l.Servers, server)
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// mergeable reports whether the given load-balancer can be merged with the receiver.
|
|
func (l *UDPServersLoadBalancer) mergeable(loadBalancer *UDPServersLoadBalancer) bool {
|
|
savedServers := l.Servers
|
|
defer func() {
|
|
l.Servers = savedServers
|
|
}()
|
|
l.Servers = nil
|
|
|
|
savedServersLB := loadBalancer.Servers
|
|
defer func() {
|
|
loadBalancer.Servers = savedServersLB
|
|
}()
|
|
loadBalancer.Servers = nil
|
|
|
|
return reflect.DeepEqual(l, loadBalancer)
|
|
}
|
|
|
|
// +k8s:deepcopy-gen=true
|
|
|
|
// UDPServer defines a UDP server configuration.
|
|
type UDPServer struct {
|
|
Address string `json:"address,omitempty" toml:"address,omitempty" yaml:"address,omitempty" label:"-"`
|
|
Port string `json:"-" toml:"-" yaml:"-" file:"-"`
|
|
}
|