阶段性提交

internal/rpc/services/service_node_grant.go

@@ -87,3 +87,56 @@ func (this *NodeGrantService) ListEnabledNodeGrants(ctx context.Context, req *pb
 
 	return &pb.ListEnabledNodeGrantsResponse{Grants: result}, nil
 }
+
+// 列出所有认证信息
+func (this *NodeGrantService) FindAllEnabledNodeGrants(ctx context.Context, req *pb.FindAllEnabledNodeGrantsRequest) (*pb.FindAllEnabledNodeGrantsResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+	grants, err := models.SharedNodeGrantDAO.FindAllEnabledGrants()
+	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.FindAllEnabledNodeGrantsResponse{Grants: result}, nil
+}
+
+func (this *NodeGrantService) FindEnabledGrant(ctx context.Context, req *pb.FindEnabledGrantRequest) (*pb.FindEnabledGrantResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	grant, err := models.SharedNodeGrantDAO.FindEnabledNodeGrant(req.GrantId)
+	if err != nil {
+		return nil, err
+	}
+	if grant == nil {
+		return &pb.FindEnabledGrantResponse{}, nil
+	}
+	return &pb.FindEnabledGrantResponse{Grant: &pb.NodeGrant{
+		Id:          int64(grant.Id),
+		Name:        grant.Name,
+		Method:      grant.Method,
+		Username:    grant.Username,
+		Password:    grant.Password,
+		Su:          grant.Su == 1,
+		PrivateKey:  grant.PrivateKey,
+		Description: grant.Description,
+		NodeId:      int64(grant.NodeId),
+	}}, nil
+}