[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)
 }