Added micro benchmarking using JMH

Author: Bytekeeper

pom.xml

@@ -10,6 +10,7 @@
 
     <properties>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+        <jmh.version>1.21</jmh.version>
     </properties>
 
     <build>
@@ -45,6 +46,30 @@
                     </execution>
                 </executions>
             </plugin>
+            <plugin>
+                <groupId>org.codehaus.mojo</groupId>
+                <artifactId>exec-maven-plugin</artifactId>
+                <version>1.6.0</version>
+                <executions>
+                    <execution>
+                        <id>run-benchmarks</id>
+                        <phase>integration-test</phase>
+                        <goals>
+                            <goal>exec</goal>
+                        </goals>
+                        <configuration>
+                            <classpathScope>test</classpathScope>
+                            <executable>java</executable>
+                            <arguments>
+                                <argument>-classpath</argument>
+                                <classpath/>
+                                <argument>org.openjdk.jmh.Main</argument>
+                                <argument>.*</argument>
+                            </arguments>
+                        </configuration>
+                    </execution>
+                </executions>
+            </plugin>
         </plugins>
     </build>
 
@@ -86,5 +111,20 @@
             <version>3.12.2</version>
             <scope>test</scope>
         </dependency>
+
+        <!-- https://mvnrepository.com/artifact/org.openjdk.jmh -->
+        <dependency>
+            <groupId>org.openjdk.jmh</groupId>
+            <artifactId>jmh-core</artifactId>
+            <version>${jmh.version}</version>
+            <scope>test</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>org.openjdk.jmh</groupId>
+            <artifactId>jmh-generator-annprocess</artifactId>
+            <version>${jmh.version}</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 </project>

src/main/java/bwapi/Client.java

@@ -45,7 +45,7 @@ class Client {
             + 4 // LastKeepAliveTime
             ;
     private static final List<Integer> SUPPORTED_BWAPI_VERSIONS = Arrays.asList(10003);
-    private static final int MAX_COUNT = 19999;
+    static final int MAX_COUNT = 19999;
 
     private static final int maxNumGames = 8;
     private static final int gameTableSize = GAME_SIZE * maxNumGames;
@@ -71,6 +71,13 @@ class Client {
         throw new Exception("All servers busy!");
     }
 
+    /**
+     * For test purposes only
+     */
+    Client(ByteBuffer buffer) {
+        data = new ClientData(buffer).new GameData(0);
+    }
+
     public GameData data() {
         return data;
     }

src/test/java/bwapi/ClientDataBenchmark.java

@@ -0,0 +1,83 @@
+package bwapi;
+
+import org.openjdk.jmh.annotations.*;
+import org.openjdk.jmh.infra.Blackhole;
+
+import java.nio.ByteBuffer;
+import java.util.SplittableRandom;
+import java.util.stream.Collectors;
+
+@Warmup(iterations = 3, time = 5)
+@Measurement(iterations = 3, time = 5)
+@Fork(3)
+public class ClientDataBenchmark {
+    @State(Scope.Thread)
+    public static class EmptyState {
+        Client client;
+        Game game;
+        String[] strings;
+
+        @Setup(Level.Invocation)
+        public void setup() {
+            client = new Client(ByteBuffer.allocateDirect(ClientData.GameData.SIZE));
+            game = new Game(client);
+            strings = buildStrings();
+        }
+
+    }
+
+    @State(Scope.Thread)
+    public static class FilledWithStrings {
+        Client client;
+        ClientData.GameData data;
+        Game game;
+
+        @Setup(Level.Invocation)
+        public void setup() {
+            client = new Client(ByteBuffer.allocateDirect(ClientData.GameData.SIZE));
+            data = client.data();
+            game = new Game(client);
+            String[] strings = buildStrings();
+            for (String s : strings) {
+                client.addString(s);
+            }
+        }
+    }
+
+    private static String[] buildStrings() {
+        SplittableRandom rnd = new SplittableRandom(987654321L);
+        String[] strings = new String[Client.MAX_COUNT];
+        for (int i = 0; i < strings.length; i++) {
+            strings[i] = rnd.ints(1022, 0, 9)
+                    .mapToObj(Integer::toString)
+                    .collect(Collectors.joining());
+        }
+        return strings;
+    }
+
+    @Benchmark
+    @OperationsPerInvocation(Client.MAX_COUNT)
+    public int addUnitCommand(EmptyState s) {
+        for (int i = 0; i < Client.MAX_COUNT; i++) {
+            s.game.addUnitCommand(0, 1, 2, 3, 4, 5);
+        }
+        return s.client.data().getCommandCount();
+    }
+
+    @Benchmark
+    @OperationsPerInvocation(Client.MAX_COUNT)
+    public int addString(EmptyState s) {
+        for (int i = 0; i < Client.MAX_COUNT; i++) {
+            s.client.addString(s.strings[i]);
+        }
+        return s.client.data().getStringCount();
+    }
+
+    @Benchmark
+    @OperationsPerInvocation(Client.MAX_COUNT)
+    public void getString(FilledWithStrings s, Blackhole blackhole) {
+        for (int i = 0; i < Client.MAX_COUNT; i++) {
+            blackhole.consume(s.data.getStrings(i));
+        }
+    }
+}