diff --git a/internal/iplibrary/library_ip2region_test.go b/internal/iplibrary/library_ip2region_test.go index d53e923..2ff394e 100644 --- a/internal/iplibrary/library_ip2region_test.go +++ b/internal/iplibrary/library_ip2region_test.go @@ -106,6 +106,7 @@ func BenchmarkIP2RegionLibrary_Lookup(b *testing.B) { if err != nil { b.Fatal(err) } + b.ResetTimer() for i := 0; i < b.N; i++ { _, _ = library.Lookup("8.8.8.8") diff --git a/internal/iplibrary/manager_city.go b/internal/iplibrary/manager_city.go index eb3b964..7531958 100644 --- a/internal/iplibrary/manager_city.go +++ b/internal/iplibrary/manager_city.go @@ -118,7 +118,7 @@ func (this *CityManager) loop() error { if err != nil { return err } - resp, err := rpcClient.RegionCityRPC().FindAllEnabledRegionCities(rpcClient.Context(), &pb.FindAllEnabledRegionCitiesRequest{}) + resp, err := rpcClient.RegionCityRPC().FindAllRegionCities(rpcClient.Context(), &pb.FindAllRegionCitiesRequest{}) if err != nil { return err } diff --git a/internal/iplibrary/manager_country.go b/internal/iplibrary/manager_country.go index 3576dea..5049667 100644 --- a/internal/iplibrary/manager_country.go +++ b/internal/iplibrary/manager_country.go @@ -117,7 +117,7 @@ func (this *CountryManager) loop() error { if err != nil { return err } - resp, err := rpcClient.RegionCountryRPC().FindAllEnabledRegionCountries(rpcClient.Context(), &pb.FindAllEnabledRegionCountriesRequest{}) + resp, err := rpcClient.RegionCountryRPC().FindAllRegionCountries(rpcClient.Context(), &pb.FindAllRegionCountriesRequest{}) if err != nil { return err } diff --git a/internal/iplibrary/manager_provider.go b/internal/iplibrary/manager_provider.go index 128faa1..3fe6239 100644 --- a/internal/iplibrary/manager_provider.go +++ b/internal/iplibrary/manager_provider.go @@ -117,7 +117,7 @@ func (this *ProviderManager) loop() error { if err != nil { return err } - resp, err := rpcClient.RegionProviderRPC().FindAllEnabledRegionProviders(rpcClient.Context(), &pb.FindAllEnabledRegionProvidersRequest{}) + resp, err := rpcClient.RegionProviderRPC().FindAllRegionProviders(rpcClient.Context(), &pb.FindAllRegionProvidersRequest{}) if err != nil { return err } diff --git a/internal/iplibrary/manager_province.go b/internal/iplibrary/manager_province.go index e668bd4..068e044 100644 --- a/internal/iplibrary/manager_province.go +++ b/internal/iplibrary/manager_province.go @@ -121,7 +121,7 @@ func (this *ProvinceManager) loop() error { if err != nil { return err } - resp, err := rpcClient.RegionProvinceRPC().FindAllEnabledRegionProvincesWithCountryId(rpcClient.Context(), &pb.FindAllEnabledRegionProvincesWithCountryIdRequest{ + resp, err := rpcClient.RegionProvinceRPC().FindAllRegionProvincesWithCountryId(rpcClient.Context(), &pb.FindAllRegionProvincesWithCountryIdRequest{ RegionCountryId: ChinaCountryId, }) if err != nil {