mirror of
https://github.com/grafana/grafana.git
synced 2025-09-20 14:02:52 +08:00
3589baac68
* Batch write events * Improve instrumentation * Measure batch phases * Detect lock contention * remove the execBatch goroutine * removing tracing prefix * detect context cancel * batch channel map
42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
infraDB "github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
|
"github.com/grafana/grafana/pkg/storage/unified/sql"
|
|
"github.com/grafana/grafana/pkg/storage/unified/sql/db/dbimpl"
|
|
test "github.com/grafana/grafana/pkg/storage/unified/testing"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newTestBackend(b *testing.B) resource.StorageBackend {
|
|
dbstore := infraDB.InitTestDB(b)
|
|
eDB, err := dbimpl.ProvideResourceDB(dbstore, setting.NewCfg(), nil)
|
|
require.NoError(b, err)
|
|
require.NotNil(b, eDB)
|
|
|
|
backend, err := sql.NewBackend(sql.BackendOptions{
|
|
DBProvider: eDB,
|
|
IsHA: true,
|
|
SimulatedNetworkLatency: 2 * time.Millisecond, // to simulate some network latency
|
|
})
|
|
require.NoError(b, err)
|
|
require.NotNil(b, backend)
|
|
err = backend.Init(context.Background())
|
|
require.NoError(b, err)
|
|
return backend
|
|
}
|
|
|
|
func BenchmarkSQLStorageBackend(b *testing.B) {
|
|
opts := test.DefaultBenchmarkOptions()
|
|
if infraDB.IsTestDbSQLite() {
|
|
opts.Concurrency = 1 // to avoid SQLite database is locked error
|
|
}
|
|
test.BenchmarkStorageBackend(b, newTestBackend(b), opts)
|
|
}
|