manual freeze

Signed-off-by: Valeriy Khorunzhin <valeriy.khorunzhin@flant.com>

Author: Valeriy Khorunzhin

images/virt-handler/werf.inc.yaml

@@ -106,6 +106,8 @@ shell:
 
   setup:
   - |
+    echo kek
+  
     /relocate_binaries.sh -i "{{ $builderDependencies.binaries | join " " }}" -o /relocate
 
     mkdir -p /relocate/etc /relocate/root

images/virtualization-artifact/pkg/controller/vm/internal/filesystem.go

@@ -18,22 +18,29 @@ package internal
 
 import (
 	"context"
+	"fmt"
 
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/controller-runtime/pkg/reconcile"
 
 	"github.com/deckhouse/virtualization-controller/pkg/controller/conditions"
 	"github.com/deckhouse/virtualization-controller/pkg/controller/vm/internal/state"
 	"github.com/deckhouse/virtualization/api/core/v1alpha2/vmcondition"
+	subv1alpha2 "github.com/deckhouse/virtualization/api/subresources/v1alpha2"
 )
 
-const nameFilesystemHandler = "FilesystemHandler"
+const (
+	nameFilesystemHandler = "FilesystemHandler"
+	annKek                = "kek"
+)
 
-func NewFilesystemHandler() *FilesystemHandler {
-	return &FilesystemHandler{}
+func NewFilesystemHandler(virtClient VirtClient) *FilesystemHandler {
+	return &FilesystemHandler{virtClient: virtClient}
 }
 
-type FilesystemHandler struct{}
+type FilesystemHandler struct {
+	virtClient VirtClient
+}
 
 func (h *FilesystemHandler) Handle(ctx context.Context, s state.VirtualMachineState) (reconcile.Result, error) {
 	if s.VirtualMachine().IsEmpty() {
@@ -67,6 +74,20 @@ func (h *FilesystemHandler) Handle(ctx context.Context, s state.VirtualMachineSt
 		return reconcile.Result{}, nil
 	}
 
+	if request, ok := changed.Annotations[annKek]; ok {
+		switch request {
+		case "freeze":
+			if err = h.virtClient.VirtualMachines(changed.Namespace).Freeze(ctx, changed.Name, subv1alpha2.VirtualMachineFreeze{}); err != nil {
+				return reconcile.Result{}, fmt.Errorf("freeze virtual machine %s/%s: %w", changed.Namespace, changed.Name, err)
+			}
+		case "unfreeze":
+			if err = h.virtClient.VirtualMachines(changed.Namespace).Unfreeze(ctx, changed.Name); err != nil {
+				return reconcile.Result{}, fmt.Errorf("unfreeze virtual machine %s/%s: %w", changed.Namespace, changed.Name, err)
+			}
+		}
+		delete(changed.Annotations, annKek)
+	}
+
 	agentReady, _ := conditions.GetCondition(vmcondition.TypeAgentReady, changed.Status.Conditions)
 	if agentReady.Status != metav1.ConditionTrue {
 		return reconcile.Result{}, nil

images/virtualization-artifact/pkg/controller/vm/vm_controller.go

@@ -71,7 +71,7 @@ func SetupController(
 		internal.NewUSBDeviceAttachHandler(client, virtClient),
 		internal.NewProvisioningHandler(client),
 		internal.NewAgentHandler(),
-		internal.NewFilesystemHandler(),
+		internal.NewFilesystemHandler(virtClient),
 		internal.NewSnapshottingHandler(client),
 		internal.NewPodHandler(client),
 		internal.NewSizePolicyHandler(),