Skip to content

Commit

Permalink
[C++] respect the package override option for C++ codecs and DTOs
Browse files Browse the repository at this point in the history
  • Loading branch information
@nbradac
nbradac committed Nov 13, 2024
1 parent d38b4bf commit b385e9c
Show file tree
Hide file tree
Showing 8 changed files with 322 additions and 41 deletions.
12 changes: 10 additions & 2 deletions sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/TargetCodeGeneratorLoader.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -107,13 +107,21 @@ public CodeGenerator newInstance(final Ir ir, final String outputDir)
final NamespaceOutputManager outputManager = new NamespaceOutputManager(
outputDir, ir.applicableNamespace());
final boolean decodeUnknownEnumValues = Boolean.getBoolean(DECODE_UNKNOWN_ENUM_VALUES);
final boolean shouldSupportTypesPackageNames = Boolean.getBoolean(TYPES_PACKAGE_OVERRIDE);

final CodeGenerator codecGenerator = new CppGenerator(ir, decodeUnknownEnumValues, precedenceChecks(),
final CodeGenerator codecGenerator = new CppGenerator(
ir,
decodeUnknownEnumValues,
precedenceChecks(),
shouldSupportTypesPackageNames,
outputManager);

if (Boolean.getBoolean(CPP_GENERATE_DTOS))
{
final CodeGenerator dtoGenerator = new CppDtoGenerator(ir, outputManager);
final CodeGenerator dtoGenerator = new CppDtoGenerator(
ir,
shouldSupportTypesPackageNames,
outputManager);
return () ->
{
codecGenerator.generate();
Expand Down
95 changes: 82 additions & 13 deletions sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/cpp/CppDtoGenerator.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -50,27 +50,65 @@ public class CppDtoGenerator implements CodeGenerator

private final Ir ir;
private final OutputManager outputManager;
private final boolean shouldSupportTypesPackageNames;
private final Map<String, String> namespaceByType = new HashMap<>();

/**
* Create a new C++ DTO {@link CodeGenerator}.
*
* @param ir for the messages and types.
* @param outputManager for generating the DTOs to.
* @param ir for the messages and types.
* @param shouldSupportTypesPackageNames generator support for types in their own package.
* @param outputManager for generating the DTOs to.
*/
public CppDtoGenerator(final Ir ir, final OutputManager outputManager)
public CppDtoGenerator(
final Ir ir,
final boolean shouldSupportTypesPackageNames,
final OutputManager outputManager)
{
Verify.notNull(ir, "ir");
Verify.notNull(outputManager, "outputManager");

this.ir = ir;
this.shouldSupportTypesPackageNames = shouldSupportTypesPackageNames;
this.outputManager = outputManager;
}

private String[] fetchTypesPackageName(final Token token, final Ir ir)
{
if (!shouldSupportTypesPackageNames)
{
return ir.namespaces();
}

if (token.packageName() != null)
{
return Ir.getNamespaces(token.packageName());
}

return ir.namespaces();
}

/**
* {@inheritDoc}
*/
public void generate() throws IOException
{
namespaceByType.clear();

if (shouldSupportTypesPackageNames)
{
for (final List<Token> tokens : ir.types())
{
final Token token = tokens.get(0);
final String packageName = token.packageName();

if (packageName != null)
{
namespaceByType.put(token.applicableTypeName(), packageName);
}
}
}

generateDtosForTypes();

for (final List<Token> tokens : ir.messages())
Expand Down Expand Up @@ -116,13 +154,14 @@ public void generate() throws IOException
final Set<String> referencedTypes = generateTypesToIncludes(beginTypeTokensInSchema);
referencedTypes.add(codecClassName);

final String[] namespaces = fetchTypesPackageName(msgToken, ir);
out.append(generateDtoFileHeader(
ir.namespaces(),
namespaces,
className,
referencedTypes));
out.append(generateDocumentation(BASE_INDENT, msgToken));
classBuilder.appendTo(out);
out.append(CppUtil.closingBraces(ir.namespaces().length));
out.append(CppUtil.closingBraces(namespaces.length));
out.append("#endif\n");
}
}
Expand Down Expand Up @@ -1732,7 +1771,8 @@ private void generateDtosForTypes() throws IOException

private void generateComposite(final List<Token> tokens) throws IOException
{
final String name = tokens.get(0).applicableTypeName();
final Token token = tokens.get(0);
final String name = token.applicableTypeName();
final String className = formatDtoClassName(name);
final String codecClassName = formatClassName(name);

Expand All @@ -1741,8 +1781,9 @@ private void generateComposite(final List<Token> tokens) throws IOException
final List<Token> compositeTokens = tokens.subList(1, tokens.size() - 1);
final Set<String> referencedTypes = generateTypesToIncludes(compositeTokens);
referencedTypes.add(codecClassName);
out.append(generateDtoFileHeader(ir.namespaces(), className, referencedTypes));
out.append(generateDocumentation(BASE_INDENT, tokens.get(0)));
final String[] namespaces = fetchTypesPackageName(token, ir);
out.append(generateDtoFileHeader(namespaces, className, referencedTypes));
out.append(generateDocumentation(BASE_INDENT, token));

final ClassBuilder classBuilder = new ClassBuilder(className, BASE_INDENT);

Expand All @@ -1754,14 +1795,15 @@ private void generateComposite(final List<Token> tokens) throws IOException
codecClassName + "::sbeSchemaVersion()", BASE_INDENT + INDENT);

classBuilder.appendTo(out);
out.append(CppUtil.closingBraces(ir.namespaces().length));
out.append(CppUtil.closingBraces(namespaces.length));
out.append("#endif\n");
}
}

private void generateChoiceSet(final List<Token> tokens) throws IOException
{
final String name = tokens.get(0).applicableTypeName();
final Token token = tokens.get(0);
final String name = token.applicableTypeName();
final String className = formatDtoClassName(name);
final String codecClassName = formatClassName(name);

Expand All @@ -1770,7 +1812,8 @@ private void generateChoiceSet(final List<Token> tokens) throws IOException
final List<Token> setTokens = tokens.subList(1, tokens.size() - 1);
final Set<String> referencedTypes = generateTypesToIncludes(setTokens);
referencedTypes.add(codecClassName);
out.append(generateDtoFileHeader(ir.namespaces(), className, referencedTypes));
final String[] namespaces = fetchTypesPackageName(token, ir);
out.append(generateDtoFileHeader(namespaces, className, referencedTypes));
out.append(generateDocumentation(BASE_INDENT, tokens.get(0)));

final ClassBuilder classBuilder = new ClassBuilder(className, BASE_INDENT);
Expand All @@ -1780,7 +1823,7 @@ private void generateChoiceSet(final List<Token> tokens) throws IOException
generateChoiceSetEncodeWith(classBuilder, className, codecClassName, setTokens, BASE_INDENT + INDENT);

classBuilder.appendTo(out);
out.append(CppUtil.closingBraces(ir.namespaces().length));
out.append(CppUtil.closingBraces(namespaces.length));
out.append("#endif\n");
}
}
Expand Down Expand Up @@ -1965,7 +2008,7 @@ private static CharSequence typeWithFieldOptionality(
}
}

private static CharSequence generateDtoFileHeader(
private CharSequence generateDtoFileHeader(
final CharSequence[] namespaces,
final String className,
final Collection<String> typesToInclude)
Expand Down Expand Up @@ -2010,6 +2053,32 @@ private static CharSequence generateDtoFileHeader(
sb.append(String.join(" {\nnamespace ", namespaces));
sb.append(" {\n\n");

if (shouldSupportTypesPackageNames && typesToInclude != null && !typesToInclude.isEmpty())
{
final Set<String> namespacesToUse = namespaceByType
.entrySet()
.stream()
.filter(e -> typesToInclude.contains(e.getKey()))
.map(Map.Entry::getValue)
.collect(Collectors.toSet());

// remove the current namespace
namespacesToUse.remove(String.join(".", namespaces));

for (final String namespace : namespacesToUse)
{
sb
.append("using namespace ")
.append(namespace.replaceAll("\\.", "::"))
.append(";\n");
}

if (!namespacesToUse.isEmpty())
{
sb.append("\n");
}
}

return sb;
}

Expand Down
2 changes: 1 addition & 1 deletion sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/cpp/CppDtos.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,6 @@ public class CppDtos implements TargetCodeGenerator
*/
public CodeGenerator newInstance(final Ir ir, final String outputDir)
{
return new CppDtoGenerator(ir, new NamespaceOutputManager(outputDir, ir.applicableNamespace()));
return new CppDtoGenerator(ir, false, new NamespaceOutputManager(outputDir, ir.applicableNamespace()));
}
}
Loading

0 comments on commit b385e9c

Please sign in to comment.