diff --git a/internal/rpc/services/service_file_chunk.go b/internal/rpc/services/service_file_chunk.go index 5771b852..67a193ce 100644 --- a/internal/rpc/services/service_file_chunk.go +++ b/internal/rpc/services/service_file_chunk.go @@ -3,6 +3,7 @@ package services import ( "context" "github.com/TeaOSLab/EdgeAPI/internal/db/models" + rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils" "github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb" ) @@ -31,7 +32,7 @@ func (this *FileChunkService) CreateFileChunk(ctx context.Context, req *pb.Creat // FindAllFileChunkIds 获取的一个文件的所有片段IDs func (this *FileChunkService) FindAllFileChunkIds(ctx context.Context, req *pb.FindAllFileChunkIdsRequest) (*pb.FindAllFileChunkIdsResponse, error) { // 校验请求 - _, _, err := this.ValidateAdminAndUser(ctx, 0, -1) + _, _, err := this.ValidateNodeId(ctx, rpcutils.UserTypeNode, rpcutils.UserTypeAdmin, rpcutils.UserTypeUser) if err != nil { return nil, err } @@ -50,7 +51,7 @@ func (this *FileChunkService) FindAllFileChunkIds(ctx context.Context, req *pb.F // DownloadFileChunk 下载文件片段 func (this *FileChunkService) DownloadFileChunk(ctx context.Context, req *pb.DownloadFileChunkRequest) (*pb.DownloadFileChunkResponse, error) { // 校验请求 - _, _, err := this.ValidateAdminAndUser(ctx, 0, -1) + _, _, err := this.ValidateNodeId(ctx, rpcutils.UserTypeNode, rpcutils.UserTypeAdmin, rpcutils.UserTypeUser) if err != nil { return nil, err }