add

Signed-off-by: yaroslavborbat <[email protected]>

images/virtualization-artifact/cmd/virtualization-controller/main.go

@@ -51,6 +51,7 @@ import (
 	"github.com/deckhouse/virtualization-controller/pkg/controller/vmrestore"
 	"github.com/deckhouse/virtualization-controller/pkg/controller/vmsnapshot"
 	"github.com/deckhouse/virtualization-controller/pkg/logger"
+	"github.com/deckhouse/virtualization-controller/pkg/version"
 	"github.com/deckhouse/virtualization/api/client/kubeclient"
 	virtv2alpha1 "github.com/deckhouse/virtualization/api/core/v1alpha2"
 )
@@ -294,4 +295,5 @@ func main() {
 func printVersion(log *slog.Logger) {
 	log.Info(fmt.Sprintf("Go Version: %s", runtime.Version()))
 	log.Info(fmt.Sprintf("Go OS/Arch: %s/%s", runtime.GOOS, runtime.GOARCH))
+	log.Info(fmt.Sprintf("Edition: %s", version.GetEdition()))
 }

images/virtualization-artifact/pkg/controller/vmchange/comparator_pod_placement.go

@@ -44,7 +44,7 @@ func compareAffinity(current, desired *v1alpha2.VirtualMachineSpec) []FieldChang
 		currentValue,
 		desiredValue,
 		reflect.DeepEqual(current.Affinity, desired.Affinity),
-		ActionRestart,
+		placementAction,
 	)
 }
 
@@ -57,7 +57,7 @@ func compareNodeSelector(current, desired *v1alpha2.VirtualMachineSpec) []FieldC
 		currentValue,
 		desiredValue,
 		reflect.DeepEqual(current.NodeSelector, desired.NodeSelector),
-		ActionRestart,
+		placementAction,
 	)
 }
 

images/virtualization-artifact/pkg/controller/vmchange/pod_placement_ce.go

@@ -0,0 +1,6 @@
+//go:build !EE
+// +build !EE
+
+package vmchange
+
+const placementAction = ActionRestart

images/virtualization-artifact/pkg/controller/vmchange/pod_placement_ee.go

@@ -0,0 +1,6 @@
+//go:build EE
+// +build EE
+
+package vmchange
+
+const placementAction = ActionApplyImmediate

images/virtualization-artifact/pkg/version/edition.go

@@ -0,0 +1,6 @@
+//go:build !EE
+// +build !EE
+
+package version
+
+const edition = "CE"

images/virtualization-artifact/pkg/version/edition_ee.go

@@ -0,0 +1,6 @@
+//go:build EE
+// +build EE
+
+package version
+
+const edition = "EE"

images/virtualization-artifact/pkg/version/version.go

@@ -0,0 +1,5 @@
+package version
+
+func GetEdition() string {
+	return edition
+}

images/virtualization-artifact/werf.inc.yaml

@@ -32,5 +32,5 @@ shell:
   - export GOOS=linux
   - export CGO_ENABLED=0
   - export GOARCH=amd64
-  - go build -v -a -o virtualization-controller ./cmd/virtualization-controller
+  - go build -tag EE -v -a -o virtualization-controller ./cmd/virtualization-controller
   - go build -v -a -o virtualization-api ./cmd/virtualization-api