package com.nq.utils.task.stock;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.nq.dao.StockSubscribeMapper;
import com.nq.dao.UserMapper;
import com.nq.dao.UserStockSubscribeMapper;
import com.nq.pojo.StockSubscribe;
import com.nq.pojo.User;
import com.nq.pojo.UserStockSubscribe;
import com.nq.service.IUserPositionService;
import com.nq.utils.DateUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.math.BigDecimal;
import java.util.List;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import java.util.stream.Collectors;

/**
 * @program: huihuangserver
 * @description: 中签定时任务
 * @create: 2024-03-25 15:21
 **/
@Component
@Slf4j
public class BallotTask {

    @Autowired
    UserStockSubscribeMapper userStockSubscribeMapper;

    @Autowired
    IUserPositionService iUserPositionService;

    @Autowired
    StockSubscribeMapper stockSubscribeMapper;

    @Autowired
    UserMapper userMapper;

    private final Lock ballotLock = new ReentrantLock();

    private final Lock transferFundsLock = new ReentrantLock();

    @Scheduled(cron = "0 0 0/1 * * ? ")
//    @Scheduled(cron = "0 0/1 * * * *")
    public void ballot() {
        if (ballotLock.tryLock()) {
            try {
                log.info("新股申购转持仓定时任务--------->开始");
                List<StockSubscribe> stockSubscribes = stockSubscribeMapper.selectList(new LambdaQueryWrapper<StockSubscribe>().lt(StockSubscribe::getListingTime, DateUtils.getDate()));
                if (CollectionUtils.isNotEmpty(stockSubscribes)) {
                    List<String> codeList = stockSubscribes.stream().map(StockSubscribe::getCode).collect(Collectors.toList());
                    List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
                            .eq(UserStockSubscribe::getStatus, 4).in(UserStockSubscribe::getNewCode, codeList));
                    //订单转持仓
                    userStockSubscribes.forEach(f -> {
                        iUserPositionService.newStockToPosition(f.getId());//转持仓
                        f.setStatus(5);
                        userStockSubscribeMapper.updateById(f);
                    });
                }
                log.info("新股申购转持仓定时任务--------->结束");
            } catch (Exception e) {
                log.error("新股申购转持仓定时任务发生异常", e);
            } finally {
                ballotLock.unlock();
            }
        } else {
            log.info("新股申购转持仓定时任务--------->上次任务还未执行完成，本次任务忽略");
        }
    }


    @Scheduled(cron = "0/10 * * * * ? ")
    public void transferFunds() {
        if (transferFundsLock.tryLock()) {
            log.info("锁定账户待补资金扣钱定时任务--------->开始");
            try {
                //查询锁定的账户
                List<User> users = userMapper.selectList(new LambdaQueryWrapper<User>().gt(User::getFundsReplenished, 0));
                for (int i = 0; i < users.size(); i++) {
                    User user = users.get(i);
                    if (user.getEnableAmt().compareTo(BigDecimal.ZERO) > 0) {
                        if (user.getEnableAmt().compareTo(user.getFundsReplenished()) >= 0) {
                            user.setEnableAmt(user.getEnableAmt().subtract(user.getFundsReplenished()));
                            user.setFundsReplenished(BigDecimal.ZERO);
//                            user.setIsLock(0);
                        } else {
                            user.setFundsReplenished(user.getFundsReplenished().subtract(user.getEnableAmt()));
                            user.setEnableAmt(BigDecimal.ZERO);
//                            user.setIsLock(1);
                        }
                        userMapper.updateById(user);
                    }
                }
            } catch (Exception e) {
                log.error("锁定账户待补资金扣钱定时任务发生异常", e);
            } finally {
                transferFundsLock.unlock();
                log.info("锁定账户待补资金扣钱定时任务--------->结束");
            }
        } else {
            log.info("锁定账户待补资金扣钱定时任务--------->上次任务还未执行完成，本次任务忽略");
        }
    }
}