| | |
| | | qty = Optional.ofNullable(qty).orElse(1); |
| | | |
| | | // reserve sta in |
| | | if (qty < staMapper.tryReserveIn(sta.getId(), qty)) { |
| | | if (staMapper.tryReserveIn(sta.getId(), qty) < qty) { |
| | | throw new BusinessException("Sta[" + sta.getStaNo() + "] can't reserve IN"); |
| | | } |
| | | Date now = new Date(); |
| | |
| | | reserve.setQty(qty); |
| | | reserve.setState(StaReserveStateType.RESERVED.toString()); |
| | | |
| | | reserve.setUniqKey(this.generateReserveUniqKey(sta.getId(), task.getId())); |
| | | reserve.setUniqKey(this.generateReserveUniqKey(sta.getId(), task.getId(), StaReserveType.IN)); |
| | | reserve.setExpireTime(new Date(now.getTime() + RESERVE_EXPIRE_TIME)); |
| | | |
| | | reserve.setStatus(StatusType.ENABLE.val); |
| | |
| | | return reserve; |
| | | } |
| | | |
| | | @Override |
| | | @Transactional(rollbackFor = Exception.class) |
| | | public StaReserve reserveStaOut(Sta sta, Task task, Integer qty) { |
| | | qty = Optional.ofNullable(qty).orElse(1); |
| | | |
| | | private String generateReserveUniqKey(Long staId, Long taskId) { |
| | | return "STA:" + staId + "-TASK:" + taskId + "-T"; |
| | | // reserve sta out |
| | | if (staMapper.tryReserveOut(sta.getId(), qty) < qty) { |
| | | throw new BusinessException("Sta[" + sta.getStaNo() + "] can't reserve OUT"); |
| | | } |
| | | |
| | | Date now = new Date(); |
| | | |
| | | // save staReserve |
| | | StaReserve reserve = new StaReserve(); |
| | | reserve.setStaId(sta.getId()); |
| | | reserve.setTaskId(task.getId()); |
| | | reserve.setType(StaReserveType.OUT.toString()); |
| | | reserve.setQty(qty); |
| | | reserve.setState(StaReserveStateType.RESERVED.toString()); |
| | | |
| | | reserve.setUniqKey(this.generateReserveUniqKey(sta.getId(), task.getId(), StaReserveType.OUT)); |
| | | reserve.setExpireTime(new Date(now.getTime() + RESERVE_EXPIRE_TIME)); |
| | | |
| | | reserve.setStatus(StatusType.ENABLE.val); |
| | | reserve.setCreateTime(now); |
| | | reserve.setUpdateTime(now); |
| | | |
| | | if (!this.save(reserve)) { |
| | | staMapper.releaseReserveOut(sta.getId(), qty); |
| | | throw new BusinessException("reserve record insert failed"); |
| | | } |
| | | |
| | | return reserve; |
| | | } |
| | | |
| | | |
| | | private String generateReserveUniqKey(Long staId, Long taskId, StaReserveType type) { |
| | | return "STA:" + staId + "-TASK:" + taskId + "-T:" + type.toString(); |
| | | } |
| | | |
| | | } |