57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
// 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)
|
|
}
|