-
Notifications
You must be signed in to change notification settings - Fork 98
/
gke.go
70 lines (61 loc) · 2.22 KB
/
gke.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// Copyright 2022 Google LLC
//
// 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
//
// https://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 gcp
import (
"fmt"
"strings"
)
const (
// If the kubernetes.default.svc service exists in the cluster,
// then the KUBERNETES_SERVICE_HOST env var will be populated.
// Use this as an indication that we are running on kubernetes.
k8sServiceHostEnv = "KUBERNETES_SERVICE_HOST"
// See the available GKE metadata:
// https://cloud.google.com/kubernetes-engine/docs/concepts/workload-identity#instance_metadata
clusterNameMetadataAttr = "cluster-name"
clusterLocationMetadataAttr = "cluster-location"
)
func (d *Detector) onGKE() bool {
_, found := d.os.LookupEnv(k8sServiceHostEnv)
return found
}
// GKEHostID returns the instance ID of the instance on which this program is running
func (d *Detector) GKEHostID() (string, error) {
return d.GCEHostID()
}
// GKEClusterName returns the name if the GKE cluster in which this program is running
func (d *Detector) GKEClusterName() (string, error) {
return d.metadata.InstanceAttributeValue(clusterNameMetadataAttr)
}
type LocationType int64
const (
UndefinedLocation LocationType = iota
Zone
Region
)
// GKEAvailabilityZoneOrRegion returns the location of the cluster and whether the cluster is zonal or regional
func (d *Detector) GKEAvailabilityZoneOrRegion() (string, LocationType, error) {
clusterLocation, err := d.metadata.InstanceAttributeValue(clusterLocationMetadataAttr)
if err != nil {
return "", UndefinedLocation, err
}
switch strings.Count(clusterLocation, "-") {
case 1:
return clusterLocation, Region, nil
case 2:
return clusterLocation, Zone, nil
default:
return "", UndefinedLocation, fmt.Errorf("unrecognized format for cluster location: %v", clusterLocation)
}
}