费用审核添加保险公司搜索

src/main/java/com/ydtech/modules/protocols/entity/vo/AgreementProductQueryVo.java

@@ -8,6 +8,9 @@ import lombok.Data;
 import java.time.LocalDate;
 @Data
 public class AgreementProductQueryVo extends PageRequest {
+    @ApiModelProperty(value = "保险公司id")
+    private String companyId;
+
     @ApiModelProperty(value = "协议id")
     private String agreementId;
 

src/main/java/com/ydtech/modules/protocols/service/impl/PtlAgreementProductCostsNewServiceImpl.java

@@ -1091,6 +1091,13 @@ public class PtlAgreementProductCostsNewServiceImpl extends ServiceImpl<PtlAgree
     public Page<PtlAgreementProductCostsNew> getCostsAuditList(Page page, AgreementProductQueryVo agreementProductQueryVo){
         LambdaQueryWrapper<PtlAgreementProductCostsNew> ptlAgreementProductCostsNewLambdaQueryWrapper = new LambdaQueryWrapper<>();
 //        ptlAgreementProductCostsNewLambdaQueryWrapper.eq(PtlAgreementProductCostsNew::getAgreementId, agreementProductQueryVo.getAgreementId());
+        //保险公司id
+        if(agreementProductQueryVo.getCompanyId() != null){
+            LambdaQueryWrapper<PtlAgreement> eq = new LambdaQueryWrapper<PtlAgreement>().eq(PtlAgreement::getPartnerCompaniesId, agreementProductQueryVo.getCompanyId());
+            List<PtlAgreement> list = ptlAgreementService.list(eq);
+            List<String> agreementIds = list.stream().map(PtlAgreement::getId).collect(Collectors.toList());
+            ptlAgreementProductCostsNewLambdaQueryWrapper.in(PtlAgreementProductCostsNew::getAgreementId, agreementIds);
+        }
 
         //协议id
         if(agreementProductQueryVo.getAgreementId() != null){