修复使用标签无线收发器时偶现退出游戏无法保存问题

src/main/java/com/extendedae_plus/ExtendedAEPlus.java

@@ -8,6 +8,8 @@ import appeng.blockentity.crafting.CraftingBlockEntity;
 import appeng.items.parts.PartModelsHelper;
 import com.extendedae_plus.api.ids.EAPComponents;
 import com.extendedae_plus.api.storage.InfinityBigIntegerCellHandler;
+import com.extendedae_plus.ae.wireless.LabelNetworkRegistry;
+import com.extendedae_plus.ae.wireless.WirelessMasterRegistry;
 import com.extendedae_plus.config.ModConfigs;
 import com.extendedae_plus.content.ae2.MirrorPatternProviderBlockEntity;
 import com.extendedae_plus.content.matrix.CrafterCorePlusBlockEntity;
@@ -29,6 +31,7 @@ import net.neoforged.fml.event.lifecycle.FMLCommonSetupEvent;
 import net.neoforged.neoforge.common.NeoForge;
 import net.neoforged.neoforge.event.server.ServerStartedEvent;
 import net.neoforged.neoforge.event.server.ServerStartingEvent;
+import net.neoforged.neoforge.event.server.ServerStoppingEvent;
 import net.neoforged.neoforge.event.server.ServerStoppedEvent;
 import org.jetbrains.annotations.Nullable;
 import org.slf4j.Logger;
@@ -42,6 +45,7 @@ public class ExtendedAEPlus {
     private static InfinityStorageManager storageManager;
     @Nullable
     private static MinecraftServer storageManagerServer;
+    private static volatile boolean serverStopping;
 
     public ExtendedAEPlus(IEventBus modEventBus, ModContainer modContainer) {
         modEventBus.addListener(this::commonSetup);
@@ -60,6 +64,7 @@ public class ExtendedAEPlus {
         EAPComponents.DR.register(modEventBus);
         NeoForge.EVENT_BUS.register(this);
         NeoForge.EVENT_BUS.addListener(ExtendedAEPlus::onServerStarted);
+        NeoForge.EVENT_BUS.addListener(ExtendedAEPlus::onServerStopping);
         NeoForge.EVENT_BUS.addListener(ExtendedAEPlus::onServerStopped);
         // 注册配置：接入自定义的 ModConfigs
         modContainer.registerConfig(ModConfig.Type.COMMON, ModConfigs.COMMON_SPEC, "extendedae_plus-common.toml");
@@ -72,17 +77,29 @@ public class ExtendedAEPlus {
     }
 
     private static void onServerStarted(ServerStartedEvent event) {
+        serverStopping = false;
         storageManagerServer = event.getServer();
         storageManager = InfinityStorageManager.getInstance(event.getServer());
     }
 
+    private static void onServerStopping(ServerStoppingEvent event) {
+        serverStopping = true;
+        LabelNetworkRegistry.get(event.getServer()).shutdownAllVirtualNodes();
+        WirelessMasterRegistry.clear();
+    }
+
     private static void onServerStopped(ServerStoppedEvent event) {
+        serverStopping = false;
         if (storageManagerServer == event.getServer()) {
             storageManagerServer = null;
             storageManager = null;
         }
     }
 
+    public static boolean isServerStopping() {
+        return serverStopping;
+    }
+
     @Nullable
     public static InfinityStorageManager currentStorageManager() {
         return storageManager;

src/main/java/com/extendedae_plus/ae/wireless/LabelLink.java

@@ -1,6 +1,7 @@
 package com.extendedae_plus.ae.wireless;
 
 import appeng.api.networking.GridHelper;
+import appeng.api.networking.IGridConnection;
 import appeng.api.networking.IGridNode;
 import appeng.me.service.helpers.ConnectionWrapper;
 import net.minecraft.resources.ResourceKey;
@@ -88,14 +89,62 @@ public class LabelLink {
     }
 
     public void onUnloadOrRemove() {
+        this.target = null;
         destroyConnection();
     }
 
     private void destroyConnection() {
         var current = connection.getConnection();
         if (current != null) {
+            var a = current.a();
+            var b = current.b();
             current.destroy();
+            try {
+                if (a != null && a.getGrid() != null) {
+                    a.getGrid().getTickManager().wakeDevice(a);
+                }
+            } catch (Throwable ignored) {}
+            try {
+                if (b != null && b.getGrid() != null) {
+                    b.getGrid().getTickManager().wakeDevice(b);
+                }
+            } catch (Throwable ignored) {}
             connection.setConnection(null);
+        } else {
+            try {
+                IGridNode hostNode = this.host.getGridNode();
+                IGridNode targetNode = this.target == null ? null : this.target.node();
+                if (hostNode != null && targetNode != null) {
+                    IGridConnection existing = this.findExistingConnection(hostNode, targetNode);
+                    if (existing != null) {
+                        existing.destroy();
+                        try {
+                            if (hostNode.getGrid() != null) {
+                                hostNode.getGrid().getTickManager().wakeDevice(hostNode);
+                            }
+                        } catch (Throwable ignored) {}
+                        try {
+                            if (targetNode.getGrid() != null) {
+                                targetNode.getGrid().getTickManager().wakeDevice(targetNode);
+                            }
+                        } catch (Throwable ignored) {}
+                    }
+                }
+            } catch (Throwable ignored) {}
         }
     }
+
+    @Nullable
+    private IGridConnection findExistingConnection(IGridNode a, IGridNode b) {
+        try {
+            for (IGridConnection gc : a.getConnections()) {
+                var ga = gc.a();
+                var gb = gc.b();
+                if ((ga == a || gb == a) && (ga == b || gb == b)) {
+                    return gc;
+                }
+            }
+        } catch (Throwable ignored) {}
+        return null;
+    }
 }

src/main/java/com/extendedae_plus/ae/wireless/LabelNetworkRegistry.java

@@ -111,10 +111,19 @@ public class LabelNetworkRegistry extends SavedData {
         BlockPos pos = endpoint.getBlockPos();
         for (LabelNetwork net : networks.values()) {
             net.endpoints.removeIf(ref -> ref.matches(dimKey, pos));
+            if (net.endpoints.isEmpty()) {
+                net.destroyVirtualNode();
+            }
         }
         setDirty();
     }
 
+    public synchronized void shutdownAllVirtualNodes() {
+        for (LabelNetwork net : networks.values()) {
+            net.destroyVirtualNode();
+        }
+    }
+
     public synchronized LabelNetwork getNetwork(ServerLevel level, String rawLabel, @Nullable UUID placerId) {
         String label = normalizeLabel(rawLabel);
         if (label == null) return null;

src/main/java/com/extendedae_plus/ae/wireless/WirelessMasterRegistry.java

@@ -85,6 +85,10 @@ public final class WirelessMasterRegistry {
         return ref == null ? null : ref.get();
     }
 
+    public static synchronized void clear() {
+        MASTERS.clear();
+    }
+
     private static void cleanupIfCleared(Key key) {
         var ref = MASTERS.get(key);
         if (ref != null && ref.get() == null) {

src/main/java/com/extendedae_plus/content/wireless/LabeledWirelessTransceiverBlockEntity.java

@@ -8,6 +8,7 @@ import appeng.api.networking.IInWorldGridNodeHost;
 import appeng.api.networking.IManagedGridNode;
 import appeng.api.util.AECableType;
 import appeng.blockentity.AEBaseBlockEntity;
+import com.extendedae_plus.ExtendedAEPlus;
 import com.extendedae_plus.ae.wireless.IWirelessEndpoint;
 import com.extendedae_plus.ae.wireless.LabelLink;
 import com.extendedae_plus.ae.wireless.LabelNetworkRegistry;
@@ -194,25 +195,49 @@ public class LabeledWirelessTransceiverBlockEntity extends AEBaseBlockEntity imp
     }
 
     public void onRemoved() {
+        cleanupForRemoval();
+    }
+
+    @Override
+    public void onChunkUnloaded() {
+        cleanupForRemoval();
+        super.onChunkUnloaded();
+    }
+
+    @Override
+    public void setRemoved() {
+        cleanupForRemoval();
+        super.setRemoved();
+    }
+
+    private void cleanupForRemoval() {
+        if (this.beingRemoved) {
+            return;
+        }
+
         this.beingRemoved = true;
-        labelLink.onUnloadOrRemove();
+        this.labelLink.onUnloadOrRemove();
+
         ServerLevel sl = getServerLevel();
         if (sl != null) {
             LabelNetworkRegistry.get(sl).unregister(this);
         }
-        if (managedNode != null) {
+
-            managedNode.destroy();
+        if (this.managedNode != null) {
+            this.managedNode.destroy();
         }
     }
 
     public static void serverTick(Level level, BlockPos pos, BlockState state, LabeledWirelessTransceiverBlockEntity be) {
         if (!(level instanceof ServerLevel)) return;
+        if (ExtendedAEPlus.isServerStopping()) return;
         be.labelLink.updateStatus();
         be.updateState();
     }
 
     private void updateState() {
         if (this.level == null || this.level.isClientSide) return;
+        if (ExtendedAEPlus.isServerStopping()) return;
         if (this.beingRemoved || this.isRemoved()) return;
         BlockState currentState = this.getBlockState();
         if (!(currentState.getBlock() instanceof LabeledWirelessTransceiverBlock)) {
@@ -307,15 +332,12 @@ public class LabeledWirelessTransceiverBlockEntity extends AEBaseBlockEntity imp
         }
         @Override
         public void onStateChanged(LabeledWirelessTransceiverBlockEntity host, IGridNode node, State state) {
-            host.updateState();
         }
         @Override
         public void onInWorldConnectionChanged(LabeledWirelessTransceiverBlockEntity host, IGridNode node) {
-            host.updateState();
         }
         @Override
         public void onGridChanged(LabeledWirelessTransceiverBlockEntity host, IGridNode node) {
-            host.updateState();
         }
         @Override
         public void onOwnerChanged(LabeledWirelessTransceiverBlockEntity host, IGridNode node) {}

src/main/java/com/extendedae_plus/content/wireless/WirelessTransceiverBlockEntity.java

@@ -42,6 +42,8 @@ public class WirelessTransceiverBlockEntity extends AEBaseBlockEntity implements
     @Nullable
     private String placerName; // 放置者名称，用于显示
 
+    private boolean beingRemoved = false;
+
     private WirelessMasterLink masterLink;
     private WirelessSlaveLink slaveLink;
 
@@ -247,11 +249,33 @@ public class WirelessTransceiverBlockEntity extends AEBaseBlockEntity implements
     }
 
     void onRemoved() {
+        cleanupForRemoval();
+    }
+
+    @Override
+    public void onChunkUnloaded() {
+        cleanupForRemoval();
+        super.onChunkUnloaded();
+    }
+
+    @Override
+    public void setRemoved() {
+        cleanupForRemoval();
+        super.setRemoved();
+    }
+
+    private void cleanupForRemoval() {
+        if (this.beingRemoved) {
+            return;
+        }
+
+        this.beingRemoved = true;
         if (this.masterMode) {
             this.masterLink.onUnloadOrRemove();
         } else {
             this.slaveLink.onUnloadOrRemove();
         }
+
         if (this.managedNode != null) {
             this.managedNode.destroy();
         }