fix: request cleanup jobs
This commit is contained in:
@@ -19,6 +19,7 @@ type EventLog interface {
|
||||
FirstIndex(context.Context) (uint64, error)
|
||||
LastIndex(context.Context) (uint64, error)
|
||||
}
|
||||
|
||||
type EventLogWithTruncate interface {
|
||||
Truncate(context.Context, int64) error
|
||||
}
|
||||
|
||||
@@ -16,10 +16,22 @@ type projector struct {
|
||||
fns []func(event.Event) []event.Event
|
||||
}
|
||||
|
||||
func New(ctx context.Context, fns ...func(event.Event) []event.Event) *projector {
|
||||
func New(_ context.Context, fns ...func(event.Event) []event.Event) *projector {
|
||||
return &projector{fns: fns}
|
||||
}
|
||||
func (p *projector) Apply(e *es.EventStore) {
|
||||
|
||||
up := e.Driver
|
||||
for up != nil {
|
||||
if op, ok := up.(*projector); ok {
|
||||
op.AddProjections(p.fns...)
|
||||
p.up = op.up
|
||||
return
|
||||
}
|
||||
|
||||
up = es.Unwrap(up)
|
||||
}
|
||||
|
||||
p.up = e.Driver
|
||||
e.Driver = p
|
||||
}
|
||||
@@ -39,6 +51,9 @@ func (s *projector) EventLog(ctx context.Context, streamID string) (driver.Event
|
||||
l, err := s.up.EventLog(ctx, streamID)
|
||||
return &wrapper{l, s}, err
|
||||
}
|
||||
func (s *projector) AddProjections(fns ...func(event.Event) []event.Event) {
|
||||
s.fns = append(s.fns, fns...)
|
||||
}
|
||||
|
||||
type wrapper struct {
|
||||
up driver.EventLog
|
||||
|
||||
@@ -17,13 +17,16 @@ type resolvelinks struct {
|
||||
func New() *resolvelinks {
|
||||
return &resolvelinks{}
|
||||
}
|
||||
|
||||
func (r *resolvelinks) Apply(es *es.EventStore) {
|
||||
r.up = es.Driver
|
||||
es.Driver = r
|
||||
}
|
||||
|
||||
func (r *resolvelinks) Unwrap() driver.Driver {
|
||||
return r.up
|
||||
}
|
||||
|
||||
func (r *resolvelinks) Open(ctx context.Context, dsn string) (driver.Driver, error) {
|
||||
ctx, span := lg.Span(ctx)
|
||||
defer span.End()
|
||||
@@ -47,6 +50,7 @@ type wrapper struct {
|
||||
func (r *wrapper) Unwrap() driver.EventLog {
|
||||
return r.up
|
||||
}
|
||||
|
||||
func (w *wrapper) Read(ctx context.Context, after int64, count int64) (event.Events, error) {
|
||||
ctx, span := lg.Span(ctx)
|
||||
defer span.End()
|
||||
@@ -56,29 +60,60 @@ func (w *wrapper) Read(ctx context.Context, after int64, count int64) (event.Eve
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for i, e := range events {
|
||||
switch e := e.(type) {
|
||||
case *event.EventPtr:
|
||||
d, err := w.resolvelinks.EventLog(ctx, e.StreamID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
lis, err := d.ReadN(ctx, e.Pos)
|
||||
if err != nil && !errors.Is(err, es.ErrNotFound) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if ne := lis.First(); ne != event.NilEvent {
|
||||
meta := ne.EventMeta()
|
||||
actual := e.EventMeta()
|
||||
meta.ActualPosition = actual.Position
|
||||
meta.ActualStreamID = actual.ActualStreamID
|
||||
ne.SetEventMeta(meta)
|
||||
events[i] = ne
|
||||
}
|
||||
idx := make(map[string][]uint64)
|
||||
ptrs := make(map[string][]int)
|
||||
for i := range events {
|
||||
e := events[i]
|
||||
if e, ok := e.(*event.EventPtr); ok {
|
||||
idx[e.StreamID] = append(idx[e.StreamID], e.Pos)
|
||||
ptrs[e.StreamID] = append(ptrs[e.StreamID], i)
|
||||
}
|
||||
}
|
||||
|
||||
for streamID, ids := range idx {
|
||||
d, err := w.resolvelinks.EventLog(ctx, streamID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
ptr := ptrs[streamID]
|
||||
lis, err := d.ReadN(ctx, ids...)
|
||||
if err != nil && !errors.Is(err, es.ErrNotFound) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for i := range lis {
|
||||
meta := lis[i].EventMeta()
|
||||
actual := events[ptr[i]].EventMeta()
|
||||
meta.ActualPosition = actual.Position
|
||||
meta.ActualStreamID = actual.ActualStreamID
|
||||
lis[i].SetEventMeta(meta)
|
||||
events[i] = lis[i]
|
||||
}
|
||||
}
|
||||
|
||||
// for i, e := range events {
|
||||
// switch e := e.(type) {
|
||||
// case *event.EventPtr:
|
||||
// d, err := w.resolvelinks.EventLog(ctx, e.StreamID)
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
// lis, err := d.ReadN(ctx, e.Pos)
|
||||
// if err != nil && !errors.Is(err, es.ErrNotFound) {
|
||||
// return nil, err
|
||||
// }
|
||||
|
||||
// if ne := lis.First(); ne != event.NilEvent {
|
||||
// meta := ne.EventMeta()
|
||||
// actual := e.EventMeta()
|
||||
// meta.ActualPosition = actual.Position
|
||||
// meta.ActualStreamID = actual.ActualStreamID
|
||||
// ne.SetEventMeta(meta)
|
||||
// events[i] = ne
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
return events, err
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user