1

kb/controller/contract_controller.go

@@ -107,14 +107,35 @@ func (c *contract) Detail(ctx context.Context, req *dto.GetContractDetailReq) (*
 		return nil, err
 	}
 	marks := make(map[int64][]*entity.ContractMark)
+	risks := make(map[int64][]*entity.ContractRisk)
 	for _, cl := range clauses {
 		list, err := dao.ContractMark.ListByClause(ctx, cl.Id)
 		if err != nil {
 			return nil, err
 		}
 		marks[cl.Id] = list
+		riskList, err := dao.ContractRisk.ListByClause(ctx, cl.Id)
+		if err != nil {
+			return nil, err
+		}
+		risks[cl.Id] = riskList
 	}
-	return &dto.GetContractDetailRes{Task: task, Clauses: clauses, Marks: marks}, nil
+	return &dto.GetContractDetailRes{Task: task, Clauses: clauses, Marks: marks, Risks: risks}, nil
+}
+
+func (c *contract) Summary(ctx context.Context, req *dto.SummaryContractReq) (*dto.SummaryContractRes, error) {
+	task, err := dao.ContractTask.GetOne(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+	if task == nil {
+		return nil, gerror.New("任务不存在")
+	}
+	sum, err := service.AnnotationService.Summary(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.SummaryContractRes{Task: task, RiskSummary: *sum}, nil
 }
 
 func (c *contract) Annotated(ctx context.Context, req *dto.AnnotatedContractReq) (*dto.AnnotatedContractRes, error) {