mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
Search: in-memory index (#47709)
* #45498: add entity events table
* #45498: add entity events service
* #45498: hook up entity events service to http server
* #45498: use `dashboards.id` rather than `uid` and `org_id` in grn
* Update pkg/services/entityevents/service.go
Co-authored-by: Ryan McKinley <ryantxu@gmail.com>
* #45498: move entityeventsservice to services/store
* #45498: add null check
* #45498: rename
* #45498: fix comment
* #45498: switch grn back to uid
* Search: listen for updates (#47719)
* #45498: wire entity event service with searchv2
* load last event id before building index for org 1
* fix service init in integration tests
* depend on required subset of event store methods
* Update pkg/services/sqlstore/migrations/entity_events_mig.go
Co-authored-by: Alexander Emelin <frvzmb@gmail.com>
* #45498: pointer receiver
* #45498: mockery!
* #45498: add entity events service to background services
* dashboard query pagination, allow queries while re-indexing
* log level cleanups, use rlock, add comments
* fix lint, check feature toggle in search v2 service
* use unix time for event created column
* add missing changes for created column
* fix integration tests init
* log re-index execution times on info level
* #45498: fix entityEventsService tests
* #45498: save events on dashboard delete
* use camel case for log labels
* formatting
* #45498: rename grn to entityid
* #45498: add `IsDisabled` to entityEventsService
* #45498: remove feature flag from migration
* better context usage, fix capacity, comments/cleanups
* replace print with logger
* Revert "#45498: remove feature flag from migration"
This reverts commit ed23968898.
* revert:revert:revert conditional feature flag
Co-authored-by: Ryan McKinley <ryantxu@gmail.com>
Co-authored-by: Alexander Emelin <frvzmb@gmail.com>
This commit is contained in:
182
pkg/services/store/entity_events_test.go
Normal file
182
pkg/services/store/entity_events_test.go
Normal file
@@ -0,0 +1,182 @@
|
||||
//go:build integration
|
||||
// +build integration
|
||||
|
||||
package store
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/grafana/grafana/pkg/infra/log"
|
||||
"github.com/grafana/grafana/pkg/services/sqlstore"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestEntityEventsService(t *testing.T) {
|
||||
var ctx context.Context
|
||||
var service EntityEventsService
|
||||
|
||||
setup := func() {
|
||||
service = &entityEventService{
|
||||
sql: sqlstore.InitTestDB(t),
|
||||
log: log.New("entity-event-test"),
|
||||
}
|
||||
ctx = context.Background()
|
||||
}
|
||||
|
||||
t.Run("Should insert an entity event", func(t *testing.T) {
|
||||
setup()
|
||||
|
||||
err := service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/1",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
})
|
||||
|
||||
t.Run("Should retrieve nil entity if database is empty", func(t *testing.T) {
|
||||
setup()
|
||||
|
||||
ev, err := service.GetLastEvent(ctx)
|
||||
require.NoError(t, err)
|
||||
require.Nil(t, ev)
|
||||
})
|
||||
|
||||
t.Run("Should retrieve last entity event", func(t *testing.T) {
|
||||
setup()
|
||||
lastEventEntityId := "database/dash/1"
|
||||
|
||||
err := service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/3",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
err = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/2",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
err = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: lastEventEntityId,
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
lastEv, err := service.GetLastEvent(ctx)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, lastEventEntityId, lastEv.EntityId)
|
||||
})
|
||||
|
||||
t.Run("Should retrieve sorted events after an id", func(t *testing.T) {
|
||||
setup()
|
||||
lastEventEntityId := "database/dash/1"
|
||||
|
||||
err := service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/3",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
firstEv, err := service.GetLastEvent(ctx)
|
||||
firstEvId := firstEv.Id
|
||||
|
||||
err = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/2",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
err = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: lastEventEntityId,
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
evs, err := service.GetAllEventsAfter(ctx, firstEvId)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, evs, 2)
|
||||
require.Equal(t, evs[0].EntityId, "database/dash/2")
|
||||
require.Equal(t, evs[1].EntityId, lastEventEntityId)
|
||||
})
|
||||
|
||||
t.Run("Should delete old events", func(t *testing.T) {
|
||||
setup()
|
||||
_ = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/3",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
_ = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/2",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
_ = service.SaveEvent(ctx, SaveEventCmd{
|
||||
EntityId: "database/dash/1",
|
||||
EventType: EntityEventTypeCreate,
|
||||
})
|
||||
|
||||
evs, err := service.GetAllEventsAfter(ctx, 0)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, evs, 3)
|
||||
|
||||
err = service.deleteEventsOlderThan(ctx, 24*time.Hour)
|
||||
require.NoError(t, err)
|
||||
|
||||
// did not delete any events
|
||||
evs, err = service.GetAllEventsAfter(ctx, 0)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, evs, 3)
|
||||
|
||||
time.Sleep(2 * time.Second)
|
||||
err = service.deleteEventsOlderThan(ctx, 1*time.Second)
|
||||
require.NoError(t, err)
|
||||
|
||||
// deleted all events
|
||||
evs, err = service.GetAllEventsAfter(ctx, 0)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, evs, 0)
|
||||
})
|
||||
}
|
||||
|
||||
func TestCreateDatabaseEntityId(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
entityType EntityType
|
||||
orgId int64
|
||||
internalId interface{}
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
name: "int64 internal id",
|
||||
entityType: EntityTypeDashboard,
|
||||
orgId: 10,
|
||||
internalId: int64(45),
|
||||
expected: "database/10/dashboard/45",
|
||||
},
|
||||
{
|
||||
name: "big-ish int64 internal id",
|
||||
entityType: EntityTypeDashboard,
|
||||
orgId: 10,
|
||||
internalId: int64(12412421),
|
||||
expected: "database/10/dashboard/12412421",
|
||||
},
|
||||
{
|
||||
name: "int internal id",
|
||||
entityType: EntityTypeDashboard,
|
||||
orgId: 10,
|
||||
internalId: int(1244),
|
||||
expected: "database/10/dashboard/1244",
|
||||
},
|
||||
{
|
||||
name: "string internal id",
|
||||
entityType: EntityTypeDashboard,
|
||||
orgId: 10,
|
||||
internalId: "string-internal-id",
|
||||
expected: "database/10/dashboard/string-internal-id",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
require.Equal(t, tt.expected, CreateDatabaseEntityId(tt.internalId, tt.orgId, tt.entityType))
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user