Skip to content

Commit acfe9c7

Browse files
committed
Add bencmark for LookupNetwork
1 parent e92e74b commit acfe9c7

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

reader_test.go

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -652,6 +652,24 @@ func BenchmarkLookup(b *testing.B) {
652652
assert.NoError(b, db.Close(), "error on close")
653653
}
654654

655+
func BenchmarkLookupNetwork(b *testing.B) {
656+
db, err := Open("GeoLite2-City.mmdb")
657+
require.NoError(b, err)
658+
659+
r := rand.New(rand.NewSource(time.Now().UnixNano()))
660+
var result interface{}
661+
662+
ip := make(net.IP, 4)
663+
for i := 0; i < b.N; i++ {
664+
randomIPv4Address(r, ip)
665+
_, _, err = db.LookupNetwork(ip, &result)
666+
if err != nil {
667+
b.Error(err)
668+
}
669+
}
670+
assert.NoError(b, db.Close(), "error on close")
671+
}
672+
655673
func BenchmarkCountryCode(b *testing.B) {
656674
db, err := Open("GeoLite2-City.mmdb")
657675
require.NoError(b, err)

0 commit comments

Comments
 (0)