gfwleak/galaxy-tsg-olap-generate-baseline
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

多线程批量进行baseline生成和写入

Browse Source
This commit is contained in:
yinjiangyi
2021-08-01 18:24:22 +08:00
parent bff209ac5a
commit f1e243ded0
18 changed files with 1375 additions and 285 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
src/main/java/cn/mesalab/dao/DruidData.java
View File

@@ -6,7 +6,6 @@ import cn.mesalab.utils.DruidUtils;
import io.vavr.Tuple;
import io.vavr.Tuple2;
import org.apache.calcite.avatica.AvaticaConnection;
import org.apache.calcite.avatica.AvaticaStatement;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -28,34 +27,39 @@ public class DruidData {
private static final Logger LOG = LoggerFactory.getLogger(DruidData.class);
private static DruidData druidData;
private static DruidUtils druidUtils;
private AvaticaConnection connection;
{
try {
connection = DruidUtils.getConn();
} catch (SQLException exception) {
exception.printStackTrace();
}
}
private String timeFilter = ApplicationConfig.DRUID_RECVTIME_COLUMN_NAME
+ " >= MILLIS_TO_TIMESTAMP(" + getTimeLimit()._2
+ ") AND " + ApplicationConfig.DRUID_RECVTIME_COLUMN_NAME
+ " < MILLIS_TO_TIMESTAMP(" + getTimeLimit()._1 + ")";
static {
druidUtils = DruidUtils.getInstance();
}
public static DruidData getInstance() {
if (druidData == null){
druidData = new DruidData();
}
druidData = new DruidData();
return druidData;
}
public ArrayList<String> getServerIpList(String attackType) {
Long startQueryIPLIstTime = System.currentTimeMillis();
ArrayList<String> serverIPs = new ArrayList<String>();
String sql = "SELECT distinct " + ApplicationConfig.DRUID_SERVERIP_COLUMN_NAME
+ " FROM " + ApplicationConfig.DRUID_TABLE
+ " WHERE " + ApplicationConfig.DRUID_ATTACKTYPE_COLUMN_NAME + " = '" + attackType + "'"
+ " AND " + timeFilter
+ " LIMIT 20"; // FOR TEST
+ " LIMIT 100"; // FOR TEST
try{
ResultSet resultSet = druidUtils.executeQuery(sql);
ResultSet resultSet = DruidUtils.executeQuery(connection,sql);
while(resultSet.next()){
String ip = resultSet.getString(ApplicationConfig.DRUID_SERVERIP_COLUMN_NAME);
serverIPs.add(ip);
@@ -63,6 +67,9 @@ public class DruidData {
} catch (Exception e){
e.printStackTrace();
}
Long endQueryIPListTime = System.currentTimeMillis();
LOG.info("性能测试：ip list查询耗时——"+(endQueryIPListTime-startQueryIPLIstTime));
return serverIPs;
}
@@ -84,9 +91,9 @@ public class DruidData {
+ " = '" + attackType + "'"
+ " AND " + timeFilter;
System.out.println("getTimeSeriesData:" + sql);
// System.out.println("getTimeSeriesData:" + sql);
try{
ResultSet resultSet = druidUtils.executeQuery(sql);
ResultSet resultSet = DruidUtils.executeQuery(connection,sql);
ResultSetToListService service = new ResultSetToListServiceImp();
rsList = service.selectAll(resultSet);
} catch (Exception e){
@@ -129,6 +136,10 @@ public class DruidData {
}
public void closeConn(){
druidUtils.closeConnection();
try {
DruidUtils.closeConnection();
} catch (SQLException exception) {
exception.printStackTrace();
}
}
}

29
src/main/java/cn/mesalab/service/BaselineGeneration.java
View File

@@ -42,9 +42,9 @@ public class BaselineGeneration {
try{
generateBaselinesThread(ApplicationConfig.DRUID_ATTACKTYPE_TCP_SYN_FLOOD);
//generateBaselines(ApplicationConfig.DRUID_ATTACKTYPE_UDP_FLOOD);
//generateBaselines(ApplicationConfig.DRUID_ATTACKTYPE_ICMP_FLOOD);
//generateBaselines(ApplicationConfig.DRUID_ATTACKTYPE_DNS_AMPL);
// generateBaselines(ApplicationConfig.DRUID_ATTACKTYPE_UDP_FLOOD);
// generateBaselines(ApplicationConfig.DRUID_ATTACKTYPE_ICMP_FLOOD);
// generateBaselines(ApplicationConfig.DRUID_ATTACKTYPE_DNS_AMPL);
long last = System.currentTimeMillis();
LOG.warn("运行时间:" + (last - start));
@@ -102,25 +102,39 @@ public class BaselineGeneration {
}
public static void generateBaselines(List<String> ipList, String attackType){
Long startGenerateTime = System.currentTimeMillis();
druidData = DruidData.getInstance();
List<Put> putList = new ArrayList<>();
for(String ip: ipList){
int[] ipBaseline = generateSingleIpBaseline(ip, attackType);
putList = hbaseUtils.cachedInPut(putList, ip, ipBaseline, attackType, ApplicationConfig.BASELINE_METRIC_TYPE);
}
Long endGenerateTime = System.currentTimeMillis();
// LOG.info("性能测试：Baseline生成耗时——"+(endGenerateTime-startGenerateTime));
try {
hbaseTable.put(putList);
LOG.info("HBase 写入数据条数 " + ApplicationConfig.GENERATE_BATCH_NUM);
} catch (IOException e) {
e.printStackTrace();
}
Long endWriteTime = System.currentTimeMillis();
// LOG.info("性能测试：Baseline写入耗时——"+(endWriteTime-endGenerateTime));
}
private static int[] generateSingleIpBaseline(String ip, String attackType){
// 查询
Long startQuerySingleIPTime = System.currentTimeMillis();
List<Map<String, Object>> originSeries = druidData.getTimeSeriesData(ip, attackType);
Long endQuerySingleIPTime = System.currentTimeMillis();
LOG.info("性能测试：单个ip查询耗时——"+(endQuerySingleIPTime-startQuerySingleIPTime));
// 时间序列缺失值补0
System.out.println("当前线程id"+Thread.currentThread().getId());
System.out.println("origin 大小"+originSeries.size());
List<Map<String, Object>> completSeries = SeriesUtils.complementSeries(originSeries);
int[] baselineArr = new int[completSeries.size()];
@@ -148,8 +162,9 @@ public class BaselineGeneration {
}
}
System.out.println(ip);
System.out.println(Arrays.toString(baselineArr));
Long endGenerateSingleIPTime = System.currentTimeMillis();
LOG.info("性能测试：单个baseline生成耗时——"+(endGenerateSingleIPTime-endQuerySingleIPTime));
return baselineArr;
}

154
src/main/java/cn/mesalab/utils/ConcurrentUtils.java
View File

@@ -1,154 +0,0 @@
package cn.mesalab.utils;
import com.google.common.collect.Lists;
import org.springframework.core.task.AsyncTaskExecutor;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Function;
/**
* 〈一句话功能简述〉:
* 〈并发工具〉
*
* @create 2020-05-28
* @since 1.0.0
*/
public class ConcurrentUtils {
/**
* @Description
* 并发执行工具类
*
* @param batchProcessNum 并发执行的个数
* @param executor 执行需要的线程池
* @param keys 批量执行的key
* @param kvConvert 从key生成value
* @param valueProcessor 对value的后置处理
* @param exceptionHook 通过key获取value时发生异常自定义处理
* @return void
**/
public static <K, V> void concurrentProcess(int batchProcessNum, AsyncTaskExecutor executor, K[] keys,
Function<K, V> kvConvert,
ValueProcessor<K, V> valueProcessor,
ExceptionHook<K> exceptionHook ){
AtomicInteger index = new AtomicInteger(1);
List<Future> futureList = Lists.newArrayListWithExpectedSize(batchProcessNum);
final int length = keys.length;
System.out.println("total jobs size:{}" + length);
for (int i = 0; i < batchProcessNum; i++) {
int finalI = i;
futureList.add(executor.submit(() -> {
System.out.println("batch process start thread num:{}"+ finalI);
int currentIndex;
while (length >= (currentIndex = index.getAndIncrement())) {
System.out.println("current job index:"+currentIndex+" of "+ length);
K key = keys[currentIndex-1];
try {
valueProcessor.process(key, kvConvert.apply(key));
} catch (Exception e) {
exceptionHook.process(key, e);
}
}
System.out.println("batch process end thread num:{}"+finalI);
}));
}
waitFutureFinished(futureList);
}
public static <K, V> void concurrentProcess(int batchProcessNum, AsyncTaskExecutor executor, K[] keys,
Function<K, V> kvConvert,
ValueProcessor<K, V> valueProcessor) {
concurrentProcess(batchProcessNum, executor, keys, kvConvert, valueProcessor,
(key,e) -> System.out.println("通过key:"+ key +" 获取value异常, e:"+e));
}
/**
* @Description
* 并发执行工具类
*
* @param batchProcessNum 并发执行的个数
* @param executor 执行需要的线程池
* @param keys 批量执行的key
* @param kvConvert 从key生成value
* @return Collection<V> 返回value的集合
**/
public static <K, V> Collection<V> concurrentGet(int batchProcessNum, AsyncTaskExecutor executor, K[] keys,
Function<K, V> kvConvert) {
List<V> rt = Lists.newArrayListWithCapacity(keys.length);
concurrentProcess(batchProcessNum, executor, keys, kvConvert, (k, v) -> {
if (v == null) {
System.out.println("key:{} apply value is null");
return;
}
rt.add(v);
});
return rt;
}
public static void waitFutureFinished(List<Future> unfinishedFuture, boolean ignoreExcetion) {
boolean interrupt = false;
while (!unfinishedFuture.isEmpty()) {
Iterator<Future> iterator = unfinishedFuture.iterator();
while (iterator.hasNext()) {
Future next = iterator.next();
if (next.isDone()) {
try {
next.get();
} catch (InterruptedException | ExecutionException e) {
System.out.println("执行异常， e:"+e);
if (!ignoreExcetion) {
interrupt = true;
break;
}
}
iterator.remove();
}
}
if (interrupt) {
break;
}
sleep();
}
if (interrupt) {
Iterator<Future> iterator = unfinishedFuture.iterator();
while (iterator.hasNext()) {
Future next = iterator.next();
if (next.isDone()) {
next.cancel(true);
}
}
throw new RuntimeException("任务异常终止");
}
}
public static void waitFutureFinished(List<Future> unfinishedFuture) {
waitFutureFinished(unfinishedFuture, false);
}
public static void sleep() {
sleep(5000);
}
public static void sleep(long millis) {
try {
Thread.sleep(millis);
} catch (InterruptedException e) {
System.out.println("sleep error, e:" + e);
}
}
@FunctionalInterface
public interface ValueProcessor<K, V> {
void process(K key, V value);
}
@FunctionalInterface
public interface ExceptionHook<K>{
void process(K key, Exception e);
}
}

65
src/main/java/cn/mesalab/utils/DruidUtils.java
View File

@@ -1,12 +1,9 @@
package cn.mesalab.utils;
import cn.mesalab.config.ApplicationConfig;
import cn.mesalab.service.BaselineGeneration;
import org.apache.calcite.avatica.AvaticaConnection;
import org.apache.calcite.avatica.AvaticaStatement;
import org.jfree.util.Log;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.hadoop.hbase.client.Table;
import java.sql.DriverManager;
import java.sql.ResultSet;
@@ -19,59 +16,39 @@ import java.util.Properties;
* @date 2021/7/23 4:50 下午
*/
public class DruidUtils {
private static final Logger LOG = LoggerFactory.getLogger(DruidUtils.class);
private static ThreadLocal<AvaticaConnection> threadLocal = new ThreadLocal<AvaticaConnection>();
private static DruidUtils druidUtils;
private static final String DRUID_URL = ApplicationConfig.DRUID_URL;
private static AvaticaConnection connection;
private static AvaticaStatement statement;
public DruidUtils() throws SQLException {
Properties properties = new Properties();
connection = (AvaticaConnection) DriverManager.getConnection(DRUID_URL, properties);
statement = connection.createStatement();
}
public static DruidUtils getInstance() {
if (connection==null){
try{
druidUtils = new DruidUtils();
} catch (SQLException e){
LOG.error("Druid 建立连接失败！");
e.printStackTrace();
}
}
return druidUtils;
}
private static AvaticaConnection getConn() throws SQLException {
/**
* 打开连接
* @throws SQLException
*/
public static AvaticaConnection getConn() throws SQLException {
Properties properties = new Properties();
AvaticaConnection connection = (AvaticaConnection) DriverManager.getConnection(DRUID_URL, properties);
threadLocal.set(connection);
return connection;
}
public void closeConnection() {
if(connection != null){
try{
connection.close();
} catch (SQLException e){
LOG.error("Druid 关闭连接失败！");
e.printStackTrace();
}
/**
* 关闭连接
*/
public static void closeConnection() throws SQLException{
AvaticaConnection conn = threadLocal.get();
if(conn != null){
conn.close();
threadLocal.remove();
}
}
public ResultSet executeQuery (String sql) throws SQLException{
System.out.println("executeQuery:"+sql);
/**
* 根据sql查询结果
*/
public static ResultSet executeQuery (AvaticaConnection connection, String sql) throws SQLException{
AvaticaStatement statement = connection.createStatement();
ResultSet resultSet = statement.executeQuery(sql);
return resultSet;
}
public AvaticaConnection getConnection() {
return connection;
}
public AvaticaStatement getStatement() {
return statement;
}
}

2
src/main/java/cn/mesalab/utils/SeriesUtils.java
View File

@@ -143,8 +143,6 @@ public class SeriesUtils {
if(denominator>0) {
result = numerator / denominator;
}
//FOR TEST
System.out.println(result);
return result;
}