Forráskód Böngészése

导出功能修改

user5 2 éve
szülő
commit
1d321e4f59

+ 4 - 0
jeeplus-module/jeeplus-test/src/main/java/com/jeeplus/test/projectRecords/controller/ProjectController.java

@@ -469,6 +469,10 @@ public class ProjectController {
         List <ProjectDTO> userProjectDTOList = new ArrayList<>();
         if ( ExportMode.current.equals ( options.getMode ( ) ) ) {
             result = projectService.selectPage ( userProjectPage, queryWrapper ).getRecords ( );
+        } else if ( ExportMode.selected.equals ( options.getMode ( ) ) ) {
+            result = projectService.selectPage ( userProjectPage, queryWrapper ).getRecords().stream ( ).filter ( reimbursement ->
+                    options.getSelectIds ( ).contains ( reimbursement.getId ( ) )
+            ).collect ( Collectors.toList ( ) );
         } else {
             userProjectPage.setSize ( -1 );
             userProjectPage.setCurrent ( 0 );