From a9c9983876234559cf1c88855603f6607bf382ea Mon Sep 17 00:00:00 2001
From: Atsuihsio <842960157@qq.com>
Date: Sat, 7 Dec 2024 18:26:21 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=85=85=E7=94=B5=E7=AB=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../entity/ChargingStationBlockEntity.java    | 24 ++++++++++---------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/atsuishio/superbwarfare/block/entity/ChargingStationBlockEntity.java b/src/main/java/com/atsuishio/superbwarfare/block/entity/ChargingStationBlockEntity.java
index cae224ee0..06d5daa5a 100644
--- a/src/main/java/com/atsuishio/superbwarfare/block/entity/ChargingStationBlockEntity.java
+++ b/src/main/java/com/atsuishio/superbwarfare/block/entity/ChargingStationBlockEntity.java
@@ -97,6 +97,19 @@ public class ChargingStationBlockEntity extends BlockEntity implements WorldlyCo
 
     public static void serverTick(Level pLevel, BlockPos pPos, BlockState pState, ChargingStationBlockEntity blockEntity) {
         blockEntity.setChanged();
+        blockEntity.energyHandler.ifPresent(handler -> blockEntity.energy = handler.getEnergyStored());
+
+        blockEntity.energyHandler.ifPresent(handler -> {
+            int energy = handler.getEnergyStored();
+            blockEntity.energy = energy;
+            if (energy > 0) {
+                blockEntity.chargeEntity(handler);
+            }
+            if (handler.getEnergyStored() > 0) {
+                blockEntity.chargeItemStack(handler);
+            }
+        });
+
         if (blockEntity.fuelTick > 0) {
             blockEntity.fuelTick--;
             blockEntity.energyHandler.ifPresent(handler -> {
@@ -140,17 +153,6 @@ public class ChargingStationBlockEntity extends BlockEntity implements WorldlyCo
                 blockEntity.maxFuelTick = tick;
             }
         }
-
-        blockEntity.energyHandler.ifPresent(handler -> {
-            int energy = handler.getEnergyStored();
-            blockEntity.energy = energy;
-            if (energy > 0) {
-                blockEntity.chargeEntity(handler);
-            }
-            if (handler.getEnergyStored() > 0) {
-                blockEntity.chargeItemStack(handler);
-            }
-        });
     }
 
     private void chargeEntity(EnergyStorage handler) {