raex_back/src/main/java/com/izouma/nineth/web/MetaLuckyDrawController.java

package com.izouma.nineth.web;

import com.izouma.nineth.config.MetaConstants;
import com.izouma.nineth.domain.MetaLuckyDraw;
import com.izouma.nineth.domain.MetaLuckyDrawAwardModel;
import com.izouma.nineth.dto.MetaRestResult;
import com.izouma.nineth.dto.PageQuery;
import com.izouma.nineth.enums.MetaAwardTypeEnum;
import com.izouma.nineth.exception.BusinessException;
import com.izouma.nineth.repo.MetaLuckyDrawRepo;
import com.izouma.nineth.service.MetaLuckyDrawService;
import com.izouma.nineth.service.MetaParamsConfigService;
import com.izouma.nineth.utils.ObjUtils;
import com.izouma.nineth.utils.excel.ExcelUtils;
import lombok.AllArgsConstructor;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.domain.Page;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicInteger;

@RestController
@RequestMapping("/metaLuckyDraw")
@AllArgsConstructor
public class MetaLuckyDrawController extends BaseController {
    private MetaLuckyDrawService    metaLuckyDrawService;
    private MetaLuckyDrawRepo       metaLuckyDrawRepo;
    private MetaParamsConfigService metaParamsConfigService;

    //@PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/save")
    public MetaLuckyDraw save(@RequestBody MetaLuckyDraw record) {
        List<MetaLuckyDrawAwardModel> metaLuckDrawAwards = record.getMetaLuckDrawAwards();
        if (CollectionUtils.isNotEmpty(metaLuckDrawAwards)) {
            AtomicInteger i = new AtomicInteger();
            List<String> list = new ArrayList<>();
            metaLuckDrawAwards.forEach(metaLuckDrawAward -> {
                if (metaLuckDrawAward.getAwardType().equals(MetaAwardTypeEnum.GOLD)) {
                    metaLuckDrawAward.setLimitNum(false);
                    metaLuckDrawAward.setMaxNum(0);
                }
                if (metaLuckDrawAward.getAwardType().equals(MetaAwardTypeEnum.NFT) && !metaLuckDrawAward.isLimitNum()) {
                    metaLuckDrawAward.setMaxNum(0);
                }
                list.add(metaLuckDrawAward.getName());
                i.set(i.get() + metaLuckDrawAward.getProbability());
            });
            long count = list.stream().distinct().count();
            if (count < list.size()) {
                throw new BusinessException("奖励名称不可以重复！");
            }
            if (i.get() != 100) {
                throw new BusinessException("奖励概率总和必须为100！");
            }
            record.setMetaLuckDrawAwards(metaLuckDrawAwards);
        }
        MetaLuckyDraw metaLuckyDraw = metaLuckyDrawRepo.findByNameAndDel(record.getName(), false);
        if (Objects.nonNull(metaLuckyDraw) && !Objects.equals(metaLuckyDraw.getId(), record.getId())) {
            throw new BusinessException("该活动名称已经存在！");
        }
        if (record.getId() != null) {
            MetaLuckyDraw orig = metaLuckyDrawRepo.findById(record.getId()).orElseThrow(new BusinessException("无记录"));
            ObjUtils.merge(orig, record);
            return metaLuckyDrawRepo.save(orig);
        }
        return metaLuckyDrawRepo.save(record);
    }


    //@PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/all")
    public Page<MetaLuckyDraw> all(@RequestBody PageQuery pageQuery) {
        return metaLuckyDrawService.all(pageQuery);
    }

    @GetMapping("/get/{id}")
    public MetaLuckyDraw get(@PathVariable Long id) {
        return metaLuckyDrawRepo.findById(id).orElseThrow(new BusinessException("无记录"));
    }

    @PostMapping("/del/{id}")
    public void del(@PathVariable Long id) {
        metaLuckyDrawRepo.softDelete(id);
    }

    @GetMapping("/excel")
    @ResponseBody
    public void excel(HttpServletResponse response, PageQuery pageQuery) throws IOException {
        List<MetaLuckyDraw> data = all(pageQuery).getContent();
        ExcelUtils.export(response, data);
    }

    @GetMapping("/findMetaLuckyDraw")
    public MetaRestResult<MetaLuckyDraw> findMetaLuckyDraw() {
        String idConfig = metaParamsConfigService.getString(MetaConstants.META_LUCKY_DRAW_ID);
        if (StringUtils.isBlank(idConfig)) {
            return MetaRestResult.returnError("抽奖活动id配置不存在！");
        }
        MetaLuckyDraw metaLuckyDraw = metaLuckyDrawRepo.findById(Long.parseLong(idConfig)).orElse(null);
        if (Objects.isNull(metaLuckyDraw)) {
            return MetaRestResult.returnError("抽奖活动不存在！");
        }
        return MetaRestResult.returnSuccess(metaLuckyDraw);
    }

}