141eb60f06
This allows for client agent to be run in a more stateless manner where they may be abruptly terminated and not expected to come back. If advertising a per-agent reconnect timeout using the advertise_reconnect_timeout configuration when that agent leaves, other agents will wait only that amount of time for the agent to come back before reaping it. This has the advantageous side effect of causing servers to deregister the node/services/checks for that agent sooner than if the global reconnect_timeout was used.
83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
package serf
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/serf/serf"
|
|
)
|
|
|
|
const (
|
|
ReconnectTimeoutTag = "rc_tm"
|
|
)
|
|
|
|
// DefaultConfig returns a Consul-flavored Serf default configuration,
|
|
// suitable as a basis for a LAN, WAN, segment, or area.
|
|
func DefaultConfig() *serf.Config {
|
|
base := serf.DefaultConfig()
|
|
|
|
// This effectively disables the annoying queue depth warnings.
|
|
base.QueueDepthWarning = 1000000
|
|
|
|
// This enables dynamic sizing of the message queue depth based on the
|
|
// cluster size.
|
|
base.MinQueueDepth = 4096
|
|
|
|
// This gives leaves some time to propagate through the cluster before
|
|
// we shut down. The value was chosen to be reasonably short, but to
|
|
// allow a leave to get to over 99.99% of the cluster with 100k nodes
|
|
// (using https://www.serf.io/docs/internals/simulator.html).
|
|
base.LeavePropagateDelay = 3 * time.Second
|
|
|
|
return base
|
|
}
|
|
|
|
func GetTags(serf *serf.Serf) map[string]string {
|
|
tags := make(map[string]string)
|
|
for tag, value := range serf.LocalMember().Tags {
|
|
tags[tag] = value
|
|
}
|
|
|
|
return tags
|
|
}
|
|
|
|
func UpdateTag(serf *serf.Serf, tag, value string) {
|
|
tags := GetTags(serf)
|
|
tags[tag] = value
|
|
|
|
serf.SetTags(tags)
|
|
}
|
|
|
|
type ReconnectOverride struct {
|
|
logger hclog.Logger
|
|
}
|
|
|
|
func NewReconnectOverride(logger hclog.Logger) *ReconnectOverride {
|
|
if logger == nil {
|
|
logger = hclog.Default()
|
|
}
|
|
|
|
return &ReconnectOverride{
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (r *ReconnectOverride) ReconnectTimeout(m *serf.Member, timeout time.Duration) time.Duration {
|
|
val, ok := m.Tags[ReconnectTimeoutTag]
|
|
if !ok {
|
|
return timeout
|
|
}
|
|
newTimeout, err := time.ParseDuration(val)
|
|
if err != nil {
|
|
r.logger.Warn("Member is advertising a malformed reconnect timeout", "member", m.Name, "rc_tm", val)
|
|
return timeout
|
|
}
|
|
|
|
// ignore a timeout of 0 as that indicates the default should be used
|
|
if newTimeout == 0 {
|
|
return timeout
|
|
}
|
|
|
|
return newTimeout
|
|
}
|