mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-26 21:17:23 +00:00
This commit is contained in:
parent
b447dc0c54
commit
83b70c4411
@ -18,6 +18,7 @@ package cloudprovider
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"net"
|
"net"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
||||||
)
|
)
|
||||||
@ -42,8 +43,17 @@ type Clusters interface {
|
|||||||
Master(clusterName string) (string, error)
|
Master(clusterName string) (string, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetLoadBalancerName(clusterName, serviceNamespace, serviceName string) string {
|
// TODO(#6812): Use a shorter name that's less likely to be longer than cloud
|
||||||
return clusterName + "-" + serviceNamespace + "-" + serviceName
|
// providers' name length limits.
|
||||||
|
func GetLoadBalancerName(service *api.Service) string {
|
||||||
|
//GCE requires that the name of a load balancer starts with a lower case letter.
|
||||||
|
ret := "a" + string(service.UID)
|
||||||
|
ret = strings.Replace(ret, "-", "", -1)
|
||||||
|
//AWS requires that the name of a load balancer is shorter than 32 bytes.
|
||||||
|
if len(ret) > 32 {
|
||||||
|
ret = ret[:32]
|
||||||
|
}
|
||||||
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
// TCPLoadBalancer is an abstract, pluggable interface for TCP load balancers.
|
// TCPLoadBalancer is an abstract, pluggable interface for TCP load balancers.
|
||||||
|
@ -438,7 +438,7 @@ func needsUpdate(oldService *api.Service, newService *api.Service) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *ServiceController) loadBalancerName(service *api.Service) string {
|
func (s *ServiceController) loadBalancerName(service *api.Service) string {
|
||||||
return cloudprovider.GetLoadBalancerName(s.clusterName, service.Namespace, service.Name)
|
return cloudprovider.GetLoadBalancerName(service)
|
||||||
}
|
}
|
||||||
|
|
||||||
func getTCPPorts(service *api.Service) ([]int, error) {
|
func getTCPPorts(service *api.Service) ([]int, error) {
|
||||||
@ -571,7 +571,7 @@ func (s *ServiceController) lockedUpdateLoadBalancerHosts(service *api.Service,
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
name := s.loadBalancerName(service)
|
name := cloudprovider.GetLoadBalancerName(service)
|
||||||
err := s.balancer.UpdateTCPLoadBalancer(name, s.zone.Region, hosts)
|
err := s.balancer.UpdateTCPLoadBalancer(name, s.zone.Region, hosts)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return nil
|
return nil
|
||||||
|
@ -27,14 +27,9 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const region = "us-central"
|
const region = "us-central"
|
||||||
const clusterName = "test-cluster"
|
|
||||||
const namespace = "namespace"
|
|
||||||
|
|
||||||
func newService(name string, uid types.UID, external bool) *api.Service {
|
func newService(name string, uid types.UID, external bool) *api.Service {
|
||||||
return &api.Service{ObjectMeta: api.ObjectMeta{Name: name, Namespace: namespace, UID: uid}, Spec: api.ServiceSpec{CreateExternalLoadBalancer: external}}
|
return &api.Service{ObjectMeta: api.ObjectMeta{Name: name, Namespace: "namespace", UID: uid}, Spec: api.ServiceSpec{CreateExternalLoadBalancer: external}}
|
||||||
}
|
|
||||||
func lbName(serviceName string) string {
|
|
||||||
return clusterName + "-" + namespace + "-" + serviceName
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCreateExternalLoadBalancer(t *testing.T) {
|
func TestCreateExternalLoadBalancer(t *testing.T) {
|
||||||
@ -96,7 +91,7 @@ func TestCreateExternalLoadBalancer(t *testing.T) {
|
|||||||
cloud := &fake_cloud.FakeCloud{}
|
cloud := &fake_cloud.FakeCloud{}
|
||||||
cloud.Region = region
|
cloud.Region = region
|
||||||
client := &testclient.Fake{}
|
client := &testclient.Fake{}
|
||||||
controller := New(cloud, client, clusterName)
|
controller := New(cloud, client, "test-cluster")
|
||||||
controller.init()
|
controller.init()
|
||||||
cloud.Calls = nil // ignore any cloud calls made in init()
|
cloud.Calls = nil // ignore any cloud calls made in init()
|
||||||
client.Actions = nil // ignore any client calls made in init()
|
client.Actions = nil // ignore any client calls made in init()
|
||||||
@ -160,7 +155,7 @@ func TestUpdateNodesInExternalLoadBalancer(t *testing.T) {
|
|||||||
newService("s0", "333", true),
|
newService("s0", "333", true),
|
||||||
},
|
},
|
||||||
expectedUpdateCalls: []fake_cloud.FakeUpdateBalancerCall{
|
expectedUpdateCalls: []fake_cloud.FakeUpdateBalancerCall{
|
||||||
{Name: lbName("s0"), Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
{Name: "a333", Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@ -171,9 +166,9 @@ func TestUpdateNodesInExternalLoadBalancer(t *testing.T) {
|
|||||||
newService("s2", "666", true),
|
newService("s2", "666", true),
|
||||||
},
|
},
|
||||||
expectedUpdateCalls: []fake_cloud.FakeUpdateBalancerCall{
|
expectedUpdateCalls: []fake_cloud.FakeUpdateBalancerCall{
|
||||||
{Name: lbName("s0"), Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
{Name: "a444", Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
||||||
{Name: lbName("s1"), Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
{Name: "a555", Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
||||||
{Name: lbName("s2"), Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
{Name: "a666", Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@ -185,8 +180,8 @@ func TestUpdateNodesInExternalLoadBalancer(t *testing.T) {
|
|||||||
newService("s4", "123", false),
|
newService("s4", "123", false),
|
||||||
},
|
},
|
||||||
expectedUpdateCalls: []fake_cloud.FakeUpdateBalancerCall{
|
expectedUpdateCalls: []fake_cloud.FakeUpdateBalancerCall{
|
||||||
{Name: lbName("s1"), Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
{Name: "a888", Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
||||||
{Name: lbName("s3"), Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
{Name: "a999", Region: region, Hosts: []string{"node0", "node1", "node73"}},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@ -195,7 +190,7 @@ func TestUpdateNodesInExternalLoadBalancer(t *testing.T) {
|
|||||||
|
|
||||||
cloud.Region = region
|
cloud.Region = region
|
||||||
client := &testclient.Fake{}
|
client := &testclient.Fake{}
|
||||||
controller := New(cloud, client, clusterName)
|
controller := New(cloud, client, "test-cluster2")
|
||||||
controller.init()
|
controller.init()
|
||||||
cloud.Calls = nil // ignore any cloud calls made in init()
|
cloud.Calls = nil // ignore any cloud calls made in init()
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user