autopilot: make n uint32 in chooseN

To avoid negative values being input, as it doesn't really make sense.
This commit is contained in:
Johan T. Halseth committed 2018-12-19 08:27:34 +01:00
1 parent 2f17030e8c
commit a202860ce1
2 files changed
+3 -3

No files matched your search

+2 -2
View File
@@ -52,7 +52,7 @@ func weightedChoice(s map[NodeID]*AttachmentDirective) (NodeID, error) {
// chooseN picks at random min[n, len(s)] nodes if from the
// AttachmentDirectives map, with a probability weighted by their score.
func chooseN(n int, s map[NodeID]*AttachmentDirective) (
func chooseN(n uint32, s map[NodeID]*AttachmentDirective) (
map[NodeID]*AttachmentDirective, error) {
// Keep a map of nodes not yet choosen.
@@ -64,7 +64,7 @@ func chooseN(n int, s map[NodeID]*AttachmentDirective) (
// Pick a weighted choice from the remaining nodes as long as there are
// nodes left, and we haven't already picked n.
chosen := make(map[NodeID]*AttachmentDirective)
for len(chosen) < n && len(rem) > 0 {
for len(chosen) < int(n) && len(rem) > 0 {
choice, err := weightedChoice(rem)
if err == ErrNoPositive {
return chosen, nil