package org.example.timedTask;

import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONArray;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.google.gson.Gson;
import lombok.extern.slf4j.Slf4j;
import org.example.dao.StockNewShareMapper;
import org.example.enums.EStockType;
import org.example.pojo.StockNewShare;
import org.example.util.HttpClientRequest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @program: webSocketProject
 * @description: 新股
 * @create: 2024-03-27 13:14
 **/
@Component
@Slf4j
public class NewShareTask {

    private final Lock lock = new ReentrantLock();

    @Autowired
    StockNewShareMapper stockNewShareMapper;

    /*
     * ipo、新股日历抓取
     * */
    @Scheduled(cron = "0 0/1 * * * ? ")
    public void get() {
        if (lock.tryLock()) {
            log.info("ipo、新股日历抓取--------->开始");
            try {
                newShare(EStockType.IN);
            } finally {
                lock.unlock();
                log.info("ipo、新股日历抓取--------->结束");
            }
        } else {
            log.info("ipo、新股日历抓取--------->上次任务还未执行完成，本次任务忽略");
        }
    }

    public void newShare(EStockType e) {
        String result = HttpClientRequest.doGet(e.stockUrl + "new-stock?country_id=" + e.getContryId() + "&key=" + e.stockKey);
        JSONArray jsonArray = JSONArray.parseArray(result);

        List<StockNewShare> list = jsonArray.stream().map(stock -> JSONUtil.toBean(stock.toString(), StockNewShare.class)).collect(Collectors.toList());
        List<String> pidList = list.stream().map(StockNewShare::getPid).collect(Collectors.toList());
        List<StockNewShare> shareList = stockNewShareMapper.selectList(new LambdaQueryWrapper<StockNewShare>().in(StockNewShare::getPid, pidList));
        Map<String, StockNewShare> resultMap = shareList.stream()
                .collect(Collectors.toMap(StockNewShare::getPid, Function.identity()));
        list.forEach(f -> {
            StockNewShare share = resultMap.get(f.getPid());
            if (share == null) {
                stockNewShareMapper.insert(f);
            } else {
                f.setId(share.getId());
                stockNewShareMapper.updateById(f);
            }
        });

    }

}