79 lines
2.4 KiB
Java
79 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 java.util.concurrent.CountDownLatch;
|
|||
|
|
|
|||
|
|
public class IpLearningApplicationTest {
|
|||
|
|
|
|||
|
|
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();
|
|||
|
|
System.out.println("读取ArangoDb时间："+(lastA - startA));
|
|||
|
|
|
|||
|
|
// UpdateGraphsData.updateVFqdn();
|
|||
|
|
// UpdateGraphsData.updateVIp();
|
|||
|
|
// UpdateGraphsData.updateEFqdnAddressIp();
|
|||
|
|
// UpdateGraphsData.updateEIpVisitFqdn();
|
|||
|
|
|
|||
|
|
long startC = System.currentTimeMillis();
|
|||
|
|
CountDownLatch countDownLatch = new CountDownLatch(4);
|
|||
|
|
new Thread(new Runnable() {
|
|||
|
|
@Override
|
|||
|
|
public void run() {
|
|||
|
|
BaseClickhouseData.BaseVFqdn();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
new Thread(new Runnable() {
|
|||
|
|
@Override
|
|||
|
|
public void run() {
|
|||
|
|
BaseClickhouseData.BaseVIp();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
new Thread(new Runnable() {
|
|||
|
|
@Override
|
|||
|
|
public void run() {
|
|||
|
|
BaseClickhouseData.BaseEFqdnAddressIp();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}
|
|||
|
|
}).start();
|
|||
|
|
|
|||
|
|
new Thread(new Runnable() {
|
|||
|
|
@Override
|
|||
|
|
public void run() {
|
|||
|
|
BaseClickhouseData.BaseEIpVisitFqdn();
|
|||
|
|
countDownLatch.countDown();
|
|||
|
|
}
|
|||
|
|
}).start();
|
|||
|
|
try {
|
|||
|
|
countDownLatch.await();
|
|||
|
|
}catch (Exception e){
|
|||
|
|
e.printStackTrace();
|
|||
|
|
}
|
|||
|
|
long lastC = System.currentTimeMillis();
|
|||
|
|
System.out.println("更新ArangoDb时间："+(lastC - startC));
|
|||
|
|
|
|||
|
|
System.out.println(BaseArangoData.v_Fqdn_Map.size());
|
|||
|
|
System.out.println(BaseArangoData.v_Ip_Map.size());
|
|||
|
|
System.out.println(BaseArangoData.e_Fqdn_Address_Ip_Map.size());
|
|||
|
|
System.out.println(BaseArangoData.e_Ip_Visit_Fqdn_Map.size());
|
|||
|
|
|
|||
|
|
ArangoDBConnect.clean();
|
|||
|
|
}
|
|||
|
|
}
|