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 {
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,

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

@ -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
}