Skip to content

Commit e487bd6

Browse files
authored
Merge pull request kubernetes#3102 from wccsama/EphemeralStorage
add EphemeralStorage in group template
2 parents e95ef27 + fbfccd8 commit e487bd6

File tree

2 files changed

+26
-20
lines changed

2 files changed

+26
-20
lines changed

cluster-autoscaler/cloudprovider/baiducloud/baiducloud-sdk-go/cce/cluster.go

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,14 @@ type CceCluster struct {
6262

6363
// CceGroup defines autoscaling group
6464
type CceGroup struct {
65-
InstanceType int `json:"instanceType"`
66-
CPU int `json:"cpu,omitempty"`
67-
Memory int `json:"memory,omitempty"`
68-
GpuCount int `json:"gpuCount,omitempty"`
69-
GpuCard string `json:"gpuCard,omitempty"`
70-
DiskSize int `json:"diskSize,omitempty"`
71-
Tags []Tag `json:"tags"`
65+
InstanceType int `json:"instanceType"`
66+
CPU int `json:"cpu,omitempty"`
67+
Memory int `json:"memory,omitempty"`
68+
GpuCount int `json:"gpuCount,omitempty"`
69+
GpuCard string `json:"gpuCard,omitempty"`
70+
DiskSize int `json:"diskSize,omitempty"`
71+
EphemeralStorage int `json:"ephemeralStorage,omitempty"`
72+
Tags []Tag `json:"tags"`
7273
}
7374

7475
// Tag defines label

cluster-autoscaler/cloudprovider/baiducloud/baiducloud_manager.go

Lines changed: 18 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -58,13 +58,14 @@ type asgInformation struct {
5858
}
5959

6060
type asgTemplate struct {
61-
InstanceType int
62-
Region string
63-
Zone string
64-
CPU int
65-
Memory int
66-
GpuCount int
67-
Tags map[string]string
61+
InstanceType int
62+
Region string
63+
Zone string
64+
CPU int
65+
Memory int
66+
GpuCount int
67+
EphemeralStorage int
68+
Tags map[string]string
6869
}
6970

7071
// CreateBaiducloudManager constructs baiducloudManager object.
@@ -230,12 +231,13 @@ func (m *BaiducloudManager) getAsgTemplate(name string) (*asgTemplate, error) {
230231
}
231232

232233
return &asgTemplate{
233-
InstanceType: cceGroup.InstanceType,
234-
Region: m.cloudConfig.Region,
235-
CPU: cceGroup.CPU,
236-
Memory: cceGroup.Memory,
237-
GpuCount: cceGroup.GpuCount,
238-
Tags: tags,
234+
InstanceType: cceGroup.InstanceType,
235+
Region: m.cloudConfig.Region,
236+
CPU: cceGroup.CPU,
237+
Memory: cceGroup.Memory,
238+
GpuCount: cceGroup.GpuCount,
239+
EphemeralStorage: cceGroup.EphemeralStorage,
240+
Tags: tags,
239241
}, nil
240242
}
241243

@@ -258,6 +260,9 @@ func (m *BaiducloudManager) buildNodeFromTemplate(asg *Asg, template *asgTemplat
258260

259261
node.Status.Capacity[gpu.ResourceNvidiaGPU] = *resource.NewQuantity(int64(template.GpuCount), resource.DecimalSI)
260262

263+
// add ephemeral-storage
264+
node.Status.Capacity[apiv1.ResourceEphemeralStorage] = *resource.NewQuantity(int64(template.EphemeralStorage*1024*1024*1024), resource.DecimalSI)
265+
261266
node.Status.Allocatable = node.Status.Capacity
262267

263268
node.Status.Conditions = cloudprovider.BuildReadyConditions()

0 commit comments

Comments
 (0)