From dd094e401282e135989f57c0ca3dee7dea3f5207 Mon Sep 17 00:00:00 2001 From: Dominik Süß Date: Thu, 6 Feb 2025 12:14:37 +0100 Subject: [chore] update otel libraries (#3740) * chore: update otel dependencies * refactor: combine tracing & metrics in observability package * chore: update example tracing compose file --- internal/observability/bun.go | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 internal/observability/bun.go (limited to 'internal/observability/bun.go') 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 . + +//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..., + ) +} -- cgit v1.2.3