From 2ac0174a008cdc2c2df29aa51277a7cbcad49ac6 Mon Sep 17 00:00:00 2001
From: EnderByEndera <707475564@qq.com>
Date: Wed, 24 Apr 2024 14:41:03 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B7=B2=E7=BB=93=E6=9D=9F=E7=8A=B6=E6=80=81?=
 =?UTF-8?q?=E7=9A=84=E4=BB=BB=E5=8A=A1=E5=8F=AF=E4=BB=A5=E8=BD=AC=E6=8D=A2?=
 =?UTF-8?q?=E5=9B=9E=E6=9C=AA=E5=90=AF=E5=8A=A8=E7=8A=B6=E6=80=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../server/task/status/states/FinishedState.java         | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/realtime/protection/server/task/status/states/FinishedState.java b/src/main/java/com/realtime/protection/server/task/status/states/FinishedState.java
index 757577f..06d9d9a 100644
--- a/src/main/java/com/realtime/protection/server/task/status/states/FinishedState.java
+++ b/src/main/java/com/realtime/protection/server/task/status/states/FinishedState.java
@@ -1,11 +1,16 @@
 package com.realtime.protection.server.task.status.states;
 
+import com.realtime.protection.configuration.utils.enums.StateEnum;
 import com.realtime.protection.server.command.CommandService;
 import com.realtime.protection.server.task.TaskService;
+import com.realtime.protection.server.task.status.StateHandler;
 
-public class FinishedState implements State {
+public class FinishedState extends StateHandler implements State {
     @Override
     public Boolean handle(State newState, CommandService commandService, TaskService taskService, Long taskId) {
-        return newState instanceof FinishedState;
+        return switch (StateEnum.getStateEnumByState(newState)) {
+            case PENDING, FINISHED -> true;
+            default -> throw new IllegalStateException("Unexpected value: " + StateEnum.getStateEnumByState(newState));
+        };
     }
 }