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
|
// Package config reads relevant SSH access information from cluster config declaration.
package config
import (
"io/ioutil"
v3 "github.com/rancher/types/apis/management.cattle.io/v3"
"gopkg.in/yaml.v2"
)
const (
defaultConfigFile = "cluster.yml"
)
// NodeInfo contains role and SSH access information for a single cluster node.
type NodeInfo struct {
Role []string
User string
Address string
Port string
SSHKeyPath string
}
// GetNodesInfo returns nodes' roles and SSH access information for a whole cluster.
func GetNodesInfo() ([]NodeInfo, error) {
config, err := readConfig(defaultConfigFile)
if err != nil {
return []NodeInfo{}, err
}
cluster, err := parseConfig(config)
if err != nil {
return []NodeInfo{}, err
}
var nodes []NodeInfo
for _, node := range cluster.Nodes {
nodes = append(nodes, NodeInfo{
node.Role, node.User, node.Address, node.Port, node.SSHKeyPath,
})
}
return nodes, nil
}
func readConfig(configFile string) (string, error) {
config, err := ioutil.ReadFile(configFile)
if err != nil {
return "", err
}
return string(config), nil
}
func parseConfig(config string) (*v3.RancherKubernetesEngineConfig, error) {
var rkeConfig v3.RancherKubernetesEngineConfig
if err := yaml.Unmarshal([]byte(config), &rkeConfig); err != nil {
return nil, err
}
return &rkeConfig, nil
}
|
