-
Notifications
You must be signed in to change notification settings - Fork 3.9k
/
alicloud_instance_types.go
156 lines (139 loc) · 4.5 KB
/
alicloud_instance_types.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
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
/*
Copyright 2018 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package alicloud
import (
"fmt"
"k8s.io/autoscaler/cluster-autoscaler/cloudprovider/alicloud/alibaba-cloud-sdk-go/services/ecs"
"k8s.io/autoscaler/cluster-autoscaler/cloudprovider/alicloud/alibaba-cloud-sdk-go/services/ess"
klog "k8s.io/klog/v2"
"time"
)
type ecsInstance interface {
DescribeInstanceTypes(req *ecs.DescribeInstanceTypesRequest) (*ecs.DescribeInstanceTypesResponse, error)
}
type instanceType struct {
instanceTypeID string
vcpu int64
memoryInBytes int64
gpu int64
}
type instanceTypeModel struct {
instanceType
// TODO add price model .
}
// instanceWrapper will provide functions about
// instance types,price model and so on.
type instanceWrapper struct {
ecsInstance
InstanceTypeCache map[string]*instanceTypeModel
}
func (iw *instanceWrapper) getInstanceTypeById(typeId string) (*instanceType, error) {
if instanceTypeModel := iw.FindInstanceType(typeId); instanceTypeModel != nil {
return &instanceTypeModel.instanceType, nil
}
err := iw.RefreshCache()
if err != nil {
klog.Errorf("failed to refresh instance type cache,because of %s", err.Error())
return nil, err
}
if instanceTypeModel := iw.FindInstanceType(typeId); instanceTypeModel != nil {
return &instanceTypeModel.instanceType, nil
}
return nil, fmt.Errorf("failed to find the specific instance type by Id: %s", typeId)
}
func (iw *instanceWrapper) getInstanceTags(tags ess.Tags) (map[string]string, error) {
tagsMap := make(map[string]string)
for _, tag := range tags.Tag {
tagsMap[tag.Key] = tag.Value
}
return tagsMap, nil
}
func (iw *instanceWrapper) FindInstanceType(typeId string) *instanceTypeModel {
if iw.InstanceTypeCache == nil || iw.InstanceTypeCache[typeId] == nil {
return nil
}
return iw.InstanceTypeCache[typeId]
}
func (iw *instanceWrapper) RefreshCache() error {
req := ecs.CreateDescribeInstanceTypesRequest()
resp, err := iw.DescribeInstanceTypes(req)
if err != nil {
return err
}
if iw.InstanceTypeCache == nil {
iw.InstanceTypeCache = make(map[string]*instanceTypeModel)
}
types := resp.InstanceTypes.InstanceType
for _, item := range types {
iw.InstanceTypeCache[item.InstanceTypeId] = &instanceTypeModel{
instanceType{
instanceTypeID: item.InstanceTypeId,
vcpu: int64(item.CpuCoreCount),
memoryInBytes: int64(item.MemorySize * 1024 * 1024 * 1024),
gpu: int64(item.GPUAmount),
},
}
}
return nil
}
func newInstanceWrapper(cfg *cloudConfig) (*instanceWrapper, error) {
if cfg.isValid() == false {
return nil, fmt.Errorf("your cloud config is not valid")
}
iw := &instanceWrapper{}
if cfg.STSEnabled {
go func(iw *instanceWrapper, cfg *cloudConfig) {
timer := time.NewTicker(refreshClientInterval)
defer timer.Stop()
for {
select {
case <-timer.C:
client, err := getEcsClient(cfg)
if err == nil {
iw.ecsInstance = client
}
}
}
}(iw, cfg)
}
client, err := getEcsClient(cfg)
if err == nil {
iw.ecsInstance = client
}
return iw, err
}
func getEcsClient(cfg *cloudConfig) (client *ecs.Client, err error) {
region := cfg.getRegion()
if cfg.STSEnabled == true {
auth, err := cfg.getSTSToken()
if err != nil {
klog.Errorf("failed to get sts token from metadata,because of %s", err.Error())
return nil, err
}
client, err = ecs.NewClientWithStsToken(region, auth.AccessKeyId, auth.AccessKeySecret, auth.SecurityToken)
if err != nil {
klog.Errorf("failed to create client with sts in metadata,because of %s", err.Error())
}
} else if cfg.RRSAEnabled {
client, err = ecs.NewClientWithRRSA(region, cfg.RoleARN, cfg.OIDCProviderARN, cfg.OIDCTokenFilePath, cfg.RoleSessionName)
if err != nil {
klog.Errorf("Failed to create ess client with RRSA, because of %s", err.Error())
}
} else {
client, err = ecs.NewClientWithAccessKey(region, cfg.AccessKeyID, cfg.AccessKeySecret)
if err != nil {
klog.Errorf("failed to create ecs client with AccessKeyId and AccessKeySecret,because of %s", err.Error())
}
}
return
}