Add UDP in providers with labels

pkg/provider/configuration.go

@@ -46,6 +46,12 @@ func Merge(ctx context.Context, configurations map[string]*dynamic.Configuration
 	routersTCPToDelete := map[string]struct{}{}
 	routersTCP := map[string][]string{}
 
+	servicesUDPToDelete := map[string]struct{}{}
+	servicesUDP := map[string][]string{}
+
+	routersUDPToDelete := map[string]struct{}{}
+	routersUDP := map[string][]string{}
+
 	middlewaresToDelete := map[string]struct{}{}
 	middlewares := map[string][]string{}
 
@@ -85,6 +91,20 @@ func Merge(ctx context.Context, configurations map[string]*dynamic.Configuration
 			}
 		}
 
+		for serviceName, service := range conf.UDP.Services {
+			servicesUDP[serviceName] = append(servicesUDP[serviceName], root)
+			if !AddServiceUDP(configuration.UDP, serviceName, service) {
+				servicesUDPToDelete[serviceName] = struct{}{}
+			}
+		}
+
+		for routerName, router := range conf.UDP.Routers {
+			routersUDP[routerName] = append(routersUDP[routerName], root)
+			if !AddRouterUDP(configuration.UDP, routerName, router) {
+				routersUDPToDelete[routerName] = struct{}{}
+			}
+		}
+
 		for middlewareName, middleware := range conf.HTTP.Middlewares {
 			middlewares[middlewareName] = append(middlewares[middlewareName], root)
 			if !AddMiddleware(configuration.HTTP, middlewareName, middleware) {
@@ -117,6 +137,18 @@ func Merge(ctx context.Context, configurations map[string]*dynamic.Configuration
 		delete(configuration.TCP.Routers, routerName)
 	}
 
+	for serviceName := range servicesUDPToDelete {
+		logger.WithField(log.ServiceName, serviceName).
+			Errorf("UDP service defined multiple times with different configurations in %v", servicesUDP[serviceName])
+		delete(configuration.UDP.Services, serviceName)
+	}
+
+	for routerName := range routersUDPToDelete {
+		logger.WithField(log.RouterName, routerName).
+			Errorf("UDP router defined multiple times with different configurations in %v", routersUDP[routerName])
+		delete(configuration.UDP.Routers, routerName)
+	}
+
 	for middlewareName := range middlewaresToDelete {
 		logger.WithField(log.MiddlewareName, middlewareName).
 			Errorf("Middleware defined multiple times with different configurations in %v", middlewares[middlewareName])
@@ -141,6 +173,21 @@ func AddServiceTCP(configuration *dynamic.TCPConfiguration, serviceName string,
 	return true
 }
 
+// AddServiceUDP adds a service to a configuration.
+func AddServiceUDP(configuration *dynamic.UDPConfiguration, serviceName string, service *dynamic.UDPService) bool {
+	if _, ok := configuration.Services[serviceName]; !ok {
+		configuration.Services[serviceName] = service
+		return true
+	}
+
+	if !configuration.Services[serviceName].LoadBalancer.Mergeable(service.LoadBalancer) {
+		return false
+	}
+
+	configuration.Services[serviceName].LoadBalancer.Servers = append(configuration.Services[serviceName].LoadBalancer.Servers, service.LoadBalancer.Servers...)
+	return true
+}
+
 // AddRouterTCP Adds a router to a configurations.
 func AddRouterTCP(configuration *dynamic.TCPConfiguration, routerName string, router *dynamic.TCPRouter) bool {
 	if _, ok := configuration.Routers[routerName]; !ok {
@@ -151,6 +198,16 @@ func AddRouterTCP(configuration *dynamic.TCPConfiguration, routerName string, ro
 	return reflect.DeepEqual(configuration.Routers[routerName], router)
 }
 
+// AddRouterUDP adds a router to a configuration.
+func AddRouterUDP(configuration *dynamic.UDPConfiguration, routerName string, router *dynamic.UDPRouter) bool {
+	if _, ok := configuration.Routers[routerName]; !ok {
+		configuration.Routers[routerName] = router
+		return true
+	}
+
+	return reflect.DeepEqual(configuration.Routers[routerName], router)
+}
+
 // AddService Adds a service to a configurations.
 func AddService(configuration *dynamic.HTTPConfiguration, serviceName string, service *dynamic.Service) bool {
 	if _, ok := configuration.Services[serviceName]; !ok {
@@ -223,6 +280,28 @@ func BuildTCPRouterConfiguration(ctx context.Context, configuration *dynamic.TCP
 	}
 }
 
+// BuildUDPRouterConfiguration Builds a router configuration.
+func BuildUDPRouterConfiguration(ctx context.Context, configuration *dynamic.UDPConfiguration) {
+	for routerName, router := range configuration.Routers {
+		loggerRouter := log.FromContext(ctx).WithField(log.RouterName, routerName)
+		if len(router.Service) > 0 {
+			continue
+		}
+
+		if len(configuration.Services) > 1 {
+			delete(configuration.Routers, routerName)
+			loggerRouter.
+				Error("Could not define the service name for the router: too many services")
+			continue
+		}
+
+		for serviceName := range configuration.Services {
+			router.Service = serviceName
+			break
+		}
+	}
+}
+
 // BuildRouterConfiguration Builds a router configuration.
 func BuildRouterConfiguration(ctx context.Context, configuration *dynamic.HTTPConfiguration, defaultRouterName string, defaultRuleTpl *template.Template, model interface{}) {
 	if len(configuration.Routers) == 0 {