frostfs-node/pkg/services/object_manager/placement/neighbours.go

package placement

import (
	"math"

	"github.com/nspcc-dev/hrw"
	"github.com/nspcc-dev/neofs-node/pkg/core/netmap"
	"github.com/nspcc-dev/neofs-node/pkg/network/peers"
	"go.uber.org/zap"
)

func calculateCount(n int) int {
	if n < 30 {
		return n
	}

	return int(1.4*math.Log(float64(n))+9) + 1
}

// Neighbours peers that which are distributed by hrw(seed)
// If full flag is set, all set of peers returns.
// Otherwise, result size depends on calculateCount function.
func (p *placement) Neighbours(seed, epoch uint64, full bool) []peers.ID {
	nm := p.nmStore.get(epoch)
	if nm == nil {
		p.log.Error("could not receive network state",
			zap.Uint64("epoch", epoch),
		)

		return nil
	}

	rPeers := p.listPeers(nm.Nodes(), !full)

	hrw.SortSliceByValue(rPeers, seed)

	if full {
		return rPeers
	}

	var (
		ln  = len(rPeers)
		cut = calculateCount(ln)
	)

	if cut > ln {
		cut = ln
	}

	return rPeers[:cut]
}

func (p *placement) listPeers(nodes []netmap.Info, exclSelf bool) []peers.ID {
	var (
		id     = p.ps.SelfID()
		result = make([]peers.ID, 0, len(nodes))
	)

	for i := range nodes {
		key := peers.IDFromBinary(nodes[i].PublicKey())
		if exclSelf && id.Equal(key) {
			continue
		}

		result = append(result, key)
	}

	return result
}