/
nodename.go
60 lines (52 loc) · 1.74 KB
/
nodename.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
// SPDX-License-Identifier: Apache-2.0
// Copyright Authors of Cilium
package types
import (
"os"
"github.com/cilium/cilium/pkg/defaults"
k8sConsts "github.com/cilium/cilium/pkg/k8s/constants"
"github.com/cilium/cilium/pkg/logging/logfields"
"github.com/cilium/cilium/pkg/option"
)
var (
nodeName = "localhost"
)
// SetName sets the name of the local node. This will overwrite the value that
// is automatically retrieved with `os.Hostname()`.
//
// Note: This function is currently designed to only be called during the
// bootstrapping procedure of the agent where no parallelism exists. If you
// want to use this function in later stages, a mutex must be added first.
func SetName(name string) {
nodeName = name
}
// GetName returns the name of the local node. The value returned was either
// previously set with SetName(), retrieved via `os.Hostname()`, or as a last
// resort is hardcoded to "localhost".
func GetName() string {
return nodeName
}
// GetAbsoluteNodeName returns the absolute node name combined of both
// (prefixed)cluster name and the local node name in case of
// clustered environments otherwise returns the name of the local node.
func GetAbsoluteNodeName() string {
if option.Config.ClusterName != "" &&
option.Config.ClusterName != defaults.ClusterName {
return option.Config.ClusterName + "/" + nodeName
} else {
return nodeName
}
}
func init() {
// Give priority to the environment variable available in the Cilium agent
if name := os.Getenv(k8sConsts.EnvNodeNameSpec); name != "" {
nodeName = name
return
}
if h, err := os.Hostname(); err != nil {
log.WithError(err).Warn("Unable to retrieve local hostname")
} else {
log.WithField(logfields.NodeName, h).Debug("os.Hostname() returned")
nodeName = h
}
}