From 830b0fbd2f5a5eeb80c33f3f9af4056ec7c48d61 Mon Sep 17 00:00:00 2001
From: wanglihui <949764788@qq.com>
Date: Mon, 2 Aug 2021 11:50:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=20attack=5Ftype=E4=B8=8Edest?=
 =?UTF-8?q?ination=5Fip=E5=AD=97=E6=AE=B5=E5=86=99=E5=8F=8Dbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/zdjizhi/etl/EtlProcessFunction.java |  2 +-
 .../java/com/zdjizhi/etl/TrafficServerIpMetrics.java  | 11 +++++------
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/zdjizhi/etl/EtlProcessFunction.java b/src/main/java/com/zdjizhi/etl/EtlProcessFunction.java
index 987d26a..1fb0985 100644
--- a/src/main/java/com/zdjizhi/etl/EtlProcessFunction.java
+++ b/src/main/java/com/zdjizhi/etl/EtlProcessFunction.java
@@ -30,7 +30,7 @@ public class EtlProcessFunction extends ProcessWindowFunction<DosSketchLog, DosS
             if (middleResult != null){
                 out.collect(middleResult);
                 logger.info("获取中间聚合结果：{}",middleResult.toString());
-                context.output(outputTag,TrafficServerIpMetrics.getOutputMetric(keys, middleResult));
+                context.output(outputTag,TrafficServerIpMetrics.getOutputMetric(middleResult));
             }
         }catch (Exception e){
             logger.error("获取中间聚合结果失败,middleResult: {}\n{}",middleResult.toString(),e);
diff --git a/src/main/java/com/zdjizhi/etl/TrafficServerIpMetrics.java b/src/main/java/com/zdjizhi/etl/TrafficServerIpMetrics.java
index a43314b..1afd244 100644
--- a/src/main/java/com/zdjizhi/etl/TrafficServerIpMetrics.java
+++ b/src/main/java/com/zdjizhi/etl/TrafficServerIpMetrics.java
@@ -3,7 +3,6 @@ package com.zdjizhi.etl;
 import com.zdjizhi.common.CommonConfig;
 import com.zdjizhi.common.DosMetricsLog;
 import com.zdjizhi.common.DosSketchLog;
-import org.apache.flink.api.java.tuple.Tuple4;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -12,13 +11,13 @@ class TrafficServerIpMetrics {
 
     private static final Logger logger = LoggerFactory.getLogger(TrafficServerIpMetrics.class);
 
-    static DosMetricsLog getOutputMetric(Tuple4<String, String, String, String> keys, DosSketchLog midResuleLog) {
+    static DosMetricsLog getOutputMetric(DosSketchLog midResuleLog) {
         DosMetricsLog dosMetricsLog = new DosMetricsLog();
         dosMetricsLog.setSketch_start_time(timeFloor(System.currentTimeMillis()/1000));
-        dosMetricsLog.setCommon_sled_ip(keys.f0);
-        dosMetricsLog.setCommon_data_center(keys.f1);
-        dosMetricsLog.setDestination_ip(keys.f2);
-        dosMetricsLog.setAttack_type(keys.f3);
+        dosMetricsLog.setCommon_sled_ip(midResuleLog.getCommon_sled_ip());
+        dosMetricsLog.setCommon_data_center(midResuleLog.getCommon_data_center());
+        dosMetricsLog.setDestination_ip(midResuleLog.getDestination_ip());
+        dosMetricsLog.setAttack_type(midResuleLog.getAttack_type());
         dosMetricsLog.setSession_rate(midResuleLog.getSketch_sessions());
         dosMetricsLog.setPacket_rate(midResuleLog.getSketch_packets());
         dosMetricsLog.setBit_rate(midResuleLog.getSketch_bytes());