1
package com.example.system.config;
2

3
import com.example.system.api.domain.ExportUser;
4
import lombok.extern.slf4j.Slf4j;
5
import org.springframework.stereotype.Component;
6

7
import java.util.LinkedList;
8

9
@Slf4j
10
@Component
11
public class ExportQueue {
12

13
 private final int MAX_CAPACITY = 10; // 队列最大容量
14
 private LinkedList<ExportUser> queue; // 用户队列
15

16
 public ExportQueue(LinkedList<ExportUser> queue) {
17
 this.queue = new LinkedList<>();
18
 }
19

20
 /**
21
 * 排队队列添加
22
 * @param sysUser
23
 */
24
 public synchronized LinkedList<ExportUser> add(ExportUser sysUser) {
25
 while (queue.size() >= MAX_CAPACITY) {
26
 try {
27
 log.info("当前排队人已满，请等待");
28
 wait();
29
 } catch (InterruptedException e) {
30
 e.getMessage();
31
 }
32
 }
33
 queue.add(sysUser);
34
 log.info("目前导出队列排队人数：" + queue.size());
35
 notifyAll();
36
 return queue;
37
 }
38

39
 /**
40
 * 获取排队队列下一个人
41
 * @return
42
 */
43
 public synchronized ExportUser getNextSysUser() {
44
 while (queue.isEmpty()) {
45
 try {
46
 wait();
47
 } catch (InterruptedException e) {
48
 e.printStackTrace();
49
 }
50
 }
51
 ExportUser sysUser = queue.remove();
52
 notifyAll(); //唤醒
53
 return sysUser;
54
 }
55
}

1
package com.example.system.config;
2

3
import cn.hutool.core.bean.BeanUtil;
4
import cn.hutool.core.util.PageUtil;
5
import com.alibaba.excel.EasyExcel;
6
import com.alibaba.excel.ExcelWriter;
7
import com.alibaba.excel.write.metadata.WriteSheet;
8
import com.example.system.api.domain.ExportUser;
9
import lombok.extern.slf4j.Slf4j;
10

11
import javax.servlet.http.HttpServletResponse;
12
import java.io.IOException;
13
import java.net.URLEncoder;
14
import java.util.List;
15

16
@Slf4j
17
public abstract class AbstractExport<T, K> {
18

19
 public abstract void export(ExportUser sysUser) throws InterruptedException;
20

21
 /**
22
 * 导出
23
 *
24
 * @param response 输出流
25
 * @param pageSize 每页大小
26
 * @param t        导出条件
27
 * @param k        Excel内容实体类
28
 * @param fileName 文件名称
29
 */
30
 public void export(HttpServletResponse response, int pageSize, T t, Class<K> k, String fileName) throws Exception {
31
 ExcelWriter writer = null;
32
 try {
33
 writer = getExcelWriter(response, fileName);
34
 //查询导出总条数
35
 int total = this.countExport(t);
36
 //页数
37
 int loopCount = PageUtil.totalPage(total, pageSize);
38
 BeanUtil.setProperty(t, "pageSize", pageSize);
39
 for (int i = 0; i < loopCount; i++) {
40
 //开始页
41
 BeanUtil.setProperty(t, "pageNum", PageUtil.getStart(i + 1, pageSize));
42
 //获取Excel导出信息
43
 List<K> kList = this.getExportDetail(t);
44
 WriteSheet writeSheet = EasyExcel.writerSheet(fileName).head(k).build();
45
 writer.write(kList, writeSheet);
46
 }
47
 } catch (Exception e) {
48
 String msg = "导出" + fileName + "异常";
49
 log.error(msg, e);
50
 throw new Exception(msg + e);
51
 } finally {
52
 if (writer != null) {
53
 writer.finish();
54
 }
55
 }
56
 }
57

58
 public com.alibaba.excel.ExcelWriter getExcelWriter(HttpServletResponse response, String fileName) throws IOException {
59
 response.setContentType("application/vnd.ms-excel");
60
 response.setCharacterEncoding("utf-8");
61
 // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
62
 String fileNameUtf = URLEncoder.encode(fileName, "UTF-8").replaceAll("\\\\+", "%20");
63
 response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileNameUtf + ".xlsx");
64
 return EasyExcel.write(response.getOutputStream()).build();
65
 }
66

67
 /**
68
 * （模版导出）
69
 *
70
 * @param t
71
 * @param fileName
72
 * @param response
73
 */
74
 public abstract void complexFillWithTable(T t, String fileName, HttpServletResponse response);
75

76
 /**
77
 * 查询导出总条数
78
 *
79
 * @param t
80
 * @return
81
 */
82
 public abstract int countExport(T t);
83

84
 /**
85
 * 查询导出数据
86
 *
87
 * @param t
88
 * @return
89
 */
90
 public abstract List<K> getExportDetail(T t);
91
}

1
package com.example.system.service.impl;
2

3
import com.alibaba.excel.ExcelWriter;
4
import com.example.system.api.domain.ExportUser;
5
import com.example.system.config.AbstractExport;
6
import com.example.system.config.ExportQueue;
7
import lombok.extern.slf4j.Slf4j;
8
import org.springframework.beans.factory.annotation.Autowired;
9
import org.springframework.stereotype.Service;
10

11
import javax.servlet.http.HttpServletResponse;
12
import java.io.IOException;
13
import java.util.LinkedList;
14
import java.util.List;
15

16
@Service
17
@Slf4j
18
public class ExportImpl extends AbstractExport {
19

20
 @Autowired
21
 private ExportQueue exportQueue;
22

23
 @Override
24
 public void export(ExportUser sysUser) throws InterruptedException {
25

26
 //导出
27
 log.info("导出文件方法执行～～～～～～～～～");
28
//        export(response,pageSize,t,k,fileName);
29
 LinkedList<ExportUser> queue = exportQueue.add(sysUser);
30
 log.info("导出队列：" + queue);
31
 //休眠时间稍微设置大点，模拟导出处理时间
32
 Thread.sleep(20000);
33
 //导出成功后移除当前导出用户
34
 ExportUser nextSysUser = exportQueue.getNextSysUser();
35
 log.info("移除后获取下一个排队的用户: " + nextSysUser.getUserName());
36

37
 }
38

39
 @Override
40
 public void export(HttpServletResponse response, int pageSize, Object o, Class k, String fileName) throws Exception {
41
 super.export(response, pageSize, o, k, fileName);
42
 }
43

44
 @Override
45
 public ExcelWriter getExcelWriter(HttpServletResponse response, String fileName) throws IOException {
46
 return super.getExcelWriter(response, fileName);
47
 }
48

49
 @Override
50
 public void complexFillWithTable(Object o, String fileName, HttpServletResponse response) {
51

52
 }
53

54
 @Override
55
 public int countExport(Object o) {
56
 return 0;
57
 }
58

59
 @Override
60
 public List getExportDetail(Object o) {
61
 return null;
62
 }
63
}

1
package com.example.system.controller;
2

3
import com.example.system.api.domain.ExportUser;
4
import com.example.system.api.domain.SysUser;
5
import com.example.system.service.impl.ExportImpl;
6
import lombok.SneakyThrows;
7
import lombok.extern.slf4j.Slf4j;
8
import org.springframework.beans.factory.annotation.Autowired;
9
import org.springframework.web.bind.annotation.PostMapping;
10
import org.springframework.web.bind.annotation.RequestBody;
11
import org.springframework.web.bind.annotation.RequestMapping;
12
import org.springframework.web.bind.annotation.RestController;
13

14
@RestController
15
@RequestMapping("/export")
16
@Slf4j
17
public class ExportController {
18

19
 @Autowired
20
 private ExportImpl export;
21

22
 @PostMapping("/exportFile")
23
 public void exportFile() {
24
 new Thread(new Runnable() {
25
 @SneakyThrows
26
 @Override
27
 public void run() {
28
 Thread thread1 = Thread.currentThread();
29
 ExportUser sysUser =new ExportUser();
30
 sysUser.setUserName(thread1.getName());
31

32
 export.export(sysUser);
33
 }
34
 }).start();
35
 }
36
}