Merge pull request #614 from kdkavanagh/master

Lookup blockLength and numInGrp rather than assuming positions
aeron-io · Nov 8, 2018 · 96521eb · 96521eb
2 parents ab42ba0 + ae10d9b
commit 96521eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/java/JavaGenerator.java b/sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/java/JavaGenerator.java
@@ -430,8 +430,9 @@ private void generateGroupEncoderClassHeader(
             sb, groupToken, groupName, parentMessageClassName, ind, dimensionsClassName, dimensionHeaderSize);
 
         final int blockLength = tokens.get(index).encodedLength();
-        final String javaTypeForBlockLength = primitiveTypeName(tokens.get(index + 2));
-        final Token numInGroupToken = tokens.get(index + 3);
+        final Token blockLengthToken = tokens.stream().filter(t -> "blockLength".equals(t.name())).findFirst().get();
+        final String javaTypeForBlockLength = primitiveTypeName(blockLengthToken);
+        final Token numInGroupToken = tokens.stream().filter(t -> "numInGroup".equals(t.name())).findFirst().get();
         final String javaTypeForNumInGroup = primitiveTypeName(numInGroupToken);
 
         sb.append(String.format(