dm/dcoshc
1
0
Fork 0
Code Issues Pull Requests Activity

lingering changes

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

124
main.go
View File

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