Instrumentation: log the total number of db queries per request (#54647)

Signed-off-by: bergquist <carl.bergquist@gmail.com>
Co-authored-by: Emil Tullstedt <emil.tullstedt@grafana.com>
Co-authored-by: Dave Henderson <dave.henderson@grafana.com>
This commit is contained in:
Carl Bergquist
2022-09-05 07:39:22 +02:00
committed by GitHub
parent 7e53bd107c
commit 78978048c3
4 changed files with 78 additions and 0 deletions

View File

@ -0,0 +1,19 @@
package log
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
)
func TestCountingDatabaseCalls(t *testing.T) {
ctx := context.Background()
ctx = IncDBCallCounter(ctx)
ctx = IncDBCallCounter(ctx)
ctx = IncDBCallCounter(ctx)
count := TotalDBCallCount(ctx)
assert.Equal(t, int64(3), count, "expect counter to increase three times")
}