[optimize] fixed gen all report data

atms-api/src/main/java/pwc/taxtech/atms/common/util/MyAsserts.java

@@ -41,4 +41,12 @@ public class MyAsserts {
     public static void assertEmpty(Collection obj, ApiException exception) {
         if (obj != null && !obj.isEmpty()) throw exception;
     }
+
+    public static <T,S> void assertEq(T t,S s,ApiException exception){
+        if(s != t) throw exception;
+    }
+
+    public static <T,S> void assertNotEq(T t,S s,ApiException exception){
+        if(s == t) throw exception;
+    }
 }

atms-api/src/main/java/pwc/taxtech/atms/vat/service/impl/ReportServiceImpl.java

@@ -10,6 +10,7 @@ import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import pwc.taxtech.atms.common.CommonUtils;
+import pwc.taxtech.atms.common.util.MyAsserts;
 import pwc.taxtech.atms.common.util.SpringContextUtil;
 import pwc.taxtech.atms.constant.Constant;
 import pwc.taxtech.atms.constant.enums.*;
@@ -18,6 +19,7 @@ import pwc.taxtech.atms.dpo.ReportDto;
 import pwc.taxtech.atms.dto.OperationResultDto;
 import pwc.taxtech.atms.dto.vatdto.*;
 import pwc.taxtech.atms.entity.*;
+import pwc.taxtech.atms.exception.NotFoundException;
 import pwc.taxtech.atms.exception.NotSupportedException;
 import pwc.taxtech.atms.service.impl.CellConfigTranslater;
 import pwc.taxtech.atms.service.impl.DistributedIdService;
@@ -292,7 +294,7 @@ public class ReportServiceImpl {
         CellTemplateConfigExample cellTemplateConfigExample = new CellTemplateConfigExample();
         cellTemplateConfigExample.createCriteria().andReportTemplateIdEqualTo(templateId);
         List<CellTemplateConfig> cellTemplateConfigList = cellTemplateConfigMapper.selectByExample(cellTemplateConfigExample);
-        if(cellTemplateConfigList.isEmpty())return;
+        if (cellTemplateConfigList.isEmpty()) return;
 
         List<PeriodCellTemplateConfig> periodCellTemplateConfigList = new ArrayList<>();
 
@@ -465,22 +467,29 @@ public class ReportServiceImpl {
                 return operationResultDto;
             }
 
-            if (serviceType != EnumServiceType.VAT) {
-                // 暂不支持非CIT/VAT service批量后端生成报表
-                throw new NotSupportedException();
-            }
+            MyAsserts.assertEq(serviceType, EnumServiceType.VAT, new NotFoundException());
+            List<Long> templateIds = queryTemplates(projectId, periodParam, reportType, serviceType.getCode().toString());
 
-            int period = periodParam;
-            String serviceTypeStr = serviceType.getCode().toString();
+            String rslt = reportGenerator.generateData(projectId, templateIds, ifDeleteManualDataSource, null, periodParam, generator);
+            if (StringUtils.isBlank(rslt)) {
+                operationResultDto.setResultMsg("ReportGenerateFailed!");
+                return operationResultDto;
+            }
+            operationResultDto.setResult(true);
+        } catch (Exception ex) {
+            operationResultDto.setResult(false);
+            logger.error(ex.getMessage(), ex);
+        }
+        return operationResultDto;
+    }
 
+    private List<Long> queryTemplates(String projectId, Integer period, Integer reportType, String serviceTypeStr) {
         ProjectServiceTypeExample projectServiceTypeExample = new ProjectServiceTypeExample();
         projectServiceTypeExample.createCriteria().andServiceTypeIdEqualTo(serviceTypeStr).andProjectIdEqualTo(projectId);
         Optional<Long> templateGroupId = projectServiceTypeMapper.selectByExample(projectServiceTypeExample).stream()
                 .map(ProjectServiceType::getTemplateGroupId).findFirst();
-            if (templateGroupId == null) {
-                operationResultDto.setResultMsg("TemplateGroupNotExist");
-                return operationResultDto;
-            }
+
+        MyAsserts.assertNotNull(templateGroupId, new NotFoundException("not found group"));
 
         List<Long> templateIds;
         if (reportType != null) {
@@ -500,17 +509,7 @@ public class ReportServiceImpl {
             templateIds = periodTemplateMapper.selectByExample(periodTemplateExample).stream().map(PeriodTemplate::getId).collect(Collectors.toList());
         }
 
-            String rslt = reportGenerator.generateData(projectId, templateIds, ifDeleteManualDataSource, null, periodParam, generator);
-            if (StringUtils.isBlank(rslt)) {
-                operationResultDto.setResultMsg("ReportGenerateFailed!");
-                return operationResultDto;
-            }
-            operationResultDto.setResult(true);
-        } catch (Exception ex) {
-            operationResultDto.setResult(false);
-            logger.error(ex.getMessage(), ex);
-        }
-        return operationResultDto;
+        return templateIds;
     }
 
     public List<CellTemplateReferenceDto> getTemplateReferences(int period) {

atms-api/src/main/java/pwc/taxtech/atms/vat/service/impl/report/functions/FormulaContext.java

@@ -2,6 +2,7 @@ package pwc.taxtech.atms.vat.service.impl.report.functions;
 
 import lombok.Getter;
 import lombok.Setter;
+import pwc.taxtech.atms.entity.Project;
 import pwc.taxtech.atms.vat.service.impl.FormulaAgent;
 
 @Setter
@@ -36,4 +37,22 @@ public class FormulaContext {
     private FormulaAgent formulaAgent;
 
     //private Map<String, FormulaResult> replaceSpecialCellFormulaDic;
+
+    public static FormulaContext extractContextFromProject(Project project){
+        FormulaContext formulaContext = new FormulaContext();
+        formulaContext.setProjectId(project.getId());
+        formulaContext.setYear(project.getYear());
+        formulaContext.setIfRound(true);
+        formulaContext.setOrganizationId(project.getOrganizationId());
+        formulaContext.setIfRound(true);
+        return formulaContext;
+    }
+
+    public FormulaContext fixedFormula(Integer period,Long reportTemplateGroupId,FormulaAgent agent){
+        this.period=period;
+        this.formulaAgent=agent;
+        this.reportTemplateGroupId=reportTemplateGroupId;
+        this.isYear=(period==0);
+        return this;
+    }
 }