b2d342aa9b
Create global config to specify a list of country codes that should be excluded from node selection during uploads. This exclusion is not implemented when the upload selection cache is disabled. Change-Id: Ic41e8b4f18857a11045668eac23107da99668a72
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package uploadselection
|
|
|
|
import (
|
|
"storj.io/common/storj"
|
|
"storj.io/common/storj/location"
|
|
)
|
|
|
|
// Criteria to filter nodes.
|
|
type Criteria struct {
|
|
ExcludeNodeIDs []storj.NodeID
|
|
AutoExcludeSubnets map[string]struct{} // initialize it with empty map to keep only one node per subnet.
|
|
Placement storj.PlacementConstraint
|
|
ExcludedCountryCodes []location.CountryCode
|
|
}
|
|
|
|
// MatchInclude returns with true if node is selected.
|
|
func (c *Criteria) MatchInclude(node *Node) bool {
|
|
if ContainsID(c.ExcludeNodeIDs, node.ID) {
|
|
return false
|
|
}
|
|
|
|
if !c.Placement.AllowedCountry(node.CountryCode) {
|
|
return false
|
|
}
|
|
|
|
if c.AutoExcludeSubnets != nil {
|
|
if _, excluded := c.AutoExcludeSubnets[node.LastNet]; excluded {
|
|
return false
|
|
}
|
|
c.AutoExcludeSubnets[node.LastNet] = struct{}{}
|
|
}
|
|
|
|
for _, code := range c.ExcludedCountryCodes {
|
|
if node.CountryCode == code {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// ContainsID returns whether ids contain id.
|
|
func ContainsID(ids []storj.NodeID, id storj.NodeID) bool {
|
|
for _, k := range ids {
|
|
if k == id {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|