mirror of
https://github.com/vmware-tanzu/velero.git
synced 2026-02-12 06:41:37 +00:00
104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
/*
|
|
Copyright 2018 the Velero contributors.
|
|
|
|
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 backup
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
corev1api "k8s.io/api/core/v1"
|
|
)
|
|
|
|
// pvcSnapshotTracker keeps track of persistent volume claims that have been snapshotted
|
|
// with pod volume backup.
|
|
type pvcSnapshotTracker struct {
|
|
pvcs map[string]pvcSnapshotStatus
|
|
}
|
|
|
|
type pvcSnapshotStatus struct {
|
|
taken bool
|
|
}
|
|
|
|
func newPVCSnapshotTracker() *pvcSnapshotTracker {
|
|
return &pvcSnapshotTracker{
|
|
pvcs: make(map[string]pvcSnapshotStatus),
|
|
}
|
|
}
|
|
|
|
// Track indicates a volume from a pod should be snapshotted by pod volume backup.
|
|
func (t *pvcSnapshotTracker) Track(pod *corev1api.Pod, volumeName string) {
|
|
// if the volume is a PVC, track it
|
|
for _, volume := range pod.Spec.Volumes {
|
|
if volume.Name == volumeName {
|
|
if volume.PersistentVolumeClaim != nil {
|
|
if _, ok := t.pvcs[key(pod.Namespace, volume.PersistentVolumeClaim.ClaimName)]; !ok {
|
|
t.pvcs[key(pod.Namespace, volume.PersistentVolumeClaim.ClaimName)] = pvcSnapshotStatus{false}
|
|
}
|
|
}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// Take indicates a volume from a pod has been taken by pod volume backup.
|
|
func (t *pvcSnapshotTracker) Take(pod *corev1api.Pod, volumeName string) {
|
|
for _, volume := range pod.Spec.Volumes {
|
|
if volume.Name == volumeName {
|
|
if volume.PersistentVolumeClaim != nil {
|
|
t.pvcs[key(pod.Namespace, volume.PersistentVolumeClaim.ClaimName)] = pvcSnapshotStatus{true}
|
|
}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// Has returns true if the PVC with the specified namespace and name has been tracked.
|
|
func (t *pvcSnapshotTracker) Has(namespace, name string) bool {
|
|
_, found := t.pvcs[key(namespace, name)]
|
|
return found
|
|
}
|
|
|
|
// TakenForPodVolume returns true and the PVC's name if the pod volume with the specified name uses a
|
|
// PVC and that PVC has been taken by pod volume backup.
|
|
func (t *pvcSnapshotTracker) TakenForPodVolume(pod *corev1api.Pod, volume string) (bool, string) {
|
|
for _, podVolume := range pod.Spec.Volumes {
|
|
if podVolume.Name != volume {
|
|
continue
|
|
}
|
|
|
|
if podVolume.PersistentVolumeClaim == nil {
|
|
return false, ""
|
|
}
|
|
|
|
status, found := t.pvcs[key(pod.Namespace, podVolume.PersistentVolumeClaim.ClaimName)]
|
|
if !found {
|
|
return false, ""
|
|
}
|
|
|
|
if !status.taken {
|
|
return false, ""
|
|
}
|
|
|
|
return true, podVolume.PersistentVolumeClaim.ClaimName
|
|
}
|
|
|
|
return false, ""
|
|
}
|
|
|
|
func key(namespace, name string) string {
|
|
return fmt.Sprintf("%s/%s", namespace, name)
|
|
}
|