TheArchive/zitadel
1
0
Fork 0
mirror of https://github.com/zitadel/zitadel.git synced 2025-08-11 19:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: push timeout (#4882) (#4885)

Browse Source 
* push with timeout

* test: config for eventstore

(cherry picked from commit b9156da76d)

Co-authored-by: Silvan <silvan.reusser@gmail.com>
This commit is contained in:
Livio Spring
2022-12-15 10:40:13 +01:00
committed by GitHub
parent 3e52beaf89
commit d21bb902f1
11 changed files with 88 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
internal/eventstore/config.go
View File

@@ -2,10 +2,24 @@ package eventstore
import (
"database/sql"
"time"
"github.com/zitadel/zitadel/internal/eventstore/repository"
z_sql "github.com/zitadel/zitadel/internal/eventstore/repository/sql"
)
func Start(sqlClient *sql.DB) (*Eventstore, error) {
return NewEventstore(z_sql.NewCRDB(sqlClient)), nil
type Config struct {
PushTimeout time.Duration
Client *sql.DB
repo repository.Repository
}
func TestConfig(repo repository.Repository) *Config {
return &Config{repo: repo}
}
func Start(config *Config) (*Eventstore, error) {
config.repo = z_sql.NewCRDB(config.Client)
return NewEventstore(config), nil
}

14
internal/eventstore/eventstore.go
View File

@@ -6,6 +6,7 @@ import (
"encoding/json"
"reflect"
"sync"
"time"
"github.com/zitadel/zitadel/internal/api/authz"
"github.com/zitadel/zitadel/internal/errors"
@@ -18,17 +19,19 @@ type Eventstore struct {
repo repository.Repository
interceptorMutex sync.Mutex
eventInterceptors map[EventType]eventTypeInterceptors
PushTimeout time.Duration
}
type eventTypeInterceptors struct {
eventMapper func(*repository.Event) (Event, error)
}
func NewEventstore(repo repository.Repository) *Eventstore {
func NewEventstore(config *Config) *Eventstore {
return &Eventstore{
repo: repo,
repo: config.repo,
eventInterceptors: map[EventType]eventTypeInterceptors{},
interceptorMutex: sync.Mutex{},
PushTimeout: config.PushTimeout,
}
}
@@ -45,6 +48,13 @@ func (es *Eventstore) Push(ctx context.Context, cmds ...Command) ([]Event, error
if err != nil {
return nil, err
}
if es.PushTimeout > 0 {
var cancel func()
ctx, cancel = context.WithTimeout(ctx, es.PushTimeout)
defer cancel()
}
err = es.repo.Push(ctx, events, constraints...)
if err != nil {
return nil, err

8
internal/eventstore/example_test.go
View File

@@ -9,7 +9,6 @@ import (
"github.com/zitadel/zitadel/internal/api/authz"
"github.com/zitadel/zitadel/internal/eventstore"
"github.com/zitadel/zitadel/internal/eventstore/repository"
"github.com/zitadel/zitadel/internal/eventstore/repository/sql"
)
// ------------------------------------------------------------
@@ -287,7 +286,12 @@ func (rm *UserReadModel) Reduce() error {
// ------------------------------------------------------------
func TestUserReadModel(t *testing.T) {
es := eventstore.NewEventstore(sql.NewCRDB(testCRDBClient))
es, err := eventstore.Start(&eventstore.Config{Client: testCRDBClient})
if err != nil {
t.Errorf("unable to start eventstore: %v", err)
t.FailNow()
}
// es := eventstore.NewEventstore(&eventstore.Config{re})
es.RegisterFilterEventMapper(UserAddedEventMapper()).
RegisterFilterEventMapper(UserFirstNameChangedMapper()).
RegisterFilterEventMapper(UserPasswordCheckedMapper()).

33
internal/eventstore/handler/crdb/handler_stmt_test.go
View File

@@ -269,10 +269,11 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "fetch previous fails",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).
ExpectFilterEventsError(errFilter),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
args: args{
@@ -301,9 +302,10 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "no successful stmts",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
args: args{
@@ -339,9 +341,10 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "update current sequence fails",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t),
),
),
aggregates: []eventstore.AggregateType{"agg"},
},
args: args{
@@ -381,9 +384,10 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "commit fails",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t),
),
),
aggregates: []eventstore.AggregateType{"agg"},
},
args: args{
@@ -423,9 +427,10 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "correct",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
args: args{
@@ -455,9 +460,10 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "fetch previous stmts no additional stmts",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
args: args{
@@ -486,7 +492,7 @@ func TestStatementHandler_Update(t *testing.T) {
{
name: "fetch previous stmts additional events",
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
AggregateType: "testAgg",
@@ -496,6 +502,7 @@ func TestStatementHandler_Update(t *testing.T) {
},
),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
args: args{
@@ -617,9 +624,10 @@ func TestProjectionHandler_fetchPreviousStmts(t *testing.T) {
stmtSeq: 6,
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEventsError(errFilter),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
want: want{
@@ -642,9 +650,10 @@ func TestProjectionHandler_fetchPreviousStmts(t *testing.T) {
stmtSeq: 6,
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
want: want{
@@ -666,7 +675,7 @@ func TestProjectionHandler_fetchPreviousStmts(t *testing.T) {
stmtSeq: 10,
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
ID: "id",
@@ -690,6 +699,7 @@ func TestProjectionHandler_fetchPreviousStmts(t *testing.T) {
},
),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
want: want{
@@ -712,7 +722,7 @@ func TestProjectionHandler_fetchPreviousStmts(t *testing.T) {
stmtSeq: 10,
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
ID: "id",
@@ -726,6 +736,7 @@ func TestProjectionHandler_fetchPreviousStmts(t *testing.T) {
},
),
),
),
aggregates: []eventstore.AggregateType{"testAgg"},
},
want: want{

34
internal/eventstore/handler/handler_projection_test.go
View File

@@ -68,7 +68,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
es_repo_mock.NewRepo(t).ExpectFilterEvents(),
eventstore.TestConfig(es_repo_mock.NewRepo(t).ExpectFilterEvents()),
)
},
query: testQuery(
@@ -93,7 +93,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
"process error",
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
ID: "id",
@@ -106,6 +106,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
AggregateType: "testAgg",
},
),
),
)
},
query: testQuery(
@@ -131,7 +132,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
"process ok",
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
ID: "id",
@@ -144,6 +145,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
AggregateType: "testAgg",
},
),
),
)
},
query: testQuery(
@@ -170,7 +172,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
"process limit exceeded ok",
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).
ExpectFilterEvents(
&repository.Event{
@@ -184,6 +186,7 @@ func TestProjectionHandler_Trigger(t *testing.T) {
AggregateType: "testAgg",
},
).ExpectFilterEvents(),
),
)
},
query: testQuery(
@@ -389,9 +392,10 @@ func TestProjectionHandler_FetchEvents(t *testing.T) {
ctx: context.Background(),
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEventsError(ErrFilter),
),
),
query: testQuery(
eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
AddQuery().
@@ -415,7 +419,7 @@ func TestProjectionHandler_FetchEvents(t *testing.T) {
},
fields: fields{
eventstore: eventstore.NewEventstore(
es_repo_mock.NewRepo(t).ExpectFilterEvents(),
eventstore.TestConfig(es_repo_mock.NewRepo(t).ExpectFilterEvents()),
),
query: testQuery(
eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
@@ -439,7 +443,7 @@ func TestProjectionHandler_FetchEvents(t *testing.T) {
ctx: context.Background(),
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
ID: "id",
@@ -463,6 +467,7 @@ func TestProjectionHandler_FetchEvents(t *testing.T) {
},
),
),
),
query: testQuery(
eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
AddQuery().
@@ -485,7 +490,7 @@ func TestProjectionHandler_FetchEvents(t *testing.T) {
ctx: context.Background(),
},
fields: fields{
eventstore: eventstore.NewEventstore(
eventstore: eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
ID: "id",
@@ -509,6 +514,7 @@ func TestProjectionHandler_FetchEvents(t *testing.T) {
},
),
),
),
query: testQuery(
eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
AddQuery().
@@ -694,8 +700,9 @@ func TestProjection_schedule(t *testing.T) {
},
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEventsError(ErrFilter),
),
)
},
triggerProjection: time.NewTimer(0),
@@ -713,7 +720,7 @@ func TestProjection_schedule(t *testing.T) {
},
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
AggregateType: "system",
@@ -723,6 +730,7 @@ func TestProjection_schedule(t *testing.T) {
Type: "system.projections.scheduler.succeeded",
}).
ExpectInstanceIDsError(ErrFilter),
),
)
},
triggerProjection: time.NewTimer(0),
@@ -740,7 +748,7 @@ func TestProjection_schedule(t *testing.T) {
},
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
AggregateType: "system",
@@ -749,6 +757,7 @@ func TestProjection_schedule(t *testing.T) {
InstanceID: "",
Type: "system.projections.scheduler.succeeded",
}).ExpectInstanceIDs("instanceID1"),
),
)
},
triggerProjection: time.NewTimer(0),
@@ -771,7 +780,7 @@ func TestProjection_schedule(t *testing.T) {
},
fields{
eventstore: func(t *testing.T) *eventstore.Eventstore {
return eventstore.NewEventstore(
return eventstore.NewEventstore(eventstore.TestConfig(
es_repo_mock.NewRepo(t).ExpectFilterEvents(
&repository.Event{
AggregateType: "system",
@@ -780,6 +789,7 @@ func TestProjection_schedule(t *testing.T) {
InstanceID: "",
Type: "system.projections.scheduler.succeeded",
}).ExpectInstanceIDs("instanceID1"),
),
)
},
triggerProjection: time.NewTimer(0),