2020-07-28 07:42:21 +00:00
|
|
|
package models
|
|
|
|
|
2021-03-25 16:26:21 +00:00
|
|
|
import (
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/caos/zitadel/internal/errors"
|
|
|
|
)
|
2020-07-28 07:42:21 +00:00
|
|
|
|
|
|
|
//SearchQuery is deprecated. Use SearchQueryFactory
|
|
|
|
type SearchQuery struct {
|
|
|
|
Limit uint64
|
|
|
|
Desc bool
|
|
|
|
Filters []*Filter
|
|
|
|
}
|
|
|
|
|
|
|
|
//NewSearchQuery is deprecated. Use SearchQueryFactory
|
|
|
|
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
|
|
|
|
return q.setFilter(NewFilter(Field_LatestSequence, sequence, sortOrder))
|
|
|
|
}
|
|
|
|
|
2020-12-18 15:47:45 +00:00
|
|
|
func (q *SearchQuery) SequenceBetween(from, to uint64) *SearchQuery {
|
|
|
|
q.setFilter(NewFilter(Field_LatestSequence, from, Operation_Greater))
|
|
|
|
q.setFilter(NewFilter(Field_LatestSequence, to, Operation_Less))
|
|
|
|
return q
|
|
|
|
}
|
|
|
|
|
2020-07-28 07:42:21 +00:00
|
|
|
func (q *SearchQuery) ResourceOwnerFilter(resourceOwner string) *SearchQuery {
|
|
|
|
return q.setFilter(NewFilter(Field_ResourceOwner, resourceOwner, Operation_Equals))
|
|
|
|
}
|
|
|
|
|
2022-03-15 06:19:02 +00:00
|
|
|
func (q *SearchQuery) TenantFilter(tenant string) *SearchQuery {
|
|
|
|
return q.setFilter(NewFilter(Field_Tenant, tenant, Operation_Equals))
|
|
|
|
}
|
|
|
|
|
2021-03-25 16:26:21 +00:00
|
|
|
func (q *SearchQuery) CreationDateNewerFilter(time time.Time) *SearchQuery {
|
|
|
|
return q.setFilter(NewFilter(Field_CreationDate, time, Operation_Greater))
|
|
|
|
}
|
|
|
|
|
2020-07-28 07:42:21 +00:00
|
|
|
func (q *SearchQuery) setFilter(filter *Filter) *SearchQuery {
|
|
|
|
for i, f := range q.Filters {
|
2020-12-21 17:42:34 +00:00
|
|
|
if f.field == filter.field && f.field != Field_LatestSequence {
|
2020-07-28 07:42:21 +00:00
|
|
|
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
|
|
|
|
}
|