forked from grafana/cortex-tools
-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add tool to deserialize alertmanager state file
Signed-off-by: Mustafain Ali Khan <[email protected]>
- Loading branch information
1 parent
76a414b
commit 3ed0c04
Showing
4 changed files
with
148 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,140 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "flag" | ||
| "fmt" | ||
| "github.com/cortexproject/cortex/pkg/alertmanager/alertspb" | ||
| "github.com/gogo/protobuf/proto" | ||
| "github.com/matttproud/golang_protobuf_extensions/pbutil" | ||
| "github.com/prometheus/alertmanager/nflog/nflogpb" | ||
| "github.com/prometheus/alertmanager/silence/silencepb" | ||
| "io" | ||
| "log" | ||
| "os" | ||
| "strings" | ||
| ) | ||
|
|
||
| var out bytes.Buffer | ||
|
|
||
| func main() { | ||
| var intputFile string | ||
| var outputFile string | ||
|
|
||
| flag.CommandLine.StringVar(&intputFile, "file", "", "path of fullstate file to deserialize") | ||
| flag.CommandLine.StringVar(&outputFile, "output", "", "file for deserialized output") | ||
|
|
||
| flag.Parse() | ||
|
|
||
| if intputFile == "" { | ||
| log.Fatalf("No full state file specified.") | ||
| } | ||
|
|
||
| decodeFullState(intputFile) | ||
|
|
||
| if outputFile == "" { | ||
| // write to stdout if output file not specified | ||
| fmt.Print(out.String()) | ||
| } else { | ||
| outputToFile(outputFile) | ||
| } | ||
| } | ||
|
|
||
| func outputToFile(file string) { | ||
| fo, err := os.Create(file) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| // close fo on exit and check for its returned error | ||
| defer func() { | ||
| if err := fo.Close(); err != nil { | ||
| panic(err) | ||
| } | ||
| }() | ||
|
|
||
| _, err = fo.Write(out.Bytes()) | ||
| if err != nil { | ||
| log.Fatalf("Failed writing output to file: %v", err) | ||
| } | ||
| } | ||
|
|
||
| func decodeFullState(path string) { | ||
| in, err := os.ReadFile(path) | ||
| if err != nil { | ||
| log.Fatalln("Error reading file:", err) | ||
| } | ||
| fs := alertspb.FullStateDesc{} | ||
| err = proto.Unmarshal(in, &fs) | ||
| if err != nil { | ||
| log.Fatalln("Error unmarshalling full state:", err) | ||
| } | ||
|
|
||
| for _, part := range fs.GetState().Parts { | ||
| out.WriteString("\n----\n") | ||
| if isNfLog(part.Key) { | ||
| parseNotifications(part.Data) | ||
| } else if isSilence(part.Key) { | ||
| parseSilences(part.Data) | ||
| } else { | ||
| out.WriteString(fmt.Sprintf("Unknown part type: %s", part.Key)) | ||
| } | ||
|
|
||
| } | ||
| } | ||
|
|
||
| func parseNotifications(data []byte) { | ||
| out.WriteString("Alerts:\n") | ||
| r := bytes.NewReader(data) | ||
| for { | ||
| nf := nflogpb.MeshEntry{} | ||
| n, err := pbutil.ReadDelimited(r, &nf) | ||
| if err != nil && err != io.EOF { | ||
| log.Fatalf("unable to read alert notifications, %v", err) | ||
| } | ||
| if n == 0 || err == io.EOF { | ||
| break | ||
| } | ||
| result, err := json.Marshal(nf) | ||
| if err != nil { | ||
| log.Fatalf("unable to marshal to json, %v", err) | ||
| } | ||
| _, err = out.WriteString(string(result) + "\n") | ||
| if err != nil { | ||
| log.Fatalf("unable to write output, %v", err) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func parseSilences(data []byte) { | ||
| out.WriteString("Silences:\n") | ||
| r := bytes.NewReader(data) | ||
| for { | ||
| silence := silencepb.MeshSilence{} | ||
| n, err := pbutil.ReadDelimited(r, &silence) | ||
| if err != nil && err != io.EOF { | ||
| log.Fatalf("unable to read silences, %v", err) | ||
| } | ||
| if n == 0 || err == io.EOF { | ||
| break | ||
| } | ||
|
|
||
| result, err := json.Marshal(silence) | ||
| if err != nil { | ||
| log.Fatalf("unable to marshal to json, %v", err) | ||
| } | ||
|
|
||
| _, err = out.WriteString(string(result) + "\n") | ||
| if err != nil { | ||
| log.Fatalf("unable to write output, %v", err) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func isNfLog(key string) bool { | ||
| return strings.HasPrefix(key, "nfl") | ||
| } | ||
|
|
||
| func isSilence(key string) bool { | ||
| return strings.HasPrefix(key, "sil") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters