From 4c277754cc6639c63bcc9fa1939bbda4103ad54f Mon Sep 17 00:00:00 2001
From: tiger_zhou <Shmily123@WJ>
Date: Mon, 4 Sep 2023 16:30:11 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../message/collect/convertor/TrainRecordConvertor.java   | 1 -
 .../xiannccda/ats/warn/TrainReacrdAlertListener.java      | 8 +++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/main/java/club/joylink/xiannccda/ats/message/collect/convertor/TrainRecordConvertor.java b/src/main/java/club/joylink/xiannccda/ats/message/collect/convertor/TrainRecordConvertor.java
index b51f512..0e990c3 100644
--- a/src/main/java/club/joylink/xiannccda/ats/message/collect/convertor/TrainRecordConvertor.java
+++ b/src/main/java/club/joylink/xiannccda/ats/message/collect/convertor/TrainRecordConvertor.java
@@ -3,7 +3,6 @@ package club.joylink.xiannccda.ats.message.collect.convertor;
 import club.joylink.xiannccda.alert.core.AlertManager;
 import club.joylink.xiannccda.ats.message.MessageId;
 import club.joylink.xiannccda.ats.message.collect.DeviceDataRepository.DataTypeEnum;
-import club.joylink.xiannccda.ats.message.line3.rep.TrainRecordResponse;
 import club.joylink.xiannccda.ats.warn.TrainRecordAlertEvent;
 import club.joylink.xiannccda.dto.protos.TrainProto.TrainRecord;
 import com.google.protobuf.GeneratedMessageV3.Builder;
diff --git a/src/main/java/club/joylink/xiannccda/ats/warn/TrainReacrdAlertListener.java b/src/main/java/club/joylink/xiannccda/ats/warn/TrainReacrdAlertListener.java
index 96889bd..f44316f 100644
--- a/src/main/java/club/joylink/xiannccda/ats/warn/TrainReacrdAlertListener.java
+++ b/src/main/java/club/joylink/xiannccda/ats/warn/TrainReacrdAlertListener.java
@@ -19,9 +19,11 @@ public class TrainReacrdAlertListener implements AlertSourceEventListener<TrainR
   public void accept(TrainRecordAlertEvent event) {
     TrainRecord.Builder trainRecord = event.getSource();
     if (trainRecord.getTrainType() == TrainTypeEnum.PlannedTrain.getType()) {
-      this.platformAlertMonitoringTask.putTrainRecord(trainRecord);
-    } else {
-      this.platformAlertMonitoringTask.remoceTrainRecord(trainRecord);
+      if (trainRecord.getRecordType()) {
+        this.platformAlertMonitoringTask.putTrainRecord(trainRecord);
+      } else {
+        this.platformAlertMonitoringTask.remoceTrainRecord(trainRecord);
+      }
     }
   }
 }