with custom marshaller

lestrrat-go · Sep 19, 2024 · c66af92 · c66af92
1 parent 2dfef78
commit c66af92
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/buffered.go b/buffered.go
@@ -17,6 +17,7 @@ import (
 //   - fluent.WithJSONMarshaler
 //   - fluent.WithMaxConnAttempts
 //   - fluent.WithMsgpackMarshaler
+//   - fluent.WithMarshaller
 //   - fluent.WithNetwork
 //   - fluent.WithTagPrefix
 //   - fluent.WithWriteThreshold

diff --git a/options.go b/options.go
@@ -56,6 +56,12 @@ func WithJSONMarshaler() Option {
 	return option.New(identMarshaler{}, marshalFunc(jsonMarshal))
 }
 
+// WithMarshaller specifies a custom marshaling to be used when
+// sending messages to fluentd. Used for `fluent.New`
+func WithMarshaller(m marshaler) Option {
+	return option.New(identMarshaler{}, m)
+}
+
 // WithMsgpackMarshaler specifies msgpack marshaling to be used when
 // sending messages to fluentd. Used in `fluent.New`
 func WithMsgpackMarshaler() Option {