init

pkg/telemetry/telemetry.go

@@ -0,0 +1,56 @@
+// Copyright (C) 2025 wangyusong
+//
+// 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 <https://www.gnu.org/licenses/>.
+
+package telemetry
+
+import (
+	"context"
+
+	"github.com/prometheus/client_golang/prometheus"
+
+	"github.com/glidea/zenfeed/pkg/telemetry/log"
+	"github.com/glidea/zenfeed/pkg/telemetry/metric"
+)
+
+type Labels []any
+
+func (l Labels) Get(key any) any {
+	for i := 0; i < len(l); i += 2 {
+		if l[i] == key {
+			return l[i+1]
+		}
+	}
+
+	return nil
+}
+
+// StartWith starts a new operation with the given key-value pairs.
+// MUST call End() to finalize the operation.
+func StartWith(ctx context.Context, keyvals ...any) context.Context {
+	ctx = log.With(ctx, keyvals...)
+	ctx = metric.StartWith(ctx, keyvals...)
+
+	return ctx
+}
+
+// End records and finalizes the operation.
+func End(ctx context.Context, err error) {
+	metric.RecordRED(ctx, err)
+}
+
+// CloseMetrics closes the metrics for the given id.
+func CloseMetrics(id prometheus.Labels) {
+	metric.Close(id)
+}