-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #2 from kedark3/working
Merging current working branch
- Loading branch information
Showing
12 changed files
with
395 additions
and
125 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1 @@ | ||
| config/prometheus.yaml | ||
| config/*.yaml |
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,13 @@ | ||
| all: update | ||
| .PHONY: all | ||
|
|
||
| update: clean build | ||
|
|
||
| OUT_DIR=bin | ||
| build: | ||
| mkdir -p "${OUT_DIR}" | ||
| go build -v -o "${OUT_DIR}/cpa" main.go | ||
|
|
||
| clean: | ||
| $(RM) ./bin/cpa | ||
| .PHONY: clean |
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
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,123 @@ | ||
| package Notify | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "io/ioutil" | ||
| "log" | ||
| "math/rand" | ||
| "os" | ||
| "os/exec" | ||
| "strings" | ||
| "time" | ||
|
|
||
| "github.com/slack-go/slack" | ||
| "gopkg.in/yaml.v2" | ||
| ) | ||
|
|
||
| const configPath = "./config/" | ||
|
|
||
| type slackConfig struct { | ||
| UserID string `json:"userid"` | ||
| ChannelID string `json:"channelid"` | ||
| SlackToken string `json:"slacktoken"` | ||
| } | ||
|
|
||
| func (c *slackConfig) Parse(data []byte) error { | ||
| return yaml.Unmarshal(data, c) | ||
| } | ||
|
|
||
| func ReadslackConfig() (config slackConfig, err error) { | ||
| data, err := ioutil.ReadFile(configPath + "slack.yaml") | ||
| msg := fmt.Sprintf("Cound't read %sslack.yaml", configPath) | ||
| if err != nil { | ||
| return config, fmt.Errorf(msg) | ||
| } | ||
| if err := config.Parse(data); err != nil { | ||
| log.Fatal(err) | ||
| return config, err | ||
| } | ||
| return config, nil | ||
| } | ||
|
|
||
| func (s slackConfig) SlackNotify(message, thread_ts string) string { | ||
| // api := slack.New(s.SlackToken, slack.OptionDebug(true)) // To debug api requests, you uncomment this line and comment the one below | ||
| api := slack.New(s.SlackToken) | ||
| msgText := slack.NewTextBlockObject("mrkdwn", fmt.Sprintf("Hi <@%s>, %s", s.UserID, message), false, false) | ||
| msgSection := slack.NewSectionBlock(msgText, nil, nil) | ||
| msgBlock := slack.MsgOptionBlocks( | ||
| msgSection, | ||
| ) | ||
| var err error | ||
| if thread_ts != "" { | ||
| // if we have thread_ts - use it to send new messages on the thread | ||
| _, _, _, err = api.SendMessage(s.ChannelID, msgBlock, slack.MsgOptionTS(thread_ts)) | ||
| } else { | ||
| // if thread_ts was empty, assume that this is first message we are sending and retrieve thread_ts and return it for subsequent requests | ||
| _, thread_ts, _, err = api.SendMessage(s.ChannelID, msgBlock) | ||
| } | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| return thread_ts | ||
| } | ||
|
|
||
| func (s slackConfig) Notify(c chan string, thread_ts string) { | ||
| waitChars := []string{"/", "-", "\\", "|"} | ||
| for { | ||
| select { | ||
| case msg := <-c: | ||
| msgFmt := fmt.Sprintf(` | ||
| %s | ||
| Received following on the channel: %s | ||
| %[1]s | ||
| `, strings.Repeat("~", 80), msg) | ||
| log.Println(msgFmt) | ||
| if s.ChannelID != "" && s.UserID != "" && s.SlackToken != "" { | ||
| s.SlackNotify("Following query failed:"+msg, thread_ts) | ||
| } | ||
| default: | ||
| fmt.Printf("\r%s Please Wait. No new message received on the channel....", waitChars[rand.Intn(4)]) | ||
| time.Sleep(time.Millisecond * 500) | ||
| } | ||
| } | ||
|
|
||
| } | ||
|
|
||
| func TerminateBenchmark(tb chan bool, processID string) { | ||
| for { | ||
| select { | ||
| case b := <-tb: | ||
| msgFmt := fmt.Sprintf(` | ||
| %s | ||
| Received signal %t to kill -SIGTERM %s | ||
| %[1]s | ||
| `, strings.Repeat("~", 80), b, processID) | ||
| log.Println(msgFmt) | ||
| // TODO: should we terminate CPA as a result of termination of benchmark | ||
| err := exec.Command("kill", "-SIGTERM", processID).Run() | ||
| if err != nil { | ||
| // if the process is already killed, subsequent kills will fail with exit status 1 | ||
| log.Println("Failed to kill the process:", processID, err) | ||
| } | ||
| os.Exit(1) | ||
| // proc, err := os.FindProcess(processID) | ||
| // if err != nil { | ||
| // // if the process is already killed, subsequent kills will fail with exit status 1 | ||
| // log.Println("Failed to find the process:", processID, err) | ||
| // break | ||
| // } | ||
| // proc.Kill() | ||
| // state, err := proc.Wait() | ||
| // if err != nil || state.ExitCode() != 0 { | ||
| // // if the process is already killed, subsequent kills will fail with exit status 1 | ||
| // log.Println("Failed to kill the process:", processID, err) | ||
| // log.Println("Exit code was", state.ExitCode()) | ||
| // } else { | ||
| // log.Println("Killed the process:", processID) | ||
| // log.Println("Exit code was: ", state.ExitCode()) | ||
| // } | ||
| default: | ||
| time.Sleep(time.Millisecond * 500) | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.