azure_dd_block.go 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. /*
  2. Copyright 2018 The Kubernetes Authors.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package azure_dd
  14. import (
  15. "fmt"
  16. "path/filepath"
  17. "k8s.io/api/core/v1"
  18. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  19. "k8s.io/apimachinery/pkg/types"
  20. "k8s.io/klog"
  21. "k8s.io/kubernetes/pkg/util/mount"
  22. "k8s.io/kubernetes/pkg/volume"
  23. "k8s.io/kubernetes/pkg/volume/util"
  24. "k8s.io/kubernetes/pkg/volume/util/volumepathhandler"
  25. utilstrings "k8s.io/utils/strings"
  26. )
  27. var _ volume.VolumePlugin = &azureDataDiskPlugin{}
  28. var _ volume.PersistentVolumePlugin = &azureDataDiskPlugin{}
  29. var _ volume.BlockVolumePlugin = &azureDataDiskPlugin{}
  30. var _ volume.DeletableVolumePlugin = &azureDataDiskPlugin{}
  31. var _ volume.ProvisionableVolumePlugin = &azureDataDiskPlugin{}
  32. func (plugin *azureDataDiskPlugin) ConstructBlockVolumeSpec(podUID types.UID, volumeName, mapPath string) (*volume.Spec, error) {
  33. pluginDir := plugin.host.GetVolumeDevicePluginDir(azureDataDiskPluginName)
  34. blkutil := volumepathhandler.NewBlockVolumePathHandler()
  35. globalMapPathUUID, err := blkutil.FindGlobalMapPathUUIDFromPod(pluginDir, mapPath, podUID)
  36. if err != nil {
  37. return nil, err
  38. }
  39. klog.V(5).Infof("constructing block volume spec from globalMapPathUUID: %s", globalMapPathUUID)
  40. globalMapPath := filepath.Dir(globalMapPathUUID)
  41. if len(globalMapPath) <= 1 {
  42. return nil, fmt.Errorf("failed to get volume plugin information from globalMapPathUUID: %v", globalMapPathUUID)
  43. }
  44. return getVolumeSpecFromGlobalMapPath(globalMapPath, volumeName)
  45. }
  46. func getVolumeSpecFromGlobalMapPath(globalMapPath, volumeName string) (*volume.Spec, error) {
  47. // Get volume spec information from globalMapPath
  48. // globalMapPath example:
  49. // plugins/kubernetes.io/{PluginName}/{DefaultKubeletVolumeDevicesDirName}/{volumeID}
  50. // plugins/kubernetes.io/azure-disk/volumeDevices/vol-XXXXXX
  51. diskName := filepath.Base(globalMapPath)
  52. if len(diskName) <= 1 {
  53. return nil, fmt.Errorf("failed to get diskName from global path=%s", globalMapPath)
  54. }
  55. klog.V(5).Infof("got diskName(%s) from globalMapPath: %s", globalMapPath, diskName)
  56. block := v1.PersistentVolumeBlock
  57. pv := &v1.PersistentVolume{
  58. ObjectMeta: metav1.ObjectMeta{
  59. Name: volumeName,
  60. },
  61. Spec: v1.PersistentVolumeSpec{
  62. PersistentVolumeSource: v1.PersistentVolumeSource{
  63. AzureDisk: &v1.AzureDiskVolumeSource{
  64. DiskName: diskName,
  65. },
  66. },
  67. VolumeMode: &block,
  68. },
  69. }
  70. return volume.NewSpecFromPersistentVolume(pv, true), nil
  71. }
  72. // NewBlockVolumeMapper creates a new volume.BlockVolumeMapper from an API specification.
  73. func (plugin *azureDataDiskPlugin) NewBlockVolumeMapper(spec *volume.Spec, pod *v1.Pod, _ volume.VolumeOptions) (volume.BlockVolumeMapper, error) {
  74. // If this is called via GenerateUnmapDeviceFunc(), pod is nil.
  75. // Pass empty string as dummy uid since uid isn't used in the case.
  76. var uid types.UID
  77. if pod != nil {
  78. uid = pod.UID
  79. }
  80. return plugin.newBlockVolumeMapperInternal(spec, uid, plugin.host.GetMounter(plugin.GetPluginName()))
  81. }
  82. func (plugin *azureDataDiskPlugin) newBlockVolumeMapperInternal(spec *volume.Spec, podUID types.UID, mounter mount.Interface) (volume.BlockVolumeMapper, error) {
  83. volumeSource, readOnly, err := getVolumeSource(spec)
  84. if err != nil {
  85. return nil, err
  86. }
  87. disk := makeDataDisk(spec.Name(), podUID, volumeSource.DiskName, plugin.host, plugin)
  88. return &azureDataDiskMapper{
  89. dataDisk: disk,
  90. readOnly: readOnly,
  91. }, nil
  92. }
  93. func (plugin *azureDataDiskPlugin) NewBlockVolumeUnmapper(volName string, podUID types.UID) (volume.BlockVolumeUnmapper, error) {
  94. return plugin.newUnmapperInternal(volName, podUID, plugin.host.GetMounter(plugin.GetPluginName()))
  95. }
  96. func (plugin *azureDataDiskPlugin) newUnmapperInternal(volName string, podUID types.UID, mounter mount.Interface) (volume.BlockVolumeUnmapper, error) {
  97. disk := makeDataDisk(volName, podUID, "", plugin.host, plugin)
  98. return &azureDataDiskUnmapper{dataDisk: disk}, nil
  99. }
  100. func (c *azureDataDiskUnmapper) TearDownDevice(mapPath, devicePath string) error {
  101. return nil
  102. }
  103. type azureDataDiskUnmapper struct {
  104. *dataDisk
  105. }
  106. var _ volume.BlockVolumeUnmapper = &azureDataDiskUnmapper{}
  107. type azureDataDiskMapper struct {
  108. *dataDisk
  109. readOnly bool
  110. }
  111. var _ volume.BlockVolumeMapper = &azureDataDiskMapper{}
  112. func (b *azureDataDiskMapper) SetUpDevice() (string, error) {
  113. return "", nil
  114. }
  115. func (b *azureDataDiskMapper) MapDevice(devicePath, globalMapPath, volumeMapPath, volumeMapName string, podUID types.UID) error {
  116. return util.MapBlockVolume(devicePath, globalMapPath, volumeMapPath, volumeMapName, podUID)
  117. }
  118. // GetGlobalMapPath returns global map path and error
  119. // path: plugins/kubernetes.io/{PluginName}/volumeDevices/volumeID
  120. // plugins/kubernetes.io/azure-disk/volumeDevices/vol-XXXXXX
  121. func (disk *dataDisk) GetGlobalMapPath(spec *volume.Spec) (string, error) {
  122. volumeSource, _, err := getVolumeSource(spec)
  123. if err != nil {
  124. return "", err
  125. }
  126. return filepath.Join(disk.plugin.host.GetVolumeDevicePluginDir(azureDataDiskPluginName), string(volumeSource.DiskName)), nil
  127. }
  128. // GetPodDeviceMapPath returns pod device map path and volume name
  129. // path: pods/{podUid}/volumeDevices/kubernetes.io~azure
  130. func (disk *dataDisk) GetPodDeviceMapPath() (string, string) {
  131. name := azureDataDiskPluginName
  132. return disk.plugin.host.GetPodVolumeDeviceDir(disk.podUID, utilstrings.EscapeQualifiedName(name)), disk.volumeName
  133. }