rancher/os
1
0
Fork 1
mirror of https://github.com/rancher/os.git synced 2025-09-18 08:06:48 +00:00
Code Issues Releases Wiki Activity

Auto resize root disk on AWS (#2191)

Browse Source
This commit is contained in:
niusmallnan
2017-12-26 17:59:08 +08:00
committed by GitHub
parent dab1c4ffb3
commit 55368a3897
5 changed files with 28 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
config/cloudinit/datasource/metadata/ec2/metadata.go
View File

@@ -32,6 +32,9 @@ const (
apiVersion = "latest/"
userdataPath = apiVersion + "user-data"
metadataPath = apiVersion + "meta-data/"
defaultXVRootDisk = "/dev/xvda"
defaultNVMeRootDisk = "/dev/nvme0n1"
)
type MetadataService struct {
@@ -137,6 +140,17 @@ func (ms MetadataService) FetchMetadata() (datasource.Metadata, error) {
}
}
// With C5 and M5 instances, EBS volumes are exposed as NVMe block devices.
// http://docs.aws.amazon.com/AWSEC2/latest/UserGuide/nvme-ebs-volumes.html
metadata.RootDisk = defaultXVRootDisk
if instanceType, err := ms.FetchAttribute("instance-type"); err == nil {
if strings.HasPrefix(instanceType, "m5") || strings.HasPrefix(instanceType, "c5") {
metadata.RootDisk = defaultNVMeRootDisk
}
} else if _, ok := err.(pkg.ErrNotFound); !ok {
return metadata, err
}
return metadata, nil
}

3
config/cloudinit/datasource/metadata/ec2/metadata_test.go
View File

@@ -67,6 +67,7 @@ func TestFetchMetadata(t *testing.T) {
PrivateIPv4: net.ParseIP("1.2.3.4"),
PublicIPv4: net.ParseIP("5.6.7.8"),
SSHPublicKeys: map[string]string{"test1": "key"},
RootDisk: "/dev/xvda",
NetworkConfig: netconf.NetworkConfig{
Interfaces: map[string]netconf.InterfaceConfig{
/* "eth0": netconf.InterfaceConfig{
@@ -89,12 +90,14 @@ func TestFetchMetadata(t *testing.T) {
"/2009-04-04/meta-data/public-keys": "0=test1\n",
"/2009-04-04/meta-data/public-keys/0": "openssh-key",
"/2009-04-04/meta-data/public-keys/0/openssh-key": "key",
"/2009-04-04/meta-data/instance-type": "m5.large",
},
expect: datasource.Metadata{
Hostname: "host",
PrivateIPv4: net.ParseIP("21.2.3.4"),
PublicIPv4: net.ParseIP("25.6.7.8"),
SSHPublicKeys: map[string]string{"test1": "key"},
RootDisk: "/dev/nvme0n1",
NetworkConfig: netconf.NetworkConfig{
Interfaces: map[string]netconf.InterfaceConfig{
/* "eth0": netconf.InterfaceConfig{