mirror of
https://github.com/andwati/zitadel.git
synced 2026-10-31 11:32:15 +00:00
feat: add quotas (#4779)
adds possibilities to cap authenticated requests and execution seconds of actions on a defined intervall
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
package errors
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
var (
|
||||
_ ResourceExhausted = (*ResourceExhaustedError)(nil)
|
||||
_ Error = (*ResourceExhaustedError)(nil)
|
||||
)
|
||||
|
||||
type ResourceExhausted interface {
|
||||
error
|
||||
IsResourceExhausted()
|
||||
}
|
||||
|
||||
type ResourceExhaustedError struct {
|
||||
*CaosError
|
||||
}
|
||||
|
||||
func ThrowResourceExhausted(parent error, id, message string) error {
|
||||
return &ResourceExhaustedError{CreateCaosError(parent, id, message)}
|
||||
}
|
||||
|
||||
func ThrowResourceExhaustedf(parent error, id, format string, a ...interface{}) error {
|
||||
return ThrowResourceExhausted(parent, id, fmt.Sprintf(format, a...))
|
||||
}
|
||||
|
||||
func (err *ResourceExhaustedError) IsResourceExhausted() {}
|
||||
|
||||
func IsResourceExhausted(err error) bool {
|
||||
//nolint:errorlint
|
||||
_, ok := err.(ResourceExhausted)
|
||||
return ok
|
||||
}
|
||||
|
||||
func (err *ResourceExhaustedError) Is(target error) bool {
|
||||
//nolint:errorlint
|
||||
t, ok := target.(*ResourceExhaustedError)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
return err.CaosError.Is(t.CaosError)
|
||||
}
|
||||
|
||||
func (err *ResourceExhaustedError) Unwrap() error {
|
||||
return err.CaosError
|
||||
}
|
||||
@@ -0,0 +1,34 @@
|
||||
package errors_test
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
|
||||
caos_errs "github.com/zitadel/zitadel/internal/errors"
|
||||
)
|
||||
|
||||
func TestResourceExhaustedError(t *testing.T) {
|
||||
var err interface{} = new(caos_errs.ResourceExhaustedError)
|
||||
_, ok := err.(caos_errs.ResourceExhausted)
|
||||
assert.True(t, ok)
|
||||
}
|
||||
|
||||
func TestThrowResourceExhaustedf(t *testing.T) {
|
||||
err := caos_errs.ThrowResourceExhaustedf(nil, "id", "msg")
|
||||
// TODO: refactor errors package
|
||||
//nolint:errorlint
|
||||
_, ok := err.(*caos_errs.ResourceExhaustedError)
|
||||
assert.True(t, ok)
|
||||
}
|
||||
|
||||
func TestIsResourceExhausted(t *testing.T) {
|
||||
err := caos_errs.ThrowResourceExhausted(nil, "id", "msg")
|
||||
ok := caos_errs.IsResourceExhausted(err)
|
||||
assert.True(t, ok)
|
||||
|
||||
err = errors.New("I am found!")
|
||||
ok = caos_errs.IsResourceExhausted(err)
|
||||
assert.False(t, ok)
|
||||
}
|
||||
Reference in New Issue
Block a user