mirror of https://github.com/k3s-io/k3s
117 lines
3.8 KiB
Go
117 lines
3.8 KiB
Go
/*
|
|
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 (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/golang/glog"
|
|
compute "google.golang.org/api/compute/v1"
|
|
)
|
|
|
|
func newAddressMetricContext(request, region string) *metricContext {
|
|
return &metricContext{
|
|
start: time.Now(),
|
|
attributes: []string{"address_" + request, region, unusedMetricLabel},
|
|
}
|
|
}
|
|
|
|
// ReserveGlobalAddress creates a global address.
|
|
// Caller is allocated a random IP if they do not specify an ipAddress. If an
|
|
// ipAddress is specified, it must belong to the current project, eg: an
|
|
// ephemeral IP associated with a global forwarding rule.
|
|
func (gce *GCECloud) ReserveGlobalAddress(addr *compute.Address) error {
|
|
mc := newAddressMetricContext("reserve", "")
|
|
op, err := gce.service.GlobalAddresses.Insert(gce.projectID, addr).Do()
|
|
if err != nil {
|
|
return mc.Observe(err)
|
|
}
|
|
return gce.waitForGlobalOp(op, mc)
|
|
}
|
|
|
|
// DeleteGlobalAddress deletes a global address by name.
|
|
func (gce *GCECloud) DeleteGlobalAddress(name string) error {
|
|
mc := newAddressMetricContext("delete", "")
|
|
op, err := gce.service.GlobalAddresses.Delete(gce.projectID, name).Do()
|
|
if err != nil {
|
|
return mc.Observe(err)
|
|
}
|
|
return gce.waitForGlobalOp(op, mc)
|
|
}
|
|
|
|
// GetGlobalAddress returns the global address by name.
|
|
func (gce *GCECloud) GetGlobalAddress(name string) (*compute.Address, error) {
|
|
mc := newAddressMetricContext("get", "")
|
|
v, err := gce.service.GlobalAddresses.Get(gce.projectID, name).Do()
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// ReserveRegionAddress creates a region address
|
|
func (gce *GCECloud) ReserveRegionAddress(addr *compute.Address, region string) error {
|
|
mc := newAddressMetricContext("reserve", region)
|
|
op, err := gce.service.Addresses.Insert(gce.projectID, region, addr).Do()
|
|
if err != nil {
|
|
return mc.Observe(err)
|
|
}
|
|
return gce.waitForRegionOp(op, region, mc)
|
|
}
|
|
|
|
// DeleteRegionAddress deletes a region address by name.
|
|
func (gce *GCECloud) DeleteRegionAddress(name, region string) error {
|
|
mc := newAddressMetricContext("delete", region)
|
|
op, err := gce.service.Addresses.Delete(gce.projectID, region, name).Do()
|
|
if err != nil {
|
|
return mc.Observe(err)
|
|
}
|
|
return gce.waitForRegionOp(op, region, mc)
|
|
}
|
|
|
|
// GetRegionAddress returns the region address by name
|
|
func (gce *GCECloud) GetRegionAddress(name, region string) (*compute.Address, error) {
|
|
mc := newAddressMetricContext("get", region)
|
|
v, err := gce.service.Addresses.Get(gce.projectID, region, name).Do()
|
|
return v, mc.Observe(err)
|
|
}
|
|
|
|
// GetRegionAddressByIP returns the regional address matching the given IP
|
|
// address.
|
|
func (gce *GCECloud) GetRegionAddressByIP(region, ipAddress string) (*compute.Address, error) {
|
|
mc := newAddressMetricContext("list", region)
|
|
addrs, err := gce.service.Addresses.List(gce.projectID, region).Filter("address eq " + ipAddress).Do()
|
|
// Record the metrics for the call.
|
|
mc.Observe(err)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(addrs.Items) > 1 {
|
|
// We don't expect more than one match.
|
|
addrsToPrint := []compute.Address{}
|
|
for _, addr := range addrs.Items {
|
|
addrsToPrint = append(addrsToPrint, *addr)
|
|
}
|
|
glog.Errorf("More than one addresses matching the IP %q: %+v", ipAddress, addrsToPrint)
|
|
}
|
|
for _, addr := range addrs.Items {
|
|
if addr.Address == ipAddress {
|
|
return addr, nil
|
|
}
|
|
}
|
|
return nil, makeGoogleAPINotFoundError(fmt.Sprintf("Address with IP %q was not found in region %q", ipAddress, region))
|
|
}
|