diff --git a/persistence/mongo/persistor.go b/persistence/mongo/persistor.go index 6401861..2904a13 100644 --- a/persistence/mongo/persistor.go +++ b/persistence/mongo/persistor.go @@ -45,6 +45,18 @@ func WithOtelEnabled(v bool) Option { } } +func WithOtelOptions(v ...otelmongo.Option) Option { + return func(o *Options) { + o.OtelOptions = append(o.OtelOptions, v...) + } +} + +func WithClientOptions(v ...ClientOption) Option { + return func(o *Options) { + o.ClientOptions = append(o.ClientOptions, v...) + } +} + func WithClientLogger(v *zap.Logger) Option { return func(o *Options) { o.ClientLoggerOptions = append(o.ClientLoggerOptions, func(o *options.LoggerOptions) { @@ -61,7 +73,7 @@ func WithClientLoggerComponentLevel(c options.LogComponent, l options.LogLevel) } } -func WithClientCompression(v *zap.Logger) Option { +func WithClientCompression() Option { return func(o *Options) { o.ClientOptions = append(o.ClientOptions, func(o *options.ClientOptions) { o.SetCompressors([]string{"snappy", "zstd"}) @@ -69,18 +81,6 @@ func WithClientCompression(v *zap.Logger) Option { } } -func WithOtelOptions(v ...otelmongo.Option) Option { - return func(o *Options) { - o.OtelOptions = append(o.OtelOptions, v...) - } -} - -func WithClientOptions(v ...ClientOption) Option { - return func(o *Options) { - o.ClientOptions = append(o.ClientOptions, v...) - } -} - func WithDatabaseOptions(v ...DatabaseOption) Option { return func(o *Options) { o.DatabaseOptions = append(o.DatabaseOptions, v...)