zitadel/internal/eventstore/models/search_query.go

package models

import (
	"github.com/caos/zitadel/internal/errors"
)

type SearchQuery struct {
	Limit   uint64
	Desc    bool
	Filters []*Filter
}

func NewSearchQuery() *SearchQuery {
	return &SearchQuery{
		Filters: make([]*Filter, 0, 4),
	}
}

func (q *SearchQuery) SetLimit(limit uint64) *SearchQuery {
	q.Limit = limit
	return q
}

func (q *SearchQuery) OrderDesc() *SearchQuery {
	q.Desc = true
	return q
}

func (q *SearchQuery) OrderAsc() *SearchQuery {
	q.Desc = false
	return q
}

func (q *SearchQuery) AggregateIDFilter(id string) *SearchQuery {
	return q.setFilter(NewFilter(Field_AggregateID, id, Operation_Equals))
}

func (q *SearchQuery) AggregateIDsFilter(ids ...string) *SearchQuery {
	return q.setFilter(NewFilter(Field_AggregateID, ids, Operation_In))
}

func (q *SearchQuery) AggregateTypeFilter(types ...AggregateType) *SearchQuery {
	return q.setFilter(NewFilter(Field_AggregateType, types, Operation_In))
}

func (q *SearchQuery) EventTypesFilter(types ...EventType) *SearchQuery {
	return q.setFilter(NewFilter(Field_EventType, types, Operation_In))
}

func (q *SearchQuery) LatestSequenceFilter(sequence uint64) *SearchQuery {
	if sequence == 0 {
		return q
	}
	sortOrder := Operation_Greater
	if q.Desc {
		sortOrder = Operation_Less
	}
	return q.setFilter(NewFilter(Field_LatestSequence, sequence, sortOrder))
}

func (q *SearchQuery) ResourceOwnerFilter(resourceOwner string) *SearchQuery {
	return q.setFilter(NewFilter(Field_ResourceOwner, resourceOwner, Operation_Equals))
}

func (q *SearchQuery) setFilter(filter *Filter) *SearchQuery {
	for i, f := range q.Filters {
		if f.field == filter.field {
			q.Filters[i] = filter
			return q
		}
	}
	q.Filters = append(q.Filters, filter)
	return q
}

func (q *SearchQuery) Validate() error {
	if q == nil {
		return errors.ThrowPreconditionFailed(nil, "MODEL-J5xQi", "search query is nil")
	}
	if len(q.Filters) == 0 {
		return errors.ThrowPreconditionFailed(nil, "MODEL-pF3DR", "no filters set")
	}
	for _, filter := range q.Filters {
		if err := filter.Validate(); err != nil {
			return err
		}
	}

	return nil
}