multiple tries

backend/v3/domain/invoke.go

@@ -0,0 +1,94 @@
+package domain
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/zitadel/zitadel/backend/v3/storage/eventstore"
+)
+
+var defaultInvoker = newEventStoreInvoker(newTraceInvoker(nil))
+
+func Invoke(ctx context.Context, cmd Commander) error {
+	invoker := newEventStoreInvoker(newTraceInvoker(nil))
+	opts := &CommandOpts{
+		Invoker: invoker.collector,
+	}
+	return invoker.Invoke(ctx, cmd, opts)
+}
+
+type eventStoreInvoker struct {
+	collector *eventCollector
+}
+
+func newEventStoreInvoker(next Invoker) *eventStoreInvoker {
+	return &eventStoreInvoker{collector: &eventCollector{next: next}}
+}
+
+func (i *eventStoreInvoker) Invoke(ctx context.Context, command Commander, opts *CommandOpts) (err error) {
+	err = i.collector.Invoke(ctx, command, opts)
+	if err != nil {
+		return err
+	}
+	if len(i.collector.events) > 0 {
+		err = eventstore.Publish(ctx, i.collector.events, opts.DB)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+type eventCollector struct {
+	next   Invoker
+	events []*eventstore.Event
+}
+
+type eventer interface {
+	Events() []*eventstore.Event
+}
+
+func (i *eventCollector) Invoke(ctx context.Context, command Commander, opts *CommandOpts) (err error) {
+	if e, ok := command.(eventer); ok && len(e.Events()) > 0 {
+		// we need to ensure all commands are executed in the same transaction
+		close, err := opts.EnsureTx(ctx)
+		if err != nil {
+			return err
+		}
+		defer func() { err = close(ctx, err) }()
+
+		i.events = append(i.events, e.Events()...)
+	}
+	if i.next != nil {
+		err = i.next.Invoke(ctx, command, opts)
+	} else {
+		err = command.Execute(ctx, opts)
+	}
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+type traceInvoker struct {
+	next Invoker
+}
+
+func newTraceInvoker(next Invoker) *traceInvoker {
+	return &traceInvoker{next: next}
+}
+
+func (i *traceInvoker) Invoke(ctx context.Context, command Commander, opts *CommandOpts) (err error) {
+	ctx, span := tracer.Start(ctx, fmt.Sprintf("%T", command))
+	defer span.End()
+
+	if i.next != nil {
+		err = i.next.Invoke(ctx, command, opts)
+	} else {
+		err = command.Execute(ctx, opts)
+	}
+	if err != nil {
+		span.RecordError(err)
+	}
+	return err
+}