diff options
author | 2025-02-06 12:14:37 +0100 | |
---|---|---|
committer | 2025-02-06 12:14:37 +0100 | |
commit | dd094e401282e135989f57c0ca3dee7dea3f5207 (patch) | |
tree | 74cb77830f621840273255a17565ced73b4fa997 /internal/observability/bun.go | |
parent | [feature] Use `X-Robots-Tag` headers to instruct scrapers/crawlers (#3737) (diff) | |
download | gotosocial-dd094e401282e135989f57c0ca3dee7dea3f5207.tar.xz |
[chore] update otel libraries (#3740)
* chore: update otel dependencies
* refactor: combine tracing & metrics in observability package
* chore: update example tracing compose file
Diffstat (limited to 'internal/observability/bun.go')
-rw-r--r-- | internal/observability/bun.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/internal/observability/bun.go b/internal/observability/bun.go new file mode 100644 index 000000000..4baf786e4 --- /dev/null +++ b/internal/observability/bun.go @@ -0,0 +1,42 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program. If not, see <http://www.gnu.org/licenses/>. + +//go:build !nootel + +package observability + +import ( + "github.com/uptrace/bun" + "github.com/uptrace/bun/extra/bunotel" + metricnoop "go.opentelemetry.io/otel/metric/noop" + tracenoop "go.opentelemetry.io/otel/trace/noop" +) + +func InstrumentBun(traces bool, metrics bool) bun.QueryHook { + opts := []bunotel.Option{ + bunotel.WithFormattedQueries(true), + } + if !traces { + opts = append(opts, bunotel.WithTracerProvider(tracenoop.NewTracerProvider())) + } + if !metrics { + opts = append(opts, bunotel.WithMeterProvider(metricnoop.NewMeterProvider())) + } + return bunotel.NewQueryHook( + opts..., + ) +} |