uk.co.real_logic.sbe.generation.csharp.CSharpDtoGenerator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sbe-tool Show documentation
Show all versions of sbe-tool Show documentation
FIX/SBE - OSI layer 6 presentation for encoding and decoding application messages in binary format for low-latency applications.
The newest version!
/*
* Copyright 2013-2024 Real Logic Limited.
* Copyright (C) 2017 MarketFactory, Inc
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package uk.co.real_logic.sbe.generation.csharp;
import uk.co.real_logic.sbe.PrimitiveType;
import uk.co.real_logic.sbe.generation.CodeGenerator;
import uk.co.real_logic.sbe.generation.Generators;
import uk.co.real_logic.sbe.ir.Encoding;
import uk.co.real_logic.sbe.ir.Ir;
import uk.co.real_logic.sbe.ir.Signal;
import uk.co.real_logic.sbe.ir.Token;
import org.agrona.Verify;
import org.agrona.generation.OutputManager;
import java.io.IOException;
import java.io.Writer;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Predicate;
import static uk.co.real_logic.sbe.generation.csharp.CSharpUtil.*;
import static uk.co.real_logic.sbe.ir.GenerationUtil.collectFields;
import static uk.co.real_logic.sbe.ir.GenerationUtil.collectGroups;
import static uk.co.real_logic.sbe.ir.GenerationUtil.collectVarData;
/**
* DTO generator for the CSharp programming language.
*/
public class CSharpDtoGenerator implements CodeGenerator
{
private static final String INDENT = " ";
private static final String BASE_INDENT = INDENT;
private static final Predicate CANNOT_EXTEND = ignored -> false;
private final Ir ir;
private final OutputManager outputManager;
/**
* Create a new C# DTO {@link CodeGenerator}.
*
* @param ir for the messages and types.
* @param outputManager for generating the DTOs to.
*/
public CSharpDtoGenerator(final Ir ir, final OutputManager outputManager)
{
Verify.notNull(ir, "ir");
Verify.notNull(outputManager, "outputManager");
this.ir = ir;
this.outputManager = outputManager;
}
/**
* {@inheritDoc}
*/
public void generate() throws IOException
{
generateDtosForTypes();
for (final List tokens : ir.messages())
{
final Token msgToken = tokens.get(0);
final String codecClassName = formatClassName(msgToken.name());
final String dtoClassName = formatDtoClassName(msgToken.name());
final List messageBody = tokens.subList(1, tokens.size() - 1);
int offset = 0;
final StringBuilder sb = new StringBuilder();
final StringBuilder ctorArgs = new StringBuilder();
final List fields = new ArrayList<>();
offset = collectFields(messageBody, offset, fields);
generateFields(sb, ctorArgs, codecClassName, fields, BASE_INDENT + INDENT);
final List groups = new ArrayList<>();
offset = collectGroups(messageBody, offset, groups);
generateGroups(sb, ctorArgs, dtoClassName, codecClassName, groups, BASE_INDENT + INDENT);
final List varData = new ArrayList<>();
collectVarData(messageBody, offset, varData);
generateVarData(sb, ctorArgs, varData, BASE_INDENT + INDENT);
generateDecodeWith(sb, dtoClassName, codecClassName, fields, groups, varData,
token -> token.version() > msgToken.version(),
BASE_INDENT + INDENT);
generateDecodeFrom(sb, dtoClassName, codecClassName, BASE_INDENT + INDENT);
generateEncodeWith(sb, dtoClassName, codecClassName, fields, groups, varData,
BASE_INDENT + INDENT);
generateEncodeInto(sb, dtoClassName, codecClassName, BASE_INDENT + INDENT);
generateDisplay(sb, codecClassName, "WrapForEncode", null, BASE_INDENT + INDENT);
removeTrailingComma(ctorArgs);
try (Writer out = outputManager.createOutput(dtoClassName))
{
out.append(generateFileHeader(
ir.applicableNamespace(),
"#nullable enable\n\n",
"using System.Collections.Generic;\n",
"using System.Linq;\n"));
out.append(generateDocumentation(BASE_INDENT, msgToken));
out.append(BASE_INDENT).append("public sealed partial record ").append(dtoClassName).append("(\n")
.append(ctorArgs)
.append(BASE_INDENT).append(")\n")
.append(BASE_INDENT).append("{")
.append(sb)
.append(BASE_INDENT).append("}\n")
.append("}\n");
}
}
}
private void generateGroups(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String qualifiedParentDtoClassName,
final String qualifiedParentCodecClassName,
final List tokens,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token groupToken = tokens.get(i);
if (groupToken.signal() != Signal.BEGIN_GROUP)
{
throw new IllegalStateException("tokens must begin with BEGIN_GROUP: token=" + groupToken);
}
final String groupName = groupToken.name();
final String groupClassName = formatDtoClassName(groupName);
final String formattedPropertyName = formatPropertyName(groupName);
final Token dimToken = tokens.get(i + 1);
if (dimToken.signal() != Signal.BEGIN_COMPOSITE)
{
throw new IllegalStateException("groups must start with BEGIN_COMPOSITE: token=" + dimToken);
}
final int sinceVersion = dimToken.version();
ctorArgs.append(indent).append("IReadOnlyList<")
.append(qualifiedParentDtoClassName).append(".").append(groupClassName).append("> ")
.append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(generateDocumentation(indent, groupToken))
.append(indent).append("public IReadOnlyList<").append(groupClassName).append("> ")
.append(formattedPropertyName)
.append(" { get; init; } = ").append(formattedPropertyName).append(";\n");
final StringBuilder groupCtorArgs = new StringBuilder();
final StringBuilder groupRecordBody = new StringBuilder();
i++;
i += tokens.get(i).componentTokenCount();
final String qualifiedDtoClassName = qualifiedParentDtoClassName + "." + groupClassName;
final String qualifiedCodecClassName =
qualifiedParentCodecClassName + "." + formatClassName(groupName) + "Group";
final List fields = new ArrayList<>();
i = collectFields(tokens, i, fields);
generateFields(groupRecordBody, groupCtorArgs, qualifiedCodecClassName, fields, indent + INDENT);
final List groups = new ArrayList<>();
i = collectGroups(tokens, i, groups);
generateGroups(groupRecordBody, groupCtorArgs, qualifiedDtoClassName,
qualifiedCodecClassName, groups, indent + INDENT);
final List varData = new ArrayList<>();
i = collectVarData(tokens, i, varData);
generateVarData(groupRecordBody, groupCtorArgs, varData, indent + INDENT);
generateDecodeListWith(
groupRecordBody, groupClassName, qualifiedCodecClassName, indent + INDENT);
final Predicate wasAddedAfterGroup = token ->
{
final boolean addedAfterParent = token.version() > sinceVersion;
if (addedAfterParent && token.signal() == Signal.BEGIN_VAR_DATA)
{
throw new IllegalStateException("Cannot extend var data inside a group.");
}
return addedAfterParent;
};
generateDecodeWith(
groupRecordBody, groupClassName, qualifiedCodecClassName, fields, groups, varData,
wasAddedAfterGroup, indent + INDENT);
generateEncodeWith(
groupRecordBody, groupClassName, qualifiedCodecClassName, fields, groups, varData,
indent + INDENT);
removeTrailingComma(groupCtorArgs);
sb.append("\n")
.append(generateDocumentation(indent, groupToken))
.append(indent).append("public sealed partial record ").append(groupClassName).append("(\n")
.append(groupCtorArgs)
.append(indent).append(")\n")
.append(indent).append("{\n")
.append(groupRecordBody)
.append(indent).append("}\n");
}
}
private void generateCompositeDecodeWith(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final List tokens,
final String indent)
{
sb.append("\n")
.append(indent).append("public static ").append(dtoClassName).append(" DecodeWith(")
.append(codecClassName).append(" codec)\n")
.append(indent).append("{\n");
sb.append(indent).append(INDENT).append("return new ").append(dtoClassName).append("(\n");
for (int i = 0; i < tokens.size(); )
{
final Token token = tokens.get(i);
generateFieldDecodeWith(
sb, CANNOT_EXTEND, token, token, codecClassName, indent + INDENT + INDENT);
i += tokens.get(i).componentTokenCount();
}
removeTrailingComma(sb);
sb.append(indent).append(INDENT).append(");\n");
sb.append(indent).append("}\n");
}
private void generateCompositeEncodeWith(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final List tokens,
final String indent)
{
sb.append("\n")
.append(indent).append("public static void EncodeWith(")
.append(codecClassName).append(" codec, ")
.append(dtoClassName).append(" dto)\n")
.append(indent).append("{\n");
for (int i = 0; i < tokens.size(); )
{
final Token token = tokens.get(i);
generateFieldEncodeWith(sb, codecClassName, token, token, indent + INDENT);
i += tokens.get(i).componentTokenCount();
}
sb.append(indent).append("}\n");
}
private void generateDecodeListWith(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final String indent)
{
sb.append("\n")
.append(indent).append("public static IReadOnlyList<").append(dtoClassName).append("> DecodeListWith(")
.append(codecClassName).append(" codec)\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("var ").append("list = new List<").append(dtoClassName)
.append(">(codec.Count);\n")
.append(indent).append(INDENT)
.append("while (codec.HasNext)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("var element = ").append(dtoClassName).append(".DecodeWith(codec.Next());\n")
.append(indent).append(INDENT).append(INDENT)
.append("list.Add(element);\n")
.append(indent).append(INDENT)
.append("}\n")
.append(indent).append(INDENT)
.append("return list.AsReadOnly();\n")
.append(indent).append("}\n");
}
private void generateDecodeWith(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final List fields,
final List groups,
final List varData,
final Predicate wasAddedAfterParent,
final String indent)
{
sb.append("\n")
.append(indent).append("public static ").append(dtoClassName)
.append(" DecodeWith(").append(codecClassName).append(" codec)\n")
.append(indent).append("{\n");
sb.append(indent).append(INDENT).append("return new ").append(dtoClassName).append("(\n");
generateMessageFieldsDecodeWith(sb, wasAddedAfterParent, fields, codecClassName, indent + INDENT + INDENT);
generateGroupsDecodeWith(sb, groups, indent + INDENT + INDENT);
generateVarDataDecodeWith(sb, varData, wasAddedAfterParent, indent + INDENT + INDENT);
removeTrailingComma(sb);
sb.append(indent).append(INDENT).append(");\n");
sb.append(indent).append("}\n");
}
private void generateDecodeFrom(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final String indent)
{
sb.append("\n")
.append(indent).append("public static ").append(dtoClassName)
.append(" DecodeFrom(DirectBuffer buffer, int offset, int length, ")
.append("int actingBlockLength, int actingVersion)\n")
.append(indent).append("{\n")
.append(indent).append(INDENT)
.append("var decoder = new ").append(codecClassName).append("();\n")
.append(indent).append(INDENT)
.append("decoder.WrapForDecode(buffer, offset, actingBlockLength, actingVersion);\n")
.append(indent).append(INDENT)
.append("return DecodeWith(decoder);\n")
.append(indent).append("}\n");
}
private void generateMessageFieldsDecodeWith(
final StringBuilder sb,
final Predicate wasAddedAfterParent,
final List tokens,
final String codecClassName,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token signalToken = tokens.get(i);
if (signalToken.signal() == Signal.BEGIN_FIELD)
{
final Token encodingToken = tokens.get(i + 1);
generateFieldDecodeWith(
sb, wasAddedAfterParent, signalToken, encodingToken, codecClassName, indent);
}
}
}
private void generateFieldDecodeWith(
final StringBuilder sb,
final Predicate wasAddedAfterParent,
final Token fieldToken,
final Token typeToken,
final String codecClassName,
final String indent)
{
switch (typeToken.signal())
{
case ENCODING:
generatePrimitiveDecodeWith(sb, fieldToken, typeToken, wasAddedAfterParent, codecClassName, indent);
break;
case BEGIN_SET:
generatePropertyDecodeWith(sb, fieldToken, wasAddedAfterParent, "0", null, indent);
break;
case BEGIN_ENUM:
final String enumName = formatClassName(typeToken.applicableTypeName());
final String nullValue = formatNamespace(ir.packageName()) + "." + enumName + ".NULL_VALUE";
generatePropertyDecodeWith(sb, fieldToken, wasAddedAfterParent, nullValue, null, indent);
break;
case BEGIN_COMPOSITE:
generateComplexDecodeWith(sb, fieldToken, typeToken, indent);
break;
default:
break;
}
}
private void generatePrimitiveDecodeWith(
final StringBuilder sb,
final Token fieldToken,
final Token typeToken,
final Predicate wasAddedAfterParent,
final String codecClassName,
final String indent)
{
if (typeToken.isConstantEncoding())
{
return;
}
final int arrayLength = typeToken.arrayLength();
if (arrayLength == 1)
{
final String codecNullValue = codecClassName + "." + formatPropertyName(fieldToken.name()) + "NullValue";
generatePropertyDecodeWith(sb, fieldToken, wasAddedAfterParent, "null", codecNullValue, indent);
}
else if (arrayLength > 1)
{
generateArrayDecodeWith(sb, fieldToken, typeToken, wasAddedAfterParent, indent);
}
}
private void generateArrayDecodeWith(
final StringBuilder sb,
final Token fieldToken,
final Token typeToken,
final Predicate wasAddedAfterParent,
final String indent)
{
if (fieldToken.isConstantEncoding())
{
return;
}
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
if (typeToken.encoding().primitiveType() == PrimitiveType.CHAR)
{
generateRecordPropertyAssignment(
sb,
fieldToken,
wasAddedAfterParent,
indent,
"codec.Get" + formattedPropertyName + "()",
"null",
null
);
}
else
{
generateRecordPropertyAssignment(
sb,
fieldToken,
wasAddedAfterParent,
indent,
"codec." + formattedPropertyName + "AsSpan().ToArray()",
"null",
null
);
}
}
private void generatePropertyDecodeWith(
final StringBuilder sb,
final Token fieldToken,
final Predicate wasAddedAfterParent,
final String dtoNullValue,
final String codecNullValue,
final String indent)
{
if (fieldToken.isConstantEncoding())
{
return;
}
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
generateRecordPropertyAssignment(
sb,
fieldToken,
wasAddedAfterParent,
indent,
"codec." + formattedPropertyName,
dtoNullValue,
codecNullValue
);
}
private void generateComplexDecodeWith(
final StringBuilder sb,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
final String dtoClassName = formatDtoClassName(typeToken.applicableTypeName());
sb.append(indent).append(formattedPropertyName).append(": ")
.append(dtoClassName).append(".DecodeWith(codec.")
.append(formattedPropertyName).append(")").append(",\n");
}
private void generateGroupsDecodeWith(
final StringBuilder sb,
final List tokens,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token groupToken = tokens.get(i);
if (groupToken.signal() != Signal.BEGIN_GROUP)
{
throw new IllegalStateException("tokens must begin with BEGIN_GROUP: token=" + groupToken);
}
final String groupName = groupToken.name();
final String formattedPropertyName = formatPropertyName(groupName);
final String groupDtoClassName = formatDtoClassName(groupName);
final Token dimToken = tokens.get(i + 1);
if (dimToken.signal() != Signal.BEGIN_COMPOSITE)
{
throw new IllegalStateException("groups must start with BEGIN_COMPOSITE: token=" + dimToken);
}
final int sinceVersion = dimToken.version();
generateRecordPropertyAssignment(
sb,
groupToken,
token -> token.version() > sinceVersion,
indent,
groupDtoClassName + ".DecodeListWith(codec." + formattedPropertyName + ")",
"new List<" + groupDtoClassName + ">(0).AsReadOnly()",
null
);
i++;
i += tokens.get(i).componentTokenCount();
final List fields = new ArrayList<>();
i = collectFields(tokens, i, fields);
final List groups = new ArrayList<>();
i = collectGroups(tokens, i, groups);
final List varData = new ArrayList<>();
i = collectVarData(tokens, i, varData);
}
}
private void generateVarDataDecodeWith(
final StringBuilder sb,
final List tokens,
final Predicate wasAddedAfterParent,
final String indent)
{
for (int i = 0; i < tokens.size(); i++)
{
final Token token = tokens.get(i);
if (token.signal() == Signal.BEGIN_VAR_DATA)
{
final String propertyName = token.name();
final Token varDataToken = Generators.findFirst("varData", tokens, i);
final String characterEncoding = varDataToken.encoding().characterEncoding();
final String formattedPropertyName = formatPropertyName(propertyName);
final String accessor = characterEncoding == null ?
"Get" + formattedPropertyName + "Bytes" :
"Get" + formattedPropertyName;
final String missingValue = characterEncoding == null ?
"new byte[0]" :
"\"\"";
sb.append(indent).append(formattedPropertyName).append(": ");
if (wasAddedAfterParent.test(token))
{
sb.append("codec.").append(formattedPropertyName).append("InActingVersion()");
sb.append(" ?\n");
sb.append(indent).append(INDENT).append("codec.").append(accessor).append("()")
.append(" :\n")
.append(indent).append(INDENT).append(missingValue).append(",\n");
}
else
{
sb.append("codec.").append(accessor).append("()").append(",\n");
}
}
}
}
private void generateRecordPropertyAssignment(
final StringBuilder sb,
final Token token,
final Predicate wasAddedAfterParent,
final String indent,
final String presentExpression,
final String notPresentExpression,
final String nullCodecValueOrNull)
{
final String propertyName = token.name();
final String formattedPropertyName = formatPropertyName(propertyName);
sb.append(indent).append(formattedPropertyName).append(": ");
boolean hasPresenceCondition = false;
// Unfortunately, we need to check whether the field is in the acting version,
// as the codec may incorrectly decode data for missing fields.
if (wasAddedAfterParent.test(token))
{
sb.append("codec.").append(formattedPropertyName).append("InActingVersion()");
hasPresenceCondition = true;
}
if (token.isOptionalEncoding() && null != nullCodecValueOrNull)
{
if (hasPresenceCondition)
{
sb.append(" && ");
}
sb.append("codec.").append(formattedPropertyName).append(" != ").append(nullCodecValueOrNull);
hasPresenceCondition = true;
}
if (hasPresenceCondition)
{
sb.append(" ?\n");
sb.append(indent).append(INDENT).append(presentExpression).append(" :\n")
.append(indent).append(INDENT).append(notPresentExpression).append(",\n");
}
else
{
sb.append(presentExpression).append(",\n");
}
}
private void generateEncodeWith(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final List fields,
final List groups,
final List varData,
final String indent)
{
sb.append("\n")
.append(indent).append("public static void EncodeWith(")
.append(codecClassName).append(" codec, ")
.append(dtoClassName).append(" dto)\n")
.append(indent).append("{\n");
generateFieldsEncodeWith(sb, codecClassName, fields, indent + INDENT);
generateGroupsEncodeWith(sb, groups, indent + INDENT);
generateVarDataEncodeWith(sb, varData, indent + INDENT);
sb.append(indent).append("}\n");
}
private void generateEncodeInto(
final StringBuilder sb,
final String dtoClassName,
final String codecClassName,
final String indent)
{
sb.append("\n")
.append(indent).append("public static int EncodeInto(")
.append("DirectBuffer buffer, int offset, ")
.append(dtoClassName).append(" dto)\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("var encoder = new ").append(codecClassName).append("();\n")
.append(indent).append(INDENT).append("encoder.WrapForEncode(buffer, offset);\n")
.append(indent).append(INDENT).append("EncodeWith(encoder, dto);\n")
.append(indent).append(INDENT).append("return encoder.Limit - offset;\n")
.append(indent).append("}\n");
sb.append("\n")
.append(indent).append("public static int EncodeWithHeaderInto(")
.append("DirectBuffer buffer, int offset, ")
.append(dtoClassName).append(" dto)\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("var encoder = new ").append(codecClassName).append("();\n")
.append(indent).append(INDENT)
.append("encoder.WrapForEncodeAndApplyHeader(buffer, offset, new MessageHeader());\n")
.append(indent).append(INDENT).append("EncodeWith(encoder, dto);\n")
.append(indent).append(INDENT).append("return encoder.Limit - offset;\n")
.append(indent).append("}\n");
}
private void generateFieldsEncodeWith(
final StringBuilder sb,
final String codecClassName,
final List tokens,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token signalToken = tokens.get(i);
if (signalToken.signal() == Signal.BEGIN_FIELD)
{
final Token encodingToken = tokens.get(i + 1);
generateFieldEncodeWith(sb, codecClassName, signalToken, encodingToken, indent);
}
}
}
private void generateFieldEncodeWith(
final StringBuilder sb,
final String codecClassName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
switch (typeToken.signal())
{
case ENCODING:
generatePrimitiveEncodeWith(sb, codecClassName, fieldToken, typeToken, indent);
break;
case BEGIN_SET:
case BEGIN_ENUM:
generateEnumEncodeWith(sb, fieldToken, indent);
break;
case BEGIN_COMPOSITE:
generateComplexEncodeWith(sb, fieldToken, typeToken, indent);
break;
default:
break;
}
}
private void generatePrimitiveEncodeWith(
final StringBuilder sb,
final String codecClassName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
if (typeToken.isConstantEncoding())
{
return;
}
final int arrayLength = typeToken.arrayLength();
if (arrayLength == 1)
{
generatePropertyEncodeWith(sb, codecClassName, fieldToken, indent);
}
else if (arrayLength > 1)
{
generateArrayEncodeWith(sb, fieldToken, typeToken, indent);
}
}
private void generateArrayEncodeWith(
final StringBuilder sb,
final Token fieldToken,
final Token typeToken,
final String indent)
{
if (fieldToken.isConstantEncoding())
{
return;
}
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
if (typeToken.encoding().primitiveType() == PrimitiveType.CHAR)
{
final String value = nullableConvertedExpression(fieldToken, "dto." + formattedPropertyName, "\"\"");
sb.append(indent).append("codec.Set").append(formattedPropertyName).append("(")
.append(value).append(");\n");
}
else
{
final String typeName = cSharpTypeName(typeToken.encoding().primitiveType());
sb.append(indent).append("new Span<").append(typeName).append(">(dto.").append(formattedPropertyName)
.append("?.ToArray()).CopyTo(codec.").append(formattedPropertyName).append("AsSpan());\n");
}
}
private String nullableConvertedExpression(
final Token fieldToken,
final String expression,
final String nullValue)
{
return fieldToken.isOptionalEncoding() ?
expression + " ?? " + nullValue :
expression;
}
private void generatePropertyEncodeWith(
final StringBuilder sb,
final String codecClassName,
final Token fieldToken,
final String indent)
{
if (fieldToken.isConstantEncoding())
{
return;
}
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
final String value = nullableConvertedExpression(
fieldToken,
"dto." + formattedPropertyName,
codecClassName + "." + formattedPropertyName + "NullValue");
sb.append(indent).append("codec.").append(formattedPropertyName).append(" = ")
.append(value).append(";\n");
}
private void generateEnumEncodeWith(
final StringBuilder sb,
final Token fieldToken,
final String indent)
{
if (fieldToken.isConstantEncoding())
{
return;
}
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
sb.append(indent).append("codec.").append(formattedPropertyName).append(" = dto.")
.append(formattedPropertyName).append(";\n");
}
private void generateComplexEncodeWith(
final StringBuilder sb,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String propertyName = fieldToken.name();
final String formattedPropertyName = formatPropertyName(propertyName);
final String dtoClassName = formatDtoClassName(typeToken.applicableTypeName());
sb.append(indent)
.append(dtoClassName).append(".EncodeWith(codec.").append(formattedPropertyName)
.append(", dto.").append(formattedPropertyName).append(");\n");
}
private void generateGroupsEncodeWith(
final StringBuilder sb,
final List tokens,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token groupToken = tokens.get(i);
if (groupToken.signal() != Signal.BEGIN_GROUP)
{
throw new IllegalStateException("tokens must begin with BEGIN_GROUP: token=" + groupToken);
}
final String groupName = groupToken.name();
final String formattedPropertyName = formatPropertyName(groupName);
final String groupCodecVarName = groupName + "Codec";
final String groupDtoClassName = formatDtoClassName(groupName);
sb.append("\n")
.append(indent).append("var ").append(groupCodecVarName)
.append(" = codec.").append(formattedPropertyName)
.append("Count(dto.").append(formattedPropertyName).append(".Count);\n\n")
.append(indent).append("foreach (var group in dto.").append(formattedPropertyName).append(")\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append(groupDtoClassName).append(".EncodeWith(")
.append(groupCodecVarName)
.append(".Next()").append(", group);\n")
.append(indent).append("}\n\n");
i++;
i += tokens.get(i).componentTokenCount();
final List fields = new ArrayList<>();
i = collectFields(tokens, i, fields);
final List groups = new ArrayList<>();
i = collectGroups(tokens, i, groups);
final List varData = new ArrayList<>();
i = collectVarData(tokens, i, varData);
}
}
private void generateVarDataEncodeWith(
final StringBuilder sb,
final List tokens,
final String indent)
{
for (final Token token : tokens)
{
if (token.signal() == Signal.BEGIN_VAR_DATA)
{
final String propertyName = token.name();
final String formattedPropertyName = formatPropertyName(propertyName);
sb.append(indent).append("codec.Set").append(formattedPropertyName)
.append("(dto.").append(formattedPropertyName).append(");\n");
}
}
}
private void generateDisplay(
final StringBuilder sb,
final String codecClassName,
final String wrapMethod,
final String actingVersion,
final String indent)
{
sb.append("\n")
.append(indent).append("public string ToSbeString()\n")
.append(indent).append("{\n")
.append(indent).append(INDENT)
.append("var buffer = new DirectBuffer(new byte[128], (ignored, newSize) => new byte[newSize]);\n")
.append(indent).append(INDENT).append("var codec = new ").append(codecClassName).append("();\n")
.append(indent).append(INDENT).append("codec.");
sb.append(wrapMethod).append("(buffer, 0");
if (null != actingVersion)
{
sb.append(", ").append(actingVersion);
}
sb.append(");\n");
sb.append(indent).append(INDENT).append("EncodeWith(codec, this);\n")
.append(indent).append(INDENT).append("StringBuilder sb = new StringBuilder();\n")
.append(indent).append(INDENT).append("codec.BuildString(sb);\n")
.append(indent).append(INDENT).append("return sb.ToString();\n")
.append(indent).append("}\n");
}
private void generateFields(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String codecClassName,
final List tokens,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token signalToken = tokens.get(i);
if (signalToken.signal() == Signal.BEGIN_FIELD)
{
final Token encodingToken = tokens.get(i + 1);
final String propertyName = signalToken.name();
switch (encodingToken.signal())
{
case ENCODING:
generatePrimitiveProperty(
sb, ctorArgs, codecClassName, propertyName, signalToken, encodingToken, indent);
break;
case BEGIN_ENUM:
generateEnumProperty(sb, ctorArgs, propertyName, signalToken, encodingToken, indent);
break;
case BEGIN_SET:
generateBitSetProperty(sb, ctorArgs, propertyName, signalToken, encodingToken, indent);
break;
case BEGIN_COMPOSITE:
generateCompositeProperty(sb, ctorArgs, propertyName, signalToken, encodingToken, indent);
break;
default:
break;
}
}
}
}
private void generateCompositeProperty(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String compositeName = formatDtoClassName(typeToken.applicableTypeName());
final String formattedPropertyName = formatPropertyName(propertyName);
ctorArgs.append(indent).append(compositeName).append(" ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public ").append(compositeName)
.append(" ").append(formattedPropertyName)
.append(" { get; init; } = ").append(formattedPropertyName).append(";\n");
}
private void generateBitSetProperty(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String enumName = formatClassName(typeToken.applicableTypeName());
final String formattedPropertyName = formatPropertyName(propertyName);
ctorArgs.append(indent).append(enumName).append(" ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public ").append(enumName).append(" ")
.append(formattedPropertyName).append(" { get; init; } = ")
.append(formattedPropertyName).append(";\n");
}
private void generateEnumProperty(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String enumName = formatClassName(typeToken.applicableTypeName());
final String formattedPropertyName = formatPropertyName(propertyName);
if (fieldToken.isConstantEncoding())
{
final String constValue = fieldToken.encoding().constValue().toString();
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public static ").append(enumName).append(" ")
.append(formattedPropertyName).append("\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("get { return ")
.append(formatNamespace(ir.packageName())).append(".").append(constValue)
.append("; }\n")
.append(indent).append("}\n");
}
else
{
ctorArgs.append(indent).append(enumName).append(" ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public ").append(enumName).append(" ")
.append(formattedPropertyName).append(" { get; init; } = ").append(formattedPropertyName).append(";\n");
}
}
private void generatePrimitiveProperty(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String codecClassName,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
if (typeToken.isConstantEncoding())
{
generateConstPropertyMethods(sb, propertyName, fieldToken, typeToken, indent);
}
else
{
generatePrimitivePropertyMethods(sb, ctorArgs, codecClassName, propertyName, fieldToken, typeToken, indent);
}
}
private void generatePrimitivePropertyMethods(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String codecClassName,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final int arrayLength = typeToken.arrayLength();
if (arrayLength == 1)
{
generateSingleValueProperty(sb, ctorArgs, codecClassName, propertyName, fieldToken, typeToken, indent);
}
else if (arrayLength > 1)
{
generateArrayProperty(sb, ctorArgs, codecClassName, propertyName, fieldToken, typeToken, indent);
}
}
private void generateArrayProperty(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String codecClassName,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String formattedPropertyName = formatPropertyName(propertyName);
if (typeToken.encoding().primitiveType() == PrimitiveType.CHAR)
{
ctorArgs.append(indent).append("string ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public string ")
.append(formattedPropertyName).append(" { get; init; } = ")
.append(formattedPropertyName).append(";\n");
}
else
{
final String typeName = cSharpTypeName(typeToken.encoding().primitiveType());
final String fieldName = "_" + toLowerFirstChar(propertyName);
final String nullableSuffix = fieldToken.isOptionalEncoding() ? "?" : "";
final String listTypeName = "IReadOnlyList<" + typeName + ">" + nullableSuffix;
ctorArgs.append(indent).append(listTypeName).append(" ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(indent).append("private ").append(listTypeName).append(" ").append(fieldName)
.append(" = Validate").append(formattedPropertyName).append("(").append(formattedPropertyName)
.append(");\n");
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public ").append(listTypeName).append(" ")
.append(formattedPropertyName).append("\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("get => ").append(fieldName).append(";\n")
.append(indent).append(INDENT).append("init => ").append(fieldName).append(" = Validate")
.append(formattedPropertyName).append("(value);\n")
.append(indent).append("}\n");
sb.append("\n")
.append(indent).append("private static ").append(listTypeName).append(" Validate")
.append(formattedPropertyName).append("(").append(listTypeName).append(" value)\n")
.append(indent).append("{\n");
if (fieldToken.isOptionalEncoding())
{
sb.append(indent).append(INDENT)
.append("if (value == null)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("return null;\n")
.append(indent).append(INDENT)
.append("}\n");
}
sb.append(indent).append(INDENT)
.append("if (value.Count > ").append(codecClassName).append(".")
.append(formattedPropertyName).append("Length)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("throw new ArgumentException(\"too many elements: \" + value.Count);\n")
.append(indent).append(INDENT)
.append("}\n")
.append(indent).append(INDENT)
.append("return value;\n")
.append(indent).append("}\n");
}
}
private void generateSingleValueProperty(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String codecClassName,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
final String nullableSuffix = fieldToken.isOptionalEncoding() ? "?" : "";
final Encoding encoding = typeToken.encoding();
final String typeName = cSharpTypeName(encoding.primitiveType()) + nullableSuffix;
final String formattedPropertyName = formatPropertyName(propertyName);
final String fieldName = "_" + toLowerFirstChar(propertyName);
ctorArgs.append(indent).append(typeName).append(" ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(indent).append("private ").append(typeName).append(" ").append(fieldName)
.append(" = Validate").append(formattedPropertyName)
.append("(").append(formattedPropertyName).append(");\n");
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public ").append(typeName).append(" ")
.append(formattedPropertyName).append("\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("get => ").append(fieldName).append(";\n")
.append(indent).append(INDENT).append("init => ").append(fieldName).append(" = Validate")
.append(formattedPropertyName).append("(value);\n")
.append(indent).append("}\n");
sb.append("\n")
.append(indent).append("private static ").append(typeName).append(" Validate")
.append(formattedPropertyName).append("(").append(typeName).append(" value)\n")
.append(indent).append("{\n");
if (fieldToken.isOptionalEncoding())
{
sb.append(indent).append(INDENT)
.append("if (value == null)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("return null;\n")
.append(indent).append(INDENT)
.append("}\n");
sb.append(indent).append(INDENT)
.append("if (value == ").append(codecClassName).append(".")
.append(formattedPropertyName).append("NullValue)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("throw new ArgumentException(\"null value is reserved: \" + value);\n")
.append(indent).append(INDENT)
.append("}\n");
}
final boolean mustPreventLesser = !encoding.applicableMinValue().equals(encoding.primitiveType().minValue());
if (mustPreventLesser)
{
sb.append(indent).append(INDENT)
.append("if (value < ")
.append(codecClassName).append(".").append(formattedPropertyName).append("MinValue)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("throw new ArgumentException(\"value is less than minimum allowed: \" + value);\n")
.append(indent).append(INDENT)
.append("}\n");
}
final boolean mustPreventGreater = !encoding.applicableMaxValue().equals(encoding.primitiveType().maxValue());
if (mustPreventGreater)
{
sb.append(indent).append(INDENT)
.append("if (value > ")
.append(codecClassName).append(".").append(formattedPropertyName).append("MaxValue)\n")
.append(indent).append(INDENT)
.append("{\n")
.append(indent).append(INDENT).append(INDENT)
.append("throw new ArgumentException(\"value is greater than maximum allowed: \" + value);\n")
.append(indent).append(INDENT)
.append("}\n");
}
sb.append(indent).append(INDENT)
.append("return value;\n")
.append(indent).append("}\n");
}
private void generateConstPropertyMethods(
final StringBuilder sb,
final String propertyName,
final Token fieldToken,
final Token typeToken,
final String indent)
{
if (typeToken.encoding().primitiveType() == PrimitiveType.CHAR)
{
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public static string ").append(toUpperFirstChar(propertyName)).append("\n")
.append(indent).append("{\n")
.append(indent).append(INDENT)
.append("get { return \"").append(typeToken.encoding().constValue().toString()).append("\"; }\n")
.append(indent).append("}\n");
}
else
{
final String literalValue =
generateLiteral(typeToken.encoding().primitiveType(), typeToken.encoding().constValue().toString());
sb.append("\n")
.append(generateDocumentation(indent, fieldToken))
.append(indent).append("public static ").append(cSharpTypeName(typeToken.encoding().primitiveType()))
.append(" ").append(formatPropertyName(propertyName)).append("\n")
.append(indent).append("{\n")
.append(indent).append(INDENT).append("get { return ").append(literalValue).append("; }\n")
.append(indent).append("}\n");
}
}
private void generateVarData(
final StringBuilder sb,
final StringBuilder ctorArgs,
final List tokens,
final String indent)
{
for (int i = 0, size = tokens.size(); i < size; i++)
{
final Token token = tokens.get(i);
if (token.signal() == Signal.BEGIN_VAR_DATA)
{
final String propertyName = token.name();
final Token varDataToken = Generators.findFirst("varData", tokens, i);
final String characterEncoding = varDataToken.encoding().characterEncoding();
final String dtoType = characterEncoding == null ? "byte[]" : "string";
final String formattedPropertyName = formatPropertyName(propertyName);
ctorArgs.append(indent).append(dtoType).append(" ").append(formattedPropertyName).append(",\n");
sb.append("\n")
.append(indent).append("public ").append(dtoType).append(" ")
.append(formattedPropertyName).append(" { get; init; } = ")
.append(formattedPropertyName).append(";\n");
}
}
}
private String formatDtoClassName(final String name)
{
return formatClassName(name + "Dto");
}
private void generateDtosForTypes() throws IOException
{
for (final List tokens : ir.types())
{
switch (tokens.get(0).signal())
{
case BEGIN_COMPOSITE:
generateComposite(tokens);
break;
default:
break;
}
}
}
private void generateComposite(final List tokens) throws IOException
{
final String name = tokens.get(0).applicableTypeName();
final String className = formatDtoClassName(name);
final String codecClassName = formatClassName(name);
try (Writer out = outputManager.createOutput(className))
{
out.append(generateFileHeader(ir.applicableNamespace(),
"#nullable enable\n",
"using System.Collections.Generic;\n",
"using System.Linq;\n"));
out.append(generateDocumentation(BASE_INDENT, tokens.get(0)));
final StringBuilder sb = new StringBuilder();
final StringBuilder ctorArgs = new StringBuilder();
final List compositeTokens = tokens.subList(1, tokens.size() - 1);
generateCompositePropertyElements(sb, ctorArgs, codecClassName, compositeTokens, BASE_INDENT + INDENT);
generateCompositeDecodeWith(sb, className, codecClassName, compositeTokens, BASE_INDENT + INDENT);
generateCompositeEncodeWith(sb, className, codecClassName, compositeTokens, BASE_INDENT + INDENT);
generateDisplay(sb, codecClassName, "Wrap", codecClassName + ".SbeSchemaVersion", BASE_INDENT + INDENT);
removeTrailingComma(ctorArgs);
out.append(BASE_INDENT).append("public sealed partial record ").append(className).append("(\n")
.append(ctorArgs)
.append(BASE_INDENT).append(")\n")
.append(BASE_INDENT).append("{")
.append(sb)
.append(BASE_INDENT).append("}\n")
.append("}\n");
}
}
private static void removeTrailingComma(final StringBuilder ctorArgs)
{
if (ctorArgs.length() < 2)
{
return;
}
if (ctorArgs.charAt(ctorArgs.length() - 1) != '\n')
{
return;
}
if (ctorArgs.charAt(ctorArgs.length() - 2) != ',')
{
return;
}
ctorArgs.setLength(ctorArgs.length() - 2);
ctorArgs.append("\n");
}
private void generateCompositePropertyElements(
final StringBuilder sb,
final StringBuilder ctorArgs,
final String codecClassName,
final List tokens,
final String indent)
{
for (int i = 0; i < tokens.size(); )
{
final Token token = tokens.get(i);
final String propertyName = formatPropertyName(token.name());
switch (token.signal())
{
case ENCODING:
generatePrimitiveProperty(sb, ctorArgs, codecClassName, propertyName, token, token, indent);
break;
case BEGIN_ENUM:
generateEnumProperty(sb, ctorArgs, propertyName, token, token, indent);
break;
case BEGIN_SET:
generateBitSetProperty(sb, ctorArgs, propertyName, token, token, indent);
break;
case BEGIN_COMPOSITE:
generateCompositeProperty(sb, ctorArgs, propertyName, token, token, indent);
break;
default:
break;
}
i += tokens.get(i).componentTokenCount();
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy