From c9a80568d5a89ec67143a94df49c4f02b93d61f7 Mon Sep 17 00:00:00 2001 From: Satoru Takeuchi Date: Fri, 26 Jan 2024 03:16:16 +0000 Subject: [PATCH] core: add flatten-rbd-pvc command Add flatten-rbd-pvc command to accomplish the following things: - Flatten an RBD image corrensponding to the target PVC. - Remove the temporary RBD image too. Closes: https://github.com/rook/kubectl-rook-ceph/issues/222 Signed-off-by: Satoru Takeuchi --- .github/workflows/go-test.yaml | 16 ++++ cmd/commands/flatten_rbd_pvc.go | 147 ++++++++++++++++++++++++++++++++ cmd/main.go | 1 + docs/flatten-rbd-pvc.md | 12 +++ tests/github-action-helper.sh | 21 +++++ 5 files changed, 197 insertions(+) create mode 100644 cmd/commands/flatten_rbd_pvc.go create mode 100644 docs/flatten-rbd-pvc.md diff --git a/.github/workflows/go-test.yaml b/.github/workflows/go-test.yaml index 83c22d2a..dc99dc50 100644 --- a/.github/workflows/go-test.yaml +++ b/.github/workflows/go-test.yaml @@ -83,6 +83,14 @@ jobs: set -ex kubectl rook-ceph rbd ls replicapool + - name: Flatten a PVC clone + run: | + set -ex + tests/github-action-helper.sh install_external_snapshotter + tests/github-action-helper.sh wait_for_rbd_pvc_clone_to_be_bound + + kubectl rook-ceph flatten-rbd-pvc rbd-pvc-clone + - name: Subvolume command run: | set -ex @@ -251,6 +259,14 @@ jobs: set -ex kubectl rook-ceph --operator-namespace test-operator -n test-cluster rbd ls replicapool + - name: Flatten a PVC clone + run: | + set -ex + tests/github-action-helper.sh install_external_snapshotter + tests/github-action-helper.sh wait_for_rbd_pvc_clone_to_be_bound + + kubectl rook-ceph --operator-namespace test-operator -n test-cluster flatten-rbd-pvc rbd-pvc-clone + - name: Subvolume command run: | set -ex diff --git a/cmd/commands/flatten_rbd_pvc.go b/cmd/commands/flatten_rbd_pvc.go new file mode 100644 index 00000000..a0fa176f --- /dev/null +++ b/cmd/commands/flatten_rbd_pvc.go @@ -0,0 +1,147 @@ +/* +Copyright 2024 The Rook Authors. All rights reserved. + +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 command + +import ( + "context" + "encoding/json" + "fmt" + + "github.com/rook/kubectl-rook-ceph/pkg/exec" + "github.com/rook/kubectl-rook-ceph/pkg/logging" + + "github.com/spf13/cobra" + corev1 "k8s.io/api/core/v1" + metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" +) + +type RBDInfoOutput struct { + ID string `json:"id"` + Name string `json:"name"` +} + +type Watcher struct { + Address string `json:"address"` +} + +type RBDStatusOutput struct { + Watchers []Watcher `json:"watchers"` +} + +var namespace string +var allowInUse bool + +// FlattenRBDPVCCmd represents the rook commands +var FlattenRBDPVCCmd = &cobra.Command{ + Use: "flatten-rbd-pvc", + Short: "Flatten the RBD image corresponding to the target RBD PVC", + Long: `Flatten the RBD image corresponding to the target RBD PVC. +The target RBD PVC must be a cloned image and must be created by ceph-csi. +This command removes the corresponding temporary cloned image[1] +if the target PVC was cloned from another PVC. + +[1]: https://github.com/ceph/ceph-csi/blob/328e4e5a0fa06521c5508b94a5b7ff250ada1681/docs/design/proposals/rbd-snap-clone.md.`, + Args: cobra.ExactArgs(1), + Run: func(cmd *cobra.Command, args []string) { + pvcName := args[0] + pvc, err := clientSets.Kube.CoreV1().PersistentVolumeClaims(namespace).Get(cmd.Context(), pvcName, metav1.GetOptions{}) + if err != nil { + logging.Fatal(err, "failed to get pvc %s/%s", namespace, pvcName) + } + if pvc.DeletionTimestamp != nil { + logging.Fatal(fmt.Errorf("pvc %s is deleting", pvcName)) + } + if pvc.Status.Phase != corev1.ClaimBound { + logging.Fatal(fmt.Errorf("pvc %s is not bound", pvcName)) + } + + var shouldDeleteTempImage bool + if pvc.Spec.DataSource != nil { + switch pvc.Spec.DataSource.Kind { + case "PersistentVolumeClaim": + shouldDeleteTempImage = true + case "VolumeSnapshot": + default: + logging.Fatal(fmt.Errorf("pvc %s is not a cloned image", pvcName)) + } + } + + pvName := pvc.Spec.VolumeName + pv, err := clientSets.Kube.CoreV1().PersistentVolumes().Get(cmd.Context(), pvName, metav1.GetOptions{}) + if err != nil { + logging.Fatal(fmt.Errorf("failed to get pv %s", pvName)) + } + imageName, ok := pv.Spec.CSI.VolumeAttributes["imageName"] + if !ok { + logging.Fatal(fmt.Errorf("pv %s would not be a CSI PVC", pvName)) + } + poolName, ok := pv.Spec.CSI.VolumeAttributes["pool"] + if !ok { + logging.Fatal(fmt.Errorf("pv %s would not be a CSI PVC", pvName)) + } + + if !allowInUse { + out, err := exec.RunCommandInOperatorPod(cmd.Context(), clientSets, "rbd", []string{"-p", poolName, "status", imageName, "--format=json"}, cephClusterNamespace, operatorNamespace, false) + if err != nil { + logging.Fatal(fmt.Errorf("failed to stat %s/%s", poolName, imageName)) + } + var status RBDStatusOutput + json.Unmarshal([]byte(out), &status) + if len(status.Watchers) >= 1 { + logging.Fatal(fmt.Errorf("flatten in-use pvc %s is not allowed. If you want to do, run with `--allow-in-use` option", pvcName)) + } + } + + if shouldDeleteTempImage { + deleteTempImage(cmd.Context(), allowInUse, poolName, imageName) + } + logging.Info("flattening the target RBD image %s/%s", poolName, imageName) + _, err = exec.RunCommandInOperatorPod(cmd.Context(), clientSets, "rbd", []string{"-p", poolName, "flatten", imageName}, cephClusterNamespace, operatorNamespace, false) + if err != nil { + logging.Fatal(fmt.Errorf("failed to flatten %s/%s", poolName, imageName)) + } + }, +} + +func deleteTempImage(ctx context.Context, allowInUse bool, poolName, imageName string) { + tempImageName := imageName + "-temp" + + out, err := exec.RunCommandInOperatorPod(ctx, clientSets, "rbd", []string{"-p", poolName, "info", "--format=json", tempImageName}, cephClusterNamespace, operatorNamespace, false) + if err != nil { + logging.Error(fmt.Errorf("failed to run `rbd info` for rbd image %s/%s", poolName, tempImageName)) + return + } + var info RBDInfoOutput + json.Unmarshal([]byte(out), &info) + id := info.ID + logging.Info("removing the temporary RBD image %s/%s if exist", poolName, tempImageName) + _, err = exec.RunCommandInOperatorPod(ctx, clientSets, "rbd", []string{"-p", poolName, "trash", "mv", tempImageName}, cephClusterNamespace, operatorNamespace, false) + if err != nil { + logging.Fatal(fmt.Errorf("failed to move rbd image %s/%s to trash", poolName, tempImageName)) + } + if id != "" { + _, err = exec.RunCommandInOperatorPod(ctx, clientSets, "ceph", []string{"rbd", "task", "add", "trash", "remove", fmt.Sprintf("%s/%s", poolName, id)}, cephClusterNamespace, operatorNamespace, false) + if err != nil { + logging.Fatal(fmt.Errorf("failed to create a task to remove %s/%s from trash", poolName, id)) + } + } +} + +func init() { + FlattenRBDPVCCmd.Flags().StringVarP(&namespace, "namespace", "n", "default", "pvc's namespace") + FlattenRBDPVCCmd.Flags().BoolVarP(&allowInUse, "allow-in-use", "a", false, "allow to flatten in-use image") +} diff --git a/cmd/main.go b/cmd/main.go index ec95eb6f..a965bba9 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -42,5 +42,6 @@ func addcommands() { command.DestroyClusterCmd, command.SubvolumeCmd, command.RadosCmd, + command.FlattenRBDPVCCmd, ) } diff --git a/docs/flatten-rbd-pvc.md b/docs/flatten-rbd-pvc.md new file mode 100644 index 00000000..6abfdd9e --- /dev/null +++ b/docs/flatten-rbd-pvc.md @@ -0,0 +1,12 @@ +# Flatten RBD PVC + +`flatten-rbd-pvc` command is to flatten the RBD image corresponding to the target RBD PVC. +Fore more details about flatten, see [the Ceph official document](https://docs.ceph.com/en/latest/rbd/rbd-snapshot/#flattening-a-cloned-image). + +By flattening RBD images, we can bypass the problems specific to non-flattened cloned image like https://github.com/ceph/ceph-csi/discussions/4360. + +## Examples. + +```bash +kubectl rook-ceph flatten-rbd-pvc rbd-pvc-clone +``` diff --git a/tests/github-action-helper.sh b/tests/github-action-helper.sh index e48a1a23..f5de3cc7 100755 --- a/tests/github-action-helper.sh +++ b/tests/github-action-helper.sh @@ -267,6 +267,27 @@ install_minikube_with_none_driver() { sudo -E minikube start --kubernetes-version="$1" --driver=none --memory 6g --cpus=2 --addons ingress --cni=calico } +install_external_snapshotter() { + curl -L https://github.com/kubernetes-csi/external-snapshotter/archive/refs/tags/v6.2.0.zip -o external-snapshotter.zip + unzip external-snapshotter.zip + cd external-snapshotter-6.2.0 + + kubectl kustomize client/config/crd | kubectl create -f - + kubectl -n kube-system kustomize deploy/kubernetes/snapshot-controller | kubectl create -f - +} + +wait_for_rbd_pvc_clone_to_be_bound() { + kubectl create -f https://raw.githubusercontent.com/rook/rook/master/deploy/examples/csi/rbd/pvc-clone.yaml + + timeout 100 bash <<-'EOF' + until [ $(kubectl get pvc rbd-pvc-clone -o jsonpath='{.status.phase}') == "Bound" ]; do + echo "waiting for the pvc clone to be in bound state" + sleep 1 + done +EOF + timeout_command_exit_code +} + ######## # MAIN # ########