TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-12-09 19:50:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

新版IP库管理阶段性提交（未完成）

Browse Source
This commit is contained in:
GoEdgeLab
2022-08-13 23:55:48 +08:00
parent 0388d6c557
commit 0e411ff59f
42 changed files with 2628 additions and 172 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
internal/rpc/services/service_region_provider.go
View File

@@ -14,6 +14,7 @@ type RegionProviderService struct {
}
// FindAllEnabledRegionProviders 查找所有ISP
// Deprecated
func (this *RegionProviderService) FindAllEnabledRegionProviders(ctx context.Context, req *pb.FindAllEnabledRegionProvidersRequest) (*pb.FindAllEnabledRegionProvidersResponse, error) {
_, _, err := this.ValidateNodeId(ctx)
if err != nil {
@@ -29,9 +30,12 @@ func (this *RegionProviderService) FindAllEnabledRegionProviders(ctx context.Con
var pbProviders = []*pb.RegionProvider{}
for _, provider := range providers {
pbProviders = append(pbProviders, &pb.RegionProvider{
Id: int64(provider.Id),
Name: provider.Name,
Codes: provider.DecodeCodes(),
Id: int64(provider.Id),
Name: provider.Name,
Codes: provider.DecodeCodes(),
CustomName: provider.CustomName,
CustomCodes: provider.DecodeCustomCodes(),
DisplayName: provider.DisplayName(),
})
}
@@ -41,6 +45,7 @@ func (this *RegionProviderService) FindAllEnabledRegionProviders(ctx context.Con
}
// FindEnabledRegionProvider 查找单个ISP信息
// Deprecated
func (this *RegionProviderService) FindEnabledRegionProvider(ctx context.Context, req *pb.FindEnabledRegionProviderRequest) (*pb.FindEnabledRegionProviderResponse, error) {
_, _, err := this.ValidateNodeId(ctx)
if err != nil {
@@ -60,9 +65,87 @@ func (this *RegionProviderService) FindEnabledRegionProvider(ctx context.Context
return &pb.FindEnabledRegionProviderResponse{
RegionProvider: &pb.RegionProvider{
Id: int64(provider.Id),
Name: provider.Name,
Codes: provider.DecodeCodes(),
Id: int64(provider.Id),
Name: provider.Name,
Codes: provider.DecodeCodes(),
CustomName: provider.CustomName,
CustomCodes: provider.DecodeCustomCodes(),
DisplayName: provider.DisplayName(),
},
}, nil
}
// FindAllRegionProviders 查找所有ISP
func (this *RegionProviderService) FindAllRegionProviders(ctx context.Context, req *pb.FindAllRegionProvidersRequest) (*pb.FindAllRegionProvidersResponse, error) {
_, _, err := this.ValidateNodeId(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
providers, err := regions.SharedRegionProviderDAO.FindAllEnabledProviders(tx)
if err != nil {
return nil, err
}
var pbProviders = []*pb.RegionProvider{}
for _, provider := range providers {
pbProviders = append(pbProviders, &pb.RegionProvider{
Id: int64(provider.Id),
Name: provider.Name,
Codes: provider.DecodeCodes(),
CustomName: provider.CustomName,
CustomCodes: provider.DecodeCustomCodes(),
DisplayName: provider.DisplayName(),
})
}
return &pb.FindAllRegionProvidersResponse{
RegionProviders: pbProviders,
}, nil
}
// FindRegionProvider 查找单个ISP信息
func (this *RegionProviderService) FindRegionProvider(ctx context.Context, req *pb.FindRegionProviderRequest) (*pb.FindRegionProviderResponse, error) {
_, _, err := this.ValidateNodeId(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
provider, err := regions.SharedRegionProviderDAO.FindEnabledRegionProvider(tx, req.RegionProviderId)
if err != nil {
return nil, err
}
if provider == nil {
return &pb.FindRegionProviderResponse{
RegionProvider: nil,
}, nil
}
return &pb.FindRegionProviderResponse{
RegionProvider: &pb.RegionProvider{
Id: int64(provider.Id),
Name: provider.Name,
Codes: provider.DecodeCodes(),
CustomName: provider.CustomName,
CustomCodes: provider.DecodeCustomCodes(),
DisplayName: provider.DisplayName(),
},
}, nil
}
// UpdateRegionProviderCustom 修改城市定制信息
func (this *RegionProviderService) UpdateRegionProviderCustom(ctx context.Context, req *pb.UpdateRegionProviderCustomRequest) (*pb.RPCSuccess, error) {
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
err = regions.SharedRegionProviderDAO.UpdateProviderCustom(tx, req.RegionProviderId, req.CustomName, req.CustomCodes)
if err != nil {
return nil, err
}
return this.Success()
}