com.ziqni.admin.sdk.streaming.stomp.StompFrame Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ziqni-admin-sdk Show documentation
ZIQNI Admin SDK Java Client
There is a newer version: 1.0.24
/*
 * Copyright (c) 2024. ZIQNI LTD registered in England and Wales, company registration number-09693684
 */

package com.ziqni.admin.sdk.streaming.stomp;

import java.util.Map;
import java.util.HashMap;

public class StompFrame {

    private final String body;
    private final StompCommand command;
    private final StompHeaders headers;

    public StompFrame(StompCommand command, Map headers, String body) {
        this.command = command;
        this.headers = new StompHeaders(headers);
        this.body = body;
    }

    public StompCommand getCommand() {
        return command;
    }

    public StompHeaders getHeaders() {
        return headers;
    }

    public String getBody() {
        return body;
    }

    public String getDestination() {
        return headers.getDestination();
    }

    public static StompFrame parse(String frame) {
        String[] lines = frame.split("\n");
        if (lines.length == 0) {
            throw new IllegalArgumentException("Invalid STOMP frame: empty");
        }

        // First line is the command
        StompCommand command = null;
        try {
            command = StompCommand.fromString(lines[0].trim());
        }
        catch (IllegalArgumentException e) {
            command = StompCommand.NOT_A_VALID_STOMP_COMMAND;
            return new StompFrame(command, Map.of(), frame);
        }

        // Parse headers
        Map headers = new HashMap<>();
        int i = 1;
        while (i < lines.length && !lines[i].isEmpty()) {
            String line = lines[i];
            int colonIndex = line.indexOf(':');
            if (colonIndex == -1) {
                throw new IllegalArgumentException("Invalid STOMP header: " + line);
            }
            String key = line.substring(0, colonIndex).trim();
            String value = line.substring(colonIndex + 1).trim();
            headers.put(key, value);
            i++;
        }

        // Skip the empty line (delimiter between headers and body)
        i++;

        // Remaining content is the body
        StringBuilder bodyBuilder = new StringBuilder();
        while (i < lines.length) {
            bodyBuilder.append(lines[i]).append("\n");
            i++;
        }

        // Remove the trailing null character (STOMP frame terminator)
        String body = bodyBuilder.toString().replaceAll("\0", "").trim();

        return new StompFrame(command, headers, body);
    }

    @Override
    public String toString() {
        return "StompFrame{" +
                "command=" + command +
                ", headers=" + headers +
                ", body='" + body + '\'' +
                '}';
    }
}