chenda
/
gmh
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

解决多线程并发问题

Browse Source
master
chenda 2 years ago
parent e796289f97
commit e428e56fc8
2 changed files with 112 additions and 53 deletions
Show Stats Download Patch File Download Diff File

36
src/main/java/com/gmh/controller/ExcelController.java
Unescape View File

@ -8,33 +8,27 @@ import com.gmh.entity.*;
import com.gmh.entity.properties.FileServerProperties; import com.gmh.entity.properties.FileServerProperties;
import com.gmh.entity.query.ExecuteTemplateQuery; import com.gmh.entity.query.ExecuteTemplateQuery;
import com.gmh.entity.vo.ReadNameVo; import com.gmh.entity.vo.ReadNameVo;
import com.gmh.service.GmhFileLogService;
import com.gmh.service.FileService; import com.gmh.service.FileService;
import com.gmh.service.GmhFileLogService;
import com.gmh.service.SysBaijiaxingService; import com.gmh.service.SysBaijiaxingService;
import com.gmh.utils.StringUtils;
import com.gmh.utils.POIExcelUtil; import com.gmh.utils.POIExcelUtil;
import com.gmh.utils.StringUtils;
import lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
import lombok.Getter; import lombok.Getter;
import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.io.IOUtils;
import org.apache.poi.ss.usermodel.Cell; import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook; import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType; import org.springframework.http.MediaType;
import org.springframework.validation.annotation.Validated; import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartFile;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletRequest;
import java.io.File; import java.io.File;
import java.io.FileOutputStream; import java.io.FileOutputStream;
import java.io.IOException; import java.io.IOException;
import java.io.OutputStream;
import java.util.*; import java.util.*;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -54,13 +48,20 @@ public class ExcelController {
@Autowired @Autowired
private SysBaijiaxingService baijiaxingService; private SysBaijiaxingService baijiaxingService;
private final ExcelInfo excelInfo = new ExcelInfo(); @Autowired
private HttpServletRequest request;
/**
* HashMap的Key为SessionId
*
* 通过SessionId为每一个会话保存自己的对象副本
*
* 不适用于集群环境
*/
private final Map<String, ExcelInfo> excelInfoMap = new HashMap<>();
private static final String KEY_XINGMING = "姓名"; private static final String KEY_XINGMING = "姓名";
@Value("${gmh.upload-path}")
private String uploadPath;
@RequestMapping("/t01") @RequestMapping("/t01")
public R test01() { public R test01() {
return R.ok("ok"); return R.ok("ok");
@ -68,6 +69,7 @@ public class ExcelController {
@RequestMapping("/dataSourceImport") @RequestMapping("/dataSourceImport")
public R dataSourceImport(MultipartFile file, String dataSourceImportModel) throws IOException { public R dataSourceImport(MultipartFile file, String dataSourceImportModel) throws IOException {
excelInfoMap.put(request.getSession().getId(), new ExcelInfo());
DataSourceImportModel importModel = DataSourceImportModel.findEnum(dataSourceImportModel); DataSourceImportModel importModel = DataSourceImportModel.findEnum(dataSourceImportModel);
Workbook workbook = POIExcelUtil.readExcelFromInputStream(file.getInputStream(), file.getOriginalFilename()); Workbook workbook = POIExcelUtil.readExcelFromInputStream(file.getInputStream(), file.getOriginalFilename());
FileEntity fileEntity = fileService.uploadFile(file); FileEntity fileEntity = fileService.uploadFile(file);
@ -113,6 +115,7 @@ public class ExcelController {
readNameVo.setSheetName(sheet.getSheetName()); readNameVo.setSheetName(sheet.getSheetName());
readNameVoList.add(readNameVo); readNameVoList.add(readNameVo);
} }
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
excelInfo.setReadNameVoList(readNameVoList); excelInfo.setReadNameVoList(readNameVoList);
return R.ok().setData(readNameVoList); return R.ok().setData(readNameVoList);
} }
@ -155,6 +158,7 @@ public class ExcelController {
Workbook workbook = POIExcelUtil.readExcelFromInputStream(file.getInputStream(), file.getOriginalFilename()); Workbook workbook = POIExcelUtil.readExcelFromInputStream(file.getInputStream(), file.getOriginalFilename());
FileEntity fileEntity = fileService.uploadFile(file); FileEntity fileEntity = fileService.uploadFile(file);
gmhFileLogService.recordLog(fileEntity, GmhFileLog.OperateType.TEMPL_IMP); gmhFileLogService.recordLog(fileEntity, GmhFileLog.OperateType.TEMPL_IMP);
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
excelInfo.setTemplateOriginalFilename(file.getOriginalFilename()); excelInfo.setTemplateOriginalFilename(file.getOriginalFilename());
excelInfo.setTemplateSheetNum(Integer.parseInt(templateSheetNum) - 1); excelInfo.setTemplateSheetNum(Integer.parseInt(templateSheetNum) - 1);
Sheet sheet = workbook.getSheetAt(excelInfo.getTemplateSheetNum()); Sheet sheet = workbook.getSheetAt(excelInfo.getTemplateSheetNum());
@ -167,6 +171,7 @@ public class ExcelController {
@RequestMapping("/cellNumSearch") @RequestMapping("/cellNumSearch")
public R analysisCell(@Validated RowCellNum rowCellNum) { public R analysisCell(@Validated RowCellNum rowCellNum) {
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
if (excelInfo.getTemplateWorkbook() == null) { if (excelInfo.getTemplateWorkbook() == null) {
return R.error("模板文件出问题了，请重新上传"); return R.error("模板文件出问题了，请重新上传");
} }
@ -190,6 +195,7 @@ public class ExcelController {
@PostMapping("/executeTemplate") @PostMapping("/executeTemplate")
public R executeTemplate(@RequestBody ExecuteTemplateQuery executeTemplateQuery) throws IOException { public R executeTemplate(@RequestBody ExecuteTemplateQuery executeTemplateQuery) throws IOException {
// TODO 有表达式和无表达式兼容处理
String newCellData = executeTemplateQuery.getNewCellData(); String newCellData = executeTemplateQuery.getNewCellData();
String expr = newCellData.substring(newCellData.indexOf("@{"), newCellData.indexOf("}") + 1); String expr = newCellData.substring(newCellData.indexOf("@{"), newCellData.indexOf("}") + 1);
String exprVal = newCellData.substring(newCellData.indexOf("@{") + 2, newCellData.indexOf("}")); String exprVal = newCellData.substring(newCellData.indexOf("@{") + 2, newCellData.indexOf("}"));
@ -197,6 +203,7 @@ public class ExcelController {
String sheetName = exprArr[0]; String sheetName = exprArr[0];
String field = exprArr[1]; String field = exprArr[1];
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
Optional<ReadNameVo> first = excelInfo.getReadNameVoList() Optional<ReadNameVo> first = excelInfo.getReadNameVoList()
.stream() .stream()
.filter(item -> item.getSheetName().equalsIgnoreCase(sheetName)) .filter(item -> item.getSheetName().equalsIgnoreCase(sheetName))
@ -269,10 +276,13 @@ public class ExcelController {
@GetMapping("/gotoDownload") @GetMapping("/gotoDownload")
public R gotoDownload() { public R gotoDownload() {
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
try { try {
excelInfo.getTemplateWorkbook().close(); excelInfo.getTemplateWorkbook().close();
} catch (IOException e) { } catch (IOException e) {
throw new RuntimeException(e); throw new RuntimeException(e);
} finally {
excelInfoMap.remove(request.getSession().getId());
} }
return R.ok(); return R.ok();
} }

129
src/main/java/com/gmh/test.java
Unescape View File

@ -11,6 +11,7 @@ import java.io.File;
import java.io.FileOutputStream; import java.io.FileOutputStream;
import java.io.IOException; import java.io.IOException;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Arrays;
import java.util.List; import java.util.List;
/** /**
@ -23,27 +24,47 @@ import java.util.List;
public class test { public class test {
public static void main(String[] args) throws IOException { public static void main(String[] args) throws IOException {
File file = new File("C:\\Users\\xxche\\Desktop\\继续教育证明\\盛华编制聘任制.xlsx"); // File file = new File("C:\\Users\\xxche\\Desktop\\继续教育证明\\盛华编制聘任制.xlsx");
Workbook workbook = POIExcelUtil.readExcelFromFile(file); // Workbook workbook = POIExcelUtil.readExcelFromFile(file);
// 编制 // // 编制
Sheet bianzhi = workbook.getSheetAt(0); // Sheet bianzhi = workbook.getSheetAt(0);
//
List<String> bianzhiList = new ArrayList<>(); // List<String> bianzhiList = new ArrayList<>();
for (int i = 0; i <= bianzhi.getLastRowNum(); i++) { // for (int i = 0; i <= bianzhi.getLastRowNum(); i++) {
Row row = bianzhi.getRow(i); // Row row = bianzhi.getRow(i);
if (row == null) { // if (row == null) {
continue; // continue;
} // }
Cell cell = row.getCell(1); // Cell cell = row.getCell(1);
String name = cell.getStringCellValue(); // String name = cell.getStringCellValue();
if ("姓名".equals(name)) { // if ("姓名".equals(name)) {
continue; // continue;
} // }
if (StrUtil.isBlankIfStr(name)) { // if (StrUtil.isBlankIfStr(name)) {
continue; // continue;
} // }
bianzhiList.add(nameFormat(name)); // bianzhiList.add(nameFormat(name));
} // }
String bz = "牟春华\n" +
"陈文硕\n" +
"邱晓\n" +
"李英\n" +
"王贺\n" +
"邱萌\n" +
"李世祺\n" +
"王槐秀\n" +
"芦健平\n" +
"刘志佳\n" +
"陈东移\n" +
"谭俊英\n" +
"杨绍华\n" +
"孙洪艳\n" +
"张庆华\n" +
"白淼\n" +
"吴铭涛\n";
List<String> bianzhiList = Arrays.asList(bz.split("\n"));
Workbook bianzhiWb = POIExcelUtil.batchCloneSheet(new File("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明.xls"), 0, bianzhiList.size()); Workbook bianzhiWb = POIExcelUtil.batchCloneSheet(new File("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明.xls"), 0, bianzhiList.size());
for (int i = 0; i < bianzhiWb.getNumberOfSheets(); i++) { for (int i = 0; i < bianzhiWb.getNumberOfSheets(); i++) {
@ -60,7 +81,7 @@ public class test {
} }
try { try {
FileOutputStream out = new FileOutputStream("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明222.xls"); FileOutputStream out = new FileOutputStream("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明1112.xls");
out.flush(); out.flush();
bianzhiWb.write(out); bianzhiWb.write(out);
out.close(); out.close();
@ -71,23 +92,51 @@ public class test {
// 聘任制 // 聘任制
Sheet pinren = workbook.getSheetAt(1); // Sheet pinren = workbook.getSheetAt(1);
List<String> pinrenList = new ArrayList<>(); // List<String> pinrenList = new ArrayList<>();
for (int i = 0; i <= pinren.getLastRowNum(); i++) { // for (int i = 0; i <= pinren.getLastRowNum(); i++) {
Row row = pinren.getRow(i); // Row row = pinren.getRow(i);
if (row == null) { // if (row == null) {
continue; // continue;
} // }
Cell cell = row.getCell(1); // Cell cell = row.getCell(1);
String name = cell.getStringCellValue(); // String name = cell.getStringCellValue();
if ("姓名".equals(name)) { // if ("姓名".equals(name)) {
continue; // continue;
} // }
if (StrUtil.isBlankIfStr(name)) { // if (StrUtil.isBlankIfStr(name)) {
continue; // continue;
} // }
pinrenList.add(name); // pinrenList.add(name);
} // }
String prz = "梁雪\n" +
"娄万明\n" +
"刘俊苹\n" +
"刘洋洋\n" +
"武广瑞\n" +
"单春飞\n" +
"罗酩禹\n" +
"王旭\n" +
"王梓\n" +
"侯佳君\n" +
"管静\n" +
"林源\n" +
"庄稀月\n" +
"宋媛媛\n" +
"苏爽\n" +
"张萌\n" +
"张霖琳\n" +
"余金萍\n" +
"孙佳馨\n" +
"石子宁\n" +
"苏欣\n" +
"李仕伟\n" +
"刘畅\n" +
"李昱霖\n" +
"闫文凝\n";
List<String> pinrenList = Arrays.asList(prz.split("\n"));
Workbook pinrenWb = POIExcelUtil.batchCloneSheet(new File("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明(聘用制).xls"), 0, pinrenList.size()); Workbook pinrenWb = POIExcelUtil.batchCloneSheet(new File("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明(聘用制).xls"), 0, pinrenList.size());
for (int i = 0; i < pinrenWb.getNumberOfSheets(); i++) { for (int i = 0; i < pinrenWb.getNumberOfSheets(); i++) {
Sheet sheet = pinrenWb.getSheetAt(i); Sheet sheet = pinrenWb.getSheetAt(i);
@ -103,7 +152,7 @@ public class test {
} }
try { try {
FileOutputStream out = new FileOutputStream("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明(聘用制)222.xls"); FileOutputStream out = new FileOutputStream("C:\\Users\\xxche\\Desktop\\20221106_test_excel\\附件3.继续教育证明(聘用制)1112.xls");
out.flush(); out.flush();
pinrenWb.write(out); pinrenWb.write(out);
out.close(); out.close();

Write Preview
Loading…
Cancel
Save