lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-30 05:56:54 -05:00
Code Issues Releases Wiki Activity

refactor test for loader benchmark

Browse Source
This commit is contained in:
Shelikhoo 2021-05-05 13:50:56 +01:00
parent 916d3857a0
commit 77a573caf1
No known key found for this signature in database
GPG Key ID: C4D5E79D22B25316
1 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
infra/conf/geodata/geodata_test.go
View File

@ -107,3 +107,52 @@ func BenchmarkMemconservativeLoaderGeoSite(b *testing.B) {
b.ReportMetric(float64(m4.Alloc-m3.Alloc)/1024, "KiB(GeoSite-Alloc)") b.ReportMetric(float64(m4.Alloc-m3.Alloc)/1024, "KiB(GeoSite-Alloc)")
b.ReportMetric(float64(m4.TotalAlloc-m3.TotalAlloc)/1024/1024, "MiB(GeoSite-TotalAlloc)") 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)")
})
})
}
}