EDIFACTメッセージ処理をJavaでスクラッチ実装した試行錯誤

private static final ObjectMapper MAPPER = new ObjectMapper();

/***
 * EDIFACTデータを読み込み、セグメントのリストを取得する.
 *
 * @param edifactStr EDIFACT形式の文字列データ
 * @throws IOException 入出力例外
 * @return セグメントのリスト
 */
public static List<JsonNode> readSegments(String edifactStr) throws IOException {
    var factory = EDIInputFactory.newFactory();

    // 実データは標準仕様と細部の違いがあるため
    factory.setProperty(EDIInputFactory.EDI_VALIDATE_CONTROL_STRUCTURE, false);
    var stream = new ByteArrayInputStream(edifactStr.getBytes(StandardCharsets.UTF_8));

    JsonNode rootNode;
    try (var ediReader = factory.createEDIStreamReader(stream);
         var jsonParser = factory.createJsonParser(ediReader, JsonParser.class)) {
        // StaediのReaderをJacksonのParserとして扱う
        rootNode = MAPPER.readValue(jsonParser, JsonNode.class);
    }

    // 各セグメントをJsonノードのリストとして取得
    List<JsonNode> segmentList = new ArrayList<>();
    for (JsonNode node : rootNode.path("data")) {
        segmentList.add(node);
    }

    return segmentList;
}

/**
 * EDIFACTデータのカーソル操作クラス.
 */
public class EdifactCursor {
    private final List<JsonNode> nodes;
    private int position = 0;

    /**
     * コンストラクタ.
     *
     * @param nodes EDIFACTセグメントのリスト
     */
    public EdifactCursor(List<JsonNode> nodes) {
        this.nodes = nodes;
    }

    /**
     * 現在位置のセグメント名を確認.
     *
     * @return セグメント名
     */
    public String peek() {
        if (position >= nodes.size()) return null;
        return nodes.get(position).path("name").asText();
    }

    /**
     * 指定したタグであれば取得して位置を進める.
     *
     * @param expectedTag 期待するセグメント名
     * @return JsonNode
     */
    public JsonNode match(String expectedTag) {
        if (expectedTag.equals(peek())) {
            return nodes.get(position++);
        }
        return null;
    }
}

/**
 * ツリー構造のノードとなるグループクラス
 */
public class EdifactGroup {
    private final String name;
    private final List<JsonNode> segments = new ArrayList<>();
    private final List<EdifactGroup> children = new ArrayList<>();

    /**
     * コンストラクタ.
     *
     * @param name グループ名
     */
    public EdifactGroup(String name) {
        this.name = name;
    }

    /**
     * ルートのパース処理.
     *
     * @param cursor EDIFACTカーソル
     */
    public static EdifactGroup parseInterchange(EdifactCursor cursor) {
        EdifactGroup interchange = new EdifactGroup("Interchange")
                .addSegment(cursor.match("UNB"));
        while ("UNH".equals(cursor.peek())) {
            interchange.addChild(parseMessage(cursor));
        }
        return interchange.addSegment(cursor.match("UNZ"));
    }

    /**
     * Message単位のパース.
     *
     * @param cursor EDIFACTカーソル
     */
    private static EdifactGroup parseMessage(EdifactCursor cursor) {
        EdifactGroup message = new EdifactGroup("Message")
                .addSegment(cursor.match("UNH"))
                .addSegment(cursor.match("BGM"))
                .addSegment(cursor.match("DTM"));

        while ("NAD".equals(cursor.peek())) {
            message.addSegment(cursor.match("NAD"));
        }

        return message.addChild(parseDetailSection(cursor))
                .addSegment(cursor.match("UNS"))
                .addSegment(cursor.match("CNT"))
                .addSegment(cursor.match("UNT"));
    }

    /**
     * 明細行単体のパース.
     *
     * @param cursor EDIFACTカーソル
     * @return 明細行グループ
     */
    private static EdifactGroup parseLineItem(EdifactCursor cursor) {
        return new EdifactGroup("LineItem")
                .addSegment(cursor.match("LIN"))
                .addSegment(cursor.match("QTY"))
                .addSegment(cursor.match("PRI"));
    }

    /**
     * 明細セクションのパース.
     *
     * @param cursor EDIFACTカーソル
     * @return 明細セクショングループ
     */
    private static EdifactGroup parseDetailSection(EdifactCursor cursor) {
        EdifactGroup detail = new EdifactGroup("Detail");
        while ("LIN".equals(cursor.peek())) {
            detail.addChild(parseLineItem(cursor));
        }
        return detail;
    }

    /**
     * セグメント追加.
     *
     * @param seg 追加するセグメント
     * @return 追加したセグメント
     */
    public EdifactGroup addSegment(JsonNode seg) {
        if (seg != null) {
            this.segments.add(seg);
        }
        return this;
    }

    /**
     * 子グループ追加.
     *
     * @param group 追加する子グループ
     * @return 追加した子グループ
     */
    public EdifactGroup addChild(EdifactGroup group) {
        if (group != null && !group.isEmpty()) {
            this.children.add(group);
        }
        return this;
    }

    private boolean isEmpty() {
        return segments.isEmpty() && children.isEmpty();
    }
}