33 lines
898 B
Go
33 lines
898 B
Go
package bunotel
|
|
|
|
import (
|
|
"go.opentelemetry.io/otel/attribute"
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.12.0"
|
|
)
|
|
|
|
type Option func(h *QueryHook)
|
|
|
|
// WithAttributes configures attributes that are used to create a span.
|
|
func WithAttributes(attrs ...attribute.KeyValue) Option {
|
|
return func(h *QueryHook) {
|
|
h.attrs = append(h.attrs, attrs...)
|
|
}
|
|
}
|
|
|
|
// WithDBName configures a db.name attribute.
|
|
func WithDBName(name string) Option {
|
|
return func(h *QueryHook) {
|
|
h.attrs = append(h.attrs, semconv.DBNameKey.String(name))
|
|
}
|
|
}
|
|
|
|
// WithFormattedQueries enables formatting of the query that is added
|
|
// as the statement attribute to the trace.
|
|
// This means that all placeholders and arguments will be filled first
|
|
// and the query will contain all information as sent to the database.
|
|
func WithFormattedQueries(format bool) Option {
|
|
return func(h *QueryHook) {
|
|
h.formatQueries = format
|
|
}
|
|
}
|