diff --git a/internal/rpc/services/service_base.go b/internal/rpc/services/service_base.go index 9574e096..f7519a75 100644 --- a/internal/rpc/services/service_base.go +++ b/internal/rpc/services/service_base.go @@ -228,6 +228,10 @@ func (this *BaseService) PermissionError() error { return errors.New("Permission Denied") } +func (this *BaseService) NotImplementedYet() error { + return errors.New("not implemented yet") +} + // NullTx 空的数据库事务 func (this *BaseService) NullTx() *dbs.Tx { return nil diff --git a/internal/rpc/services/service_node_region.go b/internal/rpc/services/service_node_region.go index f5501d08..57afaf0c 100644 --- a/internal/rpc/services/service_node_region.go +++ b/internal/rpc/services/service_node_region.go @@ -61,7 +61,7 @@ func (this *NodeRegionService) DeleteNodeRegion(ctx context.Context, req *pb.Del // FindAllEnabledNodeRegions 查找所有区域 func (this *NodeRegionService) FindAllEnabledNodeRegions(ctx context.Context, req *pb.FindAllEnabledNodeRegionsRequest) (*pb.FindAllEnabledNodeRegionsResponse, error) { - _, err := this.ValidateAdmin(ctx) + _, _, err := this.ValidateAdminAndUser(ctx, false) if err != nil { return nil, err } @@ -87,7 +87,7 @@ func (this *NodeRegionService) FindAllEnabledNodeRegions(ctx context.Context, re // FindAllAvailableNodeRegions 查找所有启用的区域 func (this *NodeRegionService) FindAllAvailableNodeRegions(ctx context.Context, req *pb.FindAllAvailableNodeRegionsRequest) (*pb.FindAllAvailableNodeRegionsResponse, error) { - _, err := this.ValidateAdmin(ctx) + _, _, err := this.ValidateAdminAndUser(ctx, false) if err != nil { return nil, err } diff --git a/internal/rpc/services/service_user_ext.go b/internal/rpc/services/service_user_ext.go new file mode 100644 index 00000000..46f9f3e8 --- /dev/null +++ b/internal/rpc/services/service_user_ext.go @@ -0,0 +1,24 @@ +// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn . +//go:build !plus + +package services + +import ( + "context" + "github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb" +) + +// FindUserPriceInfo 读取用户计费信息 +func (this *UserService) FindUserPriceInfo(ctx context.Context, req *pb.FindUserPriceInfoRequest) (*pb.FindUserPriceInfoResponse, error) { + return nil, this.NotImplementedYet() +} + +// UpdateUserPriceType 修改用户计费方式 +func (this *UserService) UpdateUserPriceType(ctx context.Context, req *pb.UpdateUserPriceTypeRequest) (*pb.RPCSuccess, error) { + return nil, this.NotImplementedYet() +} + +// UpdateUserPricePeriod 修改用户计费周期 +func (this *UserService) UpdateUserPricePeriod(ctx context.Context, req *pb.UpdateUserPricePeriodRequest) (*pb.RPCSuccess, error) { + return nil, this.NotImplementedYet() +}