mirror of
https://github.com/grafana/grafana.git
synced 2024-11-28 03:34:15 -06:00
05709ce411
* chore: add alias for InitTestDB and Session Adds an alias for the sqlstore InitTestDB and Session, and updates tests using these to reduce dependencies on the sqlstore.Store. * next pass of removing sqlstore imports * last little bit * remove mockstore where possible
96 lines
3.2 KiB
Go
96 lines
3.2 KiB
Go
package correlations
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/api/routing"
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/events"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/services/accesscontrol"
|
|
"github.com/grafana/grafana/pkg/services/datasources"
|
|
)
|
|
|
|
func ProvideService(sqlStore db.DB, routeRegister routing.RouteRegister, ds datasources.DataSourceService, ac accesscontrol.AccessControl, bus bus.Bus) *CorrelationsService {
|
|
s := &CorrelationsService{
|
|
SQLStore: sqlStore,
|
|
RouteRegister: routeRegister,
|
|
log: log.New("correlations"),
|
|
DataSourceService: ds,
|
|
AccessControl: ac,
|
|
}
|
|
|
|
s.registerAPIEndpoints()
|
|
|
|
bus.AddEventListener(s.handleDatasourceDeletion)
|
|
|
|
return s
|
|
}
|
|
|
|
type Service interface {
|
|
CreateCorrelation(ctx context.Context, cmd CreateCorrelationCommand) (Correlation, error)
|
|
DeleteCorrelation(ctx context.Context, cmd DeleteCorrelationCommand) error
|
|
DeleteCorrelationsBySourceUID(ctx context.Context, cmd DeleteCorrelationsBySourceUIDCommand) error
|
|
DeleteCorrelationsByTargetUID(ctx context.Context, cmd DeleteCorrelationsByTargetUIDCommand) error
|
|
}
|
|
|
|
type CorrelationsService struct {
|
|
SQLStore db.DB
|
|
RouteRegister routing.RouteRegister
|
|
log log.Logger
|
|
DataSourceService datasources.DataSourceService
|
|
AccessControl accesscontrol.AccessControl
|
|
}
|
|
|
|
func (s CorrelationsService) CreateCorrelation(ctx context.Context, cmd CreateCorrelationCommand) (Correlation, error) {
|
|
return s.createCorrelation(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) DeleteCorrelation(ctx context.Context, cmd DeleteCorrelationCommand) error {
|
|
return s.deleteCorrelation(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) UpdateCorrelation(ctx context.Context, cmd UpdateCorrelationCommand) (Correlation, error) {
|
|
return s.updateCorrelation(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) GetCorrelation(ctx context.Context, cmd GetCorrelationQuery) (Correlation, error) {
|
|
return s.getCorrelation(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) GetCorrelationsBySourceUID(ctx context.Context, cmd GetCorrelationsBySourceUIDQuery) ([]Correlation, error) {
|
|
return s.getCorrelationsBySourceUID(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) GetCorrelations(ctx context.Context, cmd GetCorrelationsQuery) ([]Correlation, error) {
|
|
return s.getCorrelations(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) DeleteCorrelationsBySourceUID(ctx context.Context, cmd DeleteCorrelationsBySourceUIDCommand) error {
|
|
return s.deleteCorrelationsBySourceUID(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) DeleteCorrelationsByTargetUID(ctx context.Context, cmd DeleteCorrelationsByTargetUIDCommand) error {
|
|
return s.deleteCorrelationsByTargetUID(ctx, cmd)
|
|
}
|
|
|
|
func (s CorrelationsService) handleDatasourceDeletion(ctx context.Context, event *events.DataSourceDeleted) error {
|
|
return s.SQLStore.InTransaction(ctx, func(ctx context.Context) error {
|
|
if err := s.deleteCorrelationsBySourceUID(ctx, DeleteCorrelationsBySourceUIDCommand{
|
|
SourceUID: event.UID,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := s.deleteCorrelationsByTargetUID(ctx, DeleteCorrelationsByTargetUIDCommand{
|
|
TargetUID: event.UID,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|