package com.nq.utils.task.stock;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
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.IUserStockSubscribeService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
import java.util.Optional;

@Component
@Slf4j
public class AdminStockSubscribeTask {

    boolean enableRun = false;
    @Autowired
    StockSubscribeMapper stockSubscribeMapper;
    @Autowired
    IUserStockSubscribeService iUserStockSubscribeService;
    @Autowired
    UserMapper userMapper;
    @Resource
    private UserStockSubscribeMapper userStockSubscribeMapper;

    @Scheduled(cron = "0 0/1 * * * *")
    public void run() {
        if (enableRun) return;

        try {
            log.info("【新股申购中签】::定时任务::开始执行...");
            enableRun = true;
            int limitNo = 0;
            while (true) {
                // 主查询
                List<UserStockSubscribe> list = findList(limitNo++);
                if (CollectionUtils.isEmpty(list)) return;
                for (UserStockSubscribe subscribe : list) {
                    try {
                        apply(subscribe);
                    } catch (Exception e) {
                        log.error("【新股申购中签】::执行错误::异常信息::{}", e.getMessage());
                    }
                }
            }

        } catch (Exception e) {
            log.error("【新股申购中签】::执行错误::终止执行...");
        } finally {
            log.info("【新股申购中签】::执行完成::结束任务...");
            enableRun = false;
        }
    }

    public List<UserStockSubscribe> findList(int limitNo) {
        int limit = 10;
        return userStockSubscribeMapper.selectList(new LambdaQueryWrapper<UserStockSubscribe>()
                .ne(UserStockSubscribe::getStatus, 3)
                        .isNotNull(UserStockSubscribe::getApplyLastNumber)
                .last(" limit " + limitNo * limit + "," + limit + "  "));
    }

    private void apply(UserStockSubscribe model) {
        // 查询
        User user = userMapper.selectByPrimaryKey(model.getUserId());
        StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<>(new StockSubscribe()).eq("code", model.getNewCode()));
        if (model.getApplyLastNumber() == null || model.getApplyLastNumber() == 0) return;

        Calendar c = Calendar.getInstance();
        if (model.getEndTime() != null) {
            c.setTime(model.getEndTime());
            //增加一天
            c.add(Calendar.DAY_OF_MONTH, 1);
        }

        Date date = Optional.ofNullable(model.getPushTime()).orElse(
                Optional.ofNullable(stockSubscribe.getPushTime()).orElse(model.getEndTime() == null ? null : c.getTime())
        );

        if (date != null && date.before(new Date()))
            // 更新
            iUserStockSubscribeService.edit(model, stockSubscribe, user);
    }
}