mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
3cd3a238c2
* fix: all enums same style * fix: rename process to reduce * add some missing enum renaming Co-authored-by: Livio Amstutz <livio.a@gmail.com>
127 lines
3.8 KiB
Go
127 lines
3.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"context"
|
|
"github.com/caos/logging"
|
|
"github.com/caos/zitadel/internal/eventstore/models"
|
|
es_models "github.com/caos/zitadel/internal/eventstore/models"
|
|
"github.com/caos/zitadel/internal/eventstore/spooler"
|
|
proj_es_model "github.com/caos/zitadel/internal/project/repository/eventsourcing/model"
|
|
view_model "github.com/caos/zitadel/internal/project/repository/view/model"
|
|
usr_model "github.com/caos/zitadel/internal/user/model"
|
|
usr_event "github.com/caos/zitadel/internal/user/repository/eventsourcing"
|
|
usr_es_model "github.com/caos/zitadel/internal/user/repository/eventsourcing/model"
|
|
"time"
|
|
)
|
|
|
|
type ProjectMember struct {
|
|
handler
|
|
userEvents *usr_event.UserEventstore
|
|
}
|
|
|
|
const (
|
|
projectMemberTable = "management.project_members"
|
|
)
|
|
|
|
func (p *ProjectMember) MinimumCycleDuration() time.Duration { return p.cycleDuration }
|
|
|
|
func (p *ProjectMember) ViewModel() string {
|
|
return projectMemberTable
|
|
}
|
|
|
|
func (p *ProjectMember) EventQuery() (*models.SearchQuery, error) {
|
|
sequence, err := p.view.GetLatestProjectMemberSequence()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return es_models.NewSearchQuery().
|
|
AggregateTypeFilter(proj_es_model.ProjectAggregate, usr_es_model.UserAggregate).
|
|
LatestSequenceFilter(sequence), nil
|
|
}
|
|
|
|
func (p *ProjectMember) Reduce(event *models.Event) (err error) {
|
|
switch event.AggregateType {
|
|
case proj_es_model.ProjectAggregate:
|
|
err = p.processProjectMember(event)
|
|
case usr_es_model.UserAggregate:
|
|
err = p.processUser(event)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (p *ProjectMember) processProjectMember(event *models.Event) (err error) {
|
|
member := new(view_model.ProjectMemberView)
|
|
switch event.Type {
|
|
case proj_es_model.ProjectMemberAdded:
|
|
member.AppendEvent(event)
|
|
p.fillData(member)
|
|
case proj_es_model.ProjectMemberChanged:
|
|
err := member.SetData(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
member, err = p.view.ProjectMemberByIDs(event.AggregateID, member.UserID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
member.AppendEvent(event)
|
|
case proj_es_model.ProjectMemberRemoved:
|
|
err := member.SetData(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return p.view.DeleteProjectMember(event.AggregateID, member.UserID, event.Sequence)
|
|
default:
|
|
return p.view.ProcessedProjectMemberSequence(event.Sequence)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return p.view.PutProjectMember(member, member.Sequence)
|
|
}
|
|
|
|
func (p *ProjectMember) processUser(event *models.Event) (err error) {
|
|
switch event.Type {
|
|
case usr_es_model.UserProfileChanged,
|
|
usr_es_model.UserEmailChanged:
|
|
members, err := p.view.ProjectMembersByUserID(event.AggregateID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
user, err := p.userEvents.UserByID(context.Background(), event.AggregateID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, member := range members {
|
|
p.fillUserData(member, user)
|
|
err = p.view.PutProjectMember(member, event.Sequence)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
default:
|
|
return p.view.ProcessedProjectMemberSequence(event.Sequence)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *ProjectMember) fillData(member *view_model.ProjectMemberView) (err error) {
|
|
user, err := p.userEvents.UserByID(context.Background(), member.UserID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
p.fillUserData(member, user)
|
|
return nil
|
|
}
|
|
|
|
func (p *ProjectMember) fillUserData(member *view_model.ProjectMemberView, user *usr_model.User) {
|
|
member.UserName = user.UserName
|
|
member.FirstName = user.FirstName
|
|
member.LastName = user.LastName
|
|
member.Email = user.EmailAddress
|
|
}
|
|
func (p *ProjectMember) OnError(event *models.Event, err error) error {
|
|
logging.LogWithFields("SPOOL-u73es", "id", event.AggregateID).WithError(err).Warn("something went wrong in projectmember handler")
|
|
return spooler.HandleError(event, err, p.view.GetLatestProjectMemberFailedEvent, p.view.ProcessedProjectMemberFailedEvent, p.view.ProcessedProjectMemberSequence, p.errorCountUntilSkip)
|
|
}
|