mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-11 19:07:30 +00:00
fix: move v2 pkgs (#1331)
* fix: move eventstore pkgs * fix: move eventstore pkgs * fix: remove v2 view * fix: remove v2 view
This commit is contained in:
Fabi
303
internal/eventstore/example_test.go
Normal file
303
internal/eventstore/example_test.go
Normal file
@@ -0,0 +1,303 @@
|
||||
package eventstore_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/caos/zitadel/internal/eventstore"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/caos/zitadel/internal/api/authz"
|
||||
"github.com/caos/zitadel/internal/eventstore/repository"
|
||||
"github.com/caos/zitadel/internal/eventstore/repository/sql"
|
||||
)
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// User aggregate start
|
||||
// ------------------------------------------------------------
|
||||
func NewUserAggregate(id string) *eventstore.Aggregate {
|
||||
return eventstore.NewAggregate(
|
||||
authz.NewMockContext("caos", "adlerhurst"),
|
||||
id,
|
||||
"test.user",
|
||||
"v1",
|
||||
)
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// User added event start
|
||||
// ------------------------------------------------------------
|
||||
|
||||
type UserAddedEvent struct {
|
||||
eventstore.eventstore `json:"-"`
|
||||
|
||||
FirstName string `json:"firstName"`
|
||||
}
|
||||
|
||||
func NewUserAddedEvent(id string, firstName string) *UserAddedEvent {
|
||||
return &UserAddedEvent{
|
||||
FirstName: firstName,
|
||||
BaseEvent: *eventstore.NewBaseEventForPush(
|
||||
context.Background(),
|
||||
NewUserAggregate(id),
|
||||
"user.added"),
|
||||
}
|
||||
}
|
||||
|
||||
func UserAddedEventMapper() (eventstore.EventType, func(*repository.Event) (eventstore.EventReader, error)) {
|
||||
return "user.added", func(event *repository.Event) (eventstore.EventReader, error) {
|
||||
e := &UserAddedEvent{
|
||||
BaseEvent: *eventstore.BaseEventFromRepo(event),
|
||||
}
|
||||
err := json.Unmarshal(event.Data, e)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return e, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (e *UserAddedEvent) Data() interface{} {
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *UserAddedEvent) UniqueConstraints() []*eventstore.EventUniqueConstraint {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// User first name changed event start
|
||||
// ------------------------------------------------------------
|
||||
|
||||
type UserFirstNameChangedEvent struct {
|
||||
eventstore.BaseEvent `json:"-"`
|
||||
|
||||
FirstName string `json:"firstName"`
|
||||
}
|
||||
|
||||
func NewUserFirstNameChangedEvent(id, firstName string) *UserFirstNameChangedEvent {
|
||||
return &UserFirstNameChangedEvent{
|
||||
FirstName: firstName,
|
||||
BaseEvent: *eventstore.NewBaseEventForPush(
|
||||
context.Background(),
|
||||
NewUserAggregate(id),
|
||||
"user.firstname.changed"),
|
||||
}
|
||||
}
|
||||
|
||||
func UserFirstNameChangedMapper() (eventstore.EventType, func(*repository.Event) (eventstore.EventReader, error)) {
|
||||
return "user.firstName.changed", func(event *repository.Event) (eventstore.EventReader, error) {
|
||||
e := &UserFirstNameChangedEvent{
|
||||
BaseEvent: *eventstore.BaseEventFromRepo(event),
|
||||
}
|
||||
err := json.Unmarshal(event.Data, e)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return e, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (e *UserFirstNameChangedEvent) Data() interface{} {
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *UserFirstNameChangedEvent) UniqueConstraints() []*eventstore.EventUniqueConstraint {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// User password checked event start
|
||||
// ------------------------------------------------------------
|
||||
|
||||
type UserPasswordCheckedEvent struct {
|
||||
eventstore.BaseEvent `json:"-"`
|
||||
}
|
||||
|
||||
func NewUserPasswordCheckedEvent(id string) *UserPasswordCheckedEvent {
|
||||
return &UserPasswordCheckedEvent{
|
||||
BaseEvent: *eventstore.NewBaseEventForPush(
|
||||
context.Background(),
|
||||
NewUserAggregate(id),
|
||||
"user.password.checked"),
|
||||
}
|
||||
}
|
||||
|
||||
func UserPasswordCheckedMapper() (eventstore.EventType, func(*repository.Event) (eventstore.EventReader, error)) {
|
||||
return "user.password.checked", func(event *repository.Event) (eventstore.EventReader, error) {
|
||||
return &UserPasswordCheckedEvent{
|
||||
BaseEvent: *eventstore.BaseEventFromRepo(event),
|
||||
}, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (e *UserPasswordCheckedEvent) Data() interface{} {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (e *UserPasswordCheckedEvent) UniqueConstraints() []*eventstore.EventUniqueConstraint {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// User deleted event
|
||||
// ------------------------------------------------------------
|
||||
|
||||
type UserDeletedEvent struct {
|
||||
eventstore.BaseEvent `json:"-"`
|
||||
}
|
||||
|
||||
func NewUserDeletedEvent(id string) *UserDeletedEvent {
|
||||
return &UserDeletedEvent{
|
||||
BaseEvent: *eventstore.NewBaseEventForPush(
|
||||
context.Background(),
|
||||
NewUserAggregate(id),
|
||||
"user.deleted"),
|
||||
}
|
||||
}
|
||||
|
||||
func UserDeletedMapper() (eventstore.EventType, func(*repository.Event) (eventstore.EventReader, error)) {
|
||||
return "user.deleted", func(event *repository.Event) (eventstore.EventReader, error) {
|
||||
return &UserDeletedEvent{
|
||||
BaseEvent: *eventstore.BaseEventFromRepo(event),
|
||||
}, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (e *UserDeletedEvent) Data() interface{} {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (e *UserDeletedEvent) UniqueConstraints() []*eventstore.EventUniqueConstraint {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// Users read model start
|
||||
// ------------------------------------------------------------
|
||||
|
||||
type UsersReadModel struct {
|
||||
eventstore.ReadModel
|
||||
|
||||
Users []*UserReadModel
|
||||
}
|
||||
|
||||
func (rm *UsersReadModel) AppendEvents(events ...eventstore.EventReader) {
|
||||
rm.ReadModel.AppendEvents(events...)
|
||||
for _, event := range events {
|
||||
switch e := event.(type) {
|
||||
case *UserAddedEvent:
|
||||
//insert
|
||||
user := NewUserReadModel(e.Aggregate().ID)
|
||||
rm.Users = append(rm.Users, user)
|
||||
user.AppendEvents(e)
|
||||
case *UserFirstNameChangedEvent, *UserPasswordCheckedEvent:
|
||||
//update
|
||||
_, user := rm.userByID(e.Aggregate().ID)
|
||||
if user == nil {
|
||||
return
|
||||
}
|
||||
user.AppendEvents(e)
|
||||
case *UserDeletedEvent:
|
||||
idx, _ := rm.userByID(e.Aggregate().ID)
|
||||
if idx < 0 {
|
||||
return
|
||||
}
|
||||
copy(rm.Users[idx:], rm.Users[idx+1:])
|
||||
rm.Users[len(rm.Users)-1] = nil // or the zero value of T
|
||||
rm.Users = rm.Users[:len(rm.Users)-1]
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (rm *UsersReadModel) Reduce() error {
|
||||
for _, user := range rm.Users {
|
||||
err := user.Reduce()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
rm.ReadModel.Reduce()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (rm *UsersReadModel) userByID(id string) (idx int, user *UserReadModel) {
|
||||
for idx, user = range rm.Users {
|
||||
if user.ID == id {
|
||||
return idx, user
|
||||
}
|
||||
}
|
||||
|
||||
return -1, nil
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// User read model start
|
||||
// ------------------------------------------------------------
|
||||
|
||||
type UserReadModel struct {
|
||||
eventstore.ReadModel
|
||||
|
||||
ID string
|
||||
FirstName string
|
||||
pwCheckCount int
|
||||
lastPasswordCheck time.Time
|
||||
}
|
||||
|
||||
func NewUserReadModel(id string) *UserReadModel {
|
||||
return &UserReadModel{
|
||||
ID: id,
|
||||
}
|
||||
}
|
||||
|
||||
func (rm *UserReadModel) Reduce() error {
|
||||
for _, event := range rm.ReadModel.Events {
|
||||
switch e := event.(type) {
|
||||
case *UserAddedEvent:
|
||||
rm.FirstName = e.FirstName
|
||||
case *UserFirstNameChangedEvent:
|
||||
rm.FirstName = e.FirstName
|
||||
case *UserPasswordCheckedEvent:
|
||||
rm.pwCheckCount++
|
||||
rm.lastPasswordCheck = e.CreationDate()
|
||||
}
|
||||
}
|
||||
rm.ReadModel.Reduce()
|
||||
return nil
|
||||
}
|
||||
|
||||
// ------------------------------------------------------------
|
||||
// Tests
|
||||
// ------------------------------------------------------------
|
||||
|
||||
func TestUserReadModel(t *testing.T) {
|
||||
es := eventstore.NewEventstore(sql.NewCRDB(testCRDBClient))
|
||||
es.RegisterFilterEventMapper(UserAddedEventMapper()).
|
||||
RegisterFilterEventMapper(UserFirstNameChangedMapper()).
|
||||
RegisterFilterEventMapper(UserPasswordCheckedMapper()).
|
||||
RegisterFilterEventMapper(UserDeletedMapper())
|
||||
|
||||
events, err := es.PushEvents(context.Background(),
|
||||
NewUserAddedEvent("1", "hodor"),
|
||||
NewUserAddedEvent("2", "hodor"),
|
||||
NewUserPasswordCheckedEvent("2"),
|
||||
NewUserPasswordCheckedEvent("2"),
|
||||
NewUserFirstNameChangedEvent("2", "ueli"),
|
||||
NewUserDeletedEvent("2"))
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("unexpected error on push aggregates: %v", err)
|
||||
}
|
||||
|
||||
events = append(events, nil)
|
||||
|
||||
fmt.Printf("%+v\n", events)
|
||||
|
||||
users := UsersReadModel{}
|
||||
err = es.FilterToReducer(context.Background(), eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent, "test.user"), &users)
|
||||
if err != nil {
|
||||
t.Errorf("unexpected error on filter to reducer: %v", err)
|
||||
}
|
||||
fmt.Printf("%+v", users)
|
||||
}
|
||||
Reference in New Issue
Block a user