package org.example.ssmico.demos.web.tesk;
|
|
import cn.hutool.core.date.DateUtil;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.serializer.SerializerFeature;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.mysql.cj.util.StringUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.ThreadUtils;
|
import org.example.ssmico.demos.web.entity.*;
|
import org.example.ssmico.demos.web.mapper.ExchangeApplyOrderMapper;
|
import org.example.ssmico.demos.web.mapper.PartyMapper;
|
import org.example.ssmico.demos.web.mapper.WalletExtendMapper;
|
import org.example.ssmico.demos.web.service.IcoNewCurrencySerivce;
|
import org.example.ssmico.demos.web.service.IcoOrderService;
|
import org.example.ssmico.demos.web.util.*;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
import org.springframework.transaction.annotation.Transactional;
|
import project.wallet.WalletExtend;
|
|
import java.text.SimpleDateFormat;
|
import java.util.*;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.locks.Lock;
|
import java.util.concurrent.locks.ReentrantLock;
|
import java.util.stream.Collectors;
|
import java.util.stream.Stream;
|
|
/**
|
* @program: ssm-ico
|
* @description:
|
* @create: 2024-04-18 17:32
|
**/
|
@Slf4j
|
@Component
|
public class NewCurrencyListingTask {
|
|
@Autowired
|
private IcoOrderService icoOrderService;
|
|
@Autowired
|
private IcoNewCurrencySerivce icoNewCurrencySerivce;
|
|
@Autowired
|
private ExchangeApplyOrderMapper exchangeApplyOrderMapper;
|
|
@Autowired
|
private WalletExtendMapper walletExtendMapper;
|
|
private final Lock lock = new ReentrantLock();
|
|
/**
|
* 上市冻结转资产
|
*/
|
@Scheduled(cron = "0 0/10 * * * ?")
|
public void icoListing() {
|
if (lock.tryLock()) {
|
log.info("开始执行上市冻结转资产任务");
|
try {
|
listingThaw();
|
} catch (Exception e) {
|
log.error("上市冻结转资产任务执行失败：" + e.getMessage(), e);
|
} finally {
|
lock.unlock();
|
}
|
} else {
|
log.info("上一次上市冻结转资产任务尚未完成，本次任务取消");
|
}
|
}
|
|
@Transactional
|
public void listingThaw(){
|
List<IcoNewCurrency> icoNewCurrencyList = findICOListings();
|
if(CollectionUtils.isNotEmpty(icoNewCurrencyList)){
|
List<Integer> ids = icoNewCurrencyList.stream().map(IcoNewCurrency::getId)
|
.collect(Collectors.toList());
|
List<IcoOrder> icoOrderList = findOrdersByCurrencyIds(ids);
|
if (!icoOrderList.isEmpty()) {
|
icoOrderList.forEach(order -> {
|
WalletExtend walletExtend = walletExtendMapper.selectOne(
|
new LambdaQueryWrapper<WalletExtend>()
|
.eq(WalletExtend::getPartyId, order.getUserId())
|
.eq(WalletExtend::getWallettype, order.getTokenCode()));
|
if(null != walletExtend){
|
walletExtend.setAmount(walletExtend.getAmount()+order.getLotteryQuantity());
|
if(walletExtend.getFrozenAmount()<order.getLotteryQuantity()){
|
walletExtend.setFrozenAmount(0);
|
}else{
|
walletExtend.setFrozenAmount(walletExtend.getFrozenAmount()-order.getLotteryQuantity());
|
}
|
//
|
String json = JSON.toJSONString(walletExtend, SerializerFeature.WriteClassName);
|
walletExtendMapper.update(walletExtend,new LambdaUpdateWrapper<WalletExtend>().eq(WalletExtend::getUuid,walletExtend.getUuid()));
|
RedisShardedPoolUtils.set(WalletRedisKeys.WALLET_EXTEND_PARTY_ID + walletExtend.getPartyId() + walletExtend.getWallettype(),json);
|
order.setStatus(5); // 更新订单状态
|
icoOrderService.updateById(order);
|
}
|
});
|
}
|
}
|
}
|
|
private List<IcoNewCurrency> findICOListings() {
|
return icoNewCurrencySerivce.list(new LambdaQueryWrapper<IcoNewCurrency>()
|
.lt(IcoNewCurrency::getListingTime, new Date()));
|
}
|
|
private List<IcoOrder> findOrdersByCurrencyIds(List<Integer> ids) {
|
return icoOrderService.list(new LambdaQueryWrapper<IcoOrder>()
|
.in(IcoOrder::getIcoNewCurrencyId, ids).and(a->
|
a.eq(IcoOrder::getStatus, 3).ne(IcoOrder::getStatus, 4)));
|
}
|
|
}
|
