2021-04-01 03:11:45 -05:00
package api
import (
2022-02-23 10:30:04 -06:00
"context"
2021-04-15 07:54:37 -05:00
"errors"
2021-09-02 11:38:42 -05:00
"fmt"
2021-04-01 03:11:45 -05:00
"net/http"
"time"
2022-03-21 18:20:35 -05:00
"github.com/grafana/grafana/pkg/services/accesscontrol"
2022-01-11 10:39:34 -06:00
"github.com/grafana/grafana/pkg/services/datasources"
2022-04-05 16:48:51 -05:00
"github.com/grafana/grafana/pkg/services/ngalert/provisioning"
2022-01-11 10:39:34 -06:00
"github.com/grafana/grafana/pkg/services/ngalert/store"
"github.com/grafana/grafana/pkg/services/quota"
2022-02-23 10:30:04 -06:00
"github.com/grafana/grafana/pkg/setting"
2022-03-04 15:16:33 -06:00
"github.com/grafana/grafana/pkg/util/cmputil"
2022-01-11 10:39:34 -06:00
"github.com/prometheus/common/model"
2021-08-25 08:11:22 -05:00
"github.com/grafana/grafana/pkg/api/apierrors"
2021-04-01 03:11:45 -05:00
"github.com/grafana/grafana/pkg/api/response"
"github.com/grafana/grafana/pkg/infra/log"
"github.com/grafana/grafana/pkg/models"
2021-04-19 13:26:04 -05:00
apimodels "github.com/grafana/grafana/pkg/services/ngalert/api/tooling/definitions"
2021-04-01 03:11:45 -05:00
ngmodels "github.com/grafana/grafana/pkg/services/ngalert/models"
2022-01-11 10:39:34 -06:00
"github.com/grafana/grafana/pkg/services/ngalert/schedule"
2021-04-01 03:11:45 -05:00
"github.com/grafana/grafana/pkg/util"
2021-10-11 07:30:59 -05:00
"github.com/grafana/grafana/pkg/web"
2021-04-01 03:11:45 -05:00
)
type RulerSrv struct {
2022-04-05 16:48:51 -05:00
xactManager provisioning . TransactionManager
2022-04-28 14:27:34 -05:00
provenanceStore provisioning . ProvisioningStore
2021-04-28 03:31:51 -05:00
store store . RuleStore
DatasourceCache datasources . CacheService
2021-05-04 11:16:28 -05:00
QuotaService * quota . QuotaService
2022-01-11 10:39:34 -06:00
scheduleService schedule . ScheduleService
2021-04-28 03:31:51 -05:00
log log . Logger
2022-02-23 10:30:04 -06:00
cfg * setting . UnifiedAlertingSettings
2022-03-21 18:20:35 -05:00
ac accesscontrol . AccessControl
2021-04-01 03:11:45 -05:00
}
2022-02-23 10:30:04 -06:00
var (
errQuotaReached = errors . New ( "quota has been exceeded" )
)
2022-03-25 11:39:24 -05:00
// RouteDeleteAlertRules deletes all alert rules user is authorized to access in the namespace (request parameter :Namespace)
// or, if specified, a group of rules (request parameter :Groupname) in the namespace
func ( srv RulerSrv ) RouteDeleteAlertRules ( c * models . ReqContext ) response . Response {
2021-10-11 07:30:59 -05:00
namespaceTitle := web . Params ( c . Req ) [ ":Namespace" ]
2021-09-14 09:08:04 -05:00
namespace , err := srv . store . GetNamespaceByTitle ( c . Req . Context ( ) , namespaceTitle , c . SignedInUser . OrgId , c . SignedInUser , true )
2021-04-01 03:11:45 -05:00
if err != nil {
2021-04-15 07:54:37 -05:00
return toNamespaceErrorResponse ( err )
2021-04-01 03:11:45 -05:00
}
2022-03-25 11:39:24 -05:00
var loggerCtx = [ ] interface { } {
"namespace" ,
namespace . Title ,
}
2022-04-25 05:42:42 -05:00
var ruleGroup string
2022-03-25 11:39:24 -05:00
if group , ok := web . Params ( c . Req ) [ ":Groupname" ] ; ok {
2022-04-25 05:42:42 -05:00
ruleGroup = group
2022-03-25 11:39:24 -05:00
loggerCtx = append ( loggerCtx , "group" , group )
2021-04-01 03:11:45 -05:00
}
2022-03-25 11:39:24 -05:00
logger := srv . log . New ( loggerCtx ... )
2021-05-03 13:01:33 -05:00
2022-03-25 11:39:24 -05:00
hasAccess := func ( evaluator accesscontrol . Evaluator ) bool {
return accesscontrol . HasAccess ( srv . ac , c ) ( accesscontrol . ReqOrgAdminOrEditor , evaluator )
2021-05-03 13:01:33 -05:00
}
2022-05-06 13:55:27 -05:00
provenances , err := srv . provenanceStore . GetProvenances ( c . Req . Context ( ) , c . SignedInUser . OrgId , ( & ngmodels . AlertRule { } ) . ResourceType ( ) )
if err != nil {
return ErrResp ( http . StatusInternalServerError , err , "failed to fetch provenances of alert rules" )
}
var deletableRules [ ] string
2022-03-25 11:39:24 -05:00
err = srv . xactManager . InTransaction ( c . Req . Context ( ) , func ( ctx context . Context ) error {
2022-04-25 05:42:42 -05:00
q := ngmodels . ListAlertRulesQuery {
OrgID : c . SignedInUser . OrgId ,
NamespaceUIDs : [ ] string { namespace . Uid } ,
RuleGroup : ruleGroup ,
2022-03-25 11:39:24 -05:00
}
2022-04-25 05:42:42 -05:00
if err = srv . store . ListAlertRules ( ctx , & q ) ; err != nil {
2022-03-25 11:39:24 -05:00
return err
}
2021-04-01 03:11:45 -05:00
2022-03-25 11:39:24 -05:00
if len ( q . Result ) == 0 {
logger . Debug ( "no alert rules to delete from namespace/group" )
return nil
}
2022-05-06 13:55:27 -05:00
var canDelete [ ] * ngmodels . AlertRule
var cannotDelete [ ] string
// partition will partation the given rules in two, one partition
// being the rules that fulfill the predicate the other partation being
// the ruleIDs not fulfilling it.
partition := func ( alerts [ ] * ngmodels . AlertRule , predicate func ( rule * ngmodels . AlertRule ) bool ) ( [ ] * ngmodels . AlertRule , [ ] string ) {
positive , negative := make ( [ ] * ngmodels . AlertRule , 0 , len ( alerts ) ) , make ( [ ] string , 0 , len ( alerts ) )
for _ , rule := range alerts {
if predicate ( rule ) {
positive = append ( positive , rule )
continue
}
negative = append ( negative , rule . UID )
2022-03-25 11:39:24 -05:00
}
2022-05-06 13:55:27 -05:00
return positive , negative
2022-03-25 11:39:24 -05:00
}
2022-05-06 13:55:27 -05:00
canDelete , cannotDelete = partition ( q . Result , func ( rule * ngmodels . AlertRule ) bool {
return authorizeDatasourceAccessForRule ( rule , hasAccess )
} )
2022-03-25 11:39:24 -05:00
if len ( canDelete ) == 0 {
return fmt . Errorf ( "%w to delete rules because user is not authorized to access data sources used by the rules" , ErrAuthorization )
}
if len ( cannotDelete ) > 0 {
logger . Info ( "user cannot delete one or many alert rules because it does not have access to data sources. Those rules will be skipped" , "expected" , len ( q . Result ) , "authorized" , len ( canDelete ) , "unauthorized" , cannotDelete )
}
2022-05-06 13:55:27 -05:00
canDelete , cannotDelete = partition ( canDelete , func ( rule * ngmodels . AlertRule ) bool {
provenance , exists := provenances [ rule . UID ]
return ( exists && provenance == ngmodels . ProvenanceNone ) || ! exists
} )
if len ( canDelete ) == 0 {
return fmt . Errorf ( "all rules have been provisioned and cannot be deleted through this api" )
}
if len ( cannotDelete ) > 0 {
logger . Info ( "user cannot delete one or many alert rules because it does have a provenance set. Those rules will be skipped" , "expected" , len ( q . Result ) , "provenance_none" , len ( canDelete ) , "provenance_set" , cannotDelete )
}
for _ , rule := range canDelete {
deletableRules = append ( deletableRules , rule . UID )
}
return srv . store . DeleteAlertRulesByUID ( ctx , c . SignedInUser . OrgId , deletableRules ... )
2022-03-25 11:39:24 -05:00
} )
2021-05-03 13:01:33 -05:00
if err != nil {
2022-03-25 11:39:24 -05:00
if errors . Is ( err , ErrAuthorization ) {
return ErrResp ( http . StatusUnauthorized , err , "" )
2021-04-16 07:00:07 -05:00
}
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusInternalServerError , err , "failed to delete rule group" )
2021-04-01 03:11:45 -05:00
}
2021-04-16 07:00:07 -05:00
2022-03-25 11:39:24 -05:00
logger . Debug ( "rules have been deleted from the store. updating scheduler" )
2022-05-06 13:55:27 -05:00
for _ , uid := range deletableRules {
2022-01-11 10:39:34 -06:00
srv . scheduleService . DeleteAlertRule ( ngmodels . AlertRuleKey {
OrgID : c . SignedInUser . OrgId ,
UID : uid ,
} )
2021-05-03 13:01:33 -05:00
}
2022-03-25 11:39:24 -05:00
return response . JSON ( http . StatusAccepted , util . DynMap { "message" : "rules deleted" } )
2021-04-01 03:11:45 -05:00
}
func ( srv RulerSrv ) RouteGetNamespaceRulesConfig ( c * models . ReqContext ) response . Response {
2021-10-11 07:30:59 -05:00
namespaceTitle := web . Params ( c . Req ) [ ":Namespace" ]
2021-09-14 09:08:04 -05:00
namespace , err := srv . store . GetNamespaceByTitle ( c . Req . Context ( ) , namespaceTitle , c . SignedInUser . OrgId , c . SignedInUser , false )
2021-04-01 03:11:45 -05:00
if err != nil {
2021-04-15 07:54:37 -05:00
return toNamespaceErrorResponse ( err )
2021-04-01 03:11:45 -05:00
}
2022-04-25 05:42:42 -05:00
q := ngmodels . ListAlertRulesQuery {
OrgID : c . SignedInUser . OrgId ,
NamespaceUIDs : [ ] string { namespace . Uid } ,
2021-04-01 03:11:45 -05:00
}
2022-04-25 05:42:42 -05:00
if err := srv . store . ListAlertRules ( c . Req . Context ( ) , & q ) ; err != nil {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusInternalServerError , err , "failed to update rule group" )
2021-04-01 03:11:45 -05:00
}
result := apimodels . NamespaceConfigResponse { }
ruleGroupConfigs := make ( map [ string ] apimodels . GettableRuleGroupConfig )
2022-04-11 16:37:44 -05:00
hasAccess := func ( evaluator accesscontrol . Evaluator ) bool {
return accesscontrol . HasAccess ( srv . ac , c ) ( accesscontrol . ReqSignedIn , evaluator )
}
2022-04-28 14:27:34 -05:00
provenanceRecords , err := srv . provenanceStore . GetProvenances ( c . Req . Context ( ) , c . SignedInUser . OrgId , ( & ngmodels . AlertRule { } ) . ResourceType ( ) )
if err != nil {
return ErrResp ( http . StatusInternalServerError , err , "failed to get provenance for rule group" )
}
2021-04-01 03:11:45 -05:00
for _ , r := range q . Result {
2022-04-11 16:37:44 -05:00
if ! authorizeDatasourceAccessForRule ( r , hasAccess ) {
continue
}
2021-04-01 03:11:45 -05:00
ruleGroupConfig , ok := ruleGroupConfigs [ r . RuleGroup ]
if ! ok {
ruleGroupInterval := model . Duration ( time . Duration ( r . IntervalSeconds ) * time . Second )
ruleGroupConfigs [ r . RuleGroup ] = apimodels . GettableRuleGroupConfig {
Name : r . RuleGroup ,
Interval : ruleGroupInterval ,
Rules : [ ] apimodels . GettableExtendedRuleNode {
2022-04-28 14:27:34 -05:00
toGettableExtendedRuleNode ( * r , namespace . Id , provenanceRecords ) ,
2021-04-01 03:11:45 -05:00
} ,
}
} else {
2022-04-28 14:27:34 -05:00
ruleGroupConfig . Rules = append ( ruleGroupConfig . Rules , toGettableExtendedRuleNode ( * r , namespace . Id , provenanceRecords ) )
2021-04-01 03:11:45 -05:00
ruleGroupConfigs [ r . RuleGroup ] = ruleGroupConfig
}
}
for _ , ruleGroupConfig := range ruleGroupConfigs {
2021-04-15 07:54:37 -05:00
result [ namespaceTitle ] = append ( result [ namespaceTitle ] , ruleGroupConfig )
2021-04-01 03:11:45 -05:00
}
return response . JSON ( http . StatusAccepted , result )
}
2022-04-28 14:27:34 -05:00
func ( srv RulerSrv ) RouteGetRulesGroupConfig ( c * models . ReqContext ) response . Response {
2021-10-11 07:30:59 -05:00
namespaceTitle := web . Params ( c . Req ) [ ":Namespace" ]
2021-09-14 09:08:04 -05:00
namespace , err := srv . store . GetNamespaceByTitle ( c . Req . Context ( ) , namespaceTitle , c . SignedInUser . OrgId , c . SignedInUser , false )
2021-04-01 03:11:45 -05:00
if err != nil {
2021-04-15 07:54:37 -05:00
return toNamespaceErrorResponse ( err )
2021-04-01 03:11:45 -05:00
}
2021-10-11 07:30:59 -05:00
ruleGroup := web . Params ( c . Req ) [ ":Groupname" ]
2022-04-25 05:42:42 -05:00
q := ngmodels . ListAlertRulesQuery {
OrgID : c . SignedInUser . OrgId ,
NamespaceUIDs : [ ] string { namespace . Uid } ,
RuleGroup : ruleGroup ,
2021-04-01 03:11:45 -05:00
}
2022-04-25 05:42:42 -05:00
if err := srv . store . ListAlertRules ( c . Req . Context ( ) , & q ) ; err != nil {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusInternalServerError , err , "failed to get group alert rules" )
2021-04-01 03:11:45 -05:00
}
var ruleGroupInterval model . Duration
ruleNodes := make ( [ ] apimodels . GettableExtendedRuleNode , 0 , len ( q . Result ) )
2022-04-11 16:37:44 -05:00
hasAccess := func ( evaluator accesscontrol . Evaluator ) bool {
return accesscontrol . HasAccess ( srv . ac , c ) ( accesscontrol . ReqSignedIn , evaluator )
}
2022-04-28 14:27:34 -05:00
provenanceRecords , err := srv . provenanceStore . GetProvenances ( c . Req . Context ( ) , c . SignedInUser . OrgId , ( & ngmodels . AlertRule { } ) . ResourceType ( ) )
if err != nil {
return ErrResp ( http . StatusInternalServerError , err , "failed to get group alert rules" )
}
2021-04-01 03:11:45 -05:00
for _ , r := range q . Result {
2022-04-11 16:37:44 -05:00
if ! authorizeDatasourceAccessForRule ( r , hasAccess ) {
continue
}
2021-04-01 03:11:45 -05:00
ruleGroupInterval = model . Duration ( time . Duration ( r . IntervalSeconds ) * time . Second )
2022-04-28 14:27:34 -05:00
ruleNodes = append ( ruleNodes , toGettableExtendedRuleNode ( * r , namespace . Id , provenanceRecords ) )
2021-04-01 03:11:45 -05:00
}
result := apimodels . RuleGroupConfigResponse {
GettableRuleGroupConfig : apimodels . GettableRuleGroupConfig {
Name : ruleGroup ,
Interval : ruleGroupInterval ,
Rules : ruleNodes ,
} ,
}
return response . JSON ( http . StatusAccepted , result )
}
func ( srv RulerSrv ) RouteGetRulesConfig ( c * models . ReqContext ) response . Response {
2022-04-01 18:33:26 -05:00
namespaceMap , err := srv . store . GetUserVisibleNamespaces ( c . Req . Context ( ) , c . OrgId , c . SignedInUser )
2021-07-22 01:53:14 -05:00
if err != nil {
return ErrResp ( http . StatusInternalServerError , err , "failed to get namespaces visible to the user" )
}
2021-11-08 07:26:08 -06:00
result := apimodels . NamespaceConfigResponse { }
if len ( namespaceMap ) == 0 {
srv . log . Debug ( "User has no access to any namespaces" )
return response . JSON ( http . StatusOK , result )
}
2021-07-22 01:53:14 -05:00
namespaceUIDs := make ( [ ] string , len ( namespaceMap ) )
for k := range namespaceMap {
namespaceUIDs = append ( namespaceUIDs , k )
}
2021-10-04 10:33:55 -05:00
dashboardUID := c . Query ( "dashboard_uid" )
panelID , err := getPanelIDFromRequest ( c . Req )
if err != nil {
return ErrResp ( http . StatusBadRequest , err , "invalid panel_id" )
}
if dashboardUID == "" && panelID != 0 {
return ErrResp ( http . StatusBadRequest , errors . New ( "panel_id must be set with dashboard_uid" ) , "" )
}
2021-04-01 03:11:45 -05:00
q := ngmodels . ListAlertRulesQuery {
2021-07-22 01:53:14 -05:00
OrgID : c . SignedInUser . OrgId ,
NamespaceUIDs : namespaceUIDs ,
2021-10-04 10:33:55 -05:00
DashboardUID : dashboardUID ,
PanelID : panelID ,
2021-04-01 03:11:45 -05:00
}
2021-07-22 01:53:14 -05:00
2022-04-25 05:42:42 -05:00
if err := srv . store . ListAlertRules ( c . Req . Context ( ) , & q ) ; err != nil {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusInternalServerError , err , "failed to get alert rules" )
2021-04-01 03:11:45 -05:00
}
configs := make ( map [ string ] map [ string ] apimodels . GettableRuleGroupConfig )
2022-04-11 16:37:44 -05:00
hasAccess := func ( evaluator accesscontrol . Evaluator ) bool {
return accesscontrol . HasAccess ( srv . ac , c ) ( accesscontrol . ReqSignedIn , evaluator )
}
2022-04-28 14:27:34 -05:00
provenanceRecords , err := srv . provenanceStore . GetProvenances ( c . Req . Context ( ) , c . SignedInUser . OrgId , ( & ngmodels . AlertRule { } ) . ResourceType ( ) )
if err != nil {
return ErrResp ( http . StatusInternalServerError , err , "failed to get alert rules" )
}
2021-04-01 03:11:45 -05:00
for _ , r := range q . Result {
2022-04-11 16:37:44 -05:00
if ! authorizeDatasourceAccessForRule ( r , hasAccess ) {
continue
}
2021-07-22 01:53:14 -05:00
folder , ok := namespaceMap [ r . NamespaceUID ]
if ! ok {
srv . log . Error ( "namespace not visible to the user" , "user" , c . SignedInUser . UserId , "namespace" , r . NamespaceUID , "rule" , r . UID )
continue
2021-04-01 03:11:45 -05:00
}
2021-04-15 07:54:37 -05:00
namespace := folder . Title
2021-07-22 01:53:14 -05:00
_ , ok = configs [ namespace ]
2021-04-01 03:11:45 -05:00
if ! ok {
ruleGroupInterval := model . Duration ( time . Duration ( r . IntervalSeconds ) * time . Second )
configs [ namespace ] = make ( map [ string ] apimodels . GettableRuleGroupConfig )
configs [ namespace ] [ r . RuleGroup ] = apimodels . GettableRuleGroupConfig {
Name : r . RuleGroup ,
Interval : ruleGroupInterval ,
Rules : [ ] apimodels . GettableExtendedRuleNode {
2022-04-28 14:27:34 -05:00
toGettableExtendedRuleNode ( * r , folder . Id , provenanceRecords ) ,
2021-04-01 03:11:45 -05:00
} ,
}
} else {
ruleGroupConfig , ok := configs [ namespace ] [ r . RuleGroup ]
if ! ok {
ruleGroupInterval := model . Duration ( time . Duration ( r . IntervalSeconds ) * time . Second )
configs [ namespace ] [ r . RuleGroup ] = apimodels . GettableRuleGroupConfig {
Name : r . RuleGroup ,
Interval : ruleGroupInterval ,
Rules : [ ] apimodels . GettableExtendedRuleNode {
2022-04-28 14:27:34 -05:00
toGettableExtendedRuleNode ( * r , folder . Id , provenanceRecords ) ,
2021-04-01 03:11:45 -05:00
} ,
}
} else {
2022-04-28 14:27:34 -05:00
ruleGroupConfig . Rules = append ( ruleGroupConfig . Rules , toGettableExtendedRuleNode ( * r , folder . Id , provenanceRecords ) )
2021-04-01 03:11:45 -05:00
configs [ namespace ] [ r . RuleGroup ] = ruleGroupConfig
}
}
}
for namespace , m := range configs {
for _ , ruleGroupConfig := range m {
result [ namespace ] = append ( result [ namespace ] , ruleGroupConfig )
}
}
2021-10-04 10:33:55 -05:00
return response . JSON ( http . StatusOK , result )
2021-04-01 03:11:45 -05:00
}
func ( srv RulerSrv ) RoutePostNameRulesConfig ( c * models . ReqContext , ruleGroupConfig apimodels . PostableRuleGroupConfig ) response . Response {
2021-10-11 07:30:59 -05:00
namespaceTitle := web . Params ( c . Req ) [ ":Namespace" ]
2021-09-14 09:08:04 -05:00
namespace , err := srv . store . GetNamespaceByTitle ( c . Req . Context ( ) , namespaceTitle , c . SignedInUser . OrgId , c . SignedInUser , true )
2021-04-01 03:11:45 -05:00
if err != nil {
2021-04-15 07:54:37 -05:00
return toNamespaceErrorResponse ( err )
2021-04-01 03:11:45 -05:00
}
2022-02-23 10:30:04 -06:00
rules , err := validateRuleGroup ( & ruleGroupConfig , c . SignedInUser . OrgId , namespace , conditionValidator ( c , srv . DatasourceCache ) , srv . cfg )
if err != nil {
return ErrResp ( http . StatusBadRequest , err , "" )
2021-04-16 07:00:07 -05:00
}
2022-02-23 10:30:04 -06:00
return srv . updateAlertRulesInGroup ( c , namespace , ruleGroupConfig . Name , rules )
}
2022-03-24 15:53:00 -05:00
// updateAlertRulesInGroup calculates changes (rules to add,update,delete), verifies that the user is authorized to do the calculated changes and updates database.
// All operations are performed in a single transaction
2022-05-06 13:55:27 -05:00
//nolint: gocyclo
2022-02-23 10:30:04 -06:00
func ( srv RulerSrv ) updateAlertRulesInGroup ( c * models . ReqContext , namespace * models . Folder , groupName string , rules [ ] * ngmodels . AlertRule ) response . Response {
2022-05-06 13:55:27 -05:00
var finalChanges * changes
2022-03-21 18:20:35 -05:00
hasAccess := accesscontrol . HasAccess ( srv . ac , c )
2022-03-15 11:48:42 -05:00
err := srv . xactManager . InTransaction ( c . Req . Context ( ) , func ( tranCtx context . Context ) error {
2022-03-24 15:53:00 -05:00
logger := srv . log . New ( "namespace_uid" , namespace . Uid , "group" , groupName , "org_id" , c . OrgId , "user_id" , c . UserId )
groupChanges , err := calculateChanges ( tranCtx , srv . store , c . SignedInUser . OrgId , namespace , groupName , rules )
2022-02-23 10:30:04 -06:00
if err != nil {
return err
2021-04-28 03:31:51 -05:00
}
2022-02-23 10:30:04 -06:00
2022-03-04 15:16:33 -06:00
if groupChanges . isEmpty ( ) {
2022-05-06 13:55:27 -05:00
finalChanges = groupChanges
2022-03-24 15:53:00 -05:00
logger . Info ( "no changes detected in the request. Do nothing" )
2022-03-04 15:16:33 -06:00
return nil
2021-09-02 11:38:42 -05:00
}
2022-02-23 10:30:04 -06:00
2022-05-06 13:55:27 -05:00
authorizedChanges , err := authorizeRuleChanges ( namespace , groupChanges , func ( evaluator accesscontrol . Evaluator ) bool {
2022-03-21 18:20:35 -05:00
return hasAccess ( accesscontrol . ReqOrgAdminOrEditor , evaluator )
} )
if err != nil {
return err
}
2022-03-24 15:53:00 -05:00
if authorizedChanges . isEmpty ( ) {
logger . Info ( "no authorized changes detected in the request. Do nothing" , "not_authorized_add" , len ( groupChanges . New ) , "not_authorized_update" , len ( groupChanges . Update ) , "not_authorized_delete" , len ( groupChanges . Delete ) )
return nil
}
if len ( groupChanges . Delete ) > len ( authorizedChanges . Delete ) {
logger . Info ( "user is not authorized to delete one or many rules in the group. those rules will be skipped" , "expected" , len ( groupChanges . Delete ) , "authorized" , len ( authorizedChanges . Delete ) )
}
2022-05-06 13:55:27 -05:00
provenances , err := srv . provenanceStore . GetProvenances ( c . Req . Context ( ) , c . OrgId , ( & ngmodels . AlertRule { } ) . ResourceType ( ) )
if err != nil {
return err
}
// New rules don't need to be checked for provenance, just copy the whole slice.
finalChanges = & changes { }
finalChanges . New = authorizedChanges . New
for _ , rule := range authorizedChanges . Update {
if provenance , exists := provenances [ rule . Existing . UID ] ; ( exists && provenance == ngmodels . ProvenanceNone ) || ! exists {
finalChanges . Update = append ( finalChanges . Update , rule )
}
}
for _ , rule := range authorizedChanges . Delete {
if provenance , exists := provenances [ rule . UID ] ; ( exists && provenance == ngmodels . ProvenanceNone ) || ! exists {
finalChanges . Delete = append ( finalChanges . Delete , rule )
}
}
if finalChanges . isEmpty ( ) {
logger . Info ( "no changes detected that have 'none' provenance in the request. Do nothing" ,
"provenance_invalid_add" , len ( authorizedChanges . New ) ,
"provenance_invalid_update" , len ( authorizedChanges . Update ) ,
"provenance_invalid_delete" , len ( authorizedChanges . Delete ) )
return nil
}
if len ( authorizedChanges . Delete ) > len ( finalChanges . Delete ) {
logger . Info ( "provenance is not 'none' for one or many rules in the group that should be deleted. those rules will be skipped" ,
"expected" , len ( authorizedChanges . Delete ) ,
"allowed" , len ( authorizedChanges . Delete ) )
}
if len ( authorizedChanges . Update ) > len ( finalChanges . Update ) {
logger . Info ( "provenance is not 'none' for one or many rules in the group that should be updated. those rules will be skipped" ,
"expected" , len ( authorizedChanges . Update ) ,
"allowed" , len ( authorizedChanges . Update ) )
}
logger . Debug ( "updating database with the authorized changes" , "add" , len ( finalChanges . New ) , "update" , len ( finalChanges . New ) , "delete" , len ( finalChanges . Delete ) )
2022-03-21 18:20:35 -05:00
2022-05-06 13:55:27 -05:00
if len ( finalChanges . Update ) > 0 || len ( finalChanges . New ) > 0 {
updates := make ( [ ] store . UpdateRule , 0 , len ( finalChanges . Update ) )
inserts := make ( [ ] ngmodels . AlertRule , 0 , len ( finalChanges . New ) )
for _ , update := range finalChanges . Update {
2022-03-24 15:53:00 -05:00
logger . Debug ( "updating rule" , "rule_uid" , update . New . UID , "diff" , update . Diff . String ( ) )
2022-04-14 07:21:36 -05:00
updates = append ( updates , store . UpdateRule {
2022-03-04 15:16:33 -06:00
Existing : update . Existing ,
New : * update . New ,
} )
}
2022-05-06 13:55:27 -05:00
for _ , rule := range finalChanges . New {
2022-04-14 07:21:36 -05:00
inserts = append ( inserts , * rule )
}
err = srv . store . InsertAlertRules ( tranCtx , inserts )
if err != nil {
return fmt . Errorf ( "failed to add rules: %w" , err )
2022-03-04 15:16:33 -06:00
}
2022-04-14 07:21:36 -05:00
err = srv . store . UpdateAlertRules ( tranCtx , updates )
2022-03-04 15:16:33 -06:00
if err != nil {
2022-04-14 07:21:36 -05:00
return fmt . Errorf ( "failed to update rules: %w" , err )
2022-03-04 15:16:33 -06:00
}
}
2022-05-06 13:55:27 -05:00
if len ( finalChanges . Delete ) > 0 {
UIDs := make ( [ ] string , 0 , len ( finalChanges . Delete ) )
for _ , rule := range finalChanges . Delete {
2022-03-23 15:09:53 -05:00
UIDs = append ( UIDs , rule . UID )
}
if err = srv . store . DeleteAlertRulesByUID ( tranCtx , c . SignedInUser . OrgId , UIDs ... ) ; err != nil {
return fmt . Errorf ( "failed to delete rules: %w" , err )
2021-09-02 11:38:42 -05:00
}
}
2022-05-06 13:55:27 -05:00
if len ( finalChanges . New ) > 0 {
2022-02-23 10:30:04 -06:00
limitReached , err := srv . QuotaService . CheckQuotaReached ( tranCtx , "alert_rule" , & quota . ScopeParameters {
OrgId : c . OrgId ,
UserId : c . UserId ,
} ) // alert rule is table name
if err != nil {
return fmt . Errorf ( "failed to get alert rules quota: %w" , err )
}
if limitReached {
return errQuotaReached
}
2021-04-28 03:31:51 -05:00
}
2022-02-23 10:30:04 -06:00
return nil
} )
2021-04-28 03:31:51 -05:00
2022-02-23 10:30:04 -06:00
if err != nil {
2021-04-15 07:54:37 -05:00
if errors . Is ( err , ngmodels . ErrAlertRuleNotFound ) {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusNotFound , err , "failed to update rule group" )
2021-04-21 09:22:58 -05:00
} else if errors . Is ( err , ngmodels . ErrAlertRuleFailedValidation ) {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusBadRequest , err , "failed to update rule group" )
2022-02-23 10:30:04 -06:00
} else if errors . Is ( err , errQuotaReached ) {
return ErrResp ( http . StatusForbidden , err , "" )
2022-03-21 18:20:35 -05:00
} else if errors . Is ( err , ErrAuthorization ) {
return ErrResp ( http . StatusUnauthorized , err , "" )
2021-04-15 07:54:37 -05:00
}
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusInternalServerError , err , "failed to update rule group" )
2021-04-01 03:11:45 -05:00
}
2022-05-06 13:55:27 -05:00
for _ , rule := range finalChanges . Update {
2022-03-04 15:16:33 -06:00
srv . scheduleService . UpdateAlertRule ( ngmodels . AlertRuleKey {
OrgID : c . SignedInUser . OrgId ,
UID : rule . Existing . UID ,
} )
}
2022-02-23 10:30:04 -06:00
2022-05-06 13:55:27 -05:00
for _ , rule := range finalChanges . Delete {
2022-02-23 10:30:04 -06:00
srv . scheduleService . DeleteAlertRule ( ngmodels . AlertRuleKey {
2022-01-11 10:39:34 -06:00
OrgID : c . SignedInUser . OrgId ,
2022-02-23 10:30:04 -06:00
UID : rule . UID ,
2022-01-11 10:39:34 -06:00
} )
2021-05-06 11:39:34 -05:00
}
2022-05-06 13:55:27 -05:00
if finalChanges . isEmpty ( ) {
2022-03-04 15:16:33 -06:00
return response . JSON ( http . StatusAccepted , util . DynMap { "message" : "no changes detected in the rule group" } )
}
2021-04-01 03:11:45 -05:00
return response . JSON ( http . StatusAccepted , util . DynMap { "message" : "rule group updated successfully" } )
}
2022-04-28 14:27:34 -05:00
func toGettableExtendedRuleNode ( r ngmodels . AlertRule , namespaceID int64 , provenanceRecords map [ string ] ngmodels . Provenance ) apimodels . GettableExtendedRuleNode {
provenance := ngmodels . ProvenanceNone
if prov , exists := provenanceRecords [ r . ResourceID ( ) ] ; exists {
provenance = prov
}
2021-04-15 07:54:37 -05:00
gettableExtendedRuleNode := apimodels . GettableExtendedRuleNode {
2021-04-01 03:11:45 -05:00
GrafanaManagedAlert : & apimodels . GettableGrafanaRule {
ID : r . ID ,
OrgID : r . OrgID ,
Title : r . Title ,
Condition : r . Condition ,
Data : r . Data ,
Updated : r . Updated ,
IntervalSeconds : r . IntervalSeconds ,
Version : r . Version ,
UID : r . UID ,
NamespaceUID : r . NamespaceUID ,
2021-04-15 07:54:37 -05:00
NamespaceID : namespaceID ,
2021-04-01 03:11:45 -05:00
RuleGroup : r . RuleGroup ,
NoDataState : apimodels . NoDataState ( r . NoDataState ) ,
ExecErrState : apimodels . ExecutionErrorState ( r . ExecErrState ) ,
2022-04-28 14:27:34 -05:00
Provenance : provenance ,
2021-04-01 03:11:45 -05:00
} ,
}
2021-04-15 07:54:37 -05:00
gettableExtendedRuleNode . ApiRuleNode = & apimodels . ApiRuleNode {
For : model . Duration ( r . For ) ,
Annotations : r . Annotations ,
Labels : r . Labels ,
}
return gettableExtendedRuleNode
2021-04-01 03:11:45 -05:00
}
2021-04-07 07:28:06 -05:00
2021-04-15 07:54:37 -05:00
func toNamespaceErrorResponse ( err error ) response . Response {
if errors . Is ( err , ngmodels . ErrCannotEditNamespace ) {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusForbidden , err , err . Error ( ) )
2021-04-15 07:54:37 -05:00
}
if errors . Is ( err , models . ErrDashboardIdentifierNotSet ) {
2021-05-28 10:55:03 -05:00
return ErrResp ( http . StatusBadRequest , err , err . Error ( ) )
2021-04-15 07:54:37 -05:00
}
2021-08-25 08:11:22 -05:00
return apierrors . ToFolderErrorResponse ( err )
2021-04-07 07:28:06 -05:00
}
2022-02-23 10:30:04 -06:00
2022-03-04 15:16:33 -06:00
type ruleUpdate struct {
Existing * ngmodels . AlertRule
New * ngmodels . AlertRule
Diff cmputil . DiffReport
}
type changes struct {
New [ ] * ngmodels . AlertRule
Update [ ] ruleUpdate
Delete [ ] * ngmodels . AlertRule
}
func ( c * changes ) isEmpty ( ) bool {
return len ( c . Update ) + len ( c . New ) + len ( c . Delete ) == 0
2022-02-23 10:30:04 -06:00
}
// calculateChanges calculates the difference between rules in the group in the database and the submitted rules. If a submitted rule has UID it tries to find it in the database (in other groups).
// returns a list of rules that need to be added, updated and deleted. Deleted considered rules in the database that belong to the group but do not exist in the list of submitted rules.
2022-03-04 15:16:33 -06:00
func calculateChanges ( ctx context . Context , ruleStore store . RuleStore , orgId int64 , namespace * models . Folder , ruleGroupName string , submittedRules [ ] * ngmodels . AlertRule ) ( * changes , error ) {
2022-04-25 05:42:42 -05:00
q := & ngmodels . ListAlertRulesQuery {
OrgID : orgId ,
NamespaceUIDs : [ ] string { namespace . Uid } ,
RuleGroup : ruleGroupName ,
2022-02-23 10:30:04 -06:00
}
2022-04-25 05:42:42 -05:00
if err := ruleStore . ListAlertRules ( ctx , q ) ; err != nil {
2022-02-23 10:30:04 -06:00
return nil , fmt . Errorf ( "failed to query database for rules in the group %s: %w" , ruleGroupName , err )
}
existingGroupRules := q . Result
existingGroupRulesUIDs := make ( map [ string ] * ngmodels . AlertRule , len ( existingGroupRules ) )
for _ , r := range existingGroupRules {
existingGroupRulesUIDs [ r . UID ] = r
}
2022-03-04 15:16:33 -06:00
var toAdd , toDelete [ ] * ngmodels . AlertRule
var toUpdate [ ] ruleUpdate
2022-02-23 10:30:04 -06:00
for _ , r := range submittedRules {
var existing * ngmodels . AlertRule = nil
if r . UID != "" {
if existingGroupRule , ok := existingGroupRulesUIDs [ r . UID ] ; ok {
existing = existingGroupRule
// remove the rule from existingGroupRulesUIDs
delete ( existingGroupRulesUIDs , r . UID )
} else {
// Rule can be from other group or namespace
q := & ngmodels . GetAlertRuleByUIDQuery { OrgID : orgId , UID : r . UID }
if err := ruleStore . GetAlertRuleByUID ( ctx , q ) ; err != nil || q . Result == nil {
// if rule has UID then it is considered an update. Therefore, fail if there is no rule to update
if errors . Is ( err , ngmodels . ErrAlertRuleNotFound ) || q . Result == nil && err == nil {
return nil , fmt . Errorf ( "failed to update rule with UID %s because %w" , r . UID , ngmodels . ErrAlertRuleNotFound )
}
return nil , fmt . Errorf ( "failed to query database for an alert rule with UID %s: %w" , r . UID , err )
}
existing = q . Result
}
}
if existing == nil {
2022-03-04 15:16:33 -06:00
toAdd = append ( toAdd , r )
2022-02-23 10:30:04 -06:00
continue
}
ngmodels . PatchPartialAlertRule ( existing , r )
2022-03-04 15:16:33 -06:00
diff := existing . Diff ( r , alertRuleFieldsToIgnoreInDiff ... )
if len ( diff ) == 0 {
continue
}
toUpdate = append ( toUpdate , ruleUpdate {
2022-02-23 10:30:04 -06:00
Existing : existing ,
2022-03-04 15:16:33 -06:00
New : r ,
Diff : diff ,
2022-02-23 10:30:04 -06:00
} )
continue
}
for _ , rule := range existingGroupRulesUIDs {
toDelete = append ( toDelete , rule )
}
2022-03-04 15:16:33 -06:00
return & changes {
New : toAdd ,
Delete : toDelete ,
Update : toUpdate ,
2022-02-23 10:30:04 -06:00
} , nil
}
2022-03-04 15:16:33 -06:00
// alertRuleFieldsToIgnoreInDiff contains fields that the AlertRule.Diff should ignore
var alertRuleFieldsToIgnoreInDiff = [ ] string { "ID" , "Version" , "Updated" }