package com.sandu.ximon.admin.utils;

import lombok.Data;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CountDownLatch;

/**
 * @author LiuHaoNan
 * @date 2022/9/6
 */
@Configuration
public class CountDownLatchUtil {
    private static Map<String, CountDownLatch> data = new HashMap<>();

    public void countDown(String id) {
        if (data != null && data.get(id) != null) {
            data.get(id).countDown();
        }
    }

    public void push(String id, CountDownLatch countDownLatch) {
        if (data != null) {
            data.put(id, countDownLatch);
        }
    }

    public void remove(String id) {
        if (data != null) {
            data.remove(id);
        }
    }
}