From 5877e8c486a56fba9ba05b8b49f7b25ad2e008cb Mon Sep 17 00:00:00 2001
From: Richard Liu <liuzhenglaichn@gmail.com>
Date: Tue, 3 Jul 2018 21:38:07 -0700
Subject: [PATCH] soldier on multiple ports.

---
 .../experiment/commander/main.go              | 10 +++---
 .../experiment/soldier/main.go                | 35 +++++++++----------
 2 files changed, 23 insertions(+), 22 deletions(-)

diff --git a/aws-experiment-launch/experiment/commander/main.go b/aws-experiment-launch/experiment/commander/main.go
index 34d955026..764d1d5c4 100644
--- a/aws-experiment-launch/experiment/commander/main.go
+++ b/aws-experiment-launch/experiment/commander/main.go
@@ -28,16 +28,18 @@ func SocketClient(ip string, port int) {
 
 	buff := make([]byte, 1024)
 	n, _ := conn.Read(buff)
-	log.Printf("Receive: %s", buff[:n])
+	log.Printf("Receive from %v: %s", port, buff[:n])
 }
 
 func main() {
 
 	var (
-		ip   = "127.0.0.1"
-		port = 3333
+		ip       = "127.0.0.1"
+		portList = []int{3333, 4444}
 	)
 
-	SocketClient(ip, port)
+	for _, port := range portList {
+		SocketClient(ip, port)
+	}
 
 }
diff --git a/aws-experiment-launch/experiment/soldier/main.go b/aws-experiment-launch/experiment/soldier/main.go
index 1935b47d8..1ab904b20 100644
--- a/aws-experiment-launch/experiment/soldier/main.go
+++ b/aws-experiment-launch/experiment/soldier/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"bufio"
+	"flag"
 	"io"
 	"io/ioutil"
 	"log"
@@ -13,12 +14,14 @@ import (
 )
 
 const (
-	Message       = "Pong"
 	StopCharacter = "\r\n\r\n"
 )
 
-func SocketServer(port int) {
+var (
+	port *int
+)
 
+func SocketServer(port int) {
 	listen, err := net.Listen("tcp4", ":"+strconv.Itoa(port))
 	defer listen.Close()
 	if err != nil {
@@ -35,11 +38,9 @@ func SocketServer(port int) {
 		}
 		go handler(conn)
 	}
-
 }
 
 func handler(conn net.Conn) {
-
 	defer conn.Close()
 
 	var (
@@ -63,7 +64,7 @@ ILOOP:
 				break ILOOP
 			}
 
-			go handleCommand(data)
+			go handleCommand(data, w)
 
 		default:
 			log.Fatalf("Receive data failed:%s", err)
@@ -71,13 +72,9 @@ ILOOP:
 		}
 
 	}
-	w.Write([]byte(Message))
-	w.Flush()
-	log.Printf("Send: %s", Message)
-
 }
 
-func handleCommand(command string) {
+func handleCommand(command string, w *bufio.Writer) {
 	args := strings.Split(command, " ")
 
 	if len(args) <= 0 {
@@ -87,7 +84,7 @@ func handleCommand(command string) {
 	switch command := args[0]; command {
 	case "init":
 		{
-			handleInitCommand(args[1:])
+			handleInitCommand(args[1:], w)
 		}
 	case "close":
 		{
@@ -96,10 +93,10 @@ func handleCommand(command string) {
 	}
 }
 
-func handleInitCommand(args []string) {
+func handleInitCommand(args []string, w *bufio.Writer) {
 	log.Println("Init command", args)
 	// create local config file
-	localConfig := "node_config.txt"
+	localConfig := "node_config_" + strconv.Itoa(*port) + ".txt"
 	out, err := os.Create(localConfig)
 	if err != nil {
 		log.Fatal("Failed to create local file")
@@ -124,7 +121,10 @@ func handleInitCommand(args []string) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	log.Printf("File contents: %s", content)
+	log.Println("Successfully init-ed with config", content)
+
+	w.Write([]byte("Successfully init-ed"))
+	w.Flush()
 }
 
 func isTransportOver(data string) (over bool) {
@@ -133,9 +133,8 @@ func isTransportOver(data string) (over bool) {
 }
 
 func main() {
+	port = flag.Int("port", 3333, "port of the node.")
+	flag.Parse()
 
-	port := 3333
-
-	SocketServer(port)
-
+	SocketServer(*port)
 }