org.fusesource.stompjms.channel.StompFrame Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stompjms-client Show documentation
STOMP-JMS is a JMS implementation using STOMP as the wire protocol
There is a newer version: 1.19
/**
 * Copyright (C) 2010-2011, FuseSource Corp.  All rights reserved.
 *
 *     http://fusesource.com
 *
 * The software in this package is published under the terms of the
 * CDDL license a copy of which has been included with this distribution
 * in the license.txt file.
 */

package org.fusesource.stompjms.channel;


import org.fusesource.hawtbuf.AsciiBuffer;
import org.fusesource.hawtbuf.Buffer;
import org.fusesource.hawtbuf.DataByteArrayOutputStream;

import java.io.DataOutput;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.Map;

import static org.fusesource.stompjms.channel.Stomp.*;

/**
 * Represents all the data in a STOMP frame.
 *
 * @author chirino
 */
public class StompFrame {

    public static final Buffer NO_DATA = new Buffer(new byte[]{});

    public AsciiBuffer action;
    public Map headers = new HashMap();
    public Buffer content = NO_DATA;

    public StompFrame(AsciiBuffer command) {
        this(command, null, (Buffer) null);
    }

    public StompFrame(AsciiBuffer command, Map headers) {
        this(command, headers, (Buffer) null);
    }

    public StompFrame(AsciiBuffer command, Map headers, Buffer data) {
        this.action = command;
        if (headers != null)
            this.headers = headers;
        if (data != null)
            this.content = data;
    }

    public StompFrame(AsciiBuffer command, Map headers, byte[] data) {
        this(command, headers, (data != null ? new Buffer(data) : null));
    }

    public StompFrame() {
    }

    public StompFrame clone() {
        return new StompFrame(action, new HashMap(headers), content);
    }

    public AsciiBuffer getAction() {
        return action;
    }

    public void setAction(AsciiBuffer command) {
        this.action = command;
    }

    public Buffer getContent() {
        return this.content;
    }

    public String getBody() {
        try {
            Buffer b = getContent();
            if (b != null) {
                return new String(b.getData(), b.getOffset(), b.getLength(), "UTF-8");
            }
        } catch (UnsupportedEncodingException e) {
        }
        return new String("");
    }

    public void setContent(Buffer data) {
        this.content = data;
    }

    public void clearContent() {
        this.content = NO_DATA;
    }

    public Map getHeaders() {
        return headers;
    }

    public Buffer toBuffer() {
        return toBuffer(true);
    }

    public Buffer toBuffer(boolean includeBody) {
        try {
            DataByteArrayOutputStream out = new DataByteArrayOutputStream();
            write(out, includeBody);
            return out.toBuffer();
        } catch (IOException e) {
            throw new RuntimeException(e); // not expected to occur.
        }
    }

    private void write(DataOutput out, Buffer buffer) throws IOException {
        out.write(buffer.data, buffer.offset, buffer.length);
    }

    public void write(DataOutput out) throws IOException {
        write(out, true);
    }

    public void write(DataOutput out, boolean includeBody) throws IOException {
        write(out, action);
        out.writeByte(NEWLINE_BYTE);
        for (Map.Entry entry: headers.entrySet()){
            write(out, entry.getKey());
            out.writeByte(COLON_BYTE);
            write(out, entry.getValue());
            out.writeByte(NEWLINE_BYTE);
        }

        //denotes end of headers with a new line
        out.writeByte(NEWLINE_BYTE);
        if(includeBody) {
            if (content != null) {
                write(out, content);
            }
            out.writeByte(NULL_BYTE);
        }
    }

    public String toString() {
        return toBuffer(false).ascii().toString();
    }
}