From 85045e1e4a15e0a5657d189e83dd202a2c37f2b0 Mon Sep 17 00:00:00 2001
From: Yuuta Liang <yuuta@yuuta.moe>
Date: Wed, 13 Jul 2022 11:16:27 -0700
Subject: First Commit

Signed-off-by: Trumeet <yuuta@yuuta.moe>
---
 .../moe/ymc/acron/mixin/ServerNetworkIoMixin.java  | 65 ++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 src/main/java/moe/ymc/acron/mixin/ServerNetworkIoMixin.java

(limited to 'src/main/java/moe/ymc/acron/mixin/ServerNetworkIoMixin.java')

diff --git a/src/main/java/moe/ymc/acron/mixin/ServerNetworkIoMixin.java b/src/main/java/moe/ymc/acron/mixin/ServerNetworkIoMixin.java
new file mode 100644
index 0000000..7c9b60b
--- /dev/null
+++ b/src/main/java/moe/ymc/acron/mixin/ServerNetworkIoMixin.java
@@ -0,0 +1,65 @@
+package moe.ymc.acron.mixin;
+
+import io.netty.bootstrap.ServerBootstrap;
+import io.netty.channel.ChannelFuture;
+import io.netty.channel.MultithreadEventLoopGroup;
+import io.netty.channel.ServerChannel;
+import io.netty.channel.epoll.Epoll;
+import io.netty.channel.epoll.EpollServerSocketChannel;
+import io.netty.channel.nio.NioEventLoopGroup;
+import io.netty.channel.socket.nio.NioServerSocketChannel;
+import moe.ymc.acron.config.Config;
+import moe.ymc.acron.net.AcronInitializer;
+import net.minecraft.server.ServerNetworkIo;
+import net.minecraft.util.Lazy;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+import java.util.List;
+
+import static net.minecraft.server.ServerNetworkIo.EPOLL_CHANNEL;
+
+@Mixin(ServerNetworkIo.class)
+public class ServerNetworkIoMixin {
+    private static final Logger LOGGER = LogManager.getLogger();
+
+    @Shadow
+    @Final
+    private List<ChannelFuture> channels;
+
+    @Shadow
+    @Final
+    public static Lazy<NioEventLoopGroup> DEFAULT_CHANNEL;
+
+    @Inject(at = @At("RETURN"), method = "<init>")
+    private void init(CallbackInfo info) {
+        LOGGER.debug("Adding Acron channel.");
+        Lazy<? extends MultithreadEventLoopGroup> group;
+        Class<? extends ServerChannel> channel;
+        if (Epoll.isAvailable() && Config.getGlobalConfig().useNativeTransport()) {
+            channel = EpollServerSocketChannel.class;
+            group = EPOLL_CHANNEL;
+            LOGGER.info("Using native transport.");
+        } else {
+            channel = NioServerSocketChannel.class;
+            group = DEFAULT_CHANNEL;
+            LOGGER.info("Not using native transport due to " +
+                    "it is either disabled in acron.json or not available.");
+        }
+        channels.add(new ServerBootstrap()
+                .channel(channel)
+                .childHandler(new AcronInitializer())
+                .group(group.get())
+                .localAddress(Config.getGlobalConfig().address(),
+                        Config.getGlobalConfig().port())
+                .bind()
+                .syncUninterruptibly());
+
+    }
+}
-- 
cgit v1.2.3