first commit

main.go

@@ -0,0 +1,118 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"strings"
+	"time"
+	//	"github.com/bitly/go-simplejson"
+)
+
+type sshhost struct {
+	host     string
+	port     int
+	username string
+	password string
+	cmd      []string
+}
+
+func main() {
+	hosts := flag.String("hosts", "", "host address list")
+	cmd := flag.String("cmd", "", "cmds")
+	username := flag.String("u", "", "username")
+	password := flag.String("p", "", "password")
+	port := flag.Int("port", 22, "ssh port")
+	cmdfile := flag.String("cmdfile", "", "cmdfile path")
+	hostfile := flag.String("hostfile", "", "hostfile path")
+	ipfile := flag.String("ipfile", "", "hostfile path")
+	cfg := flag.String("cfg", "", "cfg path")
+
+	flag.Parse()
+
+	var cmdlist []string
+	var hostlist []string
+	var err error
+
+	sshhosts := []sshhost{}
+	var host_struct sshhost
+
+	if *ipfile != "" {
+		hostlist, err = GetIpList(*ipfile)
+		if err != nil {
+			log.Println("load hostlist error: ", err)
+			return
+		}
+	}
+
+	if *hostfile != "" {
+		hostlist, err = Getfile(*hostfile)
+		if err != nil {
+			log.Println("load hostfile error: ", err)
+			return
+		}
+	}
+	if *hosts != "" {
+		hostlist = strings.Split(*hosts, ";")
+
+	}
+
+	if *cmdfile != "" {
+		cmdlist, err = Getfile(*cmdfile)
+		if err != nil {
+			log.Println("load cmdfile error: ", err)
+			return
+		}
+	}
+	if *cmd != "" {
+		cmdlist = strings.Split(*cmd, ";")
+	}
+
+	if *cfg == "" {
+		for _, host := range hostlist {
+			host_struct.host = host
+			host_struct.username = *username
+			host_struct.password = *password
+			host_struct.port = *port
+			host_struct.cmd = cmdlist
+			sshhosts = append(sshhosts, host_struct)
+		}
+	}
+	/*
+		 else {
+			cfgjson, err := GetfileAll(*cfg)
+			if err != nil {
+				log.Println("load cfg error: ", err)
+				return
+			}
+
+				js, js_err := simplejson.NewJson(cfgjson)
+				if js_err != nil {
+					log.Println("json format error: ", js_err)
+					return
+				}
+
+
+		}
+	*/
+	//fmt.Println(sshhosts)
+
+	chs := make([]chan string, len(sshhosts))
+	for i, host := range sshhosts {
+		chs[i] = make(chan string, 1)
+		go dossh(host.username, host.password, host.host, host.cmd, host.port, chs[i])
+	}
+	for i, ch := range chs {
+		fmt.Println(sshhosts[i].host, " ssh start")
+		select {
+		case res := <-ch:
+			if res != "" {
+				fmt.Println(res)
+			}
+		case <-time.After(30 * 1000 * 1000 * 1000):
+			log.Println("SSH run timeout")
+		}
+		fmt.Println(sshhosts[i].host, " ssh end\n")
+	}
+
+}