zxl
3 天以前 61e3917c5b0d667d23430890f45396a9bb7731ac
framework/src/main/java/cn/lili/modules/lmk/service/impl/StoreCouponServiceImpl.java
@@ -1,21 +1,40 @@
package cn.lili.modules.lmk.service.impl;
import cn.lili.common.exception.ServiceException;
import cn.lili.modules.lmk.domain.entity.CouponVirtual;
import cn.lili.modules.lmk.domain.entity.StoreCouponSingle;
import cn.lili.modules.lmk.enums.general.GenerateCouponStausEnum;
import cn.lili.modules.lmk.enums.general.StoreCouponStausEnum;
import cn.lili.modules.lmk.service.StoreCouponSingleService;
import cn.lili.modules.order.order.entity.enums.ClaimStatusEnum;
import cn.lili.modules.promotion.entity.dos.Coupon;
import cn.lili.modules.promotion.service.CouponService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import cn.lili.modules.lmk.domain.entity.StoreCoupon;
import cn.lili.modules.lmk.mapper.StoreCouponMapper;
import cn.lili.modules.lmk.service.StoreCouponService;
import cn.lili.base.Result;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import cn.lili.modules.lmk.domain.form.StoreCouponForm;
import cn.lili.modules.lmk.domain.vo.StoreCouponVO;
import cn.lili.modules.lmk.domain.query.StoreCouponQuery;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.stereotype.Service;
import lombok.RequiredArgsConstructor;
import cn.lili.utils.PageUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.support.TransactionSynchronization;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import org.springframework.util.Assert;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
/**
@@ -29,6 +48,10 @@
public class StoreCouponServiceImpl extends ServiceImpl<StoreCouponMapper, StoreCoupon> implements StoreCouponService {
    private final StoreCouponMapper storeCouponMapper;
    private final RedissonClient redissonClient;
    private static final String STORE_COUPON_GENERATE = "store_coupon_generate:";
    private final CouponService couponService;
    private final StoreCouponSingleService storeCouponSingleService;
    /**
     * 添加
@@ -87,8 +110,12 @@
     */
    @Override
    public Result page(StoreCouponQuery query) {
        IPage<StoreCouponVO> page = PageUtil.getPage(query, StoreCouponVO.class);
        baseMapper.getPage(page, query);
        IPage<StoreCoupon> page = PageUtil.getPage(query, StoreCoupon.class);
        LambdaQueryWrapper<StoreCoupon> wrapper = Wrappers.lambdaQuery();
        wrapper.eq(Objects.nonNull(query.getStoreId()), StoreCoupon::getStoreId, query.getStoreId());
        wrapper.eq(StringUtils.isNotBlank(query.getStatus()), StoreCoupon::getStatus, query.getStatus());
        wrapper.eq(StringUtils.isNotBlank(query.getGenerateStatus()), StoreCoupon::getGenerateStatus, query.getGenerateStatus());
        this.page(page, wrapper);
        return Result.ok().data(page.getRecords()).total(page.getTotal());
    }
@@ -119,7 +146,98 @@
    @Override
    public Result getCoupon(String id) {
        StoreCoupon storeCoupon = this.getById(id);
        return Result.ok().data(storeCoupon);
        StoreCouponSingle storeCouponSingle = storeCouponSingleService.getById(id);
        StoreCoupon storeCoupon = this.getById(storeCouponSingle.getStoreCoupRef());
        if (storeCoupon == null) {
            throw new ServiceException("店铺礼品卡包不存在");
        }
        if (!StoreCouponStausEnum.ENABLE.name().equals(storeCoupon.getStatus())) {
            throw new ServiceException("店铺礼品卡包没有启用无法领取");
        }
        if (ClaimStatusEnum.CLAIM.name().equals(storeCouponSingle.getClaimStatus())) {
            throw new ServiceException("当前礼品卡包已被领取");
        }
        return Result.ok().data(storeCouponSingle);
    }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result addStoreCoupon(StoreCouponForm storeCoupon) {
        StoreCoupon entity = StoreCouponForm.getEntityByForm(storeCoupon, null);
        entity.setStatus(StoreCouponStausEnum.ENABLE.name());
        entity.setCouponClaimNum(0);
        entity.setGenerateStatus(GenerateCouponStausEnum.NOT_GENERATE.name());
        Coupon coupon = couponService.getById(entity.getCouponId());
        Integer publishNum = coupon.getPublishNum();
        //表示不限制领取数量
        if (!Integer.valueOf(0).equals(publishNum)) {
            if (publishNum < storeCoupon.getCouponNum()) {
                throw new ServiceException("优惠卷发行数量小于生成二维码数量");
            }
        }
        this.save(entity);
        return Result.ok();
    }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result generateStoreCoupon(String id) {
        RLock redissonLock = redissonClient.getLock(STORE_COUPON_GENERATE + id);
        try {
            redissonLock.lock();
            LambdaQueryWrapper<StoreCoupon> forUpdate = Wrappers.<StoreCoupon>lambdaQuery().eq(StoreCoupon::getId, id).last("FOR UPDATE");
            StoreCoupon storeCoupon = this.getOne(forUpdate);
            if (storeCoupon == null) {
                throw new ServiceException("当前店铺优惠卷不存在无法生成");
            }
            if (!GenerateCouponStausEnum.NOT_GENERATE.name().equals(  storeCoupon.getGenerateStatus())) {
                throw new ServiceException("当前店铺优惠卷状态异常无法生成");
            }
            //生成优惠卷
            Integer couponNum = storeCoupon.getCouponNum();
            if (couponNum == null) {
                throw new ServiceException("发行数量为空不能生成");
            }
            List<StoreCouponSingle> storeCouponSingles = new ArrayList<>();
            for (int i = 1; i <= couponNum; i++) {
                StoreCouponSingle storeCouponSingle = getStoreCouponSingle(storeCoupon, i);
                storeCouponSingles.add(storeCouponSingle);
            }
            if (!storeCouponSingles.isEmpty()) {
                storeCouponSingleService.saveBatch(storeCouponSingles);
            }
            //更新状态生成状态
            storeCoupon.setGenerateStatus(GenerateCouponStausEnum.GENERATE.name());
            this.updateById(storeCoupon);
            return Result.ok();
        }finally {
            TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronization() {
                @Override
                public void afterCommit() {
                    if (redissonLock.isHeldByCurrentThread()) {
                        redissonLock.unlock();
                    }
                }
                @Override
                public void afterCompletion(int status) {
                    // 确保即使在事务回滚的情况下也能释放锁
                    if (redissonLock.isHeldByCurrentThread()) {
                        redissonLock.unlock();
                    }
                }
            });
        }
    }
    private static StoreCouponSingle getStoreCouponSingle(StoreCoupon storeCoupon, int i) {
        StoreCouponSingle storeCouponSingle = new StoreCouponSingle();
        storeCouponSingle.setStoreCoupRef(storeCoupon.getId());
        storeCouponSingle.setStoreId(storeCoupon.getStoreId());
        storeCouponSingle.setStoreName(storeCoupon.getStoreName());
        storeCouponSingle.setCouponName(storeCoupon.getCouponName());
        storeCouponSingle.setCouponId(storeCoupon.getCouponId());
        storeCouponSingle.setClaimStatus(ClaimStatusEnum.NOT_CLAIM.name());
        storeCouponSingle.setCouponNo(String.format("%08d", i));
        return storeCouponSingle;
    }
}