mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-10-30 07:48:08 -04:00
110 lines
3.4 KiB
Go
110 lines
3.4 KiB
Go
|
package geodata_test
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"io/fs"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"runtime"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/v2fly/v2ray-core/v4/common"
|
||
|
"github.com/v2fly/v2ray-core/v4/common/platform/filesystem"
|
||
|
"github.com/v2fly/v2ray-core/v4/infra/conf/geodata"
|
||
|
_ "github.com/v2fly/v2ray-core/v4/infra/conf/geodata/memconservative"
|
||
|
_ "github.com/v2fly/v2ray-core/v4/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, 0755))
|
||
|
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, 0755))
|
||
|
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)")
|
||
|
}
|