package com.vincent.rsf.server.manager.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.vincent.rsf.framework.common.Cools; import com.vincent.rsf.framework.common.R; import com.vincent.rsf.framework.exception.CoolException; import com.vincent.rsf.server.common.domain.PageResult; import com.vincent.rsf.server.common.utils.ExcelUtil; import com.vincent.rsf.server.common.annotation.OperationLog; import com.vincent.rsf.server.common.domain.BaseParam; import com.vincent.rsf.server.common.domain.KeyValVo; import com.vincent.rsf.server.common.domain.PageParam; import com.vincent.rsf.server.manager.entity.TaskItem; import com.vincent.rsf.server.manager.entity.TaskItemLog; import com.vincent.rsf.server.manager.entity.Wave; import com.vincent.rsf.server.manager.entity.WaveItem; import com.vincent.rsf.server.manager.enums.WaveExceStatus; import com.vincent.rsf.server.manager.enums.WaveItemExceStatus; import com.vincent.rsf.server.manager.service.OutStockService; import com.vincent.rsf.server.manager.service.TaskItemLogService; import com.vincent.rsf.server.manager.service.TaskItemService; import com.vincent.rsf.server.manager.service.WaveService; import com.vincent.rsf.server.manager.service.impl.TaskItemLogServiceImpl; import com.vincent.rsf.server.manager.service.impl.WaveItemServiceImpl; import com.vincent.rsf.server.system.controller.BaseController; import io.swagger.annotations.ApiOperation; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletResponse; import java.util.*; @RestController public class WaveController extends BaseController { @Autowired private WaveService waveService; @Autowired private TaskItemService taskItemService; @Autowired private OutStockService outStockService; @Autowired private TaskItemLogService taskItemLogService; @Autowired private WaveItemServiceImpl waveItemService; @PreAuthorize("hasAuthority('manager:wave:list')") @PostMapping("/wave/page") public R page(@RequestBody Map map) { BaseParam baseParam = buildParam(map, BaseParam.class); PageParam pageParam = new PageParam<>(baseParam, Wave.class); return R.ok().add(waveService.page(pageParam, pageParam.buildWrapper(true))); } @PreAuthorize("hasAuthority('manager:wave:list')") @PostMapping("/wave/list") public R list(@RequestBody Map map) { return R.ok().add(waveService.list()); } @PreAuthorize("hasAuthority('manager:wave:list')") @PostMapping({"/wave/many/{ids}", "/waves/many/{ids}"}) public R many(@PathVariable Long[] ids) { return R.ok().add(waveService.listByIds(Arrays.asList(ids))); } @PreAuthorize("hasAuthority('manager:wave:list')") @GetMapping("/wave/{id}") public R get(@PathVariable("id") Long id) { return R.ok().add(waveService.getById(id)); } @PreAuthorize("hasAuthority('manager:wave:save')") @OperationLog("Create 波次单据") @PostMapping("/wave/save") public R save(@RequestBody Wave wave) { wave.setCreateBy(getLoginUserId()); wave.setUpdateBy(getLoginUserId()); if (!waveService.save(wave)) { return R.error("Save Fail"); } return R.ok("Save Success").add(wave); } @PreAuthorize("hasAuthority('manager:wave:update')") @OperationLog("Update 波次单据") @PostMapping("/wave/update") public R update(@RequestBody Wave wave) { wave.setUpdateBy(getLoginUserId()); wave.setUpdateTime(new Date()); if (!waveService.updateById(wave)) { return R.error("Update Fail"); } return R.ok("Update Success").add(wave); } @PreAuthorize("hasAuthority('manager:wave:remove')") @OperationLog("Delete 波次单据") @PostMapping("/wave/remove/{ids}") public R remove(@PathVariable Long[] ids) { if (Objects.isNull(ids)) { throw new CoolException("参数不能为空!!"); } List list = Arrays.asList(ids); List taskItems = taskItemService.list(new LambdaQueryWrapper().in(TaskItem::getSourceId, list)); if (!taskItems.isEmpty()) { throw new CoolException("有未完成任务,不可执行删除操作!!"); } return waveService.cancelWave(list); } @PreAuthorize("hasAuthority('manager:wave:list')") @PostMapping("/wave/query") public R query(@RequestParam(required = false) String condition) { List vos = new ArrayList<>(); LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>(); if (!Cools.isEmpty(condition)) { wrapper.like(Wave::getId, condition); } waveService.page(new Page<>(1, 30), wrapper).getRecords().forEach( item -> vos.add(new KeyValVo(item.getId(), item.getId())) ); return R.ok().add(vos); } @PreAuthorize("hasAuthority('manager:wave:list')") @PostMapping("/wave/export") public void export(@RequestBody Map map, HttpServletResponse response) throws Exception { ExcelUtil.build(ExcelUtil.create(waveService.list(), Wave.class), response); } @PreAuthorize("hasAuthority('manager:wave:update')") @ApiOperation("波次下发任务") @PostMapping("/wave/public/task") public R publicTask(@RequestBody Map map) { if (Cools.isEmpty(map) || Cools.isEmpty(map.get("wave"))) { throw new CoolException("参数不能为空!!"); } return waveService.publicTask(map, getLoginUserId()); } @PreAuthorize("hasAuthority('manager:wave:update')") @ApiOperation("选择明细下发任务") @PostMapping("/wave/selects/task") public R waveToTask(@RequestBody Map map) { if (Cools.isEmpty(map) || Cools.isEmpty(map.get("ids"))) { throw new CoolException("参数不能为空!!"); } return waveService.waveToTask(map, getLoginUserId()); } @PreAuthorize("hasAuthority('manager:wave:list')") @ApiOperation("波次出库任务预览") @PostMapping("/wave/locs/preview/page") public R mergeWavePreview(@RequestBody Map map) { if (Cools.isEmpty(map.get("waveId")) || StringUtils.isBlank(map.get("waveId").toString())) { throw new CoolException("参数不能为空!!"); } Long waveId = Long.parseLong(map.get("waveId").toString()); List waveItems = waveService.mergeWavePreview(waveId); PageResult pageResult = new PageResult<>(); pageResult.setRecords(waveItems).setTotal(Long.parseLong(waveItems.size() + "")); return R.ok().add(pageResult); } @PreAuthorize("hasAuthority('manager:waveItem:update')") @ApiOperation("暂停下发任务") @PostMapping("/wave/pause/pub/{id}") public R pausePublicTask(@PathVariable Long id) { waveService.update(new LambdaUpdateWrapper() .eq(Wave::getId, id) .set(Wave::getExceStatus, WaveExceStatus.WAVE_EXCE_STATUS_PAUSE.val)); return R.ok(); } @PreAuthorize("hasAuthority('manager:waveItem:update')") @ApiOperation("继续下发任务") @PostMapping("/wave/continue/pub/{id}") public R continuePublicTask(@PathVariable Long id) { waveService.update(new LambdaUpdateWrapper() .eq(Wave::getId, id) .set(Wave::getExceStatus, WaveExceStatus.WAVE_EXCE_STATUS_EXCING.val)); return R.ok(); } @PreAuthorize("hasAuthority('manager:waveItem:update')") @ApiOperation("终止下发任务") @PostMapping("/wave/stop/pub/{id}") public R stopPublicTask(@PathVariable Long id) { if (Objects.isNull(id)) { return R.error("参数不能为空!!"); } waveService.update(new LambdaUpdateWrapper() .eq(Wave::getId, id) .set(Wave::getExceStatus, WaveExceStatus.WAVE_EXCE_STATUS_PAUSE.val)); return waveService.stopPubTask(id); } }