You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
130 lines
2.6 KiB
130 lines
2.6 KiB
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"flag"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type commanderSetting struct {
|
|
ip string
|
|
port string
|
|
configFile string
|
|
configs [][]string
|
|
}
|
|
|
|
var (
|
|
setting commanderSetting
|
|
)
|
|
|
|
func readConfigFile() [][]string {
|
|
file, err := os.Open(setting.configFile)
|
|
if err != nil {
|
|
log.Println("Failed to read config file")
|
|
return nil
|
|
}
|
|
fscanner := bufio.NewScanner(file)
|
|
|
|
result := [][]string{}
|
|
for fscanner.Scan() {
|
|
p := strings.Split(fscanner.Text(), " ")
|
|
result = append(result, p)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func handleCommand(command string) {
|
|
args := strings.Split(command, " ")
|
|
if len(args) <= 0 {
|
|
return
|
|
}
|
|
|
|
switch cmd := args[0]; cmd {
|
|
case "init":
|
|
{
|
|
dictateNodes("init http://" + setting.ip + ":" + setting.port + "/" + setting.configFile)
|
|
}
|
|
case "ping":
|
|
fallthrough
|
|
case "kill":
|
|
{
|
|
dictateNodes(command)
|
|
}
|
|
default:
|
|
{
|
|
log.Println("Unknown command")
|
|
}
|
|
}
|
|
}
|
|
|
|
func config(ip string, port string, configFile string) {
|
|
setting.ip = ip
|
|
setting.port = port
|
|
setting.configFile = configFile
|
|
setting.configs = readConfigFile()
|
|
log.Println("Loaded config file", setting.configs)
|
|
}
|
|
|
|
func dictateNodes(command string) {
|
|
for _, config := range setting.configs {
|
|
ip := config[0]
|
|
port := "1" + config[1] // the port number of solider is "1" + node port
|
|
addr := strings.Join([]string{ip, port}, ":")
|
|
|
|
go dictateNode(addr, command)
|
|
}
|
|
}
|
|
|
|
func dictateNode(addr string, command string) {
|
|
// creates client
|
|
conn, err := net.Dial("tcp", addr)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
// send command
|
|
_, err = conn.Write([]byte(command))
|
|
if err != nil {
|
|
log.Printf("Failed to send command to %s", addr)
|
|
return
|
|
}
|
|
log.Printf("Send \"%s\" to %s", command, addr)
|
|
|
|
// read response
|
|
buff := make([]byte, 1024)
|
|
n, _ := conn.Read(buff)
|
|
log.Printf("Receive from %s: %s", addr, buff[:n])
|
|
}
|
|
|
|
func hostConfigFile() {
|
|
err := http.ListenAndServe(":"+setting.port, http.FileServer(http.Dir("./")))
|
|
if err != nil {
|
|
panic("Failed to host config file!")
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
ip := flag.String("ip", "127.0.0.1", "ip of commander")
|
|
port := flag.String("port", "8080", "port of config file")
|
|
configFile := flag.String("config_file", "test.txt", "file name of config file")
|
|
|
|
config(*ip, *port, *configFile)
|
|
|
|
log.Println("Start to host config file at http://" + setting.ip + ":" + setting.port + "/" + setting.configFile)
|
|
go hostConfigFile()
|
|
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
for true {
|
|
log.Printf("Listening to Your Command:")
|
|
if !scanner.Scan() {
|
|
break
|
|
}
|
|
handleCommand(scanner.Text())
|
|
}
|
|
}
|
|
|