package com.linghu.service;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.read.listener.ReadListener;
|
import com.linghu.model.entity.Platform;
|
import com.linghu.model.entity.Type;
|
import com.linghu.model.excel.PlatformExcel;
|
import com.linghu.service.TypeService;
|
import com.linghu.service.PlatformService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
import java.time.LocalDateTime;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
@Service
|
public class PlatformExcelService {
|
|
@Autowired
|
private PlatformService platformService;
|
|
@Autowired
|
private TypeService typeService;
|
|
/**
|
* 下载平台导入模板
|
*/
|
public void downloadTemplate(HttpServletResponse response) throws IOException {
|
response.setContentType("application/vnd.ms-excel");
|
response.setCharacterEncoding("utf-8");
|
String fileName = URLEncoder.encode("平台导入模板", "UTF-8");
|
response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
|
EasyExcel.write(response.getOutputStream(), PlatformExcel.class).sheet("平台信息").doWrite(new ArrayList<>());
|
}
|
|
/**
|
* 导入平台信息
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void importPlatform(MultipartFile file) throws IOException {
|
EasyExcel.read(file.getInputStream(), PlatformExcel.class, new ReadListener<PlatformExcel>() {
|
private List<Platform> platforms = new ArrayList<>();
|
|
@Override
|
public void invoke(PlatformExcel data, AnalysisContext context) {
|
// 根据类型名称获取类型ID
|
Type type = typeService.getTypeByName(data.getType_name());
|
if (type == null) {
|
throw new RuntimeException("类型名称不存在：" + data.getType_name());
|
}
|
|
Platform platform = new Platform();
|
platform.setPlatform_name(data.getPlatform_name());
|
platform.setDomain(data.getDomain());
|
platform.setType_id(type.getType_id());
|
platform.setCreate_time(LocalDateTime.now());
|
platforms.add(platform);
|
|
// 每100条保存一次
|
if (platforms.size() >= 100) {
|
saveData();
|
platforms.clear();
|
}
|
}
|
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext context) {
|
saveData();
|
}
|
|
private void saveData() {
|
if (!platforms.isEmpty()) {
|
platformService.saveBatch(platforms);
|
}
|
}
|
}).sheet().doRead();
|
}
|
}
|
