init2

Signed-off-by: Yaroslav Borbat <[email protected]>

Author: yaroslavborbat

api/core/v1alpha2/vdcondition/condition.go

@@ -178,5 +178,6 @@ const (
 )
 
 const (
-	MigratingReason MigrationReason = "Migrating"
+	MigratingReason        MigrationReason = "Migrating"
+	PendingMigratingReason MigrationReason = "Pending"
 )

api/core/v1alpha2/virtual_disk.go

@@ -81,7 +81,7 @@ type VirtualDiskStatus struct {
 	StorageClassName string `json:"storageClassName,omitempty"`
 
 	// Migration information.
-	MigrationInfo *VirtualDiskMigrationInfo `json:"migrationInfo,omitempty"`
+	MigrationInfo VirtualDiskMigrationInfo `json:"migrationInfo,omitempty"`
 	// Migration requests.
 	MigrationRequests []VirtualDiskMigrationRequest `json:"migrationRequest,omitempty"`
 }
@@ -225,9 +225,12 @@ type VirtualDiskMigrationInfo struct {
 	// Source PersistentVolumeClaim name.
 	SourcePVC string `json:"sourcePVC,omitempty"`
 	// Target PersistentVolumeClaim name.
-	TargetPVC string `json:"targetPVC,omitempty"`
-	Completed bool   `json:"completed"`
-	Failed    bool   `json:"failed"`
+	TargetPVC      string      `json:"targetPVC,omitempty"`
+	Completed      bool        `json:"completed"`
+	Failed         bool        `json:"failed"`
+	Message        string      `json:"message,omitempty"`
+	StartTimestamp metav1.Time `json:"startTimestamp,omitempty"`
+	EndTimestamp   metav1.Time `json:"endTimestamp,omitempty"`
 }
 
 type VirtualDiskMigrationRequest struct {
@@ -239,6 +242,7 @@ type VirtualDiskMigrationRequest struct {
 type VirtualDiskMigrationRequestAction string
 
 const (
-	VirtualDiskMigrationRequestActionMigrate VirtualDiskMigrationRequestAction = "migrate"
-	VirtualDiskMigrationRequestActionRevert  VirtualDiskMigrationRequestAction = "revert"
+	VirtualDiskMigrationRequestActionMigrate  VirtualDiskMigrationRequestAction = "migrate"
+	VirtualDiskMigrationRequestActionRevert   VirtualDiskMigrationRequestAction = "revert"
+	VirtualDiskMigrationRequestActionComplete VirtualDiskMigrationRequestAction = "complete"
 )

api/core/v1alpha2/zz_generated.deepcopy.go

@@ -371,11 +371,19 @@ spec:
                   properties:
                     completed:
                       type: boolean
+                    endTimestamp:
+                      format: date-time
+                      type: string
                     failed:
                       type: boolean
+                    message:
+                      type: string
                     sourcePVC:
                       description: Source PersistentVolumeClaim name.
                       type: string
+                    startTimestamp:
+                      format: date-time
+                      type: string
                     targetPVC:
                       description: Target PersistentVolumeClaim name.
                       type: string

crds/virtualdisks.yaml

@@ -0,0 +1,96 @@
+package volumemigrator
+
+import (
+	"context"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+
+	"github.com/deckhouse/virtualization-controller/pkg/common/patch"
+	"github.com/deckhouse/virtualization-controller/pkg/controller/conditions"
+	"github.com/deckhouse/virtualization/api/core/v1alpha2"
+	"github.com/deckhouse/virtualization/api/core/v1alpha2/vdcondition"
+)
+
+//go:generate go tool moq -rm -out mock.go . VirtualDiskMigrationService
+type VirtualDiskMigrationService interface {
+	Migrate(ctx context.Context, vd *v1alpha2.VirtualDisk, opts ...MigrationOption) error
+}
+
+type virtualDiskMigrationService struct {
+	client client.Client
+}
+
+func NewVirtualDiskMigrationService(client client.Client) VirtualDiskMigrationService {
+	return &virtualDiskMigrationService{
+		client: client,
+	}
+}
+
+type migrationOptions struct {
+	storageClassName *string
+}
+
+func newMigrationOptions(opts ...MigrationOption) *migrationOptions {
+	o := &migrationOptions{}
+	for _, opt := range opts {
+		opt(o)
+	}
+	return o
+}
+
+type MigrationOption func(o *migrationOptions)
+
+func WithStorageClassName(name string) MigrationOption {
+	return func(o *migrationOptions) {
+		o.storageClassName = &name
+	}
+}
+
+func (v *virtualDiskMigrationService) Migrate(ctx context.Context, vd *v1alpha2.VirtualDisk, opts ...MigrationOption) error {
+	if !v.requestNeeded(vd) {
+		return nil
+	}
+
+	options := newMigrationOptions(opts...)
+
+	req := v1alpha2.VirtualDiskMigrationRequest{
+		Action: v1alpha2.VirtualDiskMigrationRequestActionMigrate,
+	}
+	if options.storageClassName != nil {
+		req.StorageClassName = *options.storageClassName
+	}
+
+	requests := make([]v1alpha2.VirtualDiskMigrationRequest, 0, len(vd.Status.MigrationRequests)+1)
+	for _, r := range vd.Status.MigrationRequests {
+		requests = append(requests, *r.DeepCopy())
+	}
+	requests = append(requests, req)
+
+	patchBytes, err := patch.NewJSONPatch(patch.WithReplace("/status/migrationRequests", requests)).Bytes()
+	if err != nil {
+		return err
+	}
+
+	return v.client.Patch(ctx, vd, client.RawPatch(types.JSONPatchType, patchBytes))
+}
+
+func (v *virtualDiskMigrationService) requestNeeded(vd *v1alpha2.VirtualDisk) bool {
+	if vd == nil {
+		return false
+	}
+
+	cond, _ := conditions.GetCondition(vdcondition.MigrationType, vd.Status.Conditions)
+	if cond.Status == metav1.ConditionTrue {
+		return false
+	}
+
+	for _, req := range vd.Status.MigrationRequests {
+		if req.Action == v1alpha2.VirtualDiskMigrationRequestActionMigrate {
+			return false
+		}
+	}
+
+	return true
+}