apmelastic

Documentation

Index

• Variables

Variables

var ConsumerElasticApmInterceptor = func(captureErrors bool) consumer.UnaryInterceptorFunc {
	opts := serverOptions{
		tracer: apm.DefaultTracer(),
	}

	return func(next consumer.UnaryFunc) consumer.UnaryFunc {
		return func(ctx context.Context, req consumer.MessageRequest) (resp consumer.ConfirmationType, err error) {
			if !opts.tracer.Recording() {
				return next(ctx, req)
			}

			spec := req.Spec()
			requestName := fmt.Sprintf("%v with stream %v", spec.ConsumerName, spec.ConsumerQueue)

			tx, ctx := startTransaction(ctx, opts.tracer, requestName, req.Header(), spec.Version)
			defer tx.End()

			defer func() {
				r := recover()
				if r != nil {
					e := opts.tracer.Recovered(r)
					e.SetTransaction(tx)
					tx.Context.SetFramework(frameworkName, spec.Version)
					e.Handled = false
					e.Send()

					panic(r)
				}

				setTransactionResult(tx, resp)

				if err != nil && captureErrors {
					logError(err, ctx)
				}
			}()

			resp, err = next(ctx, req)

			return resp, err
		}
	}
}

var PublisherElasticApmInterceptor = func() publisher.UnaryPublisherInterceptorFunc {
	return func(next publisher.UnaryPublisherFunc) publisher.UnaryPublisherFunc {
		return func(ctx context.Context, req publisher.AnyEvent) {
			if apmTx := apm.TransactionFromContext(ctx); apmTx != nil {
				req.SetHeader(elasticTraceparentHeader, apmhttp.FormatTraceparentHeader(apmTx.TraceContext()))

				span := apmTx.StartSpan(
					fmt.Sprintf("publish event to %v", req.GetDestination()),
					"eventsourcing",
					apm.SpanFromContext(ctx),
				)

				span.Context.SetLabel("destination", req.GetDestination())
				span.Context.SetServiceTarget(apm.ServiceTargetSpanContext{
					Name: req.GetDestinationType(),
					Type: req.GetDestination(),
				})

				defer span.End()
			}

			next(ctx, req)
		}
	}
}