From 0ff0a3156736006913648047ba7a0093f55bb7ed Mon Sep 17 00:00:00 2001
From: Derek McQuay <derekmcquay@gmail.com>
Date: Wed, 16 May 2018 16:06:02 -0700
Subject: [PATCH] lingering changes

Signed-off-by: Derek McQuay <derekmcquay@gmail.com>
---
 main.go | 124 +++++++++++++++++++++++++++++++++++++++-----------------
 1 file changed, 86 insertions(+), 38 deletions(-)

diff --git a/main.go b/main.go
index 9a3d252..1d8eb34 100644
--- a/main.go
+++ b/main.go
@@ -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()
 }