lingering changes

Signed-off-by: Derek McQuay <derekmcquay@gmail.com>
This commit is contained in:
Derek McQuay 2018-05-16 16:06:02 -07:00
parent 849279a07f
commit 0ff0a31567
No known key found for this signature in database
GPG Key ID: 50CA472F40A5AFED
1 changed files with 86 additions and 38 deletions

124
main.go
View File

@ -6,8 +6,9 @@ import (
"fmt" "fmt"
"log" "log"
"os/exec" "os/exec"
"regexp"
"strconv"
"strings" "strings"
"sync"
"github.com/fatih/color" "github.com/fatih/color"
) )
@ -73,11 +74,11 @@ func getServiceIPs() []serviceIPs {
} }
// get rid of kubernetes pod // get rid of kubernetes pod
for i, n := range s { //for i, n := range s {
if n.Name == "kubernetes" { // if n.Name == "kubernetes" {
s = append(s[:i], s[i+1:]...) // s = append(s[:i], s[i+1:]...)
} // }
} //}
return s return s
} }
@ -102,53 +103,100 @@ func stripHealthCheck(n string, hcs healthCheck) string {
return "" return ""
} }
func getPodInstanceIndex(n string) (int, error) {
re := regexp.MustCompile("[0-9]")
index := re.FindString(n)
i, err := strconv.Atoi(index)
return i, err
}
func buildCommands(hcs healthCheck, ips []serviceIPs) []command { func buildCommands(hcs healthCheck, ips []serviceIPs) []command {
commands := []command{} commands := []command{}
for _, s := range ips { for _, s := range ips {
if strings.Contains(s.Name, "apiserver") || strings.Contains(s.Name, "etcd") { if strings.Contains(s.Name, "apiserver") {
ip := s.Statuses[0].ContainerStatus.NetworkInfos[0].IPAddresses[0].IPAddress index, err := getPodInstanceIndex(s.Name)
c := strings.Replace(stripHealthCheck(s.Name, hcs), "$LIBPROCESS_IP", ip, -1) if err != nil {
log.Fatal("failed to get pod index for build command: ", err)
}
c := strings.Replace(stripHealthCheck(s.Name, hcs), "$POD_INSTANCE_INDEX", strconv.Itoa(index), -1)
commands = append(commands, command{node: s.Name, hc: c}) commands = append(commands, command{node: s.Name, hc: c})
} else if strings.Contains(s.Name, "etcd") {
index, err := getPodInstanceIndex(s.Name)
if err != nil {
log.Fatal("failed to get pod index for build command: ", err)
}
c := strings.Replace(stripHealthCheck(s.Name, hcs), "$POD_INSTANCE_INDEX", strconv.Itoa(index), -1)
c = strings.Replace(c, "$ETCD_LISTEN_CLIENT_PORT", "2379", -1)
commands = append(commands, command{node: s.Name, hc: c})
//https://etcd-$POD_INSTANCE_INDEX-peer.{{FRAMEWORK_NAME}}.mesos:$ETCD_LISTEN_CLIENT_PORT/health
} else { } else {
commands = append(commands, command{node: s.Name, hc: stripHealthCheck(s.Name, hcs)}) //commands = append(commands, command{node: s.Name, hc: stripHealthCheck(s.Name, hcs)})
} }
} }
fmt.Println(commands)
return commands return commands
} }
func main() { func main() {
var wg sync.WaitGroup //var wg sync.WaitGroup
hcs := getHealthCheck() hcs := getHealthCheck()
ips := getServiceIPs() ips := getServiceIPs()
//for _, i := range hcs.PodSpecs {
// fmt.Println(i)
//}
//fmt.Println("\n\n\n")
//fmt.Printf("%+v\n", hcs)
//fmt.Println("\n\n\n")
//fmt.Printf("%+v\n", ips)
//fmt.Println("\n\n\n")
hosts := buildCommands(hcs, ips) hosts := buildCommands(hcs, ips)
//for i, j := range hosts {
// fmt.Printf("%d %+v\n", i, j)
//}
//fmt.Println(hosts)
//time.Sleep(time.Second * 5)
for _, n := range hosts { for _, n := range hosts {
wg.Add(1) //wg.Add(1)
go func(c command) { //go func(c command) {
defer wg.Done() //defer wg.Done()
s := strings.Split(c.hc, " ") // s := strings.Split(n.hc, " ")
cmd := exec.Command( //if len(s) != 3 {
"dcos", // log.Printf("could not be split: %v", s)
"task", // return
"exec", //}
c.node, //for i, j := range s {
s[0], // fmt.Println(i, j)
s[1], //}
s[2], cmd := exec.Command(
) "dcos",
var out bytes.Buffer "task",
cmd.Stdout = &out "exec",
err := cmd.Run() n.node,
if err != nil { n.hc,
log.Fatal(err) //s[0],
} //s[1],
if strings.Contains(out.String(), "ok") || strings.Contains(out.String(), "true") { //s[2],
color.Green(fmt.Sprintf("%v: %q\n", c, out.String())) //s[3],
} else { //s[4],
color.Red(fmt.Sprintf("%v: %q\n", c, out.String())) //s[5],
} //s[6],
}(n) //s[7],
//s[8],
)
var out bytes.Buffer
cmd.Stdout = &out
err := cmd.Run()
if err != nil {
log.Printf("failed for %v with:", n, err)
return
}
if strings.Contains(out.String(), "ok") || strings.Contains(out.String(), "true") {
color.Green("%s", n.node)
} else {
color.Red(fmt.Sprintf("%s: \t%q\n", n.node, out.String()))
}
//}(n)
} }
wg.Wait() // wg.Wait()
} }