mirror of https://github.com/k3s-io/k3s
107 lines
3.1 KiB
Go
107 lines
3.1 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 (
|
|
"context"
|
|
"fmt"
|
|
|
|
"google.golang.org/api/container/v1"
|
|
"k8s.io/klog/v2"
|
|
)
|
|
|
|
func newClustersMetricContext(request, zone string) *metricContext {
|
|
return newGenericMetricContext("clusters", request, unusedMetricLabel, zone, computeV1Version)
|
|
}
|
|
|
|
// ListClusters will return a list of cluster names for the associated project
|
|
func (g *Cloud) ListClusters(ctx context.Context) ([]string, error) {
|
|
allClusters := []string{}
|
|
|
|
for _, zone := range g.managedZones {
|
|
clusters, err := g.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
|
|
}
|
|
|
|
// GetManagedClusters will return the cluster objects associated to this project
|
|
func (g *Cloud) GetManagedClusters(ctx context.Context) ([]*container.Cluster, error) {
|
|
managedClusters := []*container.Cluster{}
|
|
|
|
if g.regional {
|
|
var err error
|
|
managedClusters, err = g.getClustersInLocation(g.region)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else if len(g.managedZones) >= 1 {
|
|
for _, zone := range g.managedZones {
|
|
clusters, err := g.getClustersInLocation(zone)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
managedClusters = append(managedClusters, clusters...)
|
|
}
|
|
} else {
|
|
return nil, fmt.Errorf("no zones associated with this cluster(%s)", g.ProjectID())
|
|
}
|
|
|
|
return managedClusters, nil
|
|
}
|
|
|
|
// Master returned the dns address of the master
|
|
func (g *Cloud) Master(ctx context.Context, clusterName string) (string, error) {
|
|
return "k8s-" + clusterName + "-master.internal", nil
|
|
}
|
|
|
|
func (g *Cloud) listClustersInZone(zone string) ([]string, error) {
|
|
clusters, err := g.getClustersInLocation(zone)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := []string{}
|
|
for _, cluster := range clusters {
|
|
result = append(result, cluster.Name)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (g *Cloud) getClustersInLocation(zoneOrRegion string) ([]*container.Cluster, error) {
|
|
// TODO: Issue/68913 migrate metric to list_location instead of list_zone.
|
|
mc := newClustersMetricContext("list_zone", zoneOrRegion)
|
|
// TODO: use PageToken to list all not just the first 500
|
|
location := getLocationName(g.projectID, zoneOrRegion)
|
|
list, err := g.containerService.Projects.Locations.Clusters.List(location).Do()
|
|
if err != nil {
|
|
return nil, mc.Observe(err)
|
|
}
|
|
if list.Header.Get("nextPageToken") != "" {
|
|
klog.Errorf("Failed to get all clusters for request, received next page token %s", list.Header.Get("nextPageToken"))
|
|
}
|
|
|
|
return list.Clusters, mc.Observe(nil)
|
|
}
|