Files
hertzbeat/alerter/src/main/java/com/usthe/alert/AlerterDataQueue.java
2022-01-29 17:11:22 +08:00

34 lines
734 B
Java

package com.usthe.alert;
import com.usthe.alert.pojo.entity.Alert;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
/**
* 采集数据队列
* @author tom
* @date 2021/11/24 17:58
*/
@Component
@Slf4j
public class AlerterDataQueue {
private final LinkedBlockingQueue<Alert> alertDataQueue;
public AlerterDataQueue() {
alertDataQueue = new LinkedBlockingQueue<>();
}
public void addAlertData(Alert alert) {
alertDataQueue.offer(alert);
}
public Alert pollAlertData() throws InterruptedException {
return alertDataQueue.poll(2, TimeUnit.SECONDS);
}
}