mirror of https://github.com/k3s-io/k3s
238 lines
9.8 KiB
Go
238 lines
9.8 KiB
Go
// +build !providerless
|
|
|
|
/*
|
|
Copyright 2017 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package gce
|
|
|
|
import (
|
|
computealpha "google.golang.org/api/compute/v0.alpha"
|
|
computebeta "google.golang.org/api/compute/v0.beta"
|
|
compute "google.golang.org/api/compute/v1"
|
|
|
|
"github.com/GoogleCloudPlatform/k8s-cloud-provider/pkg/cloud"
|
|
"github.com/GoogleCloudPlatform/k8s-cloud-provider/pkg/cloud/filter"
|
|
"github.com/GoogleCloudPlatform/k8s-cloud-provider/pkg/cloud/meta"
|
|
)
|
|
|
|
func newBackendServiceMetricContext(request, region string) *metricContext {
|
|
return newBackendServiceMetricContextWithVersion(request, region, computeV1Version)
|
|
}
|
|
|
|
func newBackendServiceMetricContextWithVersion(request, region, version string) *metricContext {
|
|
return newGenericMetricContext("backendservice", request, region, unusedMetricLabel, version)
|
|
}
|
|
|
|
// GetGlobalBackendService retrieves a backend by name.
|
|
func (g *Cloud) GetGlobalBackendService(name string) (*compute.BackendService, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("get", "")
|
|
v, err := g.c.BackendServices().Get(ctx, meta.GlobalKey(name))
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// GetBetaGlobalBackendService retrieves beta backend by name.
|
|
func (g *Cloud) GetBetaGlobalBackendService(name string) (*computebeta.BackendService, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("get", "", computeBetaVersion)
|
|
v, err := g.c.BetaBackendServices().Get(ctx, meta.GlobalKey(name))
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// GetAlphaGlobalBackendService retrieves alpha backend by name.
|
|
func (g *Cloud) GetAlphaGlobalBackendService(name string) (*computealpha.BackendService, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("get", "", computeAlphaVersion)
|
|
v, err := g.c.AlphaBackendServices().Get(ctx, meta.GlobalKey(name))
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// UpdateGlobalBackendService applies the given BackendService as an update to
|
|
// an existing service.
|
|
func (g *Cloud) UpdateGlobalBackendService(bg *compute.BackendService) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("update", "")
|
|
return mc.Observe(g.c.BackendServices().Update(ctx, meta.GlobalKey(bg.Name), bg))
|
|
}
|
|
|
|
// UpdateBetaGlobalBackendService applies the given beta BackendService as an
|
|
// update to an existing service.
|
|
func (g *Cloud) UpdateBetaGlobalBackendService(bg *computebeta.BackendService) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("update", "", computeBetaVersion)
|
|
return mc.Observe(g.c.BetaBackendServices().Update(ctx, meta.GlobalKey(bg.Name), bg))
|
|
}
|
|
|
|
// UpdateAlphaGlobalBackendService applies the given alpha BackendService as an
|
|
// update to an existing service.
|
|
func (g *Cloud) UpdateAlphaGlobalBackendService(bg *computealpha.BackendService) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("update", "", computeAlphaVersion)
|
|
return mc.Observe(g.c.AlphaBackendServices().Update(ctx, meta.GlobalKey(bg.Name), bg))
|
|
}
|
|
|
|
// DeleteGlobalBackendService deletes the given BackendService by name.
|
|
func (g *Cloud) DeleteGlobalBackendService(name string) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("delete", "")
|
|
return mc.Observe(g.c.BackendServices().Delete(ctx, meta.GlobalKey(name)))
|
|
}
|
|
|
|
// CreateGlobalBackendService creates the given BackendService.
|
|
func (g *Cloud) CreateGlobalBackendService(bg *compute.BackendService) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("create", "")
|
|
return mc.Observe(g.c.BackendServices().Insert(ctx, meta.GlobalKey(bg.Name), bg))
|
|
}
|
|
|
|
// CreateBetaGlobalBackendService creates the given beta BackendService.
|
|
func (g *Cloud) CreateBetaGlobalBackendService(bg *computebeta.BackendService) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("create", "", computeBetaVersion)
|
|
return mc.Observe(g.c.BetaBackendServices().Insert(ctx, meta.GlobalKey(bg.Name), bg))
|
|
}
|
|
|
|
// CreateAlphaGlobalBackendService creates the given alpha BackendService.
|
|
func (g *Cloud) CreateAlphaGlobalBackendService(bg *computealpha.BackendService) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("create", "", computeAlphaVersion)
|
|
return mc.Observe(g.c.AlphaBackendServices().Insert(ctx, meta.GlobalKey(bg.Name), bg))
|
|
}
|
|
|
|
// ListGlobalBackendServices lists all backend services in the project.
|
|
func (g *Cloud) ListGlobalBackendServices() ([]*compute.BackendService, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("list", "")
|
|
v, err := g.c.BackendServices().List(ctx, filter.None)
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// GetGlobalBackendServiceHealth returns the health of the BackendService
|
|
// identified by the given name, in the given instanceGroup. The
|
|
// instanceGroupLink is the fully qualified self link of an instance group.
|
|
func (g *Cloud) GetGlobalBackendServiceHealth(name string, instanceGroupLink string) (*compute.BackendServiceGroupHealth, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("get_health", "")
|
|
groupRef := &compute.ResourceGroupReference{Group: instanceGroupLink}
|
|
v, err := g.c.BackendServices().GetHealth(ctx, meta.GlobalKey(name), groupRef)
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// GetRegionBackendService retrieves a backend by name.
|
|
func (g *Cloud) GetRegionBackendService(name, region string) (*compute.BackendService, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("get", region)
|
|
v, err := g.c.RegionBackendServices().Get(ctx, meta.RegionalKey(name, region))
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// UpdateRegionBackendService applies the given BackendService as an update to
|
|
// an existing service.
|
|
func (g *Cloud) UpdateRegionBackendService(bg *compute.BackendService, region string) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("update", region)
|
|
return mc.Observe(g.c.RegionBackendServices().Update(ctx, meta.RegionalKey(bg.Name, region), bg))
|
|
}
|
|
|
|
// DeleteRegionBackendService deletes the given BackendService by name.
|
|
func (g *Cloud) DeleteRegionBackendService(name, region string) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("delete", region)
|
|
return mc.Observe(g.c.RegionBackendServices().Delete(ctx, meta.RegionalKey(name, region)))
|
|
}
|
|
|
|
// CreateRegionBackendService creates the given BackendService.
|
|
func (g *Cloud) CreateRegionBackendService(bg *compute.BackendService, region string) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("create", region)
|
|
return mc.Observe(g.c.RegionBackendServices().Insert(ctx, meta.RegionalKey(bg.Name, region), bg))
|
|
}
|
|
|
|
// ListRegionBackendServices lists all backend services in the project.
|
|
func (g *Cloud) ListRegionBackendServices(region string) ([]*compute.BackendService, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("list", region)
|
|
v, err := g.c.RegionBackendServices().List(ctx, region, filter.None)
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// GetRegionalBackendServiceHealth returns the health of the BackendService
|
|
// identified by the given name, in the given instanceGroup. The
|
|
// instanceGroupLink is the fully qualified self link of an instance group.
|
|
func (g *Cloud) GetRegionalBackendServiceHealth(name, region string, instanceGroupLink string) (*compute.BackendServiceGroupHealth, error) {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContext("get_health", region)
|
|
ref := &compute.ResourceGroupReference{Group: instanceGroupLink}
|
|
v, err := g.c.RegionBackendServices().GetHealth(ctx, meta.RegionalKey(name, region), ref)
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// SetSecurityPolicyForBetaGlobalBackendService sets the given
|
|
// SecurityPolicyReference for the BackendService identified by the given name.
|
|
func (g *Cloud) SetSecurityPolicyForBetaGlobalBackendService(backendServiceName string, securityPolicyReference *computebeta.SecurityPolicyReference) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("set_security_policy", "", computeBetaVersion)
|
|
return mc.Observe(g.c.BetaBackendServices().SetSecurityPolicy(ctx, meta.GlobalKey(backendServiceName), securityPolicyReference))
|
|
}
|
|
|
|
// SetSecurityPolicyForAlphaGlobalBackendService sets the given
|
|
// SecurityPolicyReference for the BackendService identified by the given name.
|
|
func (g *Cloud) SetSecurityPolicyForAlphaGlobalBackendService(backendServiceName string, securityPolicyReference *computealpha.SecurityPolicyReference) error {
|
|
ctx, cancel := cloud.ContextWithCallTimeout()
|
|
defer cancel()
|
|
|
|
mc := newBackendServiceMetricContextWithVersion("set_security_policy", "", computeAlphaVersion)
|
|
return mc.Observe(g.c.AlphaBackendServices().SetSecurityPolicy(ctx, meta.GlobalKey(backendServiceName), securityPolicyReference))
|
|
}
|