80422c0dfe
Fixes a bug whereby servers present in multiple network areas would be properly segmented in the Router, but not in the gRPC mirror. This would lead servers in the current datacenter leaving from a network area (possibly during the network area's removal) from deleting their own records that still exist in the standard WAN area. The gRPC client stack uses the gRPC server tracker to execute all RPCs, even those targeting members of the current datacenter (which is unlike the net/rpc stack which has a bypass mechanism). This would manifest as a gRPC method call never opening a socket because it would block forever waiting for the current datacenter's pool of servers to be non-empty.
34 lines
925 B
Go
34 lines
925 B
Go
package router
|
|
|
|
import (
|
|
"github.com/hashicorp/consul/agent/metadata"
|
|
"github.com/hashicorp/consul/types"
|
|
)
|
|
|
|
// ServerTracker is called when Router is notified of a server being added or
|
|
// removed.
|
|
type ServerTracker interface {
|
|
NewRebalancer(dc string) func()
|
|
AddServer(types.AreaID, *metadata.Server)
|
|
RemoveServer(types.AreaID, *metadata.Server)
|
|
}
|
|
|
|
// Rebalancer is called periodically to re-order the servers so that the load on the
|
|
// servers is evenly balanced.
|
|
type Rebalancer func()
|
|
|
|
// NoOpServerTracker is a ServerTracker that does nothing. Used when gRPC is not
|
|
// enabled.
|
|
type NoOpServerTracker struct{}
|
|
|
|
// Rebalance does nothing
|
|
func (NoOpServerTracker) NewRebalancer(string) func() {
|
|
return func() {}
|
|
}
|
|
|
|
// AddServer does nothing
|
|
func (NoOpServerTracker) AddServer(types.AreaID, *metadata.Server) {}
|
|
|
|
// RemoveServer does nothing
|
|
func (NoOpServerTracker) RemoveServer(types.AreaID, *metadata.Server) {}
|