Add more options

Author: lzap

writer.go

@@ -153,20 +153,24 @@ type optionFunc func(*config)
 
 func (fn optionFunc) apply(c *config) { fn(c) }
 
+type EventHintCallback func(event *sentry.Event, hint *sentry.EventHint) *sentry.Event
+
 type config struct {
-	levels       []zerolog.Level
-	sampleRate   float64
-	release      string
-	environment  string
-	serverName   string
-	ignoreErrors []string
-	debug        bool
-	tracing      bool
-	debugWriter  io.Writer
-	httpProxy    string
-	httpsProxy   string
-	caCerts      *x509.CertPool
-	flushTimeout time.Duration
+	levels           []zerolog.Level
+	sampleRate       float64
+	release          string
+	environment      string
+	serverName       string
+	ignoreErrors     []string
+	debug            bool
+	tracing          bool
+	debugWriter      io.Writer
+	httpProxy        string
+	httpsProxy       string
+	caCerts          *x509.CertPool
+	flushTimeout     time.Duration
+	beforeSend       sentry.EventProcessor
+	tracesSampleRate float64
 }
 
 // WithLevels configures zerolog levels that have to be sent to Sentry.
@@ -227,6 +231,20 @@ func WithTracing() WriterOption {
 	})
 }
 
+// WithTracingSampleRate sets tracing sample rate.
+func WithTracingSampleRate(tsr float64) WriterOption {
+	return optionFunc(func(cfg *config) {
+		cfg.tracesSampleRate = tsr
+	})
+}
+
+// WithBeforeSend sets a callback which is called before event is sent.
+func WithBeforeSend(beforeSend sentry.EventProcessor) WriterOption {
+	return optionFunc(func(cfg *config) {
+		cfg.beforeSend = beforeSend
+	})
+}
+
 // WithDebugWriter enables sentry client tracing.
 func WithDebugWriter(w io.Writer) WriterOption {
 	return optionFunc(func(cfg *config) {
@@ -263,18 +281,20 @@ func New(dsn string, opts ...WriterOption) (*Writer, error) {
 	}
 
 	err := sentry.Init(sentry.ClientOptions{
-		Dsn:           dsn,
-		SampleRate:    cfg.sampleRate,
-		Release:       cfg.release,
-		Environment:   cfg.environment,
-		ServerName:    cfg.serverName,
-		IgnoreErrors:  cfg.ignoreErrors,
-		Debug:         cfg.debug,
-		EnableTracing: cfg.tracing,
-		DebugWriter:   cfg.debugWriter,
-		HTTPProxy:     cfg.httpProxy,
-		HTTPSProxy:    cfg.httpsProxy,
-		CaCerts:       cfg.caCerts,
+		Dsn:              dsn,
+		SampleRate:       cfg.sampleRate,
+		Release:          cfg.release,
+		Environment:      cfg.environment,
+		ServerName:       cfg.serverName,
+		IgnoreErrors:     cfg.ignoreErrors,
+		Debug:            cfg.debug,
+		EnableTracing:    cfg.tracing,
+		DebugWriter:      cfg.debugWriter,
+		HTTPProxy:        cfg.httpProxy,
+		HTTPSProxy:       cfg.httpsProxy,
+		CaCerts:          cfg.caCerts,
+		BeforeSend:       cfg.beforeSend,
+		TracesSampleRate: cfg.tracesSampleRate,
 	})
 
 	if err != nil {