graph/internal/uid: test and fix overflow bug
MODIFY
graph/internal/uid/uid.go
graph/internal/uid/uid.go
// Package uid implements unique ID provision for graphs.
package uid
-import "gonum.org/v1/gonum/graph/internal/set"
+import (
+ "math"
-// Max is the maximum value of int64.
-const Max = int64(^uint64(0) >> 1)
+ "gonum.org/v1/gonum/graph/internal/set"
+)
+
+// Max is the maximum ID value.
+const Max = math.MaxInt64
// Set implements available ID storage.
type Set struct {
@@ -16,7 +20,8 @@ type Set struct {
used, free set.Int64s
}
-// NewSet returns a new Set. The returned value should not be passed except by pointer.
+// NewSet returns a new Set. The returned value should not be passed
+// except by pointer.
func NewSet() Set {
return Set{maxID: -1, used: make(set.Int64s), free: make(set.Int64s)}
}
@@ -27,10 +32,10 @@ func (s *Set) NewID() int64 {
for id := range s.free {
return id
}
- if s.maxID != Max {
+ if s.maxID != math.MaxInt64 {
return s.maxID + 1
}
- for id := int64(0); id <= s.maxID+1; id++ {
+ for id := int64(0); id <= s.maxID; id++ {
if !s.used.Has(id) {
return id
}

graph/internal/uid: test and fix overflow bug