阶段性提交

internal/rpc/services/service_node_grant.go

@@ -0,0 +1,89 @@
+package services
+
+import (
+	"context"
+	"errors"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/rpc/pb"
+	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
+)
+
+type NodeGrantService struct {
+}
+
+func (this *NodeGrantService) CreateNodeGrant(ctx context.Context, req *pb.CreateNodeGrantRequest) (*pb.CreateNodeGrantResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	grantId, err := models.SharedNodeGrantDAO.CreateGrant(req.Name, req.Method, req.Username, req.Password, req.PrivateKey, req.Description, req.NodeId)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CreateNodeGrantResponse{
+		GrantId: grantId,
+	}, err
+}
+
+func (this *NodeGrantService) UpdateNodeGrant(ctx context.Context, req *pb.UpdateNodeGrantRequest) (*pb.UpdateNodeGrantResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	if req.GrantId <= 0 {
+		return nil, errors.New("wrong grantId")
+	}
+
+	err = models.SharedNodeGrantDAO.UpdateGrant(req.GrantId, req.Name, req.Method, req.Username, req.Password, req.PrivateKey, req.Description, req.NodeId)
+	return &pb.UpdateNodeGrantResponse{}, err
+}
+
+func (this *NodeGrantService) DisableNodeGrant(ctx context.Context, req *pb.DisableNodeGrantRequest) (*pb.DisableNodeGrantResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedNodeGrantDAO.DisableNodeGrant(req.GrantId)
+	return &pb.DisableNodeGrantResponse{}, err
+}
+
+func (this *NodeGrantService) CountAllEnabledNodeGrants(ctx context.Context, req *pb.CountAllEnabledNodeGrantsRequest) (*pb.CountAllEnabledNodeGrantsResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+	count, err := models.SharedNodeGrantDAO.CountAllEnabledGrants()
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CountAllEnabledNodeGrantsResponse{Count: count}, nil
+}
+
+func (this *NodeGrantService) ListEnabledNodeGrants(ctx context.Context, req *pb.ListEnabledNodeGrantsRequest) (*pb.ListEnabledNodeGrantsResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+	grants, err := models.SharedNodeGrantDAO.ListEnabledGrants(req.Offset, req.Size)
+	if err != nil {
+		return nil, err
+	}
+	result := []*pb.NodeGrant{}
+	for _, grant := range grants {
+		result = append(result, &pb.NodeGrant{
+			Id:          int64(grant.Id),
+			Name:        grant.Name,
+			Method:      grant.Method,
+			Password:    grant.Password,
+			Su:          grant.Su == 1,
+			PrivateKey:  grant.PrivateKey,
+			Description: grant.Description,
+			NodeId:      int64(grant.NodeId),
+		})
+	}
+
+	return &pb.ListEnabledNodeGrantsResponse{Grants: result}, nil
+}