statusupdate.go

// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0

package provider

import (
	"github.com/gardener/controller-manager-library/pkg/logger"
	"github.com/gardener/controller-manager-library/pkg/resources"
	api "github.com/gardener/external-dns-management/pkg/apis/dns/v1alpha1"
)

type FinalizerHandler interface {
	SetFinalizer(name resources.Object) error
	RemoveFinalizer(name resources.Object) error
}

type StatusUpdate struct {
	*Entry
	logger   logger.LogContext
	delete   bool
	done     bool
	fhandler FinalizerHandler
}

func NewStatusUpdate(logger logger.LogContext, e *Entry, f FinalizerHandler) DoneHandler {
	// logger.Infof("request update for %s (delete=%t)", e.DNSName(), e.IsDeleting())
	return &StatusUpdate{Entry: e, logger: logger, delete: e.IsDeleting(), fhandler: f}
}

func (this *StatusUpdate) SetInvalid(err error) {
	if !this.done {
		this.done = true
		this.modified = false
		if err := this.fhandler.RemoveFinalizer(this.Entry.object); err != nil {
			this.logger.Errorf("cannot remove finalizer: %s", err)
		}
		_, err := this.UpdateStatus(this.logger, api.STATE_INVALID, err.Error())
		if err != nil {
			this.logger.Errorf("cannot update: %s", err)
		}
	}
}

func (this *StatusUpdate) Failed(err error) {
	if !this.done {
		this.done = true
		this.modified = false
		newState := api.STATE_ERROR
		if this.Entry.status.State != api.STATE_READY && this.Entry.status.State != api.STATE_STALE {
			if err2 := this.fhandler.RemoveFinalizer(this.Entry.Object()); err2 != nil {
				this.logger.Errorf("cannot remove finalizer: %s", err2)
			}
		} else {
			newState = api.STATE_STALE
		}
		_, err := this.UpdateStatus(this.logger, newState, err.Error())
		if err != nil {
			this.logger.Errorf("cannot update: %s", err)
		}
	}
}

func (this *StatusUpdate) Succeeded() {
	if !this.done {
		this.done = true
		this.modified = false
		if this.delete {
			this.logger.Infof("removing finalizer for deleted entry %s", this.ZonedDNSName())
			if err2 := this.fhandler.RemoveFinalizer(this.Entry.Object()); err2 != nil {
				this.logger.Errorf("cannot remove finalizer: %s", err2)
			}
		} else {
			this.Entry.activezone = this.ZoneId()
			if err2 := this.fhandler.SetFinalizer(this.Entry.Object()); err2 != nil {
				this.logger.Errorf("cannot set finalizer: %s", err2)
			}
			_, err := this.UpdateStatus(this.logger, api.STATE_READY, "dns entry active")
			if err != nil {
				this.logger.Errorf("cannot update: %s", err)
			}
		}
	}
}

func (this *StatusUpdate) Throttled() {
	_, err := this.UpdateState(this.logger, api.STATE_PENDING, MSG_THROTTLING)
	if err != nil {
		this.logger.Errorf("cannot update: %s", err)
	}
}