package cc.mrbird.febs.pay.util;

import cc.mrbird.febs.common.entity.FebsResponse;
import lombok.extern.slf4j.Slf4j;

import java.util.HashMap;
import java.util.Map;

@Slf4j
public class PayThreadPool {

    private static Map<Integer, Object> PAY_WAIT_POOL = new HashMap<>();
    private static Map<Integer, FebsResponse> PAY_THREAD_MSG = new HashMap<>();

    public static void waitThread(Integer orderId, Object lock) {
        PAY_WAIT_POOL.put(orderId, lock);
        synchronized (lock) {
            try {
                // 设置线程最多等待15s
                lock.wait(1000 * 15);
            } catch (InterruptedException e) {
                log.debug("线程等待失败", e);
            }
        }

    }

    public static FebsResponse getThreadResult(Integer orderId) {
        //获取一次后删除结果对象
		// TODO 需要优化可能存在对象积压过多内存溢出风险
        return PAY_THREAD_MSG.remove(orderId);
    }

    public static void notifyThread(Integer orderId, FebsResponse result) {
        Object lock = PAY_WAIT_POOL.get(orderId);
        if (lock != null) {
            synchronized (lock) {
                log.debug("尝试释放锁{}", orderId);
                PAY_THREAD_MSG.put(orderId, result);
                PAY_WAIT_POOL.remove(orderId);
                lock.notifyAll();
            }
        }
    }

}