update lb to support cluster_id annotation

Author: olegy89

serverscom/loadbalancers.go

@@ -18,6 +18,7 @@ const (
 	loadBalancerHostnameAnnotation      = "servers.com/load-balancer-hostname"
 	loadBalancerLocationIdAnnotation    = "servers.com/load-balancer-location-id"
 	loadBalancerProxyProtocolAnnotation = "servers.com/proxy-protocol"
+	loadBalancerClusterAnnotation       = "servers.com/cluster-id"
 )
 
 type loadBalancers struct {
@@ -67,6 +68,8 @@ func (l *loadBalancers) EnsureLoadBalancer(ctx context.Context, clusterName stri
 		return nil, err
 	}
 
+	lbClusterID := l.extractLBClusterID(service)
+
 	if loadBalancer == nil {
 		locationID, err := l.extractLocationID(service)
 		if err != nil {
@@ -78,6 +81,7 @@ func (l *loadBalancers) EnsureLoadBalancer(ctx context.Context, clusterName stri
 		input.UpstreamZones = upstreamZones
 		input.LocationID = locationID
 		input.Name = l.GetLoadBalancerName(ctx, clusterName, service)
+		input.ClusterID = lbClusterID
 
 		loadBalancer, err = l.client.LoadBalancers.CreateL4LoadBalancer(ctx, input)
 		if err != nil {
@@ -94,6 +98,7 @@ func (l *loadBalancers) EnsureLoadBalancer(ctx context.Context, clusterName stri
 		input.VHostZones = vhostZones
 		input.UpstreamZones = upstreamZones
 		input.Name = &name
+		input.ClusterID = lbClusterID
 
 		loadBalancer, err = l.client.LoadBalancers.UpdateL4LoadBalancer(ctx, loadBalancer.ID, input)
 		if err != nil {
@@ -285,3 +290,13 @@ func (l *loadBalancers) buildResult(service *v1.Service, loadBalancer *cli.L4Loa
 
 	return &v1.LoadBalancerStatus{Ingress: ingresses}
 }
+
+func (l *loadBalancers) extractLBClusterID(service *v1.Service) string {
+	var clusterID string
+
+	if id, ok := service.Annotations[loadBalancerClusterAnnotation]; ok {
+		clusterID = id
+	}
+
+	return clusterID
+}