mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-12 19:14:23 +00:00
e38abdcdf3
Queries the data in the storage layser at the timestamp when the call hit the API layer
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package query
|
|
|
|
import (
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
|
"github.com/zitadel/zitadel/internal/repository/quota"
|
|
)
|
|
|
|
type quotaReadModel struct {
|
|
eventstore.ReadModel
|
|
unit quota.Unit
|
|
active bool
|
|
config *quota.AddedEvent
|
|
}
|
|
|
|
// newQuotaReadModel aggregateId is filled by reducing unit matching events
|
|
func newQuotaReadModel(instanceId, resourceOwner string, unit quota.Unit) *quotaReadModel {
|
|
return "aReadModel{
|
|
ReadModel: eventstore.ReadModel{
|
|
InstanceID: instanceId,
|
|
ResourceOwner: resourceOwner,
|
|
},
|
|
unit: unit,
|
|
}
|
|
}
|
|
|
|
func (rm *quotaReadModel) Query() *eventstore.SearchQueryBuilder {
|
|
query := eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
|
|
ResourceOwner(rm.ResourceOwner).
|
|
AllowTimeTravel().
|
|
AddQuery().
|
|
InstanceID(rm.InstanceID).
|
|
AggregateTypes(quota.AggregateType).
|
|
EventTypes(
|
|
quota.AddedEventType,
|
|
quota.RemovedEventType,
|
|
).EventData(map[string]interface{}{"unit": rm.unit})
|
|
|
|
return query.Builder()
|
|
}
|
|
|
|
func (rm *quotaReadModel) Reduce() error {
|
|
for _, event := range rm.Events {
|
|
switch e := event.(type) {
|
|
case *quota.AddedEvent:
|
|
rm.AggregateID = e.Aggregate().ID
|
|
rm.active = true
|
|
rm.config = e
|
|
case *quota.RemovedEvent:
|
|
rm.AggregateID = e.Aggregate().ID
|
|
rm.active = false
|
|
rm.config = nil
|
|
}
|
|
}
|
|
return rm.ReadModel.Reduce()
|
|
}
|