Even nicer throttling + encoding.

CraftBukkit-Patches/0025-Netty.patch

@@ -1,4 +1,4 @@
-From a05a6c78a40c45b5b882ccd9877b7fea231c83eb Mon Sep 17 00:00:00 2001
+From ac55b7e35bf3f589fda99e820f8f0a8d9df866d1 Mon Sep 17 00:00:00 2001
 From: md_5 <md_5@live.com.au>
 Date: Sun, 23 Jun 2013 16:32:51 +1000
 Subject: [PATCH] Netty
@@ -569,7 +569,7 @@ index 0000000..2eb1dcb
 +}
 diff --git a/src/main/java/org/spigotmc/netty/NettyNetworkManager.java b/src/main/java/org/spigotmc/netty/NettyNetworkManager.java
 new file mode 100644
-index 0000000..313e3ea
+index 0000000..c4e1153
 --- /dev/null
 +++ b/src/main/java/org/spigotmc/netty/NettyNetworkManager.java
 @@ -0,0 +1,316 @@
@@ -774,7 +774,7 @@ index 0000000..313e3ea
 +            ChannelPromise promise = channel.newPromise();
 +            if ( packet instanceof Packet255KickDisconnect )
 +            {
-+                channel.pipeline().get( OutboundManager.class ).lastFlush = 0;
++                channel.pipeline().get( PacketEncoder.class ).lastFlush = 0;
 +            }
 +
 +            channel.write( packet, promise );
@@ -891,10 +891,10 @@ index 0000000..313e3ea
 +}
 diff --git a/src/main/java/org/spigotmc/netty/NettyServerConnection.java b/src/main/java/org/spigotmc/netty/NettyServerConnection.java
 new file mode 100644
-index 0000000..9eecd59
+index 0000000..e00bc6a
 --- /dev/null
 +++ b/src/main/java/org/spigotmc/netty/NettyServerConnection.java
-@@ -0,0 +1,106 @@
+@@ -0,0 +1,105 @@
 +package org.spigotmc.netty;
 +
 +import com.google.common.util.concurrent.ThreadFactoryBuilder;
@@ -961,7 +961,6 @@ index 0000000..9eecd59
 +
 +                NettyNetworkManager networkManager = new NettyNetworkManager();
 +                ch.pipeline()
-+                        .addLast( "flusher", new OutboundManager( networkManager ) )
 +                        .addLast( "timer", new ReadTimeoutHandler( 30 ) )
 +                        .addLast( "decoder", new PacketDecoder() )
 +                        .addLast( "encoder", new PacketEncoder( networkManager ) )
@@ -1301,46 +1300,6 @@ index 0000000..5da8a59
 +        return ch.toString();
 +    }
 +}
-diff --git a/src/main/java/org/spigotmc/netty/OutboundManager.java b/src/main/java/org/spigotmc/netty/OutboundManager.java
-new file mode 100644
-index 0000000..728f260
---- /dev/null
-+++ b/src/main/java/org/spigotmc/netty/OutboundManager.java
-@@ -0,0 +1,34 @@
-+package org.spigotmc.netty;
-+
-+import io.netty.channel.ChannelHandlerContext;
-+import io.netty.channel.ChannelOutboundHandlerAdapter;
-+import io.netty.channel.ChannelPromise;
-+import io.netty.channel.MessageList;
-+import net.minecraft.server.PendingConnection;
-+
-+class OutboundManager extends ChannelOutboundHandlerAdapter
-+{
-+
-+    private static final int FLUSH_TIME = 1;
-+    /*========================================================================*/
-+    public long lastFlush;
-+    private final NettyNetworkManager manager;
-+    private final MessageList<Object> pending = MessageList.newInstance();
-+
-+    OutboundManager(NettyNetworkManager manager)
-+    {
-+        this.manager = manager;
-+    }
-+
-+    @Override
-+    public void write(ChannelHandlerContext ctx, MessageList<Object> msgs, ChannelPromise promise) throws Exception
-+    {
-+        pending.add( msgs );
-+        if ( manager.connection instanceof PendingConnection || System.currentTimeMillis() - lastFlush > FLUSH_TIME )
-+        {
-+            lastFlush = System.currentTimeMillis();
-+            ctx.write( pending.copy() );
-+            pending.clear();
-+        }
-+    }
-+}
 diff --git a/src/main/java/org/spigotmc/netty/PacketDecoder.java b/src/main/java/org/spigotmc/netty/PacketDecoder.java
 new file mode 100644
 index 0000000..3adc8d6
@@ -1417,10 +1376,10 @@ index 0000000..3adc8d6
 +}
 diff --git a/src/main/java/org/spigotmc/netty/PacketEncoder.java b/src/main/java/org/spigotmc/netty/PacketEncoder.java
 new file mode 100644
-index 0000000..5b35ab0
+index 0000000..383fd46
 --- /dev/null
 +++ b/src/main/java/org/spigotmc/netty/PacketEncoder.java
-@@ -0,0 +1,71 @@
+@@ -0,0 +1,88 @@
 +package org.spigotmc.netty;
 +
 +import io.netty.buffer.ByteBuf;
@@ -1431,6 +1390,7 @@ index 0000000..5b35ab0
 +import io.netty.channel.MessageList;
 +import java.io.DataOutputStream;
 +import net.minecraft.server.Packet;
++import net.minecraft.server.PendingConnection;
 +
 +/**
 + * Netty encoder which takes a packet and encodes it, and adds a byte packet id
@@ -1439,7 +1399,11 @@ index 0000000..5b35ab0
 +public class PacketEncoder extends ChannelOutboundHandlerAdapter
 +{
 +
++    private static final int FLUSH_TIME = 1;
++    /*========================================================================*/
++    public long lastFlush;
 +    private final NettyNetworkManager networkManager;
++    private final MessageList<Object> pending = MessageList.newInstance();
 +
 +    public PacketEncoder(NettyNetworkManager networkManager)
 +    {
@@ -1449,9 +1413,21 @@ index 0000000..5b35ab0
 +    @Override
 +    public void write(ChannelHandlerContext ctx, MessageList<Object> msgs, ChannelPromise promise) throws Exception
 +    {
++        // Append messages to queue and then recycle - we don't need to bother about recycling if the worst happens, since MessageList is GC safe
++        pending.add( msgs );
++        msgs.recycle();
++
++        // If we are not in the pending connect phase, and we have not reached our timer
++        if ( !( networkManager.connection instanceof PendingConnection ) && System.currentTimeMillis() - lastFlush < FLUSH_TIME )
++        {
++            return;
++        }
++        // Update our last write time
++        lastFlush = System.currentTimeMillis();
++
 +        // Since we are writing in batches it can be useful to guess the size of our output to limit memcpy
 +        int estimatedSize = 0;
-+        for ( Object msg : msgs )
++        for ( Object msg : pending )
 +        {
 +            if ( msg instanceof Packet )
 +            {
@@ -1469,7 +1445,7 @@ index 0000000..5b35ab0
 +        try
 +        {
 +            // Iterate through all packets, this is safe as we know we will only ever get packets in the pipeline
-+            for ( Packet packet : (MessageList<Packet>) (MessageList) msgs )
++            for ( Packet packet : (MessageList<Packet>) (MessageList) pending )
 +            {
 +                // Write packet ID
 +                outBuf.writeByte( packet.n() );
@@ -1482,8 +1458,8 @@ index 0000000..5b35ab0
 +            ctx.write( outBuf, promise );
 +        } finally
 +        {
-+            // Very important that we do this for fast memory reclamation
-+            msgs.recycle();
++            // Reset packet queue
++            pending.clear();
 +            // Since we are now in the event loop, the bytes have been written, we can free them if this was not the case
 +            if ( outBuf.refCnt() != 0 )
 +            {