3333import static uk .co .real_logic .sbe .generation .java .JavaUtil .formatPropertyName ;
3434import static uk .co .real_logic .sbe .generation .java .JavaUtil .javaTypeName ;
3535
36- public class JavaPojoGenerator implements CodeGenerator
36+ public class JavaMockPojoGenerator implements CodeGenerator
3737{
3838 private static final String BASE_INDENT = "" ;
3939 private static final String INDENT = " " ;
40- public static final String MOCK = "Mock" ;
40+ private static final String MOCK = "Mock" ;
4141
4242 private final Ir ir ;
4343 private final OutputManager outputManager ;
4444
45- public JavaPojoGenerator (final Ir ir , final OutputManager outputManager )
45+ public JavaMockPojoGenerator (final Ir ir , final OutputManager outputManager )
4646 throws IOException
4747 {
4848 Verify .notNull (ir , "ir" );
@@ -109,20 +109,20 @@ private CharSequence generateChoices(final String bitSetClassName, final List<To
109109 {
110110 final String choiceName = token .name ();
111111 sb .append (String .format (
112- "\n " +
113- " private boolean %s;\n " +
114- " public boolean %s()\n " +
115- " {\n " +
116- " return %s;\n " +
117- " }\n \n " +
118- " public %s %s(final boolean value)\n " +
119- " {\n " +
120- " %s = value;\n " +
121- " return this;\n " +
122- " }\n " ,
123- choiceName , choiceName , choiceName ,
124- bitSetClassName ,
125- choiceName , choiceName
112+ "\n " +
113+ " private boolean %s;\n " +
114+ " public boolean %s()\n " +
115+ " {\n " +
116+ " return %s;\n " +
117+ " }\n \n " +
118+ " public %s %s(final boolean value)\n " +
119+ " {\n " +
120+ " %s = value;\n " +
121+ " return this;\n " +
122+ " }\n " ,
123+ choiceName , choiceName , choiceName ,
124+ bitSetClassName ,
125+ choiceName , choiceName
126126 ));
127127 }
128128 }
@@ -197,11 +197,7 @@ private int collectRootFields(final List<Token> tokens, int index, final List<To
197197 }
198198
199199 private int generateGroups (
200- final StringBuilder sb ,
201- final String parentMessageClassName ,
202- final List <Token > tokens ,
203- int index ,
204- final String indent )
200+ final StringBuilder sb , final String parentMessageClassName , final List <Token > tokens , int index , final String indent )
205201 {
206202 for (int size = tokens .size (); index < size ; index ++)
207203 {
@@ -211,7 +207,7 @@ private int generateGroups(
211207 final String groupName = groupToken .name ();
212208 final String groupClassName = formatClassName (groupName );
213209 sb .append (generateSingleValueProperty (parentMessageClassName ,
214- JavaUtil .toLowerFirstChar (groupName ), groupClassName , indent ));
210+ JavaUtil .toLowerFirstChar (groupName ), groupClassName , indent ));
215211
216212 generateGroupClassHeader (sb , groupName , indent + INDENT );
217213
@@ -231,10 +227,7 @@ private int generateGroups(
231227 return index ;
232228 }
233229
234- private void generateGroupClassHeader (
235- final StringBuilder sb ,
236- final String groupName ,
237- final String indent )
230+ private void generateGroupClassHeader (final StringBuilder sb , final String groupName , final String indent )
238231 {
239232 sb .append (String .format (
240233 "\n " +
@@ -369,69 +362,71 @@ else if (arrayLength > 1)
369362 }
370363
371364 private CharSequence generateArrayProperty (
372- final Encoding encoding ,
373- final String containingClassName , final String propertyName ,
374- final String javaTypeName , final String indent )
365+ final Encoding encoding ,
366+ final String containingClassName ,
367+ final String propertyName ,
368+ final String javaTypeName ,
369+ final String indent )
375370 {
376371 final StringBuilder sb = new StringBuilder ();
377372
378373 sb .append (String .format (
379- "\n " + indent + " private %s[] %s;\n " ,
380- javaTypeName ,
381- propertyName
374+ "\n " + indent + " private %s[] %s;\n " ,
375+ javaTypeName ,
376+ propertyName
382377 ));
383378
384379 sb .append (String .format (
385- indent + " public %1$s %2$s(final int index)\n " +
386- indent + " {\n " +
387- indent + " return %2$s[index];\n " +
388- indent + " }\n \n " ,
389- javaTypeName ,
390- propertyName
380+ indent + " public %1$s %2$s(final int index)\n " +
381+ indent + " {\n " +
382+ indent + " return %2$s[index];\n " +
383+ indent + " }\n \n " ,
384+ javaTypeName ,
385+ propertyName
391386 ));
392387
393388 sb .append (String .format (
394- indent + " public void %1$s(final int index, final %2$s value)\n " +
395- indent + " {\n " +
396- indent + " %1$s[index] = value;\n " +
397- indent + " }\n " ,
398- propertyName ,
399- javaTypeName
389+ indent + " public void %1$s(final int index, final %2$s value)\n " +
390+ indent + " {\n " +
391+ indent + " %1$s[index] = value;\n " +
392+ indent + " }\n " ,
393+ propertyName ,
394+ javaTypeName
400395 ));
401396
402397 if (encoding .primitiveType () == PrimitiveType .CHAR )
403398 {
404399 sb .append (String .format (
405- "\n " +
406- indent + " public int get%1$s(final byte[] dst, final int dstOffset)\n " +
407- indent + " {\n " +
408- indent + " System.arraycopy(%2$s, 0, dst, dstOffset, %2$sLength());\n " +
409- indent + " return %2$sLength();\n " +
410- indent + " }\n \n " ,
411- JavaUtil .toUpperFirstChar (propertyName ),
412- propertyName
400+ "\n " +
401+ indent + " public int get%1$s(final byte[] dst, final int dstOffset)\n " +
402+ indent + " {\n " +
403+ indent + " System.arraycopy(%2$s, 0, dst, dstOffset, %2$sLength());\n " +
404+ indent + " return %2$sLength();\n " +
405+ indent + " }\n \n " ,
406+ JavaUtil .toUpperFirstChar (propertyName ),
407+ propertyName
413408 ));
414409
415410 sb .append (String .format (
416- indent + " public %1$s put%2$s(final byte[] src, final int srcOffset)\n " +
417- indent + " {\n " +
418- indent + " System.arraycopy(src, srcOffset, %3$s, 0, src.length - srcOffset);\n " +
419- indent + " return this;\n " +
420- indent + " }\n " ,
421- containingClassName ,
422- JavaUtil .toUpperFirstChar (propertyName ),
423- propertyName
411+ indent + " public %1$s put%2$s(final byte[] src, final int srcOffset)\n " +
412+ indent + " {\n " +
413+ indent + " System.arraycopy(src, srcOffset, %3$s, 0, src.length - srcOffset);\n " +
414+ indent + " return this;\n " +
415+ indent + " }\n " ,
416+ containingClassName ,
417+ JavaUtil .toUpperFirstChar (propertyName ),
418+ propertyName
424419 ));
425420
426421 sb .append (String .format (
427- indent + " public %1$s put%2$s(final byte[] src)\n " +
428- indent + " {\n " +
429- indent + " %3$s = Arrays.copyOf(src, %3$sLength());\n " +
430- indent + " return this;\n " +
431- indent + " }\n " ,
432- containingClassName ,
433- JavaUtil .toUpperFirstChar (propertyName ),
434- propertyName
422+ indent + " public %1$s put%2$s(final byte[] src)\n " +
423+ indent + " {\n " +
424+ indent + " %3$s = Arrays.copyOf(src, %3$sLength());\n " +
425+ indent + " return this;\n " +
426+ indent + " }\n " ,
427+ containingClassName ,
428+ JavaUtil .toUpperFirstChar (propertyName ),
429+ propertyName
435430 ));
436431 }
437432
@@ -444,9 +439,9 @@ private CharSequence generateSingleValueProperty(
444439 final StringBuilder sb = new StringBuilder ();
445440
446441 sb .append (String .format (
447- "\n " + indent + " private %s %s;\n " ,
448- javaTypeName ,
449- propertyName
442+ "\n " + indent + " private %s %s;\n " ,
443+ javaTypeName ,
444+ propertyName
450445 ));
451446
452447 sb .append (String .format (
@@ -506,10 +501,10 @@ private CharSequence generateFields(final String containingClassName, final List
506501 }
507502
508503 private CharSequence generateEnumProperty (
509- final String containingClassName , final String propertyName , final Token token , final String indent )
504+ final String containingClassName , final String propertyName , final Token token , final String indent )
510505 {
511-
512506 final String enumName = token .name ();
507+
513508 return generateSingleValueProperty (containingClassName , propertyName , enumName , indent );
514509 }
515510}
0 commit comments