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

package com.izouma.nineth.web;

import com.izouma.nineth.domain.Collection;
import com.izouma.nineth.domain.MintActivity;
import com.izouma.nineth.domain.Tag;
import com.izouma.nineth.dto.MintActivityRule;
import com.izouma.nineth.enums.CollectionType;
import com.izouma.nineth.repo.CollectionRepo;
import com.izouma.nineth.service.MintActivityService;
import com.izouma.nineth.dto.PageQuery;
import com.izouma.nineth.exception.BusinessException;
import com.izouma.nineth.repo.MintActivityRepo;
import com.izouma.nineth.service.SysConfigService;
import com.izouma.nineth.utils.ObjUtils;
import com.izouma.nineth.utils.excel.ExcelUtils;
import freemarker.ext.beans.BeansWrapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.data.domain.Page;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

@Slf4j
@RestController
@RequestMapping("/mintActivity")
@AllArgsConstructor
public class MintActivityController extends BaseController {
    private MintActivityService mintActivityService;
    private MintActivityRepo    mintActivityRepo;
    private CollectionRepo      collectionRepo;
    private SysConfigService    sysConfigService;

    @PreAuthorize("hasAnyRole('ADMIN','SAAS')")
    @PostMapping("/save")
    public MintActivity save(@RequestBody MintActivity record) {
        if (record.getRule() != null) mintActivityService.checkRule(record.getRule());
        if (record.getId() != null) {
            MintActivity orig = mintActivityRepo.findById(record.getId()).orElseThrow(new BusinessException("无记录"));
            orig.setName(record.getName());
            orig.setCover(record.getCover());
            orig.setMinterId(record.getMinterId());
            orig.setMinter(record.getMinter());
            orig.setMinterAvatar(record.getMinterAvatar());
            orig.setDetail(record.getDetail());
            orig.setAudit(record.isAudit());
            orig.setNum(record.getNum());
            orig.setCollectionName(record.getCollectionName());
            orig.setRule(record.getRule());
            orig.setConsume(record.isConsume());
            orig.setGasPrice(record.getGasPrice());
            orig.setAirDrop(record.isAirDrop());
            orig.setAutoDrop(record.isAutoDrop());
            orig.setOnShelf(record.isOnShelf());
            orig.setScheduleSale(record.isScheduleSale());
            orig.setStartTime(record.getStartTime());
            orig.setAutoDrop(record.isAutoDrop());
            orig.setAirDropCollectionId(record.getAirDropCollectionId());

            orig = mintActivityRepo.save(orig);
            mintActivityService.syncStock(record.getId());
            return orig;
        } else {
            record.setStock(record.getTotal());
            record = mintActivityRepo.save(record);
            mintActivityService.syncStock(record.getId());
            return record;
        }
    }


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

    @GetMapping("/get/{id}")
//    @Cacheable(cacheNames = "mintActivity", key = "#id")
    public MintActivity get(@PathVariable Long id) {
        MintActivity mintActivity = mintActivityRepo.findById(id).orElseThrow(new BusinessException("无记录"));
//        String mintCountCollection = sysConfigService.getString("mint_countCollection");
//        String regex = ",";
//        String[] split = mintCountCollection.split(regex);
//        if (Arrays.asList(split).contains(String.valueOf(id))) {
//            // Id在split数组中
//            return mintActivity;
//        } else {
//            // Id不在split数组中
//        List<Long> tags = mintActivity.getHoldingTags();
//        tags.forEach(tagId -> {
//            List<Collection> collections = collectionRepo.findByTag(tagId);
//            if (collections.stream().anyMatch(collection -> collection.getType().equals(CollectionType.BLIND_BOX)))
//        });
        try {
            MintActivityRule mintActivityRule = mintActivity.getRule();
            List<MintActivityRule> and = mintActivityRule.getAnd();
            mintActivityRule.setAnd(mintActivityService.mintActivityRules(and, mintActivity.getId()));
            mintActivity.setRule(mintActivityRule);
            if (mintActivity.getAirDropCollectionId() != null) {
                mintActivity.setAirDropCollection(collectionRepo.findById(mintActivity.getAirDropCollectionId())
                                                                .orElse(null));
            }
        } catch (Exception e) {
            log.error("生成铸造内容出错" + id, e);
        }

        mintActivity.setVTotal(mintActivity.getTotal());
        mintActivity.setTotal(0);
        return mintActivity;
//        }
    }

    @PreAuthorize("hasAnyRole('ADMIN','SAAS')")
    @PostMapping("/del/{id}")
    public void del(@PathVariable Long id) {
        mintActivityRepo.softDelete(id);
    }

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