update example

Author: faruqisan

Makefile

@@ -5,7 +5,7 @@ deploy:
 	kubectl apply -f files/k8s/deployment.yaml
 
 run_example:
-	go run example/main.go
+	cd example && go run main.go
 
 deploy_example:
 	docker build -f example_app.Dockerfile . -t faruqisan/resilia:0.0.1

example/files/deployments/alpine.yaml

@@ -0,0 +1,19 @@
+apiVersion: apps/v1 # for versions before 1.9.0 use apps/v1beta2
+kind: Deployment
+metadata:
+  name: client-deployment
+spec:
+  selector:
+    matchLabels:
+      app: client
+  replicas: 1 # tells deployment to run 2 pods matching the template
+  template:
+    metadata:
+      labels:
+        app: client
+    spec:
+      containers:
+      - name: client
+        image: alpine
+        command: ["/bin/sh"]
+        args: ["-c", "while true; do sleep 1; echo `date`; echo -e '*1\r\n$4\r\nPING\r\n' | nc redis-service.default 6379; done"]

example/files/deployments/redis.yaml

@@ -6,7 +6,7 @@ spec:
   selector:
     matchLabels:
       app: redis
-  replicas: 2 # tells deployment to run 2 pods matching the template
+  replicas: 1
   template:
     metadata:
       labels:

example/main.go

@@ -38,12 +38,12 @@ func main() {
 
 	// setup k8s
 	switch inCluster {
-	case inCluster:
+	case true:
 		kubeEngine, err = kube.New()
 		if err != nil {
 			log.Fatal(err)
 		}
-	case !inCluster:
+	case false:
 		kubeEngine, err = kube.New(kube.WithOutsideClusterConfig())
 		if err != nil {
 			log.Fatal(err)
@@ -130,57 +130,57 @@ func main() {
 		}
 	}
 
-	// spawn pumba!
-	pumbaEngine := pumba.New(kubeEngine)
-
-	// netem example
-	// create netem worker to hit 1 redis
-	// opt := pumbaEngine.NetEm(pumba.CommandNetEmLoss, pumba.NetEmOptions{
-	// 	TCImage:     "gaiadocker/iproute2",
-	// 	Duration:    "30s",
-	// 	LossPercent: "90",
-	// })
-
-	opt := pumbaEngine.Pause(pumba.PauseOptions{
-		Duration: "30s",
-	})
-
 	// get pods
 	// wait for pods spawned
-	var pods []string
+	var (
+		pods     []string
+		redisPod string
+	)
 	for {
 		pods, err = kubeEngine.GetPods()
 		if err != nil {
 			log.Println(err)
 		}
-		log.Println("pods : ", pods)
 
 		if len(pods) == 0 {
 			log.Println("no pods found, waiting ...")
 			time.Sleep(time.Second)
-		} else {
-			break
+			continue
 		}
+		// target redis pods
+		for _, pod := range pods {
+			// check if pod name contain redis
+			if strings.Contains(pod, "redis") {
+				redisPod = pod
+				break
+			}
+		}
+
+		if redisPod == "" {
+			continue
+		}
+
+		break
 	}
 
-	// select 1st pod
-	log.Println("targeting pods : ", pods[0])
-	// to test file deployment
-	// fData, err := ioutil.ReadFile(path.Join(exampleDaemonSetsPath, "pumba.yaml"))
+	log.Println("targeting pods : ", redisPod)
+
+	// spawn pumba!
+	pumbaEngine := pumba.New(kubeEngine)
+	opt := pumbaEngine.Pause(pumba.PauseOptions{
+		Duration: "10s",
+	})
 
-	pauseWorker := pumbaEngine.NewPumbaWorker(pods[0], "1m", pumba.CommandPause, opt)
-	// netEmWorker, err := kubeEngine.LoadDaemonSetFromFile(fData)
+	pauseWorker := pumbaEngine.NewPumbaWorker(pods[0], "20s", pumba.CommandPause, opt)
 	if err != nil {
 		log.Fatal(err)
 	}
 
 	dName, err := pumbaEngine.RunWorker(pauseWorker)
-	// dName, err := kubeEngine.CreateDaemonSet(netEmWorker)
 	if err != nil {
-		log.Println("fail to run pumba : ", err)
-	} else {
-		log.Printf("daemon set %s created", dName)
+		log.Fatal("fail to run pumba : ", err)
 	}
+	log.Printf("daemon set %s created", dName)
 
 	for {
 		select {