mirror of https://github.com/status-im/consul.git
256 lines
7.0 KiB
Go
256 lines
7.0 KiB
Go
package consul
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/go-memdb"
|
|
|
|
"github.com/hashicorp/consul/acl"
|
|
"github.com/hashicorp/consul/agent/consul/state"
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"github.com/hashicorp/consul/logging"
|
|
"github.com/hashicorp/consul/types"
|
|
)
|
|
|
|
// Coordinate manages queries and updates for network coordinates.
|
|
type Coordinate struct {
|
|
// srv is a pointer back to the server.
|
|
srv *Server
|
|
|
|
logger hclog.Logger
|
|
|
|
// updates holds pending coordinate updates for the given nodes. This is
|
|
// keyed by node:segment so we can get a coordinate for each segment for
|
|
// servers, and we only track the latest update per node:segment.
|
|
updates map[string]*structs.CoordinateUpdateRequest
|
|
|
|
// updatesLock synchronizes access to the updates map.
|
|
updatesLock sync.Mutex
|
|
}
|
|
|
|
// NewCoordinate returns a new Coordinate endpoint.
|
|
func NewCoordinate(srv *Server, logger hclog.Logger) *Coordinate {
|
|
c := &Coordinate{
|
|
srv: srv,
|
|
logger: logger.Named(logging.Coordinate),
|
|
updates: make(map[string]*structs.CoordinateUpdateRequest),
|
|
}
|
|
|
|
go c.batchUpdate()
|
|
return c
|
|
}
|
|
|
|
// batchUpdate is a long-running routine that flushes pending coordinates to the
|
|
// Raft log in batches.
|
|
func (c *Coordinate) batchUpdate() {
|
|
for {
|
|
select {
|
|
case <-time.After(c.srv.config.CoordinateUpdatePeriod):
|
|
if err := c.batchApplyUpdates(); err != nil {
|
|
c.logger.Warn("Batch update failed", "error", err)
|
|
}
|
|
case <-c.srv.shutdownCh:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// batchApplyUpdates applies all pending updates to the Raft log in a series of
|
|
// batches.
|
|
func (c *Coordinate) batchApplyUpdates() error {
|
|
// Grab the pending updates and release the lock so we can still handle
|
|
// incoming messages.
|
|
c.updatesLock.Lock()
|
|
pending := c.updates
|
|
c.updates = make(map[string]*structs.CoordinateUpdateRequest)
|
|
c.updatesLock.Unlock()
|
|
|
|
// Enforce the rate limit.
|
|
limit := c.srv.config.CoordinateUpdateBatchSize * c.srv.config.CoordinateUpdateMaxBatches
|
|
size := len(pending)
|
|
if size > limit {
|
|
c.logger.Warn("Discarded coordinate updates", "number_discarded", size-limit)
|
|
size = limit
|
|
}
|
|
|
|
// Transform the map into a slice that we can feed to the Raft log in
|
|
// batches.
|
|
i := 0
|
|
updates := make(structs.Coordinates, size)
|
|
for _, update := range pending {
|
|
if !(i < size) {
|
|
break
|
|
}
|
|
|
|
updates[i] = &structs.Coordinate{
|
|
Node: update.Node,
|
|
Segment: update.Segment,
|
|
Coord: update.Coord,
|
|
}
|
|
i++
|
|
}
|
|
|
|
// Apply the updates to the Raft log in batches.
|
|
for start := 0; start < size; start += c.srv.config.CoordinateUpdateBatchSize {
|
|
end := start + c.srv.config.CoordinateUpdateBatchSize
|
|
if end > size {
|
|
end = size
|
|
}
|
|
|
|
// We set the "safe to ignore" flag on this update type so old
|
|
// servers don't crash if they see one of these.
|
|
t := structs.CoordinateBatchUpdateType | structs.IgnoreUnknownTypeFlag
|
|
|
|
slice := updates[start:end]
|
|
_, err := c.srv.raftApply(t, slice)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Update inserts or updates the LAN coordinate of a node.
|
|
func (c *Coordinate) Update(args *structs.CoordinateUpdateRequest, reply *struct{}) (err error) {
|
|
if done, err := c.srv.ForwardRPC("Coordinate.Update", args, reply); done {
|
|
return err
|
|
}
|
|
|
|
// Older clients can send coordinates with invalid numeric values like
|
|
// NaN and Inf. We guard against these coming in, though newer clients
|
|
// should never send these.
|
|
if !args.Coord.IsValid() {
|
|
return fmt.Errorf("invalid coordinate")
|
|
}
|
|
|
|
// Since this is a coordinate coming from some place else we harden this
|
|
// and look for dimensionality problems proactively.
|
|
coord, err := c.srv.serfLAN.GetCoordinate()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !coord.IsCompatibleWith(args.Coord) {
|
|
return fmt.Errorf("incompatible coordinate")
|
|
}
|
|
|
|
// Fetch the ACL token, if any, and enforce the node policy if enabled.
|
|
authz, err := c.srv.ResolveToken(args.Token)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if authz != nil {
|
|
var authzContext acl.AuthorizerContext
|
|
structs.DefaultEnterpriseMeta().FillAuthzContext(&authzContext)
|
|
if authz.NodeWrite(args.Node, &authzContext) != acl.Allow {
|
|
return acl.ErrPermissionDenied
|
|
}
|
|
}
|
|
|
|
// Add the coordinate to the map of pending updates.
|
|
key := fmt.Sprintf("%s:%s", args.Node, args.Segment)
|
|
c.updatesLock.Lock()
|
|
c.updates[key] = args
|
|
c.updatesLock.Unlock()
|
|
return nil
|
|
}
|
|
|
|
// ListDatacenters returns the list of datacenters and their respective nodes
|
|
// and the raw coordinates of those nodes (if no coordinates are available for
|
|
// any of the nodes, the node list may be empty). This endpoint will not return
|
|
// information about the LAN network area.
|
|
func (c *Coordinate) ListDatacenters(args *struct{}, reply *[]structs.DatacenterMap) error {
|
|
maps, err := c.srv.router.GetDatacenterMaps()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var out []structs.DatacenterMap
|
|
|
|
// Strip the datacenter suffixes from all the node names.
|
|
for _, dcMap := range maps {
|
|
if dcMap.AreaID == types.AreaLAN {
|
|
continue
|
|
}
|
|
|
|
suffix := fmt.Sprintf(".%s", dcMap.Datacenter)
|
|
for j := range dcMap.Coordinates {
|
|
node := dcMap.Coordinates[j].Node
|
|
dcMap.Coordinates[j].Node = strings.TrimSuffix(node, suffix)
|
|
}
|
|
|
|
out = append(out, dcMap)
|
|
}
|
|
|
|
*reply = out
|
|
return nil
|
|
}
|
|
|
|
// ListNodes returns the list of nodes with their raw network coordinates (if no
|
|
// coordinates are available for a node it won't appear in this list).
|
|
func (c *Coordinate) ListNodes(args *structs.DCSpecificRequest, reply *structs.IndexedCoordinates) error {
|
|
if done, err := c.srv.ForwardRPC("Coordinate.ListNodes", args, reply); done {
|
|
return err
|
|
}
|
|
|
|
return c.srv.blockingQuery(&args.QueryOptions,
|
|
&reply.QueryMeta,
|
|
func(ws memdb.WatchSet, state *state.Store) error {
|
|
index, coords, err := state.Coordinates(ws)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
reply.Index, reply.Coordinates = index, coords
|
|
if err := c.srv.filterACL(args.Token, reply); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// Node returns the raw coordinates for a single node.
|
|
func (c *Coordinate) Node(args *structs.NodeSpecificRequest, reply *structs.IndexedCoordinates) error {
|
|
if done, err := c.srv.ForwardRPC("Coordinate.Node", args, reply); done {
|
|
return err
|
|
}
|
|
|
|
// Fetch the ACL token, if any, and enforce the node policy if enabled.
|
|
|
|
authz, err := c.srv.ResolveToken(args.Token)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if authz != nil {
|
|
var authzContext acl.AuthorizerContext
|
|
structs.WildcardEnterpriseMeta().FillAuthzContext(&authzContext)
|
|
if authz.NodeRead(args.Node, &authzContext) != acl.Allow {
|
|
return acl.ErrPermissionDenied
|
|
}
|
|
}
|
|
|
|
return c.srv.blockingQuery(&args.QueryOptions,
|
|
&reply.QueryMeta,
|
|
func(ws memdb.WatchSet, state *state.Store) error {
|
|
index, nodeCoords, err := state.Coordinate(args.Node, ws)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var coords structs.Coordinates
|
|
for segment, coord := range nodeCoords {
|
|
coords = append(coords, &structs.Coordinate{
|
|
Node: args.Node,
|
|
Segment: segment,
|
|
Coord: coord,
|
|
})
|
|
}
|
|
reply.Index, reply.Coordinates = index, coords
|
|
return nil
|
|
})
|
|
}
|