From da60653963f4afb69881aa6b7ee3e12185999c48 Mon Sep 17 00:00:00 2001
From: wzp <minecraftwzpmc@163.com>
Date: Thu, 19 Dec 2024 12:41:12 +0800
Subject: [PATCH] fix: cannot reset room

---
 src/main/java/org/mmga/clubs/entities/chess/Room.java | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/mmga/clubs/entities/chess/Room.java b/src/main/java/org/mmga/clubs/entities/chess/Room.java
index ec1a5c1..8122188 100644
--- a/src/main/java/org/mmga/clubs/entities/chess/Room.java
+++ b/src/main/java/org/mmga/clubs/entities/chess/Room.java
@@ -156,12 +156,15 @@ public class Room {
 
     public void requestRestart(Session session) {
         String sessionId = session.getId();
-        if (sessionId.equals(whiteSession.getId())) {
+        if (whiteSession == null || sessionId.equals(whiteSession.getId())) {
             isWhiteAcceptRestart = true;
         }
-        if (sessionId.equals(blackSession.getId())) {
+        if (blackSession == null || sessionId.equals(blackSession.getId())) {
             isBlackAcceptRestart = true;
         }
+        if (isBlackAcceptRestart && isWhiteAcceptRestart) {
+            this.resetPieces();
+        }
     }
 
     public void replace(String existsUser, Session session) {