hide domain if only single host in config.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu 2017-05-10 21:05:16 +08:00
parent efdac217bd
commit b4613ae4db
No known key found for this signature in database
GPG Key ID: 0F84B2110C500B1F

View File

@ -40,7 +40,11 @@ type (
) )
func (p Plugin) log(host string, message ...interface{}) { func (p Plugin) log(host string, message ...interface{}) {
if len(host) == 0 {
fmt.Printf("%s", fmt.Sprintln(message...))
} else {
fmt.Printf("%s: %s", host, fmt.Sprintln(message...)) fmt.Printf("%s: %s", host, fmt.Sprintln(message...))
}
} }
// Exec executes the plugin. // Exec executes the plugin.
@ -61,8 +65,12 @@ func (p Plugin) Exec() error {
wg.Add(len(p.Config.Host)) wg.Add(len(p.Config.Host))
errChannel := make(chan error, 1) errChannel := make(chan error, 1)
finished := make(chan bool, 1) finished := make(chan bool, 1)
countHosts := len(p.Config.Host)
for _, host := range p.Config.Host { for _, host := range p.Config.Host {
go func(host string) { go func(host string, counts int) {
if counts == 1 {
host = ""
}
// Create MakeConfig instance with remote username, server address and path to private key. // Create MakeConfig instance with remote username, server address and path to private key.
ssh := &easyssh.MakeConfig{ ssh := &easyssh.MakeConfig{
Server: host, Server: host,
@ -115,7 +123,7 @@ func (p Plugin) Exec() error {
} }
wg.Done() wg.Done()
}(host) }(host, countHosts)
} }
go func() { go func() {