76 lines
2.4 KiB
Java
76 lines
2.4 KiB
Java
|
package cn.ac.iie.test;
|
|||
|
|
|
|||
|
|
import cn.ac.iie.dao.BaseArangoData;
|
|||
|
|
import cn.ac.iie.dao.BaseClickhouseData;
|
|||
|
|
import cn.ac.iie.utils.ArangoDBConnect;
|
|||
|
|
import cn.ac.iie.utils.ExecutorThreadPool;
|
|||
|
|
import org.slf4j.Logger;
|
|||
|
|
import org.slf4j.LoggerFactory;
|
|||
|
|
|
|||
|
|
import java.util.concurrent.CountDownLatch;
|
|||
|
|
|
|||
|
|
public class IpLearningApplicationTest {
|
|||
|
|
private static final Logger LOG = LoggerFactory.getLogger(IpLearningApplicationTest.class);
|
|||
|
|
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
|
|||
|
|
long startA = System.currentTimeMillis();
|
|||
|
|
BaseArangoData.BaseVFqdnDataMap();
|
|||
|
|
BaseArangoData.BaseVIpDataMap();
|
|||
|
|
BaseArangoData.BaseEFqdnAddressIpDataMap();
|
|||
|
|
BaseArangoData.BaseEIpVisitFqdnDataMap();
|
|||
|
|
|
|||
|
|
|
|||
|
|
ExecutorThreadPool.shutdown();
|
|||
|
|
ExecutorThreadPool.awaitThreadTask();
|
|||
|
|
long lastA = System.currentTimeMillis();
|
|||
|
|
LOG.info("读取ArangoDb时间："+(lastA - startA));
|
|||
|
|
|
|||
|
|
long startC = System.currentTimeMillis();
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
CountDownLatch countDownLatch = new CountDownLatch(4);
|
|||
|
|
|
|||
|
|
new Thread(() -> {
|
|||
|
|
BaseClickhouseData.BaseVFqdn();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
new Thread(() -> {
|
|||
|
|
BaseClickhouseData.BaseVIp();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
new Thread(() -> {
|
|||
|
|
BaseClickhouseData.BaseEFqdnAddressIp();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
new Thread(() -> {
|
|||
|
|
BaseClickhouseData.BaseEIpVisitFqdn();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
countDownLatch.await();
|
|||
|
|
LOG.info("主线程等待完毕");
|
|||
|
|
}catch (Exception e){
|
|||
|
|
LOG.error("主线程阻塞异常：\n"+e.toString());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
long lastC = System.currentTimeMillis();
|
|||
|
|
LOG.info("更新ArangoDb时间："+(lastC - startC));
|
|||
|
|
}catch (Exception e){
|
|||
|
|
e.printStackTrace();
|
|||
|
|
}finally {
|
|||
|
|
ArangoDBConnect.clean();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
LOG.info("v_Fqdn_Map大小："+BaseArangoData.v_Fqdn_Map.size());
|
|||
|
|
LOG.info("v_Ip_Map大小："+BaseArangoData.v_Ip_Map.size());
|
|||
|
|
LOG.info("e_Fqdn_Address_Ip_Map大小："+BaseArangoData.e_Fqdn_Address_Ip_Map.size());
|
|||
|
|
LOG.info("e_Ip_Visit_Fqdn_Map大小："+BaseArangoData.e_Ip_Visit_Fqdn_Map.size());
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|