wehub-resource-sync
e071084ebe
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package opentelemetry
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go-micro.dev/v6/client"
|
|
"go-micro.dev/v6/server"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
type Options struct {
|
|
TraceProvider trace.TracerProvider
|
|
|
|
CallFilter CallFilter
|
|
StreamFilter StreamFilter
|
|
PublishFilter PublishFilter
|
|
SubscriberFilter SubscriberFilter
|
|
HandlerFilter HandlerFilter
|
|
}
|
|
|
|
// CallFilter used to filter client.Call, return true to skip call trace.
|
|
type CallFilter func(context.Context, client.Request) bool
|
|
|
|
// StreamFilter used to filter client.Stream, return true to skip stream trace.
|
|
type StreamFilter func(context.Context, client.Request) bool
|
|
|
|
// PublishFilter used to filter client.Publish, return true to skip publish trace.
|
|
type PublishFilter func(context.Context, client.Message) bool
|
|
|
|
// SubscriberFilter used to filter server.Subscribe, return true to skip subcribe trace.
|
|
type SubscriberFilter func(context.Context, server.Message) bool
|
|
|
|
// HandlerFilter used to filter server.Handle, return true to skip handle trace.
|
|
type HandlerFilter func(context.Context, server.Request) bool
|
|
|
|
type Option func(*Options)
|
|
|
|
func WithTraceProvider(tp trace.TracerProvider) Option {
|
|
return func(o *Options) {
|
|
o.TraceProvider = tp
|
|
}
|
|
}
|
|
|
|
func WithCallFilter(filter CallFilter) Option {
|
|
return func(o *Options) {
|
|
o.CallFilter = filter
|
|
}
|
|
}
|
|
|
|
func WithStreamFilter(filter StreamFilter) Option {
|
|
return func(o *Options) {
|
|
o.StreamFilter = filter
|
|
}
|
|
}
|
|
|
|
func WithPublishFilter(filter PublishFilter) Option {
|
|
return func(o *Options) {
|
|
o.PublishFilter = filter
|
|
}
|
|
}
|
|
|
|
func WithSubscribeFilter(filter SubscriberFilter) Option {
|
|
return func(o *Options) {
|
|
o.SubscriberFilter = filter
|
|
}
|
|
}
|
|
|
|
func WithHandleFilter(filter HandlerFilter) Option {
|
|
return func(o *Options) {
|
|
o.HandlerFilter = filter
|
|
}
|
|
}
|