mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-14 03:54:21 +00:00
681541f41b
adds possibilities to cap authenticated requests and execution seconds of actions on a defined intervall
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package command
|
|
|
|
import (
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
|
"github.com/zitadel/zitadel/internal/repository/quota"
|
|
)
|
|
|
|
type quotaWriteModel struct {
|
|
eventstore.WriteModel
|
|
unit quota.Unit
|
|
active bool
|
|
config *quota.AddedEvent
|
|
}
|
|
|
|
// newQuotaWriteModel aggregateId is filled by reducing unit matching events
|
|
func newQuotaWriteModel(instanceId, resourceOwner string, unit quota.Unit) *quotaWriteModel {
|
|
return "aWriteModel{
|
|
WriteModel: eventstore.WriteModel{
|
|
InstanceID: instanceId,
|
|
ResourceOwner: resourceOwner,
|
|
},
|
|
unit: unit,
|
|
}
|
|
}
|
|
|
|
func (wm *quotaWriteModel) Query() *eventstore.SearchQueryBuilder {
|
|
query := eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
|
|
ResourceOwner(wm.ResourceOwner).
|
|
AddQuery().
|
|
InstanceID(wm.InstanceID).
|
|
AggregateTypes(quota.AggregateType).
|
|
EventTypes(
|
|
quota.AddedEventType,
|
|
quota.RemovedEventType,
|
|
).EventData(map[string]interface{}{"unit": wm.unit})
|
|
|
|
return query.Builder()
|
|
}
|
|
|
|
func (wm *quotaWriteModel) Reduce() error {
|
|
for _, event := range wm.Events {
|
|
switch e := event.(type) {
|
|
case *quota.AddedEvent:
|
|
wm.AggregateID = e.Aggregate().ID
|
|
wm.active = true
|
|
wm.config = e
|
|
case *quota.RemovedEvent:
|
|
wm.AggregateID = e.Aggregate().ID
|
|
wm.active = false
|
|
wm.config = nil
|
|
}
|
|
}
|
|
return wm.WriteModel.Reduce()
|
|
}
|