36 Star 412 Fork 76

GVPrancher/rancher

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
.github
app
chart
cleanup
package
pkg
agent
api
audit
auth
catalog
cert
clustermanager
clusterprovisioninglogger
clusterrouter
configfield
controllers
cron
dialer
dynamiclistener
encryptedstore
httpproxy
image
jailer
k8slookup
k8sproxy
kubeconfig
kubectl
librke
logserver
monitoring
namespace
node
node.go
nodeconfig
notifiers
pipeline
project
randomtoken
rbac
ref
resourcelink
resourcequota
rkecerts
rkedialerfactory
rkenodeconfigclient
rkenodeconfigserver
rkeworker
settings
systemaccount
systemtemplate
telemetry
ticker
tls
tunnelserver
rke-templates
scripts
server
tests
vendor
.dockerignore
.drone.yml
.gitignore
.gometalinter.json
CONTRIBUTING.md
Dockerfile.dapper
LICENSE
Makefile
README.md
README_1_6.md
code-of-conduct.md
keybase.md
main.go
vendor.conf
克隆/下载
node.go 3.31 KB
一键复制 编辑 原始数据 按行查看 历史
package node
import (
"github.com/rancher/types/apis/core/v1"
"github.com/rancher/types/apis/management.cattle.io/v3"
corev1 "k8s.io/api/core/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/labels"
)
const (
externalAddressAnnotation = "rke.cattle.io/external-ip"
LabelNodeName = "management.cattle.io/nodename"
)
func GetNodeName(node *v3.Node) string {
if node.Status.NodeName != "" {
return node.Status.NodeName
}
// to handle the case when node was provisioned first
if node.Status.NodeConfig != nil {
if node.Status.NodeConfig.HostnameOverride != "" {
return node.Status.NodeConfig.HostnameOverride
}
}
return ""
}
func IsNodeForNode(node *corev1.Node, machine *v3.Node) bool {
nodeName := GetNodeName(machine)
if nodeName == node.Name {
return true
}
machineAddress := ""
if machine.Status.NodeConfig != nil {
if machine.Status.NodeConfig.InternalAddress == "" {
// rke defaults internal address to address
machineAddress = machine.Status.NodeConfig.Address
} else {
machineAddress = machine.Status.NodeConfig.InternalAddress
}
}
if machineAddress == "" {
return false
}
if machineAddress == GetNodeInternalAddress(node) {
return true
}
return false
}
func GetNodeInternalAddress(node *corev1.Node) string {
for _, address := range node.Status.Addresses {
if address.Type == corev1.NodeInternalIP {
return address.Address
}
}
return ""
}
func GetEndpointNodeIP(node *v3.Node) string {
externalIP := ""
internalIP := ""
for _, ip := range node.Status.InternalNodeStatus.Addresses {
if ip.Type == "ExternalIP" && ip.Address != "" {
externalIP = ip.Address
break
} else if ip.Type == "InternalIP" && ip.Address != "" {
internalIP = ip.Address
}
}
if externalIP != "" {
return externalIP
}
if node.Annotations != nil {
externalIP = node.Status.NodeAnnotations[externalAddressAnnotation]
if externalIP != "" {
return externalIP
}
}
return internalIP
}
func GetNodeByNodeName(nodes []*v3.Node, nodeName string) *v3.Node {
for _, m := range nodes {
if GetNodeName(m) == nodeName {
return m
}
}
return nil
}
func GetNodeForMachine(machine *v3.Node, nodeLister v1.NodeLister) (*corev1.Node, error) {
nodeName := ""
if machine.Labels != nil {
nodeName = machine.Labels[LabelNodeName]
}
var nodes []*corev1.Node
var err error
if nodeName != "" {
node, err := nodeLister.Get("", nodeName)
if err != nil && !apierrors.IsNotFound(err) {
return nil, err
}
if node != nil {
nodes = append(nodes, node)
}
}
if len(nodes) == 0 {
nodes, err = nodeLister.List("", labels.NewSelector())
if err != nil {
return nil, err
}
}
for _, n := range nodes {
if IsNodeForNode(n, machine) {
return n, nil
}
}
return nil, nil
}
func GetMachineForNode(node *corev1.Node, clusterNamespace string, machineLister v3.NodeLister) (*v3.Node, error) {
labelsSearchSet := labels.Set{LabelNodeName: node.Name}
machines, err := machineLister.List(clusterNamespace, labels.SelectorFromSet(labelsSearchSet))
if err != nil {
return nil, err
}
if len(machines) == 0 {
machines, err = machineLister.List(clusterNamespace, labels.NewSelector())
if err != nil {
return nil, err
}
}
for _, machine := range machines {
if IsNodeForNode(node, machine) {
return machine, nil
}
}
return nil, nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/rancher/rancher.git
git@gitee.com:rancher/rancher.git
rancher
rancher
rancher
v2.2.5-rc1

搜索帮助