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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Package aliyun provides node discovery for Aliyun.
package aliyun
import (
"fmt"
"io/ioutil"
"log"
"github.com/denverdino/aliyungo/common"
"github.com/denverdino/aliyungo/ecs"
)
type Provider struct {
userAgent string
}
func (p *Provider) SetUserAgent(s string) {
p.userAgent = s
}
func (p *Provider) Help() string {
return `Aliyun(Alibaba Cloud):
provider: "aliyun"
region: The Aliyun region.
tag_key: The tag key to filter on
tag_value: The tag value to filter on
access_key_id: The Aliyun access key to use
access_key_secret: The Aliyun access key secret to use
The required RAM permission is 'ecs:DescribeInstances'.
It is recommended you make a dedicated key used only for auto-joining.
`
}
func (p *Provider) Addrs(args map[string]string, l *log.Logger) ([]string, error) {
if args["provider"] != "aliyun" {
return nil, fmt.Errorf("discover-aliyun: invalid provider " + args["provider"])
}
if l == nil {
l = log.New(ioutil.Discard, "", 0)
}
region := args["region"]
tagKey := args["tag_key"]
tagValue := args["tag_value"]
accessKeyID := args["access_key_id"]
accessKeySecret := args["access_key_secret"]
log.Printf("[DEBUG] discover-aliyun: Using region=%s tag_key=%s tag_value=%s", region, tagKey, tagValue)
if accessKeyID == "" && accessKeySecret == "" {
log.Printf("[DEBUG] discover-aliyun: No static credentials")
} else {
log.Printf("[DEBUG] discover-aliyun: Static credentials provided")
}
if region == "" {
l.Printf("[DEBUG] discover-aliyun: Region not provided")
return nil, fmt.Errorf("discover-aliyun: invalid region")
}
l.Printf("[INFO] discover-aliyun: Region is %s", region)
svc := ecs.NewClient(accessKeyID, accessKeySecret)
if p.userAgent != "" {
svc.SetUserAgent(p.userAgent)
}
l.Printf("[INFO] discover-aliyun: Filter instances with %s=%s", tagKey, tagValue)
resp, err := svc.DescribeInstancesWithRaw(&ecs.DescribeInstancesArgs{
RegionId: common.Region(region),
Status: ecs.Running,
Tag: map[string]string{
tagKey: tagValue,
}},
)
if err != nil {
return nil, fmt.Errorf("discover-aliyun: DescribeInstancesWithRaw failed: %s", err)
}
l.Printf("[DEBUG] discover-aliyun: Found total %d instances", resp.TotalCount)
var addrs []string
for _, instanceAttributesType := range resp.Instances.Instance {
switch instanceAttributesType.InstanceNetworkType {
case "classic":
for _, ipAddress := range instanceAttributesType.InnerIpAddress.IpAddress {
l.Printf("[DEBUG] discover-aliyun: Instance %s has innner ip %s ", instanceAttributesType.InstanceId, ipAddress)
addrs = append(addrs, ipAddress)
}
case "vpc":
for _, ipAddress := range instanceAttributesType.VpcAttributes.PrivateIpAddress.IpAddress {
l.Printf("[DEBUG] discover-aliyun: Instance %s has private ip %s ", instanceAttributesType.InstanceId, ipAddress)
addrs = append(addrs, ipAddress)
}
}
}
l.Printf("[DEBUG] discover-aliyun: Found ip addresses: %v", addrs)
return addrs, nil
}
|
