TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-16 01:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加批量查询IP信息API

Browse Source
This commit is contained in:
刘祥超
2021-01-13 17:00:29 +08:00
parent 42e0103b86
commit 2a0bc1f338
2 changed files with 53 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
internal/rpc/services/service_ip_library.go
View File

@@ -187,7 +187,7 @@ func (this *IPLibraryService) LookupIPRegion(ctx context.Context, req *pb.Lookup
return nil, err
}
if result == nil {
return &pb.LookupIPRegionResponse{Region: nil}, nil
return &pb.LookupIPRegionResponse{IpRegion: nil}, nil
}
tx := this.NullTx()
@@ -202,7 +202,7 @@ func (this *IPLibraryService) LookupIPRegion(ctx context.Context, req *pb.Lookup
return nil, err
}
return &pb.LookupIPRegionResponse{Region: &pb.IPRegion{
return &pb.LookupIPRegionResponse{IpRegion: &pb.IPRegion{
Country: result.Country,
Region: result.Region,
Province: result.Province,
@@ -210,5 +210,37 @@ func (this *IPLibraryService) LookupIPRegion(ctx context.Context, req *pb.Lookup
Isp: result.ISP,
CountryId: countryId,
ProvinceId: provinceId,
Summary: result.Summary(),
}}, nil
}
// 查询一组IP信息
func (this *IPLibraryService) LookupIPRegions(ctx context.Context, req *pb.LookupIPRegionsRequest) (*pb.LookupIPRegionsResponse, error) {
// 校验请求
_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
if err != nil {
return nil, err
}
result := map[string]*pb.IPRegion{}
if len(req.IpList) > 0 {
for _, ip := range req.IpList {
info, err := iplibrary.SharedLibrary.Lookup(ip)
if err != nil {
return nil, err
}
if info != nil {
result[ip] = &pb.IPRegion{
Country: info.Country,
Region: info.Region,
Province: info.Province,
City: info.City,
Isp: info.ISP,
Summary: info.Summary(),
}
}
}
}
return &pb.LookupIPRegionsResponse{IpRegionMap: result}, nil
}