From 6994c0d47f9bb9c5df52542b5b8b7404bfe0cd12 Mon Sep 17 00:00:00 2001 From: Hans Hasselberg Date: Mon, 27 Apr 2020 09:53:40 +0200 Subject: [PATCH] agent: rename local/global to src/dst --- agent/consul/flood.go | 10 ++++----- agent/router/serf_flooder.go | 40 +++++++++++++++++++++--------------- 2 files changed, 27 insertions(+), 23 deletions(-) diff --git a/agent/consul/flood.go b/agent/consul/flood.go index 1287d0d7ad..d8de89b324 100644 --- a/agent/consul/flood.go +++ b/agent/consul/flood.go @@ -24,7 +24,7 @@ func (s *Server) FloodNotify() { // 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 // 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() floodCh := make(chan struct{}) 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(): return - case <-global.ShutdownCh(): + case <-dstSerf.ShutdownCh(): return case <-ticker.C: - goto FLOOD + router.FloodJoins(s.logger, addrFn, portFn, s.config.Datacenter, s.serfLAN, dstSerf) 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) } } diff --git a/agent/router/serf_flooder.go b/agent/router/serf_flooder.go index 35a4141e49..9a6f242f34 100644 --- a/agent/router/serf_flooder.go +++ b/agent/router/serf_flooder.go @@ -18,21 +18,21 @@ type FloodAddrFn func(*metadata.Server) (string, bool) // will return false if it doesn't have one. type FloodPortFn func(*metadata.Server) (int, bool) -// FloodJoins attempts to make sure all Consul servers in the local Serf -// instance are joined in the global Serf instance. It assumes names in the -// local area are of the form and those in the global area are of the +// FloodJoins attempts to make sure all Consul servers in the src Serf +// instance are joined in the dst Serf instance. It assumes names in the +// src area are of the form and those in the dst area are of the // form . as is done for WAN and general network areas in Consul // Enterprise. 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) - // 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. index := make(map[string]*metadata.Server) - for _, m := range globalSerf.Members() { + for _, m := range dstSerf.Members() { ok, server := metadata.IsConsulServer(m) if !ok { continue @@ -42,12 +42,12 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn, continue } - localName := strings.TrimSuffix(server.Name, suffix) - index[localName] = server + srcName := strings.TrimSuffix(server.Name, suffix) + index[srcName] = server } - // Now run through the local side and look for joins. - for _, m := range localSerf.Members() { + // Now run through the src side and look for joins. + for _, m := range srcSerf.Members() { if m.Status != serf.StatusAlive { continue } @@ -61,7 +61,11 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn, 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. addr, _, err := net.SplitHostPort(server.Addr.String()) if err != nil { @@ -83,7 +87,7 @@ func FloodJoins(logger hclog.Logger, addrFn FloodAddrFn, portFn FloodPortFn, addr = net.JoinHostPort(addr, fmt.Sprintf("%d", port)) } else { // 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.To4() == nil { 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! - n, err := globalSerf.Join([]string{globalServerName + "/" + addr}, true) + n, err := dstSerf.Join([]string{dstServerName + "/" + addr}, true) if err != nil { logger.Debug("Failed to flood-join server at address", - "server", globalServerName, + "server", dstServerName, "address", addr, "error", err, ) } else if n > 0 { logger.Debug("Successfully performed flood-join for server at address", - "server", globalServerName, + "server", dstServerName, "address", addr, ) }