agent: rename local/global to src/dst

This commit is contained in:
Hans Hasselberg 2020-04-27 09:53:40 +02:00
parent 7361e11bc5
commit 6994c0d47f
2 changed files with 27 additions and 23 deletions

View File

@ -24,7 +24,7 @@ func (s *Server) FloodNotify() {
// Flood is a long-running goroutine that floods servers from the LAN to the // Flood is a long-running goroutine that floods servers from the LAN to the
// given global Serf instance, such as the WAN. This will exit once either of // given global Serf instance, such as the WAN. This will exit once either of
// the Serf instances are shut down. // the Serf instances are shut down.
func (s *Server) Flood(addrFn router.FloodAddrFn, portFn router.FloodPortFn, global *serf.Serf) { func (s *Server) Flood(addrFn router.FloodAddrFn, portFn router.FloodPortFn, dstSerf *serf.Serf) {
s.floodLock.Lock() s.floodLock.Lock()
floodCh := make(chan struct{}) floodCh := make(chan struct{})
s.floodCh = append(s.floodCh, floodCh) s.floodCh = append(s.floodCh, floodCh)
@ -50,17 +50,15 @@ func (s *Server) Flood(addrFn router.FloodAddrFn, portFn router.FloodPortFn, glo
case <-s.serfLAN.ShutdownCh(): case <-s.serfLAN.ShutdownCh():
return return
case <-global.ShutdownCh(): case <-dstSerf.ShutdownCh():
return return
case <-ticker.C: case <-ticker.C:
goto FLOOD router.FloodJoins(s.logger, addrFn, portFn, s.config.Datacenter, s.serfLAN, dstSerf)
case <-floodCh: case <-floodCh:
goto FLOOD router.FloodJoins(s.logger, addrFn, portFn, s.config.Datacenter, s.serfLAN, dstSerf)
} }
FLOOD:
router.FloodJoins(s.logger, addrFn, portFn, s.config.Datacenter, s.serfLAN, global)
} }
} }

View File

@ -18,21 +18,21 @@ type FloodAddrFn func(*metadata.Server) (string, bool)
// will return false if it doesn't have one. // will return false if it doesn't have one.
type FloodPortFn func(*metadata.Server) (int, bool) type FloodPortFn func(*metadata.Server) (int, bool)
// FloodJoins attempts to make sure all Consul servers in the local Serf // FloodJoins attempts to make sure all Consul servers in the src Serf
// instance are joined in the global Serf instance. It assumes names in the // instance are joined in the dst Serf instance. It assumes names in the
// local area are of the form <node> and those in the global area are of the // src area are of the form <node> and those in the dst area are of the
// form <node>.<dc> as is done for WAN and general network areas in Consul // form <node>.<dc> as is done for WAN and general network areas in Consul
// Enterprise. // Enterprise.
func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn, func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn,
localDatacenter string, localSerf *serf.Serf, globalSerf *serf.Serf) { localDatacenter string, srcSerf *serf.Serf, dstSerf *serf.Serf) {
// Names in the global Serf have the datacenter suffixed. // Names in the dst Serf have the datacenter suffixed.
suffix := fmt.Sprintf(".%s", localDatacenter) suffix := fmt.Sprintf(".%s", localDatacenter)
// Index the global side so we can do one pass through the local side // Index the dst side so we can do one pass through the src side
// with cheap lookups. // with cheap lookups.
index := make(map[string]*metadata.Server) index := make(map[string]*metadata.Server)
for _, m := range globalSerf.Members() { for _, m := range dstSerf.Members() {
ok, server := metadata.IsConsulServer(m) ok, server := metadata.IsConsulServer(m)
if !ok { if !ok {
continue continue
@ -42,12 +42,12 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn,
continue continue
} }
localName := strings.TrimSuffix(server.Name, suffix) srcName := strings.TrimSuffix(server.Name, suffix)
index[localName] = server index[srcName] = server
} }
// Now run through the local side and look for joins. // Now run through the src side and look for joins.
for _, m := range localSerf.Members() { for _, m := range srcSerf.Members() {
if m.Status != serf.StatusAlive { if m.Status != serf.StatusAlive {
continue continue
} }
@ -61,7 +61,11 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn,
continue continue
} }
// We can't use the port number from the local Serf, so we just // TODO make RPC
// TODO refactor into one function:
// We can't use the port number from the src Serf, so we just
// get the host part. // get the host part.
addr, _, err := net.SplitHostPort(server.Addr.String()) addr, _, err := net.SplitHostPort(server.Addr.String())
if err != nil { if err != nil {
@ -83,7 +87,7 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn,
addr = net.JoinHostPort(addr, fmt.Sprintf("%d", port)) addr = net.JoinHostPort(addr, fmt.Sprintf("%d", port))
} else { } else {
// If we have an IPv6 address, we should add brackets, // If we have an IPv6 address, we should add brackets,
// single globalSerf.Join expects that. // single dstSerf.Join expects that.
if ip := net.ParseIP(addr); ip != nil { if ip := net.ParseIP(addr); ip != nil {
if ip.To4() == nil { if ip.To4() == nil {
addr = fmt.Sprintf("[%s]", addr) addr = fmt.Sprintf("[%s]", addr)
@ -93,19 +97,21 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn,
} }
} }
globalServerName := fmt.Sprintf("%s.%s", server.Name, server.Datacenter) // end refactor
dstServerName := fmt.Sprintf("%s.%s", server.Name, server.Datacenter)
// Do the join! // Do the join!
n, err := globalSerf.Join([]string{globalServerName + "/" + addr}, true) n, err := dstSerf.Join([]string{dstServerName + "/" + addr}, true)
if err != nil { if err != nil {
logger.Debug("Failed to flood-join server at address", logger.Debug("Failed to flood-join server at address",
"server", globalServerName, "server", dstServerName,
"address", addr, "address", addr,
"error", err, "error", err,
) )
} else if n > 0 { } else if n > 0 {
logger.Debug("Successfully performed flood-join for server at address", logger.Debug("Successfully performed flood-join for server at address",
"server", globalServerName, "server", dstServerName,
"address", addr, "address", addr,
) )
} }