Merge pull request #26 from simple-rules/ric-command

trying to run benchmark in soldier.
pull/27/head
7z7 6 years ago committed by GitHub
commit b011fd8972
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 3
      aws-experiment-launch/experiment/commander/main.go
  2. 45
      aws-experiment-launch/experiment/soldier/main.go

@ -10,8 +10,7 @@ import (
) )
const ( const (
message = "init http://localhost:8080/configuration.txt" message = "init http://localhost:8080/config.txt"
defaultPort = 6000
StopCharacter = "\r\n\r\n" StopCharacter = "\r\n\r\n"
) )

@ -9,8 +9,9 @@ import (
"net" "net"
"net/http" "net/http"
"os" "os"
"strconv" "os/exec"
"strings" "strings"
"time"
) )
const ( const (
@ -18,17 +19,19 @@ const (
) )
var ( var (
port *int ip *string
port *string
localConfig string
) )
func SocketServer(port int) { func SocketServer() {
listen, err := net.Listen("tcp4", ":"+strconv.Itoa(port)) listen, err := net.Listen("tcp4", ":"+*port)
defer listen.Close() defer listen.Close()
if err != nil { if err != nil {
log.Fatalf("Socket listen port %d failed,%s", port, err) log.Fatalf("Socket listen port %s failed,%s", *port, err)
os.Exit(1) os.Exit(1)
} }
log.Printf("Begin listen for command on port: %d", port) log.Printf("Begin listen for command on port: %s", *port)
for { for {
conn, err := listen.Accept() conn, err := listen.Accept()
@ -96,7 +99,6 @@ func handleCommand(command string, w *bufio.Writer) {
func handleInitCommand(args []string, w *bufio.Writer) { func handleInitCommand(args []string, w *bufio.Writer) {
log.Println("Init command", args) log.Println("Init command", args)
// create local config file // create local config file
localConfig := "node_config_" + strconv.Itoa(*port) + ".txt"
out, err := os.Create(localConfig) out, err := os.Create(localConfig)
if err != nil { if err != nil {
log.Fatal("Failed to create local file") log.Fatal("Failed to create local file")
@ -117,25 +119,48 @@ func handleInitCommand(args []string, w *bufio.Writer) {
log.Fatal("Failed to copy file") log.Fatal("Failed to copy file")
} }
// log config file
content, err := ioutil.ReadFile(localConfig) content, err := ioutil.ReadFile(localConfig)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
log.Println("Successfully init-ed with config") log.Println("Successfully downloaded config")
log.Println(string(content)) log.Println(string(content))
runInstance()
w.Write([]byte("Successfully init-ed")) w.Write([]byte("Successfully init-ed"))
w.Flush() w.Flush()
} }
func runInstance() {
t := time.Now().Format("20060102-150405")
logFolder := "../tmp_log/log-" + t
err := os.MkdirAll(logFolder, os.ModePerm)
if err != nil {
log.Fatal("Failed to create log folder")
}
cmdName := "./benchmark"
cmdArgs := []string{"-ip", *ip, "-port", *port, "-config_file", localConfig, "-log_folder", logFolder}
log.Println(cmdName, cmdArgs)
cmdOut, err := exec.Command(cmdName, cmdArgs...).Output()
if err != nil {
log.Fatal("Failed to run command: ", err)
}
log.Println(string(cmdOut))
}
func isTransportOver(data string) (over bool) { func isTransportOver(data string) (over bool) {
over = strings.HasSuffix(data, "\r\n\r\n") over = strings.HasSuffix(data, "\r\n\r\n")
return return
} }
func main() { func main() {
port = flag.Int("port", 3333, "port of the node.") ip = flag.String("ip", "127.0.0.1", "IP of the node.")
port = flag.String("port", "3000", "port of the node.")
flag.Parse() flag.Parse()
SocketServer(*port) localConfig = "node_config_" + *port + ".txt"
SocketServer()
} }

Loading…
Cancel
Save