go run ./testConfig.go --username="aaa" --passwd='aaa' --ip_port="192.168.6.87" --cmd='display version'
-bash-4.3$ go run ./testConfig.go -h
Usage of /tmp/go-build918643740/command-line-arguments/_obj/exe/testConfig:
-cmd string
cmdstring (default "display arp statistics all")
-ip_port string
ip and port (default "1.1.1.1:22")
-passwd string
password (default "aaa")
-username string
username (default "aaa")
exit status 2
重點(diǎn)code說明
...
in <- "display version" //執(zhí)行輸入goroutine
in <- "display arp statistics all"
...
go func() {
for cmd := range in {
wg.Add(1)
w.Write([]byte(cmd + "\n"))
wg.Wait() //控制每次goroutine執(zhí)行一條命令
}
}()
go func() {
var (
buf [1024 * 1024]byte
t int
)
for {
n, err := r.Read(buf[t:])
if err != nil {
fmt.Println(err.Error())
close(in)
close(out)
return
}
t += n
result := string(buf[:t])
//循環(huán)處理設(shè)備分頁符
if strings.Contains(string(buf[t-n:t]), "More") {
w.Write([]byte("\n"))
}
//匹配等待符完成一條操作清空goroutine
if strings.Contains(result, "username:") ||
strings.Contains(result, "password:") ||
strings.Contains(result, ">") {
out <- string(buf[:t])
t = 0
wg.Done()
}
}
}()
...
