Avoid overwriting status changes

Author: ewoutp

pkg/apis/deployment/v1alpha/member_status_list_test.go

@@ -0,0 +1,61 @@
+//
+// DISCLAIMER
+//
+// Copyright 2018 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
+//
+// Author Ewout Prangsma
+//
+
+package v1alpha
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// TestMemberStatusList tests modifying a MemberStatusList.
+func TestMemberStatusList(t *testing.T) {
+	list := &MemberStatusList{}
+	m1 := MemberStatus{ID: "m1"}
+	m2 := MemberStatus{ID: "m2"}
+	m3 := MemberStatus{ID: "m3"}
+	assert.Equal(t, 0, len(*list))
+
+	assert.NoError(t, list.Add(m1))
+	assert.Equal(t, 1, len(*list))
+
+	assert.NoError(t, list.Add(m2))
+	assert.NoError(t, list.Add(m3))
+	assert.Equal(t, 3, len(*list))
+
+	assert.Error(t, list.Add(m2))
+	assert.Equal(t, 3, len(*list))
+
+	assert.NoError(t, list.RemoveByID(m3.ID))
+	assert.Equal(t, 2, len(*list))
+	assert.False(t, list.ContainsID(m3.ID))
+
+	m2.PodName = "foo"
+	assert.NoError(t, list.Update(m2))
+	assert.Equal(t, 2, len(*list))
+	assert.True(t, list.ContainsID(m2.ID))
+	x, found := list.ElementByPodName("foo")
+	assert.True(t, found)
+	assert.Equal(t, "foo", x.PodName)
+	assert.Equal(t, m2.ID, x.ID)
+}

pkg/deployment/cluster_scaling_integration.go

@@ -71,7 +71,7 @@ func (ci *clusterScalingIntegration) ListenForClusterEvents(stopCh <-chan struct
 		delay := time.Second * 2
 
 		// Is deployment in running state
-		if ci.depl.status.Phase == api.DeploymentPhaseRunning {
+		if ci.depl.GetPhase() == api.DeploymentPhaseRunning {
 			// Update cluster with our state
 			ctx := context.Background()
 			safeToAskCluster, err := ci.updateClusterServerCount(ctx)

pkg/deployment/context_impl.go

@@ -24,6 +24,8 @@ package deployment
 
 import (
 	"context"
+	"fmt"
+	"sync/atomic"
 
 	"github.com/arangodb/arangosync/client"
 	"github.com/arangodb/arangosync/tasks"
@@ -63,20 +65,31 @@ func (d *Deployment) GetNamespace() string {
 	return d.apiObject.GetNamespace()
 }
 
+// GetPhase returns the current phase of the deployment
+func (d *Deployment) GetPhase() api.DeploymentPhase {
+	return d.status.last.Phase
+}
+
 // GetSpec returns the current specification
 func (d *Deployment) GetSpec() api.DeploymentSpec {
 	return d.apiObject.Spec
 }
 
 // GetStatus returns the current status of the deployment
-func (d *Deployment) GetStatus() api.DeploymentStatus {
-	return d.status
+// together with the current version of that status.
+func (d *Deployment) GetStatus() (api.DeploymentStatus, int32) {
+	version := atomic.LoadInt32(&d.status.version)
+	return *d.status.last.DeepCopy(), version
 }
 
 // UpdateStatus replaces the status of the deployment with the given status and
 // updates the resources in k8s.
-func (d *Deployment) UpdateStatus(status api.DeploymentStatus, force ...bool) error {
-	d.status = status
+func (d *Deployment) UpdateStatus(status api.DeploymentStatus, lastVersion int32, force ...bool) error {
+	if !atomic.CompareAndSwapInt32(&d.status.version, lastVersion, lastVersion+1) {
+		// Status is obsolete
+		return maskAny(fmt.Errorf("Status conflict error. Expected version %d, got %d", lastVersion, d.status.version))
+	}
+	d.status.last = *status.DeepCopy()
 	if err := d.updateCRStatus(force...); err != nil {
 		return maskAny(err)
 	}
@@ -105,7 +118,7 @@ func (d *Deployment) GetServerClient(ctx context.Context, group api.ServerGroup,
 // GetAgencyClients returns a client connection for every agency member.
 // If the given predicate is not nil, only agents are included where the given predicate returns true.
 func (d *Deployment) GetAgencyClients(ctx context.Context, predicate func(id string) bool) ([]driver.Connection, error) {
-	agencyMembers := d.status.Members.Agents
+	agencyMembers := d.status.last.Members.Agents
 	result := make([]driver.Connection, 0, len(agencyMembers))
 	for _, m := range agencyMembers {
 		if predicate != nil && !predicate(m.ID) {
@@ -157,13 +170,14 @@ func (d *Deployment) GetSyncServerClient(ctx context.Context, group api.ServerGr
 // If ID is non-empty, it will be used, otherwise a new ID is created.
 func (d *Deployment) CreateMember(group api.ServerGroup, id string) error {
 	log := d.deps.Log
-	id, err := d.createMember(group, id, d.apiObject)
+	status, lastVersion := d.GetStatus()
+	id, err := createMember(log, &status, group, id, d.apiObject)
 	if err != nil {
 		log.Debug().Err(err).Str("group", group.AsRole()).Msg("Failed to create member")
 		return maskAny(err)
 	}
 	// Save added member
-	if err := d.updateCRStatus(); err != nil {
+	if err := d.UpdateStatus(status, lastVersion); err != nil {
 		log.Debug().Err(err).Msg("Updating CR status failed")
 		return maskAny(err)
 	}

pkg/deployment/deployment.go

@@ -83,9 +83,12 @@ const (
 // Deployment is the in process state of an ArangoDeployment.
 type Deployment struct {
 	apiObject *api.ArangoDeployment // API object
-	status    api.DeploymentStatus  // Internal status of the CR
-	config    Config
-	deps      Dependencies
+	status    struct {
+		version int32
+		last    api.DeploymentStatus // Internal status copy of the CR
+	}
+	config Config
+	deps   Dependencies
 
 	eventCh chan *deploymentEvent
 	stopCh  chan struct{}
@@ -112,20 +115,20 @@ func New(config Config, deps Dependencies, apiObject *api.ArangoDeployment) (*De
 	}
 	d := &Deployment{
 		apiObject:   apiObject,
-		status:      *(apiObject.Status.DeepCopy()),
 		config:      config,
 		deps:        deps,
 		eventCh:     make(chan *deploymentEvent, deploymentEventQueueSize),
 		stopCh:      make(chan struct{}),
 		eventsCli:   deps.KubeCli.Core().Events(apiObject.GetNamespace()),
 		clientCache: newClientCache(deps.KubeCli, apiObject),
 	}
+	d.status.last = *(apiObject.Status.DeepCopy())
 	d.reconciler = reconcile.NewReconciler(deps.Log, d)
 	d.resilience = resilience.NewResilience(deps.Log, d)
 	d.resources = resources.NewResources(deps.Log, d)
-	if d.status.AcceptedSpec == nil {
+	if d.status.last.AcceptedSpec == nil {
 		// We've validated the spec, so let's use it from now.
-		d.status.AcceptedSpec = apiObject.Spec.DeepCopy()
+		d.status.last.AcceptedSpec = apiObject.Spec.DeepCopy()
 	}
 
 	go d.run()
@@ -185,7 +188,7 @@ func (d *Deployment) send(ev *deploymentEvent) {
 func (d *Deployment) run() {
 	log := d.deps.Log
 
-	if d.status.Phase == api.DeploymentPhaseNone {
+	if d.GetPhase() == api.DeploymentPhaseNone {
 		// Create secrets
 		if err := d.resources.EnsureSecrets(); err != nil {
 			d.CreateEvent(k8sutil.NewErrorEvent("Failed to create secrets", err, d.GetAPIObject()))
@@ -211,8 +214,9 @@ func (d *Deployment) run() {
 			d.CreateEvent(k8sutil.NewErrorEvent("Failed to create pods", err, d.GetAPIObject()))
 		}
 
-		d.status.Phase = api.DeploymentPhaseRunning
-		if err := d.updateCRStatus(); err != nil {
+		status, lastVersion := d.GetStatus()
+		status.Phase = api.DeploymentPhaseRunning
+		if err := d.UpdateStatus(status, lastVersion); err != nil {
 			log.Warn().Err(err).Msg("update initial CR status failed")
 		}
 		log.Info().Msg("start running...")
@@ -277,13 +281,14 @@ func (d *Deployment) handleArangoDeploymentUpdatedEvent() error {
 	}
 
 	specBefore := d.apiObject.Spec
-	if d.status.AcceptedSpec != nil {
-		specBefore = *d.status.AcceptedSpec
+	status := d.status.last
+	if d.status.last.AcceptedSpec != nil {
+		specBefore = *status.AcceptedSpec.DeepCopy()
 	}
 	newAPIObject := current.DeepCopy()
 	newAPIObject.Spec.SetDefaultsFrom(specBefore)
 	newAPIObject.Spec.SetDefaults(d.apiObject.GetName())
-	newAPIObject.Status = d.status
+	newAPIObject.Status = status
 	resetFields := specBefore.ResetImmutableFields(&newAPIObject.Spec)
 	if len(resetFields) > 0 {
 		log.Debug().Strs("fields", resetFields).Msg("Found modified immutable fields")
@@ -309,9 +314,12 @@ func (d *Deployment) handleArangoDeploymentUpdatedEvent() error {
 		return maskAny(fmt.Errorf("failed to update ArangoDeployment spec: %v", err))
 	}
 	// Save updated accepted spec
-	d.status.AcceptedSpec = newAPIObject.Spec.DeepCopy()
-	if err := d.updateCRStatus(); err != nil {
-		return maskAny(fmt.Errorf("failed to update ArangoDeployment status: %v", err))
+	{
+		status, lastVersion := d.GetStatus()
+		status.AcceptedSpec = newAPIObject.Spec.DeepCopy()
+		if err := d.UpdateStatus(status, lastVersion); err != nil {
+			return maskAny(fmt.Errorf("failed to update ArangoDeployment status: %v", err))
+		}
 	}
 
 	// Notify cluster of desired server count
@@ -351,7 +359,7 @@ func (d *Deployment) updateCRStatus(force ...bool) error {
 	attempt := 0
 	for {
 		attempt++
-		update.Status = d.status
+		update.Status = d.status.last
 		if update.GetDeletionTimestamp() == nil {
 			ensureFinalizers(update)
 		}
@@ -388,7 +396,7 @@ func (d *Deployment) updateCRSpec(newSpec api.DeploymentSpec) error {
 	for {
 		attempt++
 		update.Spec = newSpec
-		update.Status = d.status
+		update.Status = d.status.last
 		ns := d.apiObject.GetNamespace()
 		newAPIObject, err := d.deps.DatabaseCRCli.DatabaseV1alpha().ArangoDeployments(ns).Update(update)
 		if err == nil {
@@ -417,7 +425,7 @@ func (d *Deployment) updateCRSpec(newSpec api.DeploymentSpec) error {
 // Since there is no recovery from a failed deployment, use with care!
 func (d *Deployment) failOnError(err error, msg string) {
 	log.Error().Err(err).Msg(msg)
-	d.status.Reason = err.Error()
+	d.status.last.Reason = err.Error()
 	d.reportFailedStatus()
 }
 
@@ -428,7 +436,7 @@ func (d *Deployment) reportFailedStatus() {
 	log.Info().Msg("deployment failed. Reporting failed reason...")
 
 	op := func() error {
-		d.status.Phase = api.DeploymentPhaseFailed
+		d.status.last.Phase = api.DeploymentPhaseFailed
 		err := d.updateCRStatus()
 		if err == nil || k8sutil.IsNotFound(err) {
 			// Status has been updated

pkg/deployment/deployment_inspector.go

@@ -60,7 +60,7 @@ func (d *Deployment) inspectDeployment(lastInterval time.Duration) time.Duration
 		}
 	} else {
 		// Is the deployment in failed state, if so, give up.
-		if d.status.Phase == api.DeploymentPhaseFailed {
+		if d.GetPhase() == api.DeploymentPhaseFailed {
 			log.Debug().Msg("Deployment is in Failed state.")
 			return nextInterval
 		}
@@ -72,7 +72,8 @@ func (d *Deployment) inspectDeployment(lastInterval time.Duration) time.Duration
 		}
 
 		// Is the deployment in a good state?
-		if d.status.Conditions.IsTrue(api.ConditionTypeSecretsChanged) {
+		status, _ := d.GetStatus()
+		if status.Conditions.IsTrue(api.ConditionTypeSecretsChanged) {
 			log.Debug().Msg("Condition SecretsChanged is true. Revert secrets before we can continue")
 			return nextInterval
 		}

pkg/deployment/images.go

@@ -55,15 +55,15 @@ type imagesBuilder struct {
 // ensureImages creates pods needed to detect ImageID for specified images.
 // Returns: retrySoon, error
 func (d *Deployment) ensureImages(apiObject *api.ArangoDeployment) (bool, error) {
+	status, lastVersion := d.GetStatus()
 	ib := imagesBuilder{
 		APIObject: apiObject,
 		Spec:      apiObject.Spec,
-		Status:    d.status,
+		Status:    status,
 		Log:       d.deps.Log,
 		KubeCli:   d.deps.KubeCli,
 		UpdateCRStatus: func(status api.DeploymentStatus) error {
-			d.status = status
-			if err := d.updateCRStatus(); err != nil {
+			if err := d.UpdateStatus(status, lastVersion); err != nil {
 				return maskAny(err)
 			}
 			return nil