Initial commit 单线程成功，并行报错

2021-08-01 17:28:31 +08:00
commit bff209ac5a
46 changed files with 3994 additions and 0 deletions
--- a/src/main/java/cn/mesalab/dao/DruidData.java
+++ b/src/main/java/cn/mesalab/dao/DruidData.java
@@ -0,0 +1,134 @@
+package cn.mesalab.dao;
+
+import cn.mesalab.config.ApplicationConfig;
+import cn.mesalab.dao.Impl.ResultSetToListServiceImp;
+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;
+
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.Calendar;
+import java.util.List;
+import java.util.Map;
+
+
+/**
+ * @author yjy
+ * @version 1.0
+ * @date 2021/7/23 4:56 下午
+ */
+public class DruidData {
+
+    private static final Logger LOG = LoggerFactory.getLogger(DruidData.class);
+    private static DruidData druidData;
+
+    private static DruidUtils druidUtils;
+
+    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();
+        }
+        return druidData;
+    }
+
+    public ArrayList<String> getServerIpList(String attackType) {
+        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
+        try{
+            ResultSet resultSet = druidUtils.executeQuery(sql);
+            while(resultSet.next()){
+                String ip = resultSet.getString(ApplicationConfig.DRUID_SERVERIP_COLUMN_NAME);
+                serverIPs.add(ip);
+            }
+        } catch (Exception e){
+            e.printStackTrace();
+        }
+        return serverIPs;
+    }
+
+    public ArrayList<String> getServerIpList(String attackType, String test) {
+        ArrayList<String> serverIPs = new ArrayList<String>();
+        serverIPs.add("153.99.250.54");
+        return serverIPs;
+    }
+
+    public List<Map<String, Object>> getTimeSeriesData(String ip, String attackType){
+        List<Map<String, Object>> rsList = null;
+        String sql = "SELECT "+ ApplicationConfig.DRUID_SERVERIP_COLUMN_NAME
+                + ", "+ ApplicationConfig.BASELINE_METRIC_TYPE
+                + ", " + ApplicationConfig.DRUID_RECVTIME_COLUMN_NAME
+                + " FROM " + ApplicationConfig.DRUID_TABLE
+                + " WHERE " + ApplicationConfig.DRUID_SERVERIP_COLUMN_NAME
+                + " = '" + ip + "'"
+                + " AND " + ApplicationConfig.DRUID_ATTACKTYPE_COLUMN_NAME
+                + " = '" + attackType + "'"
+                + " AND " + timeFilter;
+
+        System.out.println("getTimeSeriesData:" + sql);
+        try{
+            ResultSet resultSet = druidUtils.executeQuery(sql);
+            ResultSetToListService service  = new ResultSetToListServiceImp();
+            rsList = service.selectAll(resultSet);
+        } catch (Exception e){
+            e.printStackTrace();
+        }
+        return rsList;
+    }
+
+    public Tuple2<Long, Long> getTimeLimit(){
+        long maxTime = 0L;
+        long minTime = 0L;
+        switch(ApplicationConfig.DRUID_TIME_LIMIT_TYPE){
+            case 0:
+                maxTime = getCurrentDay();
+                minTime = getCurrentDay(-ApplicationConfig.READ_HISTORICAL_DAYS);
+                break;
+            case 1:
+                maxTime = ApplicationConfig.READ_DRUID_MAX_TIME;
+                minTime = ApplicationConfig.READ_DRUID_MIN_TIME;
+                break;
+            default:
+                LOG.warn("没有设置Druid数据读取方式");
+        }
+        return Tuple.of(maxTime, minTime);
+    }
+
+    private long getCurrentDay(int bias) {
+        Calendar calendar = Calendar.getInstance();
+        calendar.set(Calendar.DAY_OF_YEAR, calendar.get(Calendar.DAY_OF_YEAR) + bias);
+        calendar.set(Calendar.HOUR_OF_DAY, 0);
+        calendar.set(Calendar.HOUR_OF_DAY, 0);
+        calendar.set(Calendar.MINUTE, 0);
+        calendar.set(Calendar.SECOND, 0);
+        calendar.set(Calendar.MILLISECOND, 0);
+        return calendar.getTimeInMillis();
+    }
+
+    private long getCurrentDay(){
+        return getCurrentDay(0);
+    }
+
+    public void closeConn(){
+        druidUtils.closeConnection();
+    }
+}