package com.mzl.flower.service.impl.wallet;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.mzl.flower.config.exception.ValidationException;
import com.mzl.flower.config.security.SecurityUtils;
import com.mzl.flower.constant.Constants;
import com.mzl.flower.constant.LockConstants;
import com.mzl.flower.dto.request.wallet.CreateWalletWithdrawRecordDTO;
import com.mzl.flower.dto.request.wallet.QueryWalletWithdrawDTO;
import com.mzl.flower.dto.request.wallet.WalletWithdrawRecordDTO;
import com.mzl.flower.dto.response.supplier.SupplierDTO;
import com.mzl.flower.dto.response.wallet.WalletWithdrawRecordVO;
import com.mzl.flower.entity.wallet.WalletDO;
import com.mzl.flower.entity.wallet.WalletWithdrawRecordDO;
import com.mzl.flower.mapper.wallet.WalletWithdrawRecordMapper;
import com.mzl.flower.service.supplier.SupplierService;
import com.mzl.flower.service.wallet.WalletBillRecordService;
import com.mzl.flower.service.wallet.WalletService;
import com.mzl.flower.service.wallet.WalletWithdrawRecordService;
import lombok.RequiredArgsConstructor;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
import java.time.LocalDateTime;
import java.util.List;
import java.util.concurrent.TimeUnit;
/**
*
* 服务实现类
*
*
* @author @TaoJie
* @since 2024-10-22
*/
@Service
@Transactional
@RequiredArgsConstructor
public class WalletWithdrawRecordServiceImpl extends ServiceImpl implements WalletWithdrawRecordService {
private final WalletWithdrawRecordMapper walletWithdrawRecordMapper;
private final WalletService walletService;
private final SupplierService supplierService;
private final WalletBillRecordService walletBillRecordService;
@Autowired
RedissonClient redissonClient;
@Transactional
@Override
public boolean create(CreateWalletWithdrawRecordDTO dto) {
SupplierDTO supplierDTO = supplierService.getCurrentSupplier();
if (null == supplierDTO) {
throw new ValidationException("供应商不能为空");
}
WalletDO walletDO = walletService.getCurrentSupplier();
if (null == walletDO) {
throw new ValidationException("钱包不能为空");
}
// 查看钱包的金额是不是大于要提现的金额
if (null != walletDO.getWithdrawableAmount() && null != dto.getAmount() && walletDO.getWithdrawableAmount().compareTo(dto.getAmount()) < 0) {
throw new ValidationException("钱包金额不足");
}
RLock lock = redissonClient.getLock(String.format(LockConstants.WALLET_ID_KEY, walletDO.getId()));
// 获取锁,最多等待 10 秒,锁自动释放时间 30 秒
try {
if (lock.tryLock(10, 30, TimeUnit.SECONDS)) {
try {
// 加锁了之后需要二次查询,防止未加锁之前 someone else 已经修改了
walletDO = walletService.getCurrentSupplier();
if (null == walletDO) {
throw new ValidationException("钱包不能为空");
}
// 查看钱包的金额是不是大于要提现的金额
if (null != walletDO.getWithdrawableAmount() && null != dto.getAmount() && walletDO.getWithdrawableAmount().compareTo(dto.getAmount()) < 0) {
throw new ValidationException("钱包金额不足");
}
// 1.新增一条余额提现记录
WalletWithdrawRecordDO withdrawRecordDO = new WalletWithdrawRecordDO();
withdrawRecordDO.setAmount(dto.getAmount());
withdrawRecordDO.setSupplierId(supplierDTO.getId());
withdrawRecordDO.setWithdrawState(Constants.WALLET_WITHDRAW_STATE.WAITING.name());
withdrawRecordDO.setMethod(Constants.WALLET_WITHDRAW_METHOD.WEIXIN.name());
withdrawRecordDO.create(SecurityUtils.getUserId());
// 保存余额提现记录
save(withdrawRecordDO);
/*
// 2.新增一条账单明细
WalletBillRecordDO walletBillRecordDO=new WalletBillRecordDO();
walletBillRecordDO.setSupplierId(supplierDTO.getId());
walletBillRecordDO.setWalletId(walletDO.getId());
// 提现
walletBillRecordDO.setType(Constants.BILL_CHANGE_TYPE.withdraw.name());
walletBillRecordDO.setMethod(Constants.BILL_CHANGE_METHOD.add.name());
walletBillRecordDO.setOriginalAmount(walletDO.getWithdrawableAmount());
walletBillRecordDO.setChangeAmount(dto.getAmount());
walletBillRecordDO.setBalance(walletDO.getWithdrawableAmount().subtract(dto.getAmount()));
// 保存账单明细
walletBillRecordService.save(walletBillRecordDO);
*/
// 3. 钱包更新
// 钱包可提现的钱等于当前钱包可提现的钱-提现的钱
walletDO.setWithdrawableAmount(walletDO.getWithdrawableAmount().subtract(dto.getAmount()));
// 提现中的钱等于原有可提现的钱+提现的钱
walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().add(dto.getAmount()));
walletDO.update(SecurityUtils.getUserId());
// 更新钱包
return walletService.updateById(walletDO);
} finally {
lock.unlock();
}
}
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
return false;
}
@Override
public Page queryPage(QueryWalletWithdrawDTO queryWalletDTO, Page page) {
List list = walletWithdrawRecordMapper.queryPage(queryWalletDTO, page);
page.setRecords(list);
return page;
}
@Override
public void updateWallet(WalletWithdrawRecordDTO walletWithdrawRecordDTO) {
WalletWithdrawRecordDO withdrawRecordDO = walletWithdrawRecordMapper.selectById(walletWithdrawRecordDTO.getId());
if (withdrawRecordDO == null) {
throw new ValidationException("提现记录信息不存在");
}
if (StringUtils.isEmpty(walletWithdrawRecordDTO.getApproveState())) {
throw new ValidationException("审批状态不能为空");
}
BeanUtils.copyProperties(walletWithdrawRecordDTO, withdrawRecordDO);
withdrawRecordDO.update(SecurityUtils.getUserId());
withdrawRecordDO.setApproveBy(SecurityUtils.getUserId());
withdrawRecordDO.setApproveTime(LocalDateTime.now());
walletWithdrawRecordMapper.updateById(withdrawRecordDO);
}
}