package geodata_test
|
|
import (
|
"errors"
|
"io/fs"
|
"os"
|
"path/filepath"
|
"runtime"
|
"testing"
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
"github.com/v2fly/v2ray-core/v5/common/platform/filesystem"
|
"github.com/v2fly/v2ray-core/v5/infra/conf/geodata"
|
_ "github.com/v2fly/v2ray-core/v5/infra/conf/geodata/memconservative"
|
_ "github.com/v2fly/v2ray-core/v5/infra/conf/geodata/standard"
|
)
|
|
func init() {
|
const (
|
geoipURL = "https://raw.githubusercontent.com/v2fly/geoip/release/geoip.dat"
|
geositeURL = "https://raw.githubusercontent.com/v2fly/domain-list-community/release/dlc.dat"
|
)
|
|
wd, err := os.Getwd()
|
common.Must(err)
|
|
tempPath := filepath.Join(wd, "..", "..", "..", "testing", "temp")
|
geoipPath := filepath.Join(tempPath, "geoip.dat")
|
geositePath := filepath.Join(tempPath, "geosite.dat")
|
|
os.Setenv("v2ray.location.asset", tempPath)
|
|
if _, err := os.Stat(geoipPath); err != nil && errors.Is(err, fs.ErrNotExist) {
|
common.Must(os.MkdirAll(tempPath, 0o755))
|
geoipBytes, err := common.FetchHTTPContent(geoipURL)
|
common.Must(err)
|
common.Must(filesystem.WriteFile(geoipPath, geoipBytes))
|
}
|
if _, err := os.Stat(geositePath); err != nil && errors.Is(err, fs.ErrNotExist) {
|
common.Must(os.MkdirAll(tempPath, 0o755))
|
geositeBytes, err := common.FetchHTTPContent(geositeURL)
|
common.Must(err)
|
common.Must(filesystem.WriteFile(geositePath, geositeBytes))
|
}
|
}
|
|
func BenchmarkStandardLoaderGeoIP(b *testing.B) {
|
standardLoader, err := geodata.GetGeoDataLoader("standard")
|
common.Must(err)
|
|
m1 := runtime.MemStats{}
|
m2 := runtime.MemStats{}
|
runtime.ReadMemStats(&m1)
|
standardLoader.LoadGeoIP("cn")
|
standardLoader.LoadGeoIP("us")
|
standardLoader.LoadGeoIP("private")
|
runtime.ReadMemStats(&m2)
|
|
b.ReportMetric(float64(m2.Alloc-m1.Alloc)/1024/1024, "MiB(GeoIP-Alloc)")
|
b.ReportMetric(float64(m2.TotalAlloc-m1.TotalAlloc)/1024/1024, "MiB(GeoIP-TotalAlloc)")
|
}
|
|
func BenchmarkStandardLoaderGeoSite(b *testing.B) {
|
standardLoader, err := geodata.GetGeoDataLoader("standard")
|
common.Must(err)
|
|
m3 := runtime.MemStats{}
|
m4 := runtime.MemStats{}
|
runtime.ReadMemStats(&m3)
|
standardLoader.LoadGeoSite("cn")
|
standardLoader.LoadGeoSite("geolocation-!cn")
|
standardLoader.LoadGeoSite("private")
|
runtime.ReadMemStats(&m4)
|
|
b.ReportMetric(float64(m4.Alloc-m3.Alloc)/1024/1024, "MiB(GeoSite-Alloc)")
|
b.ReportMetric(float64(m4.TotalAlloc-m3.TotalAlloc)/1024/1024, "MiB(GeoSite-TotalAlloc)")
|
}
|
|
func BenchmarkMemconservativeLoaderGeoIP(b *testing.B) {
|
standardLoader, err := geodata.GetGeoDataLoader("memconservative")
|
common.Must(err)
|
|
m1 := runtime.MemStats{}
|
m2 := runtime.MemStats{}
|
runtime.ReadMemStats(&m1)
|
standardLoader.LoadGeoIP("cn")
|
standardLoader.LoadGeoIP("us")
|
standardLoader.LoadGeoIP("private")
|
runtime.ReadMemStats(&m2)
|
|
b.ReportMetric(float64(m2.Alloc-m1.Alloc)/1024, "KiB(GeoIP-Alloc)")
|
b.ReportMetric(float64(m2.TotalAlloc-m1.TotalAlloc)/1024/1024, "MiB(GeoIP-TotalAlloc)")
|
}
|
|
func BenchmarkMemconservativeLoaderGeoSite(b *testing.B) {
|
standardLoader, err := geodata.GetGeoDataLoader("memconservative")
|
common.Must(err)
|
|
m3 := runtime.MemStats{}
|
m4 := runtime.MemStats{}
|
runtime.ReadMemStats(&m3)
|
standardLoader.LoadGeoSite("cn")
|
standardLoader.LoadGeoSite("geolocation-!cn")
|
standardLoader.LoadGeoSite("private")
|
runtime.ReadMemStats(&m4)
|
|
b.ReportMetric(float64(m4.Alloc-m3.Alloc)/1024, "KiB(GeoSite-Alloc)")
|
b.ReportMetric(float64(m4.TotalAlloc-m3.TotalAlloc)/1024/1024, "MiB(GeoSite-TotalAlloc)")
|
}
|
|
func BenchmarkAllLoader(b *testing.B) {
|
type testingProfileForLoader struct {
|
name string
|
}
|
testCase := []testingProfileForLoader{
|
{"standard"},
|
{"memconservative"},
|
}
|
for _, v := range testCase {
|
b.Run(v.name, func(b *testing.B) {
|
b.Run("Geosite", func(b *testing.B) {
|
loader, err := geodata.GetGeoDataLoader(v.name)
|
if err != nil {
|
b.Fatal(err)
|
}
|
|
m3 := runtime.MemStats{}
|
m4 := runtime.MemStats{}
|
runtime.ReadMemStats(&m3)
|
loader.LoadGeoSite("cn")
|
loader.LoadGeoSite("geolocation-!cn")
|
loader.LoadGeoSite("private")
|
runtime.ReadMemStats(&m4)
|
|
b.ReportMetric(float64(m4.Alloc-m3.Alloc)/1024, "KiB(GeoSite-Alloc)")
|
b.ReportMetric(float64(m4.TotalAlloc-m3.TotalAlloc)/1024/1024, "MiB(GeoSite-TotalAlloc)")
|
})
|
|
b.Run("GeoIP", func(b *testing.B) {
|
loader, err := geodata.GetGeoDataLoader(v.name)
|
if err != nil {
|
b.Fatal(err)
|
}
|
|
m1 := runtime.MemStats{}
|
m2 := runtime.MemStats{}
|
runtime.ReadMemStats(&m1)
|
loader.LoadGeoIP("cn")
|
loader.LoadGeoIP("us")
|
loader.LoadGeoIP("private")
|
runtime.ReadMemStats(&m2)
|
|
b.ReportMetric(float64(m2.Alloc-m1.Alloc)/1024/1024, "MiB(GeoIP-Alloc)")
|
b.ReportMetric(float64(m2.TotalAlloc-m1.TotalAlloc)/1024/1024, "MiB(GeoIP-TotalAlloc)")
|
})
|
})
|
}
|
}
|
