feat(contrib/jackc/pgx.v5): Add WithErrCheck option

contrib/jackc/pgx.v5/option.go

@@ -18,6 +18,7 @@ type config struct {
 	traceConnect  bool
 	traceAcquire  bool
 	poolStats     bool
+	ignoreError   func(error) bool
 	statsdClient  instrumentation.StatsdClient
 }
 
@@ -30,6 +31,7 @@ func defaultConfig() *config {
 		tracePrepare:  true,
 		traceConnect:  true,
 		traceAcquire:  true,
+		ignoreError:   func(err error) bool { return false },
 	}
 }
 
@@ -107,3 +109,10 @@ func WithPoolStats() Option {
 		cfg.poolStats = true
 	}
 }
+
+// WithIgnoreError specifies a function that determines whether the error should be ignored.
+func WithIgnoreError(fn func(err error) bool) Option {
+	return func(cfg *config) {
+		cfg.ignoreError = fn
+	}
+}

contrib/jackc/pgx.v5/pgx_tracer.go

@@ -124,7 +124,7 @@ func (t *pgxTracer) TraceQueryEnd(ctx context.Context, conn *pgx.Conn, data pgx.
 	if ok {
 		span.SetTag(tagRowsAffected, data.CommandTag.RowsAffected())
 	}
-	finishSpan(ctx, data.Err)
+	t.finishSpan(ctx, data.Err)
 }
 
 func (t *pgxTracer) TraceBatchStart(ctx context.Context, conn *pgx.Conn, data pgx.TraceBatchStartData) context.Context {
@@ -175,7 +175,7 @@ func (t *pgxTracer) TraceBatchEnd(ctx context.Context, conn *pgx.Conn, data pgx.
 		t.prevBatchQuery.finish()
 		t.prevBatchQuery = nil
 	}
-	finishSpan(ctx, data.Err)
+	t.finishSpan(ctx, data.Err)
 }
 
 func (t *pgxTracer) TraceCopyFromStart(ctx context.Context, conn *pgx.Conn, data pgx.TraceCopyFromStartData) context.Context {
@@ -200,7 +200,7 @@ func (t *pgxTracer) TraceCopyFromEnd(ctx context.Context, conn *pgx.Conn, data p
 	if t.wrapped.copyFrom != nil {
 		t.wrapped.copyFrom.TraceCopyFromEnd(ctx, conn, data)
 	}
-	finishSpan(ctx, data.Err)
+	t.finishSpan(ctx, data.Err)
 }
 
 func (t *pgxTracer) TracePrepareStart(ctx context.Context, conn *pgx.Conn, data pgx.TracePrepareStartData) context.Context {
@@ -222,7 +222,7 @@ func (t *pgxTracer) TracePrepareEnd(ctx context.Context, conn *pgx.Conn, data pg
 	if t.wrapped.prepare != nil {
 		t.wrapped.prepare.TracePrepareEnd(ctx, conn, data)
 	}
-	finishSpan(ctx, data.Err)
+	t.finishSpan(ctx, data.Err)
 }
 
 func (t *pgxTracer) TraceConnectStart(ctx context.Context, data pgx.TraceConnectStartData) context.Context {
@@ -244,7 +244,7 @@ func (t *pgxTracer) TraceConnectEnd(ctx context.Context, data pgx.TraceConnectEn
 	if t.wrapped.connect != nil {
 		t.wrapped.connect.TraceConnectEnd(ctx, data)
 	}
-	finishSpan(ctx, data.Err)
+	t.finishSpan(ctx, data.Err)
 }
 
 func (t *pgxTracer) TraceAcquireStart(ctx context.Context, pool *pgxpool.Pool, data pgxpool.TraceAcquireStartData) context.Context {
@@ -266,7 +266,7 @@ func (t *pgxTracer) TraceAcquireEnd(ctx context.Context, pool *pgxpool.Pool, dat
 	if t.wrapped.poolAcquire != nil {
 		t.wrapped.poolAcquire.TraceAcquireEnd(ctx, pool, data)
 	}
-	finishSpan(ctx, data.Err)
+	t.finishSpan(ctx, data.Err)
 }
 
 func (t *pgxTracer) spanOptions(connConfig *pgx.ConnConfig, op operationType, sqlStatement string, extraOpts ...tracer.StartSpanOption) []tracer.StartSpanOption {
@@ -300,10 +300,14 @@ func (t *pgxTracer) spanOptions(connConfig *pgx.ConnConfig, op operationType, sq
 	return opts
 }
 
-func finishSpan(ctx context.Context, err error) {
+func (t *pgxTracer) finishSpan(ctx context.Context, err error) {
 	span, ok := tracer.SpanFromContext(ctx)
 	if !ok {
 		return
 	}
+	if t.cfg.ignoreError != nil && t.cfg.ignoreError(err) {
+		span.Finish()
+		return
+	}
 	span.Finish(tracer.WithError(err))
 }

contrib/jackc/pgx.v5/pgx_tracer_test.go

@@ -342,6 +342,7 @@ func tracingAllDisabled() []Option {
 		WithTraceBatch(false),
 		WithTraceCopyFrom(false),
 		WithTraceAcquire(false),
+		WithIgnoreError(func(err error) bool { return false }),
 	}
 }
 

ddtrace/tracer/span_config.go

@@ -75,7 +75,7 @@ type FinishConfig struct {
 	SkipStackFrames uint
 }
 
-// NewFinishConfig allows to build a base finish config struct. It accepts the same options as Finish.
+// NewFinishConfig allows building a base finish config struct. It accepts the same options as Finish.
 // It's useful to reduce the number of operations in any hot path and update it for request/operation specifics.
 func NewFinishConfig(opts ...FinishOption) *FinishConfig {
 	cfg := new(FinishConfig)