56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DiscoverInterface interface {
|
||
|
|
Addrs(string, *log.Logger) ([]string, error)
|
||
|
|
|
||
|
|
Help() string
|
||
|
|
|
||
|
|
Names() []string
|
||
|
|
}
|
||
|
|
|
||
|
|
type retryJoiner struct {
|
||
|
|
attempt int
|
||
|
|
|
||
|
|
join func([]string) (int, error)
|
||
|
|
|
||
|
|
discover DiscoverInterface
|
||
|
|
|
||
|
|
errCh chan struct{}
|
||
|
|
|
||
|
|
logger *log.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryJoiner) RetryJoin(config *Config) {
|
||
|
|
if len(config.Server.RetryJoin) == 0 || !config.Server.Enabled {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
r.logger.Printf("[INFO] agent: Joining cluster...")
|
||
|
|
|
||
|
|
for {
|
||
|
|
addrs := config.Server.RetryJoin
|
||
|
|
|
||
|
|
n, err := r.join(addrs)
|
||
|
|
if err == nil {
|
||
|
|
r.logger.Printf("[INFO] agent: Join completed. Synced with %d initial agents", n)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
r.attempt++
|
||
|
|
if config.Server.RetryMaxAttempts > 0 && r.attempt > config.Server.RetryMaxAttempts {
|
||
|
|
r.logger.Printf("[ERR] agent: max join retry exhausted, exiting")
|
||
|
|
close(r.errCh)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
r.logger.Printf("[WARN] agent: Join failed: %v, retrying in %v", err,
|
||
|
|
config.Server.RetryInterval)
|
||
|
|
time.Sleep(config.Server.retryInterval)
|
||
|
|
}
|
||
|
|
}
|