mirror of
https://github.com/openziti/zrok.git
synced 2024-12-04 22:11:13 +01:00
145 lines
3.9 KiB
Go
145 lines
3.9 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/openziti/zrok/controller/config"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type maintenanceRegistrationAgent struct {
|
|
cfg *config.RegistrationMaintenanceConfig
|
|
ctx context.Context
|
|
}
|
|
|
|
func newRegistrationMaintenanceAgent(ctx context.Context, cfg *config.RegistrationMaintenanceConfig) *maintenanceRegistrationAgent {
|
|
return &maintenanceRegistrationAgent{
|
|
cfg: cfg,
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
func (ma *maintenanceRegistrationAgent) run() {
|
|
logrus.Infof("started")
|
|
defer logrus.Info("exited")
|
|
|
|
ticker := time.NewTicker(ma.cfg.CheckFrequency)
|
|
for {
|
|
select {
|
|
case <-ma.ctx.Done():
|
|
{
|
|
ticker.Stop()
|
|
return
|
|
}
|
|
case <-ticker.C:
|
|
{
|
|
if err := ma.deleteExpiredAccountRequests(); err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ma *maintenanceRegistrationAgent) deleteExpiredAccountRequests() error {
|
|
tx, err := str.Begin()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
timeout := time.Now().UTC().Add(-ma.cfg.ExpirationTimeout)
|
|
accountRequests, err := str.FindExpiredAccountRequests(timeout, ma.cfg.BatchLimit, tx)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "error finding expire account requests before %v", timeout)
|
|
}
|
|
if len(accountRequests) > 0 {
|
|
logrus.Infof("found %d expired account requests to remove", len(accountRequests))
|
|
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("deleting 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", timeout)
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return errors.Wrapf(err, "error committing expired acount requests deletion")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type maintenanceResetPasswordAgent struct {
|
|
cfg *config.ResetPasswordMaintenanceConfig
|
|
ctx context.Context
|
|
}
|
|
|
|
func newMaintenanceResetPasswordAgent(ctx context.Context, cfg *config.ResetPasswordMaintenanceConfig) *maintenanceResetPasswordAgent {
|
|
return &maintenanceResetPasswordAgent{
|
|
cfg: cfg,
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
func (ma *maintenanceResetPasswordAgent) run() {
|
|
logrus.Infof("started")
|
|
defer logrus.Info("exited")
|
|
|
|
ticker := time.NewTicker(ma.cfg.CheckFrequency)
|
|
for {
|
|
select {
|
|
case <-ma.ctx.Done():
|
|
{
|
|
ticker.Stop()
|
|
return
|
|
}
|
|
case <-ticker.C:
|
|
{
|
|
if err := ma.deleteExpiredForgetPasswordRequests(); err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
func (ma *maintenanceResetPasswordAgent) deleteExpiredForgetPasswordRequests() error {
|
|
tx, err := str.Begin()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
timeout := time.Now().UTC().Add(-ma.cfg.ExpirationTimeout)
|
|
passwordResetRequests, err := str.FindExpiredPasswordResetRequests(timeout, ma.cfg.BatchLimit, tx)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "error finding expired password reset requests before %v", timeout)
|
|
}
|
|
if len(passwordResetRequests) > 0 {
|
|
logrus.Infof("found %d expired password reset requests to remove", len(passwordResetRequests))
|
|
acctStrings := make([]string, len(passwordResetRequests))
|
|
ids := make([]int, len(passwordResetRequests))
|
|
for i, acct := range passwordResetRequests {
|
|
ids[i] = acct.Id
|
|
acctStrings[i] = fmt.Sprintf("{id:%d}", acct.Id)
|
|
}
|
|
|
|
logrus.Infof("deleting expired password reset requests: %v", strings.Join(acctStrings, ","))
|
|
if err := str.DeleteMultiplePasswordResetRequests(ids, tx); err != nil {
|
|
return errors.Wrapf(err, "error deleting expired password reset requests before %v", timeout)
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return errors.Wrapf(err, "error committing expired password reset requests deletion")
|
|
}
|
|
}
|
|
return nil
|
|
}
|