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.query.ExecuteTemplateQuery;
import com.gmh.entity.vo.ReadNameVo;
import com.gmh.service.GmhFileLogService;
import com.gmh.service.FileService;
import com.gmh.service.GmhFileLogService;
import com.gmh.service.SysBaijiaxingService;
import com.gmh.utils.StringUtils;
import com.gmh.utils.POIExcelUtil;
import com.gmh.utils.StringUtils;
import lombok.AllArgsConstructor;
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.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
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.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.*;
import java.util.stream.Collectors;
@ -54,13 +48,20 @@ public class ExcelController {
@Autowired
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 = "姓名";
@Value("${gmh.upload-path}")
private String uploadPath;
@RequestMapping("/t01")
public R test01() {
return R.ok("ok");
@ -68,6 +69,7 @@ public class ExcelController {
@RequestMapping("/dataSourceImport")
public R dataSourceImport(MultipartFile file, String dataSourceImportModel) throws IOException {
excelInfoMap.put(request.getSession().getId(), new ExcelInfo());
DataSourceImportModel importModel = DataSourceImportModel.findEnum(dataSourceImportModel);
Workbook workbook = POIExcelUtil.readExcelFromInputStream(file.getInputStream(), file.getOriginalFilename());
FileEntity fileEntity = fileService.uploadFile(file);
@ -113,6 +115,7 @@ public class ExcelController {
readNameVo.setSheetName(sheet.getSheetName());
readNameVoList.add(readNameVo);
}
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
excelInfo.setReadNameVoList(readNameVoList);
return R.ok().setData(readNameVoList);
}
@ -155,6 +158,7 @@ public class ExcelController {
Workbook workbook = POIExcelUtil.readExcelFromInputStream(file.getInputStream(), file.getOriginalFilename());
FileEntity fileEntity = fileService.uploadFile(file);
gmhFileLogService.recordLog(fileEntity, GmhFileLog.OperateType.TEMPL_IMP);
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
excelInfo.setTemplateOriginalFilename(file.getOriginalFilename());
excelInfo.setTemplateSheetNum(Integer.parseInt(templateSheetNum) - 1);
Sheet sheet = workbook.getSheetAt(excelInfo.getTemplateSheetNum());
@ -167,6 +171,7 @@ public class ExcelController {
@RequestMapping("/cellNumSearch")
public R analysisCell(@Validated RowCellNum rowCellNum) {
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
if (excelInfo.getTemplateWorkbook() == null) {
return R.error("模板文件出问题了，请重新上传");
}
@ -190,6 +195,7 @@ public class ExcelController {
@PostMapping("/executeTemplate")
public R executeTemplate(@RequestBody ExecuteTemplateQuery executeTemplateQuery) throws IOException {
// TODO 有表达式和无表达式兼容处理
String newCellData = executeTemplateQuery.getNewCellData();
String expr = newCellData.substring(newCellData.indexOf("@{"), newCellData.indexOf("}") + 1);
String exprVal = newCellData.substring(newCellData.indexOf("@{") + 2, newCellData.indexOf("}"));
@ -197,6 +203,7 @@ public class ExcelController {
String sheetName = exprArr[0];
String field = exprArr[1];
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
Optional<ReadNameVo> first = excelInfo.getReadNameVoList()
.stream()
.filter(item -> item.getSheetName().equalsIgnoreCase(sheetName))
@ -269,10 +276,13 @@ public class ExcelController {
@GetMapping("/gotoDownload")
public R gotoDownload() {
ExcelInfo excelInfo = excelInfoMap.get(request.getSession().getId());
try {
excelInfo.getTemplateWorkbook().close();
} catch (IOException e) {
throw new RuntimeException(e);
} finally {
excelInfoMap.remove(request.getSession().getId());
}
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.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**
@ -23,27 +24,47 @@ import java.util.List;
public class test {
public static void main(String[] args) throws IOException {
File file = new File("C:\\Users\\xxche\\Desktop\\继续教育证明\\盛华编制聘任制.xlsx");
Workbook workbook = POIExcelUtil.readExcelFromFile(file);
// 编制
Sheet bianzhi = workbook.getSheetAt(0);
List<String> bianzhiList = new ArrayList<>();
for (int i = 0; i <= bianzhi.getLastRowNum(); i++) {
Row row = bianzhi.getRow(i);
if (row == null) {
continue;
}
Cell cell = row.getCell(1);
String name = cell.getStringCellValue();
if ("姓名".equals(name)) {
continue;
}
if (StrUtil.isBlankIfStr(name)) {
continue;
}
bianzhiList.add(nameFormat(name));
}
// File file = new File("C:\\Users\\xxche\\Desktop\\继续教育证明\\盛华编制聘任制.xlsx");
// Workbook workbook = POIExcelUtil.readExcelFromFile(file);
// // 编制
// Sheet bianzhi = workbook.getSheetAt(0);
//
// List<String> bianzhiList = new ArrayList<>();
// for (int i = 0; i <= bianzhi.getLastRowNum(); i++) {
// Row row = bianzhi.getRow(i);
// if (row == null) {
// continue;
// }
// Cell cell = row.getCell(1);
// String name = cell.getStringCellValue();
// if ("姓名".equals(name)) {
// continue;
// }
// if (StrUtil.isBlankIfStr(name)) {
// continue;
// }
// 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());
for (int i = 0; i < bianzhiWb.getNumberOfSheets(); i++) {
@ -60,7 +81,7 @@ public class test {
}
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();
bianzhiWb.write(out);
out.close();
@ -71,23 +92,51 @@ public class test {
// 聘任制
Sheet pinren = workbook.getSheetAt(1);
List<String> pinrenList = new ArrayList<>();
for (int i = 0; i <= pinren.getLastRowNum(); i++) {
Row row = pinren.getRow(i);
if (row == null) {
continue;
}
Cell cell = row.getCell(1);
String name = cell.getStringCellValue();
if ("姓名".equals(name)) {
continue;
}
if (StrUtil.isBlankIfStr(name)) {
continue;
}
pinrenList.add(name);
}
// Sheet pinren = workbook.getSheetAt(1);
// List<String> pinrenList = new ArrayList<>();
// for (int i = 0; i <= pinren.getLastRowNum(); i++) {
// Row row = pinren.getRow(i);
// if (row == null) {
// continue;
// }
// Cell cell = row.getCell(1);
// String name = cell.getStringCellValue();
// if ("姓名".equals(name)) {
// continue;
// }
// if (StrUtil.isBlankIfStr(name)) {
// continue;
// }
// 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());
for (int i = 0; i < pinrenWb.getNumberOfSheets(); i++) {
Sheet sheet = pinrenWb.getSheetAt(i);
@ -103,7 +152,7 @@ public class test {
}
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();
pinrenWb.write(out);
out.close();

Write Preview
Loading…
Cancel
Save