larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reintroduce gauge

Browse Source
This commit is contained in:
Abhishek Pandey 2023-10-07 01:10:26 +05:30
parent a08c36c910
commit c63b793b65
2 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/internal/events/adot_sdk_client.go
View File

@ -75,15 +75,15 @@ func NewCollector(mp metric.MeterProvider) {
} }
func CreateGauge(ctx context.Context, name string) metric.Int64ObservableCounter { func CreateGauge(ctx context.Context, name string) metric.Int64ObservableGauge {
RLGauge, _ := globalMeter.Int64ObservableCounter(name) rlg, _ := globalMeter.Int64ObservableGauge(name)
return RLGauge return rlg
} }
func RegisterGauge( func RegisterGauge(
ctx context.Context, ctx context.Context,
rlg metric.Int64ObservableCounter, rlg metric.Int64ObservableGauge,
cb func(_ context.Context, o metric.Observer) error) { cb func(_ context.Context, o metric.Observer) error) {
_, err := globalMeter.RegisterCallback( _, err := globalMeter.RegisterCallback(
cb, cb,

10
src/internal/m365/graph/concurrency_middleware.go
View File

@ -161,14 +161,16 @@ const (
) )
var twonce sync.Once var twonce sync.Once
var token int64 = 0
//var token int64 = 0
func RegisterRLMetrics(ctx context.Context) { func RegisterRLMetrics(ctx context.Context) {
twonce.Do(func() { twonce.Do(func() {
rlg := events.CreateGauge(ctx, events.RLTokens) rlg := events.CreateGauge(ctx, events.RLTokens)
cb := func(ctx context.Context, o metric.Observer) error { cb := func(_ context.Context, o metric.Observer) error {
token += int64(ctxLimiter(ctx).Tokens()) //token += int64(driveLimiter.Tokens())
o.ObserveInt64(rlg, token) //o.ObserveInt64(rlg, token)
o.ObserveInt64(rlg, int64(driveLimiter.Tokens()))
return nil return nil
} }