package com.example.service.impl;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.read.listener.ReadListener;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.example.entity.DataExcel;
|
import com.example.entity.Department;
|
import com.example.excel.DataExcelImport;
|
import com.example.mapper.DataExcelMapper;
|
import com.example.service.DataExcelService;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
@Service
|
public class DataExcelServiceImpl extends ServiceImpl<DataExcelMapper, DataExcel> implements DataExcelService {
|
|
@Override
|
public IPage<DataExcel> getDataExcelPage(Long current, Long size, String unitCode, String transactionNo, String accountingPeriod, List<String> permissionDeptCodes) {
|
Page<DataExcel> page = new Page<>(current, size);
|
page.setOptimizeJoinOfCountSql(false);
|
page.setOptimizeCountSql(false);
|
LambdaQueryWrapper<DataExcel> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(unitCode != null && !unitCode.isEmpty(), DataExcel::getUnitCode, unitCode)
|
.like(transactionNo != null && !transactionNo.isEmpty(), DataExcel::getTransactionNo, transactionNo)
|
.eq(accountingPeriod != null && !accountingPeriod.isEmpty(), DataExcel::getAccountingPeriod, accountingPeriod)
|
.in(permissionDeptCodes != null && !permissionDeptCodes.isEmpty(), DataExcel::getUnitCode, permissionDeptCodes)
|
.orderByDesc(DataExcel::getId);
|
return page(page, wrapper);
|
}
|
|
@Override
|
public IPage<DataExcel> getDataExcelPageNew(Long current, Long size, String unitCode, String transactionNo, String accountingPeriod) {
|
Page<DataExcel> page = new Page<>(current, size);
|
page.setOptimizeJoinOfCountSql(false);
|
page.setOptimizeCountSql(false);
|
LambdaQueryWrapper<DataExcel> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(unitCode != null && !unitCode.isEmpty(), DataExcel::getUnitCode, unitCode)
|
.like(transactionNo != null && !transactionNo.isEmpty(), DataExcel::getTransactionNo, transactionNo)
|
.eq(accountingPeriod != null && !accountingPeriod.isEmpty(), DataExcel::getAccountingPeriod, accountingPeriod)
|
.orderByDesc(DataExcel::getId);
|
return page(page, wrapper);
|
}
|
|
@Override
|
public List<Department> getDepartmentTree() {
|
QueryWrapper<DataExcel> wrapper = new QueryWrapper<>();
|
// 关键：两个字段去重（DISTINCT 作用于这两个字段的组合）
|
wrapper.select("distinct unit_code, unit_name");
|
List<DataExcel> list = baseMapper.selectList(wrapper);
|
List<Department> resultList = new ArrayList<>();
|
for (DataExcel dataExcel : list) {
|
Department department = new Department();
|
department.setDeptCode(dataExcel.getUnitCode());
|
department.setDeptName(dataExcel.getUnitName());
|
resultList.add(department);
|
}
|
return resultList;
|
}
|
|
@Override
|
public String importData(MultipartFile file, List<String> permissionDeptCodes) {
|
List<DataExcel> insertList = new ArrayList<>();
|
List<DataExcel> updateList = new ArrayList<>();
|
List<String> errorMsgs = new ArrayList<>();
|
int[] counts = new int[2];
|
int[] counts2 = new int[2];
|
|
try {
|
EasyExcel.read(file.getInputStream(), DataExcelImport.class, new ReadListener<DataExcelImport>() {
|
@Override
|
public void invoke(DataExcelImport data, AnalysisContext context) {
|
if (data.getUnitCode() == null || data.getUnitCode().isEmpty()) {
|
errorMsgs.add("第" + data.getSeqNo() + "行单位编码为空");
|
return;
|
}
|
DataExcel entity = new DataExcel();
|
entity.setSortNo(data.getSortNo());
|
entity.setSeqNo(data.getSeqNo());
|
entity.setSecondaryUnit(data.getSecondaryUnit());
|
entity.setUnitCode(data.getUnitCode());
|
entity.setUnitName(data.getUnitName());
|
entity.setTransactionNo(data.getTransactionNo());
|
entity.setSummary(data.getSummary());
|
entity.setAmount(data.getAmount());
|
entity.setAccountingPeriod(data.getAccountingPeriod());
|
entity.setVoucherNo(data.getVoucherNo());
|
entity.setBpContractNo(data.getBpContractNo());
|
entity.setContractCenterNo(data.getContractCenterNo());
|
entity.setBpInvoiceNo(data.getBpInvoiceNo());
|
entity.setGkInvoiceNo(data.getGkInvoiceNo());
|
entity.setBusinessRelation(data.getBusinessRelation());
|
entity.setRemark(data.getRemark());
|
entity.setStatus(1);
|
entity.setPkVoucher(data.getPkVoucher());
|
|
if (data.getSortNo() != null) {
|
DataExcel existData = baseMapper.selectBySortNo(data.getSortNo());
|
if (existData != null) {
|
entity.setId(existData.getId());
|
updateList.add(entity);
|
counts2[0]++;
|
} else {
|
insertList.add(entity);
|
counts[0]++;
|
}
|
} else {
|
insertList.add(entity);
|
counts[0]++;
|
}
|
}
|
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext context) {
|
}
|
}).sheet().doRead();
|
} catch (IOException e) {
|
return "导入失败: " + e.getMessage();
|
}
|
|
if (!insertList.isEmpty()) {
|
saveBatch(insertList);
|
}
|
if (!updateList.isEmpty()) {
|
updateBatchById(updateList);
|
}
|
|
StringBuilder result = new StringBuilder();
|
if (counts[0] > 0) {
|
result.append("成功新增 ").append(counts[0]).append(" 条记录");
|
}
|
if (counts2[0] > 0) {
|
if (result.length() > 0) {
|
result.append("，");
|
}
|
result.append("成功更新 ").append(counts2[0]).append(" 条记录");
|
}
|
if (!errorMsgs.isEmpty()) {
|
if (result.length() > 0) {
|
result.append("。");
|
}
|
result.append("导入部分失败: ").append(String.join(", ", errorMsgs));
|
}
|
|
if (result.length() == 0) {
|
return "导入失败";
|
}
|
|
return result.toString();
|
}
|
}
|
