36 Star 411 Fork 76

GVPrancher/rancher

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
cluster.go 1.64 KB
一键复制 编辑 原始数据 按行查看 历史
package cluster
import (
"encoding/base64"
"fmt"
"io/ioutil"
"os"
"path"
"k8s.io/client-go/rest"
)
const (
rancherCredentialsFolder = "/cattle-credentials"
urlFilename = "url"
tokenFilename = "token"
kubernetesServiceHostKey = "KUBERNETES_SERVICE_HOST"
kubernetesServicePortKey = "KUBERNETES_SERVICE_PORT"
)
func TokenAndURL() (string, string, error) {
url, err := readKey(urlFilename)
if err != nil {
return "", "", err
}
token, err := readKey(tokenFilename)
return token, url, err
}
func Params() (map[string]interface{}, error) {
cfg, err := rest.InClusterConfig()
if err != nil {
return nil, err
}
if err := populateCAData(cfg); err != nil {
return nil, err
}
kubernetesServiceHost, err := getenv(kubernetesServiceHostKey)
if err != nil {
return nil, err
}
kubernetesServicePort, err := getenv(kubernetesServicePortKey)
if err != nil {
return nil, err
}
return map[string]interface{}{
"cluster": map[string]interface{}{
"address": fmt.Sprintf("%s:%s", kubernetesServiceHost, kubernetesServicePort),
"token": cfg.BearerToken,
"caCert": base64.StdEncoding.EncodeToString(cfg.CAData),
},
}, nil
}
func getenv(env string) (string, error) {
value := os.Getenv(env)
if value == "" {
return "", fmt.Errorf("%s is empty", env)
}
return value, nil
}
func populateCAData(cfg *rest.Config) error {
bytes, err := ioutil.ReadFile(cfg.CAFile)
if err != nil {
return err
}
cfg.CAData = bytes
return nil
}
func readKey(key string) (string, error) {
bytes, err := ioutil.ReadFile(path.Join(rancherCredentialsFolder, key))
if err != nil {
return "", err
}
return string(bytes), nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/rancher/rancher.git
git@gitee.com:rancher/rancher.git
rancher
rancher
rancher
v2.0.16-rc1

搜索帮助