feat: block instances (#7129)

* docs: fix init description typos * feat: block instances using limits * translate * unit tests * fix translations * redirect /ui/login * fix http interceptor * cleanup * fix http interceptor * fix: delete cookies on gateway 200 * add integration tests * add command test * docs * fix integration tests * add bulk api and integration test * optimize bulk set limits * unit test bulk limits * fix broken link * fix assets middleware * fix broken link * validate instance id format * Update internal/eventstore/search_query.go Co-authored-by: Livio Spring <[email protected]> * remove support for owner bulk limit commands * project limits to instances * migrate instances projection * Revert "migrate instances projection" This reverts commit 214218732a56e6df823beac1972adfcf8beeded5. * join limits, remove owner * remove todo * use optional bool * normally validate instance ids * use 302 * cleanup * cleanup * Update internal/api/grpc/system/limits_converter.go Co-authored-by: Livio Spring <[email protected]> * remove owner * remove owner from reset --------- Co-authored-by: Livio Spring <[email protected]>
2026-11-07 22:55:12 +00:00 · 2024-01-17 10:16:48 +00:00
parent d9d376a275
commit ed0bc39ea4
80 changed files with 1609 additions and 438 deletions
@@ -0,0 +1,55 @@
+package command
+
+import (
+	"github.com/zitadel/zitadel/internal/eventstore"
+	"github.com/zitadel/zitadel/internal/repository/limits"
+)
+
+type limitsBulkWriteModel struct {
+	eventstore.WriteModel
+	writeModels       map[string]*limitsWriteModel
+	filterInstanceIDs []string
+}
+
+// newLimitsBulkWriteModel should be followed by limitsBulkWriteModel.addWriteModel before querying and reducing it.
+func newLimitsBulkWriteModel() *limitsBulkWriteModel {
+	return &limitsBulkWriteModel{
+		writeModels:       make(map[string]*limitsWriteModel),
+		filterInstanceIDs: make([]string, 0),
+	}
+}
+
+func (wm *limitsBulkWriteModel) addWriteModel(instanceID string) {
+	if _, ok := wm.writeModels[instanceID]; !ok {
+		wm.writeModels[instanceID] = newLimitsWriteModel(instanceID)
+	}
+	wm.filterInstanceIDs = append(wm.filterInstanceIDs, instanceID)
+}
+
+func (wm *limitsBulkWriteModel) Query() *eventstore.SearchQueryBuilder {
+	query := eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
+		InstanceIDs(wm.filterInstanceIDs).
+		AddQuery().
+		AggregateTypes(limits.AggregateType).
+		EventTypes(
+			limits.SetEventType,
+			limits.ResetEventType,
+		)
+
+	return query.Builder()
+}
+
+func (wm *limitsBulkWriteModel) Reduce() error {
+	for _, event := range wm.Events {
+		instanceID := event.Aggregate().InstanceID
+		limitsWm, ok := wm.writeModels[instanceID]
+		if !ok {
+			continue
+		}
+		limitsWm.AppendEvents(event)
+		if err := limitsWm.Reduce(); err != nil {
+			return err
+		}
+	}
+	return nil
+}