95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package agent
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
discover "github.com/hashicorp/go-discover"
|
|
)
|
|
|
|
// RetryJoin is used to handle retrying a join until it succeeds or all
|
|
// retries are exhausted.
|
|
func (a *Agent) retryJoin() {
|
|
cfg := a.config
|
|
if len(cfg.RetryJoin) == 0 {
|
|
return
|
|
}
|
|
|
|
disco := discover.Discover{}
|
|
a.logger.Printf("[INFO] agent: Retry join is supported for: %s", strings.Join(disco.Names(), " "))
|
|
a.logger.Printf("[INFO] agent: Joining cluster...")
|
|
attempt := 0
|
|
for {
|
|
var addrs []string
|
|
var err error
|
|
|
|
for _, addr := range cfg.RetryJoin {
|
|
switch {
|
|
case strings.Contains(addr, "provider="):
|
|
servers, err := disco.Addrs(addr, a.logger)
|
|
if err != nil {
|
|
a.logger.Printf("[ERR] agent: %s", err)
|
|
} else {
|
|
addrs = append(addrs, servers...)
|
|
a.logger.Printf("[INFO] agent: Discovered servers: %s", strings.Join(servers, " "))
|
|
}
|
|
|
|
default:
|
|
addrs = append(addrs, addr)
|
|
}
|
|
}
|
|
|
|
if len(addrs) > 0 {
|
|
n, err := a.JoinLAN(addrs)
|
|
if err == nil {
|
|
a.logger.Printf("[INFO] agent: Join completed. Synced with %d initial agents", n)
|
|
return
|
|
}
|
|
}
|
|
|
|
if len(addrs) == 0 {
|
|
err = fmt.Errorf("No servers to join")
|
|
}
|
|
|
|
attempt++
|
|
if cfg.RetryMaxAttempts > 0 && attempt > cfg.RetryMaxAttempts {
|
|
a.retryJoinCh <- fmt.Errorf("agent: max join retry exhausted, exiting")
|
|
return
|
|
}
|
|
|
|
a.logger.Printf("[WARN] agent: Join failed: %v, retrying in %v", err, cfg.RetryInterval)
|
|
time.Sleep(cfg.RetryInterval)
|
|
}
|
|
}
|
|
|
|
// RetryJoinWan is used to handle retrying a join -wan until it succeeds or all
|
|
// retries are exhausted.
|
|
func (a *Agent) retryJoinWan() {
|
|
cfg := a.config
|
|
|
|
if len(cfg.RetryJoinWan) == 0 {
|
|
return
|
|
}
|
|
|
|
a.logger.Printf("[INFO] agent: Joining WAN cluster...")
|
|
|
|
attempt := 0
|
|
for {
|
|
n, err := a.JoinWAN(cfg.RetryJoinWan)
|
|
if err == nil {
|
|
a.logger.Printf("[INFO] agent: Join -wan completed. Synced with %d initial agents", n)
|
|
return
|
|
}
|
|
|
|
attempt++
|
|
if cfg.RetryMaxAttemptsWan > 0 && attempt > cfg.RetryMaxAttemptsWan {
|
|
a.retryJoinCh <- fmt.Errorf("agent: max join -wan retry exhausted, exiting")
|
|
return
|
|
}
|
|
|
|
a.logger.Printf("[WARN] agent: Join -wan failed: %v, retrying in %v", err, cfg.RetryIntervalWan)
|
|
time.Sleep(cfg.RetryIntervalWan)
|
|
}
|
|
}
|