update 随机数 可以选择导出

jero-boot-incoming-payment/src/main/java/com/jero/meeting/controller/PayRandomCodeController.java

@@ -1,16 +1,20 @@
 package com.jero.meeting.controller;
 
+import com.jero.common.util.oConvertUtils;
 import com.jero.meeting.entity.PayRandomCode;
 import com.jero.meeting.entity.PayRandomCodeLink;
 import com.jero.meeting.service.PayRandomCodeLinkService;
 import com.jero.meeting.service.PayRandomCodeService;
 
+import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
+import java.util.stream.Collectors;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
 import com.jero.meeting.vo.GenerateDTO;
+import org.apache.poi.ss.formula.functions.T;
 import org.apache.shiro.authz.annotation.RequiresPermissions;
 import org.springframework.beans.BeanUtils;
 import org.springframework.web.bind.annotation.*;
@@ -127,12 +131,22 @@ public class PayRandomCodeController  extends JeroController<PayRandomCode, PayR
     public ModelAndView exportXls(HttpServletRequest request, PayRandomCode randomCode) {
 		QueryWrapper<PayRandomCode> queryWrapper = QueryGenerator.initQueryWrapper(randomCode, request.getParameterMap());
 		List<PayRandomCode> list = payRandomCodeService.list(queryWrapper);
-		list.forEach(item -> {
+		// 过滤选中数据
+		List<PayRandomCode> exportList;
+		String selections = request.getParameter("selections");
+		if (oConvertUtils.isNotEmpty(selections)) {
+			List<String> selectionList = Arrays.asList(selections.split(","));
+			exportList = list.stream().filter(item -> selectionList.contains(item.getRandomCodeId())).collect(Collectors.toList());
+		} else {
+			exportList = list;
+		}
+		//处理数据
+		exportList.forEach(item -> {
 			String randomCodeId = item.getRandomCodeId();
 			List<PayRandomCodeLink> randomCodeLinkList = payRandomCodeLinkService.selectByRandomCodeId(randomCodeId);
 			item.setRandomCodeLinkList(randomCodeLinkList);
 		});
-        return super.exportListXls(list, PayRandomCode.class, "随机码表");
+        return super.exportListXls(exportList, PayRandomCode.class, "随机码表");
     }
 
 }