From c63b793b65f45a429b2d8b822453e311f693ad42 Mon Sep 17 00:00:00 2001 From: Abhishek Pandey Date: Sat, 7 Oct 2023 01:10:26 +0530 Subject: [PATCH] Reintroduce gauge --- src/internal/events/adot_sdk_client.go | 8 ++++---- src/internal/m365/graph/concurrency_middleware.go | 10 ++++++---- 2 files changed, 10 insertions(+), 8 deletions(-) diff --git a/src/internal/events/adot_sdk_client.go b/src/internal/events/adot_sdk_client.go index b93208772..da914e5d4 100644 --- a/src/internal/events/adot_sdk_client.go +++ b/src/internal/events/adot_sdk_client.go @@ -75,15 +75,15 @@ func NewCollector(mp metric.MeterProvider) { } -func CreateGauge(ctx context.Context, name string) metric.Int64ObservableCounter { - RLGauge, _ := globalMeter.Int64ObservableCounter(name) +func CreateGauge(ctx context.Context, name string) metric.Int64ObservableGauge { + rlg, _ := globalMeter.Int64ObservableGauge(name) - return RLGauge + return rlg } func RegisterGauge( ctx context.Context, - rlg metric.Int64ObservableCounter, + rlg metric.Int64ObservableGauge, cb func(_ context.Context, o metric.Observer) error) { _, err := globalMeter.RegisterCallback( cb, diff --git a/src/internal/m365/graph/concurrency_middleware.go b/src/internal/m365/graph/concurrency_middleware.go index 1f3cf4b57..979ce306e 100644 --- a/src/internal/m365/graph/concurrency_middleware.go +++ b/src/internal/m365/graph/concurrency_middleware.go @@ -161,14 +161,16 @@ const ( ) var twonce sync.Once -var token int64 = 0 + +//var token int64 = 0 func RegisterRLMetrics(ctx context.Context) { twonce.Do(func() { rlg := events.CreateGauge(ctx, events.RLTokens) - cb := func(ctx context.Context, o metric.Observer) error { - token += int64(ctxLimiter(ctx).Tokens()) - o.ObserveInt64(rlg, token) + cb := func(_ context.Context, o metric.Observer) error { + //token += int64(driveLimiter.Tokens()) + //o.ObserveInt64(rlg, token) + o.ObserveInt64(rlg, int64(driveLimiter.Tokens())) return nil }