c0b7c42461
Implemented the call to get all cluster objects in a zone for a project. Also added code to allow the container api to be set in the gce.conf file. Requested fix for @lavalamp. Fixed GetClusters to be GetManagedClusters. Leaving ListClusters as ListClusters as it is part of the Cloud Clusters interface, despite also being a "managed" call. Remove copy pasta :D Fixed method variable name.
83 lines
2.3 KiB
Go
83 lines
2.3 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 (
|
|
"context"
|
|
container "google.golang.org/api/container/v1"
|
|
)
|
|
|
|
func newClustersMetricContext(request, zone string) *metricContext {
|
|
return newGenericMetricContext("clusters", request, unusedMetricLabel, zone, computeV1Version)
|
|
}
|
|
|
|
func (gce *GCECloud) ListClusters(ctx context.Context) ([]string, error) {
|
|
allClusters := []string{}
|
|
|
|
for _, zone := range gce.managedZones {
|
|
clusters, err := gce.listClustersInZone(zone)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// TODO: Scoping? Do we need to qualify the cluster name?
|
|
allClusters = append(allClusters, clusters...)
|
|
}
|
|
|
|
return allClusters, nil
|
|
}
|
|
|
|
func (gce *GCECloud) GetManagedClusters(ctx context.Context) ([]*container.Cluster, error) {
|
|
managedClusters := []*container.Cluster{}
|
|
for _, zone := range gce.managedZones {
|
|
clusters, err := gce.getClustersInZone(zone)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
managedClusters = append(managedClusters, clusters...)
|
|
}
|
|
|
|
return managedClusters, nil
|
|
}
|
|
|
|
func (gce *GCECloud) Master(ctx context.Context, clusterName string) (string, error) {
|
|
return "k8s-" + clusterName + "-master.internal", nil
|
|
}
|
|
|
|
func (gce *GCECloud) listClustersInZone(zone string) ([]string, error) {
|
|
clusters, err := gce.getClustersInZone(zone)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := []string{}
|
|
for _, cluster := range clusters {
|
|
result = append(result, cluster.Name)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (gce *GCECloud) getClustersInZone(zone string) ([]*container.Cluster, error) {
|
|
mc := newClustersMetricContext("list_zone", zone)
|
|
// TODO: use PageToken to list all not just the first 500
|
|
list, err := gce.containerService.Projects.Zones.Clusters.List(gce.projectID, zone).Do()
|
|
if err != nil {
|
|
return nil, mc.Observe(err)
|
|
}
|
|
|
|
return list.Clusters, mc.Observe(nil)
|
|
}
|