zrok/controller/maintenance.go

package controller

import (
	"context"
	"fmt"
	"strings"
	"time"

	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"
)

type maintenanceAgent struct {
	ctx        context.Context
	frequency  time.Duration
	expiration time.Duration
}

func newMaintenanceAgent(ctx context.Context, frequency, expiration time.Duration) *maintenanceAgent {
	return &maintenanceAgent{
		ctx:        ctx,
		frequency:  frequency,
		expiration: expiration,
	}
}

func (ma *maintenanceAgent) run() {
	ticker := time.NewTicker(ma.frequency)
	for {
		select {
		case <-ma.ctx.Done():
			{
				logrus.Info("stopping maintenance loop...")
				ticker.Stop()
				return
			}
		case <-ticker.C:
			{
				if err := ma.deleteExpiredAccountRequests(); err != nil {
					logrus.Error(err)
				}
			}
		}
	}
}

func (ma *maintenanceAgent) deleteExpiredAccountRequests() error {
	tx, err := str.Begin()
	if err != nil {
		return err
	}
	defer func() { _ = tx.Rollback() }()

	expir := time.Now().UTC().Add(-ma.expiration)
	accountRequests, err := str.FindExpiredAccountRequests(expir, tx)
	if err != nil {
		return errors.Wrapf(err, "error finding expire account requests before %v", expir)
	}
	if len(accountRequests) > 0 {
		acctStrings := make([]string, len(accountRequests))
		ids := make([]int, len(accountRequests))
		for i, acct := range accountRequests {
			ids[i] = acct.Id
			acctStrings[i] = fmt.Sprintf("{%d:%s}", acct.Id, acct.Email)
		}
		logrus.Infof("starting deleting for expired account requests: %v", strings.Join(acctStrings, ","))

		if err := str.DeleteMultipleAccountRequests(ids, tx); err != nil {
			return errors.Wrapf(err, "error deleting expired account requests before %v", expir)
		}
	}

	if err := tx.Commit(); err != nil {
		return errors.Wrapf(err, "error committing expired acount requests deletion")
	}

	return nil
}