mirror of
https://github.com/grafana/grafana.git
synced 2025-02-14 17:43:35 -06:00
09c4dbdb9f
* Correlations: add UpdateCorrelation HTTP API * handle correlation not found * add tests * fix lint errors * add bad request to API spec * change casing * fix casing in docs * fix tests * update spec
84 lines
2.8 KiB
Go
84 lines
2.8 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/log"
|
|
"github.com/grafana/grafana/pkg/services/accesscontrol"
|
|
"github.com/grafana/grafana/pkg/services/datasources"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
)
|
|
|
|
func ProvideService(sqlStore *sqlstore.SQLStore, 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 *sqlstore.SQLStore
|
|
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) 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
|
|
})
|
|
}
|