Добавлено сохранение tcp пакетов по таймауту

This commit is contained in:
serega6531
2019-05-14 01:44:23 +03:00
parent adba3aa18c
commit 6bbfa49b7a
4 changed files with 48 additions and 36 deletions

View File

@@ -0,0 +1,42 @@
package ru.serega6531.packmate;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import ru.serega6531.packmate.model.Protocol;
import java.util.concurrent.TimeUnit;
@Component
@Slf4j
public class TimeoutStreamsSaver {
private final PcapWorker pcapWorker;
private final long udpStreamTimeoutMillis;
private final long tcpStreamTimeoutMillis;
@Autowired
public TimeoutStreamsSaver(PcapWorker pcapWorker,
@Value("${udp-stream-timeout}") int udpStreamTimeout,
@Value("${tcp-stream-timeout}") int tcpStreamTimeout) {
this.pcapWorker = pcapWorker;
this.udpStreamTimeoutMillis = TimeUnit.SECONDS.toMillis(udpStreamTimeout);
this.tcpStreamTimeoutMillis = TimeUnit.SECONDS.toMillis(tcpStreamTimeout);
}
@Scheduled(fixedRateString = "PT${timeout-stream-check-interval}S", initialDelayString = "PT${timeout-stream-check-interval}S")
public void saveStreams() {
int streamsClosed = pcapWorker.closeTimeoutStreams(Protocol.UDP, udpStreamTimeoutMillis);
if(streamsClosed > 0) {
log.info("Закрыто {} udp стримов", streamsClosed);
}
streamsClosed = pcapWorker.closeTimeoutStreams(Protocol.TCP, tcpStreamTimeoutMillis);
if(streamsClosed > 0) {
log.info("Закрыто {} tcp стримов", streamsClosed);
}
}
}