package com.ruoyi.sales.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.ruoyi.common.utils.bean.BeanUtils;
|
import com.ruoyi.framework.aspectj.lang.annotation.Log;
|
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
|
import com.ruoyi.framework.web.controller.BaseController;
|
import com.ruoyi.framework.web.domain.R;
|
import com.ruoyi.sales.dto.WeighingRecordExcelDto;
|
import com.ruoyi.sales.pojo.WeighingRecord;
|
import com.ruoyi.sales.service.IWeighingRecordService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 过磅记录（磅单台账）Controller
|
*
|
* @author ruoyi
|
* @date 2026-04-07
|
*/
|
@Api(tags = "磅单台账管理")
|
@RestController
|
@RequestMapping("/sales/weighbridgeLedger")
|
public class WeighingRecordController extends BaseController {
|
|
@Autowired
|
private IWeighingRecordService weighingRecordService;
|
|
/**
|
* 分页查询磅单台账列表
|
*/
|
@ApiOperation("分页查询磅单台账列表")
|
@GetMapping("/listPage")
|
public R<IPage<WeighingRecord>> listPage(Page<WeighingRecord> page, WeighingRecord weighingRecord) {
|
IPage<WeighingRecord> list = weighingRecordService.listPage(page, weighingRecord);
|
return R.ok(list);
|
}
|
|
/**
|
* 导入 Excel（新磅单台账）
|
*/
|
@ApiOperation("导入Excel")
|
@Log(title = "磅单台账", businessType = BusinessType.IMPORT)
|
@PostMapping("/import")
|
public R importWeighbridgeLedgerExcel(@RequestParam("file") MultipartFile file) throws Exception {
|
try {
|
List<WeighingRecordExcelDto> excelDataList = EasyExcel.read(file.getInputStream())
|
.head(WeighingRecordExcelDto.class)
|
.sheet()
|
.doReadSync();
|
|
if (CollectionUtils.isEmpty(excelDataList)) {
|
return R.fail("模板错误或导入数据为空");
|
}
|
|
// 转换为实体对象
|
List<WeighingRecord> weighingRecordList = new ArrayList<>();
|
for (WeighingRecordExcelDto dto : excelDataList) {
|
WeighingRecord record = new WeighingRecord();
|
BeanUtils.copyProperties(dto, record);
|
weighingRecordList.add(record);
|
}
|
|
// 去重检查
|
List<WeighingRecord> newRecordList = new ArrayList<>();
|
int duplicateCount = 0;
|
|
for (WeighingRecord record : weighingRecordList) {
|
LambdaQueryWrapper<WeighingRecord> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(WeighingRecord::getSerialNo, record.getSerialNo());
|
|
long count = weighingRecordService.count(wrapper);
|
if (count == 0) {
|
newRecordList.add(record);
|
} else {
|
duplicateCount++;
|
}
|
}
|
|
if (newRecordList.isEmpty()) {
|
return R.fail("所有数据都已存在，共" + duplicateCount + "条重复数据");
|
}
|
|
// 批量插入新数据
|
weighingRecordService.saveBatch(newRecordList);
|
|
String message = String.format("导入成功：%d条，跳过重复：%d条", newRecordList.size(), duplicateCount);
|
return R.ok(message);
|
|
} catch (Exception e) {
|
return R.fail("导入失败：" + e.getMessage());
|
}
|
}
|
|
/**
|
* 删除（新磅单台账）
|
*/
|
@ApiOperation("删除磅单台账")
|
@Log(title = "磅单台账", businessType = BusinessType.DELETE)
|
@DeleteMapping("/delete")
|
public R delWeighbridgeLedger(@RequestBody List<Long> ids) {
|
boolean success = weighingRecordService.removeByIds(ids);
|
if (success) {
|
return R.ok("删除成功");
|
} else {
|
return R.fail("删除失败");
|
}
|
}
|
|
}
|
