[Feature] [ML] Extension STS update propagation (#1548)

Author: nikita-vanyasin

docs/api/ArangoMLExtension.V1Alpha1.md

@@ -1427,6 +1427,18 @@ UID keeps the information about object UID
 
 ***
 
+### .status.reconciliation.serviceChecksum
+
+Type: `string` <sup>[\[ref\]](https://github.com/arangodb/kube-arangodb/blob/1.2.35/pkg/apis/ml/v1alpha1/extension_status_reconciliation.go#L25)</sup>
+
+***
+
+### .status.reconciliation.statefulSetChecksum
+
+Type: `string` <sup>[\[ref\]](https://github.com/arangodb/kube-arangodb/blob/1.2.35/pkg/apis/ml/v1alpha1/extension_status_reconciliation.go#L24)</sup>
+
+***
+
 ### .status.serviceAccount.cluster.binding.name
 
 Type: `string` <sup>[\[ref\]](https://github.com/arangodb/kube-arangodb/blob/1.2.35/pkg/apis/shared/v1/object.go#L46)</sup>

pkg/apis/deployment/v1/plan.go

@@ -82,9 +82,9 @@ type Action struct {
 	MemberID string `json:"memberID,omitempty"`
 	// Group involved in this action
 	Group ServerGroup `json:"group,omitempty"`
-	// CreationTime is set the when the action is created.
+	// CreationTime is set when the action is created.
 	CreationTime meta.Time `json:"creationTime"`
-	// StartTime is set the when the action has been started, but needs to wait to be finished.
+	// StartTime is set when the action has been started, but needs to wait to be finished.
 	StartTime *meta.Time `json:"startTime,omitempty"`
 	// Reason for this action
 	Reason string `json:"reason,omitempty"`
@@ -98,7 +98,7 @@ type Action struct {
 	TaskID types.UID `json:"taskID,omitempty"`
 	// Architecture of the member involved in this action (if any)
 	Architecture ArangoDeploymentArchitectureType `json:"arch,omitempty"`
-	// Progress describes what is a status of the current action.
+	// Progress describes the status of the current action.
 	Progress string `json:"progress,omitempty"`
 }
 

pkg/apis/deployment/v2alpha1/plan.go

@@ -82,9 +82,9 @@ type Action struct {
 	MemberID string `json:"memberID,omitempty"`
 	// Group involved in this action
 	Group ServerGroup `json:"group,omitempty"`
-	// CreationTime is set the when the action is created.
+	// CreationTime is set when the action is created.
 	CreationTime meta.Time `json:"creationTime"`
-	// StartTime is set the when the action has been started, but needs to wait to be finished.
+	// StartTime is set when the action has been started, but needs to wait to be finished.
 	StartTime *meta.Time `json:"startTime,omitempty"`
 	// Reason for this action
 	Reason string `json:"reason,omitempty"`
@@ -98,7 +98,7 @@ type Action struct {
 	TaskID types.UID `json:"taskID,omitempty"`
 	// Architecture of the member involved in this action (if any)
 	Architecture ArangoDeploymentArchitectureType `json:"arch,omitempty"`
-	// Progress describes what is a status of the current action.
+	// Progress describes the status of the current action.
 	Progress string `json:"progress,omitempty"`
 }
 

pkg/apis/ml/v1alpha1/extension_spec_deployment.go

@@ -165,7 +165,10 @@ func (s *ArangoMLExtensionSpecDeployment) Validate() error {
 				continue
 			}
 
-			if usedPorts.IndexOf(port) >= 0 {
+			duplicateCount := usedPorts.Count(func(i int32) bool {
+				return i == port
+			})
+			if duplicateCount > 0 {
 				errs = append(errs, shared.PrefixResourceErrors(prefix, errors.Newf("port %d already specified for other component", port)))
 			} else {
 				usedPorts.Append(port)

pkg/apis/ml/v1alpha1/extension_status.go

@@ -38,4 +38,7 @@ type ArangoMLExtensionStatus struct {
 
 	// ArangoDB keeps the information about local arangodb reference
 	ArangoDB *ArangoMLExtensionStatusArangoDBRef `json:"arangoDB,omitempty"`
+
+	// Reconciliation keeps the information about reconciliation process. For internal use.
+	Reconciliation *ArangoMLExtensionStatusReconciliation `json:"reconciliation"`
 }

pkg/apis/ml/v1alpha1/extension_status_reconciliation.go

@@ -0,0 +1,40 @@
+//
+// DISCLAIMER
+//
+// Copyright 2023 ArangoDB GmbH, Cologne, Germany
+//
+// 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.
+//
+// Copyright holder is ArangoDB GmbH, Cologne, Germany
+//
+
+package v1alpha1
+
+type ArangoMLExtensionStatusReconciliation struct {
+	StatefulSetChecksum string `json:"statefulSetChecksum,omitempty"`
+	ServiceChecksum     string `json:"serviceChecksum,omitempty"`
+}
+
+func (r *ArangoMLExtensionStatusReconciliation) GetStatefulSetChecksum() string {
+	if r == nil {
+		return ""
+	}
+	return r.StatefulSetChecksum
+}
+
+func (r *ArangoMLExtensionStatusReconciliation) GetServiceChecksum() string {
+	if r == nil {
+		return ""
+	}
+	return r.ServiceChecksum
+}

pkg/apis/ml/v1alpha1/zz_generated.deepcopy.go

@@ -123,7 +123,7 @@ func P2[T interface{}, P1, P2 interface{}](
 	}
 
 	if m, p, err := Evaluate(actionBuilder, evaluatorsFunc...); err != nil {
-		log.Err(err).Error("Error while getting diff")
+		log.Err(err).Error("Error while running evaluators")
 		return SkippedRotation, nil, err
 	} else {
 		mode = mode.And(m)

pkg/util/compare/compare.go

@@ -26,13 +26,7 @@ import (
 
 type Template[T interface{}] interface {
 	GetTemplate() *T
-	SetTemplate(*T)
-
-	GetTemplateChecksum() string
-	SetTemplateChecksum(string)
-
 	GetChecksum() string
-	SetChecksum(string)
 }
 
 type Checksum[T interface{}] func(in *T) (string, error)

pkg/util/compare/interfaces.go

@@ -18,12 +18,24 @@
 // Copyright holder is ArangoDB GmbH, Cologne, Germany
 //
 
-package util
+package k8s
 
-// Ter implements a ternary operation: return cond ? a : b;
-func Ter[T any](cond bool, a T, b T) T {
-	if cond {
-		return a
+import (
+	core "k8s.io/api/core/v1"
+
+	"github.com/arangodb/kube-arangodb/pkg/util"
+)
+
+func ChecksumService(s *core.Service) (string, error) {
+	return checksumServiceSpec(&s.Spec)
+}
+
+func checksumServiceSpec(s *core.ServiceSpec) (string, error) {
+	parts := map[string]interface{}{
+		"type":     s.Type,
+		"ports":    s.Ports,
+		"selector": s.Selector,
+		// add here more fields when needed
 	}
-	return b
+	return util.SHA256FromJSON(parts)
 }