services: add maintenance mode option to snooze alert notifications

alert/store.go

@@ -197,9 +197,13 @@ func NewStore(ctx context.Context, db *sql.DB, logDB *alertlog.Store) (*Store, e
 		escalate: p(`
 			UPDATE escalation_policy_state state
 			SET force_escalation = true
+			FROM alerts as a, services as svc
 			WHERE
 				state.alert_id = ANY($1) AND
-				state.force_escalation = false
+				state.force_escalation = false AND
+				a.id = state.alert_id AND
+				svc.id = a.service_id AND
+				svc.maintenance_expires_at ISNULL
 			RETURNING state.alert_id
 		`),
 

engine/escalationmanager/db.go

@@ -13,7 +13,8 @@ import (
 type DB struct {
 	lock *processinglock.Lock
 
-	cleanupNoSteps *sql.Stmt
+	clearMaintExpiredSvc *sql.Stmt
+	cleanupNoSteps       *sql.Stmt
 
 	lockStmt     *sql.Stmt
 	updateOnCall *sql.Stmt
@@ -31,7 +32,7 @@ func (db *DB) Name() string { return "Engine.EscalationManager" }
 // NewDB creates a new DB.
 func NewDB(ctx context.Context, db *sql.DB, log *alertlog.Store) (*DB, error) {
 	lock, err := processinglock.NewLock(ctx, db, processinglock.Config{
-		Version: 3,
+		Version: 4,
 		Type:    processinglock.TypeEscalation,
 	})
 	if err != nil {
@@ -82,6 +83,12 @@ func NewDB(ctx context.Context, db *sql.DB, log *alertlog.Store) (*DB, error) {
 			returning ep_step_id, user_id
 		`),
 
+		clearMaintExpiredSvc: p.P(`
+				update services s
+				set maintenance_expires_at = null
+				where s.maintenance_expires_at <= now()
+		`),
+
 		cleanupNoSteps: p.P(`
 			delete from escalation_policy_state state
 			using escalation_policies pol
@@ -99,6 +106,7 @@ func NewDB(ctx context.Context, db *sql.DB, log *alertlog.Store) (*DB, error) {
 					step.escalation_policy_id = state.escalation_policy_id and
 					step.step_number = 0
 				join alerts a on a.id = state.alert_id and (a.status = 'triggered' or state.force_escalation)
+				join services s on a.service_id = s.id and s.maintenance_expires_at isnull
 				where state.last_escalation isnull
 				for update skip locked
 				limit 1000
@@ -169,6 +177,7 @@ func NewDB(ctx context.Context, db *sql.DB, log *alertlog.Store) (*DB, error) {
 						WHEN state.escalation_policy_step_number >= ep.step_count THEN 0
 						ELSE state.escalation_policy_step_number
 						END
+				join services s on a.service_id = s.id and s.maintenance_expires_at isnull
 				where
 					state.last_escalation notnull and
 					escalation_policy_step_id isnull
@@ -248,6 +257,7 @@ func NewDB(ctx context.Context, db *sql.DB, log *alertlog.Store) (*DB, error) {
 						WHEN state.loop_count < ep.repeat THEN 0
 						ELSE -1
 					END
+				join services s on a.service_id = s.id and s.maintenance_expires_at isnull
 				where
 					state.last_escalation notnull and
 					escalation_policy_step_id notnull and

engine/escalationmanager/update.go

@@ -43,6 +43,11 @@ func (db *DB) update(ctx context.Context, all bool, alertID *int) error {
 		return errors.Wrap(err, "commit on-call update")
 	}
 
+	_, err = db.lock.Exec(ctx, db.clearMaintExpiredSvc)
+	if err != nil {
+		return errors.Wrap(err, "set maintenance_expires_at to null where expired")
+	}
+
 	_, err = db.lock.Exec(ctx, db.cleanupNoSteps)
 	if err != nil {
 		return errors.Wrap(err, "end policies with no steps")

graphql2/graphqlapp/service.go

@@ -94,6 +94,7 @@ func (s *Service) EscalationPolicy(ctx context.Context, raw *service.Service) (*
 func (s *Service) IsFavorite(ctx context.Context, raw *service.Service) (bool, error) {
 	return raw.IsUserFavorite(), nil
 }
+
 func (s *Service) OnCallUsers(ctx context.Context, raw *service.Service) ([]oncall.ServiceOnCallUser, error) {
 	return s.OnCallStore.OnCallUsersByService(ctx, raw.ID)
 }
@@ -213,6 +214,10 @@ func (a *Mutation) UpdateService(ctx context.Context, input graphql2.UpdateServi
 		svc.EscalationPolicyID = *input.EscalationPolicyID
 	}
 
+	if input.MaintenanceExpiresAt != nil {
+		svc.MaintenanceExpiresAt = *input.MaintenanceExpiresAt
+	}
+
 	err = a.ServiceStore.UpdateTx(ctx, tx, svc)
 	if err != nil {
 		return false, err

graphql2/schema.graphql

@@ -600,6 +600,7 @@ input UpdateServiceInput {
   name: String
   description: String
   escalationPolicyID: ID
+  maintenanceExpiresAt: ISOTimestamp
 }
 
 input UpdateEscalationPolicyInput {
@@ -968,6 +969,7 @@ type Service {
   escalationPolicyID: ID!
   escalationPolicy: EscalationPolicy
   isFavorite: Boolean!
+  maintenanceExpiresAt: ISOTimestamp
 
   onCallUsers: [ServiceOnCallUser!]!
   integrationKeys: [IntegrationKey!]!

migrate/migrations/20220705124007-add-service-maintenance-expires-at.sql

@@ -0,0 +1,7 @@
+-- +migrate Up
+UPDATE engine_processing_versions SET "version" = 4 WHERE type_id = 'escalation';
+ALTER TABLE services ADD COLUMN maintenance_expires_at TIMESTAMPTZ;
+
+-- +migrate Down
+UPDATE engine_processing_versions SET "version" = 3 WHERE type_id = 'escalation';
+ALTER TABLE services DROP COLUMN maintenance_expires_at;

service/service.go

@@ -1,12 +1,17 @@
 package service
 
-import "github.com/target/goalert/validation/validate"
+import (
+	"time"
+
+	"github.com/target/goalert/validation/validate"
+)
 
 type Service struct {
-	ID                 string `json:"id"`
-	Name               string `json:"name"`
-	Description        string `json:"description"`
-	EscalationPolicyID string `json:"escalation_policy_id"`
+	ID                   string
+	Name                 string
+	Description          string
+	EscalationPolicyID   string
+	MaintenanceExpiresAt time.Time
 
 	epName         string
 	isUserFavorite bool
@@ -24,10 +29,18 @@ func (s Service) IsUserFavorite() bool {
 // Normalize will validate and 'normalize' the ContactMethod -- such as making email lower-case
 // and setting carrier to "" (for non-phone types).
 func (s Service) Normalize() (*Service, error) {
+	dur := time.Until(s.MaintenanceExpiresAt)
+
+	if dur <= 0 {
+		dur = 0
+		s.MaintenanceExpiresAt = time.Time{}
+	}
+
 	err := validate.Many(
 		validate.IDName("Name", s.Name),
 		validate.Text("Description", s.Description, 1, 255),
 		validate.UUID("EscalationPolicyID", s.EscalationPolicyID),
+		validate.Duration("MaintenanceExpiresAt", dur, 0, 8*time.Hour),
 	)
 	if err != nil {
 		return nil, err