zitadel/internal/query/quota_notifications.go

package query

import (
	"context"
	"math"
	"time"

	"github.com/zitadel/zitadel/internal/eventstore"
	"github.com/zitadel/zitadel/internal/repository/quota"
)

func (q *Queries) GetDueQuotaNotifications(ctx context.Context, config *quota.AddedEvent, periodStart time.Time, usedAbs uint64) ([]*quota.NotificationDueEvent, error) {
	if len(config.Notifications) == 0 {
		return nil, nil
	}

	aggregate := config.Aggregate()
	wm, err := q.getQuotaNotificationsReadModel(ctx, aggregate, periodStart)
	if err != nil {
		return nil, err
	}

	usedRel := uint16(math.Floor(float64(usedAbs*100) / float64(config.Amount)))

	var dueNotifications []*quota.NotificationDueEvent
	for _, notification := range config.Notifications {
		if notification.Percent > usedRel {
			continue
		}

		threshold := notification.Percent
		if notification.Repeat {
			threshold = uint16(math.Max(1, math.Floor(float64(usedRel)/float64(notification.Percent)))) * notification.Percent
		}

		if wm.latestDueThresholds[notification.ID] < threshold {
			dueNotifications = append(
				dueNotifications,
				quota.NewNotificationDueEvent(
					ctx,
					&aggregate,
					config.Unit,
					notification.ID,
					notification.CallURL,
					periodStart,
					threshold,
					usedAbs,
				),
			)
		}
	}

	return dueNotifications, nil
}

func (q *Queries) getQuotaNotificationsReadModel(ctx context.Context, aggregate eventstore.Aggregate, periodStart time.Time) (*quotaNotificationsReadModel, error) {
	wm := newQuotaNotificationsReadModel(aggregate.ID, aggregate.InstanceID, aggregate.ResourceOwner, periodStart)
	return wm, q.eventstore.FilterToQueryReducer(ctx, wm)
}