go-libs/pkg/testing/testservice/instrumentation_nats.go

package testservice

import (
	"context"
	"fmt"

	"github.com/hanzo-fi/go-libs/v5/pkg/messaging/publish"
	"github.com/hanzo-fi/go-libs/v5/pkg/testing/deferred"
)

func NatsInstrumentation(url *deferred.Deferred[string]) Instrumentation {
	return InstrumentationFunc(func(ctx context.Context, runConfiguration *RunConfiguration) error {
		url, err := url.Wait(ctx)
		if err != nil {
			return err
		}
		runConfiguration.AppendArgs(
			"--"+publish.PublisherNatsEnabledFlag,
			"--"+publish.PublisherNatsURLFlag, url,
			"--"+publish.PublisherTopicMappingFlag, fmt.Sprintf("*:%s", runConfiguration.GetID()),
		)

		return nil
	})
}