mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 23:20:25 +08:00
109 lines
2.1 KiB
Go
109 lines
2.1 KiB
Go
package iplibrary
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeNode/internal/utils"
|
|
"github.com/iwind/TeaGo/assert"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestIPItem_Contains(t *testing.T) {
|
|
a := assert.NewAssertion(t)
|
|
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.100"),
|
|
IPTo: 0,
|
|
ExpiredAt: 0,
|
|
}
|
|
a.IsTrue(item.Contains(utils.IP2Long("192.168.1.100")))
|
|
}
|
|
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.100"),
|
|
IPTo: 0,
|
|
ExpiredAt: time.Now().Unix() + 1,
|
|
}
|
|
a.IsTrue(item.Contains(utils.IP2Long("192.168.1.100")))
|
|
}
|
|
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.100"),
|
|
IPTo: 0,
|
|
ExpiredAt: time.Now().Unix() - 1,
|
|
}
|
|
a.IsFalse(item.Contains(utils.IP2Long("192.168.1.100")))
|
|
}
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.100"),
|
|
IPTo: 0,
|
|
ExpiredAt: 0,
|
|
}
|
|
a.IsFalse(item.Contains(utils.IP2Long("192.168.1.101")))
|
|
}
|
|
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.1"),
|
|
IPTo: utils.IP2Long("192.168.1.101"),
|
|
ExpiredAt: 0,
|
|
}
|
|
a.IsTrue(item.Contains(utils.IP2Long("192.168.1.100")))
|
|
}
|
|
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.1"),
|
|
IPTo: utils.IP2Long("192.168.1.100"),
|
|
ExpiredAt: 0,
|
|
}
|
|
a.IsTrue(item.Contains(utils.IP2Long("192.168.1.100")))
|
|
}
|
|
|
|
{
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.1"),
|
|
IPTo: utils.IP2Long("192.168.1.101"),
|
|
ExpiredAt: 0,
|
|
}
|
|
a.IsTrue(item.Contains(utils.IP2Long("192.168.1.1")))
|
|
}
|
|
}
|
|
|
|
func TestIPItem_Memory(t *testing.T) {
|
|
var list = NewIPList()
|
|
for i := 0; i < 2_000_000; i ++ {
|
|
list.Add(&IPItem{
|
|
Type: "ip",
|
|
Id: uint64(i),
|
|
IPFrom: utils.IP2Long("192.168.1.1"),
|
|
IPTo: 0,
|
|
ExpiredAt: time.Now().Unix(),
|
|
EventLevel: "",
|
|
})
|
|
}
|
|
t.Log("waiting")
|
|
time.Sleep(10 * time.Second)
|
|
}
|
|
|
|
func BenchmarkIPItem_Contains(b *testing.B) {
|
|
runtime.GOMAXPROCS(1)
|
|
|
|
item := &IPItem{
|
|
IPFrom: utils.IP2Long("192.168.1.1"),
|
|
IPTo: utils.IP2Long("192.168.1.101"),
|
|
ExpiredAt: 0,
|
|
}
|
|
ip := utils.IP2Long("192.168.1.1")
|
|
for i := 0; i < b.N; i++ {
|
|
for j := 0; j < 10_000; j++ {
|
|
item.Contains(ip)
|
|
}
|
|
}
|
|
}
|
|
|