139 lines
3.9 KiB
Go
139 lines
3.9 KiB
Go
// Copyright 2018 Prometheus Team
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package cli
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/alecthomas/kingpin/v2"
|
|
"github.com/go-openapi/strfmt"
|
|
"github.com/prometheus/common/model"
|
|
|
|
"github.com/prometheus/alertmanager/api/v2/client/silence"
|
|
"github.com/prometheus/alertmanager/api/v2/models"
|
|
"github.com/prometheus/alertmanager/cli/format"
|
|
)
|
|
|
|
type silenceUpdateCmd struct {
|
|
quiet bool
|
|
duration string
|
|
start string
|
|
end string
|
|
comment string
|
|
ids []string
|
|
}
|
|
|
|
func configureSilenceUpdateCmd(cc *kingpin.CmdClause) {
|
|
var (
|
|
c = &silenceUpdateCmd{}
|
|
updateCmd = cc.Command("update", "Update silences")
|
|
)
|
|
updateCmd.Flag("quiet", "Only show silence ids").Short('q').BoolVar(&c.quiet)
|
|
updateCmd.Flag("duration", "Duration of silence").Short('d').StringVar(&c.duration)
|
|
updateCmd.Flag("start", "Set when the silence should start. RFC3339 format 2006-01-02T15:04:05-07:00").StringVar(&c.start)
|
|
updateCmd.Flag("end", "Set when the silence should end (overwrites duration). RFC3339 format 2006-01-02T15:04:05-07:00").StringVar(&c.end)
|
|
updateCmd.Flag("comment", "A comment to help describe the silence").Short('c').StringVar(&c.comment)
|
|
updateCmd.Arg("update-ids", "Silence IDs to update").StringsVar(&c.ids)
|
|
|
|
updateCmd.Action(execWithTimeout(c.update))
|
|
}
|
|
|
|
func (c *silenceUpdateCmd) update(ctx context.Context, _ *kingpin.ParseContext) error {
|
|
if len(c.ids) < 1 {
|
|
return fmt.Errorf("no silence IDs specified")
|
|
}
|
|
|
|
amclient := NewAlertmanagerClient(alertmanagerURL)
|
|
|
|
var updatedSilences []models.GettableSilence
|
|
for _, silenceID := range c.ids {
|
|
params := silence.NewGetSilenceParams()
|
|
params.SilenceID = strfmt.UUID(silenceID)
|
|
response, err := amclient.Silence.GetSilence(params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
sil := response.Payload
|
|
if c.start != "" {
|
|
startsAtTime, err := time.Parse(time.RFC3339, c.start)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
startsAt := strfmt.DateTime(startsAtTime)
|
|
sil.StartsAt = &startsAt
|
|
}
|
|
|
|
if c.end != "" {
|
|
endsAtTime, err := time.Parse(time.RFC3339, c.end)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
endsAt := strfmt.DateTime(endsAtTime)
|
|
sil.EndsAt = &endsAt
|
|
} else if c.duration != "" {
|
|
d, err := model.ParseDuration(c.duration)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if d == 0 {
|
|
return fmt.Errorf("silence duration must be greater than 0")
|
|
}
|
|
endsAt := strfmt.DateTime(time.Time(*sil.StartsAt).UTC().Add(time.Duration(d)))
|
|
sil.EndsAt = &endsAt
|
|
}
|
|
|
|
if time.Time(*sil.StartsAt).After(time.Time(*sil.EndsAt)) {
|
|
return errors.New("silence cannot start after it ends")
|
|
}
|
|
|
|
if c.comment != "" {
|
|
sil.Comment = &c.comment
|
|
}
|
|
|
|
ps := &models.PostableSilence{
|
|
ID: *sil.ID,
|
|
Silence: sil.Silence,
|
|
}
|
|
|
|
amclient := NewAlertmanagerClient(alertmanagerURL)
|
|
|
|
silenceParams := silence.NewPostSilencesParams().WithContext(ctx).WithSilence(ps)
|
|
postOk, err := amclient.Silence.PostSilences(silenceParams)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sil.ID = &postOk.Payload.SilenceID
|
|
updatedSilences = append(updatedSilences, *sil)
|
|
}
|
|
|
|
if c.quiet {
|
|
for _, silence := range updatedSilences {
|
|
fmt.Println(silence.ID)
|
|
}
|
|
} else {
|
|
formatter, found := format.Formatters[output]
|
|
if !found {
|
|
return fmt.Errorf("unknown output formatter")
|
|
}
|
|
if err := formatter.FormatSilences(updatedSilences); err != nil {
|
|
return fmt.Errorf("error formatting silences: %w", err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|