zmq.Msg Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeromq Show documentation
zeromq POJO implementation
The newest version!
/*
    Copyright (c) 2007-2012 iMatix Corporation
    Copyright (c) 2009-2011 250bpm s.r.o.
    Copyright (c) 2007-2011 Other contributors as noted in the AUTHORS file

    This file is part of 0MQ.

    0MQ is free software; you can redistribute it and/or modify it under
    the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.

    0MQ is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with this program.  If not, see .
*/

package zmq;

import java.nio.ByteBuffer;
import java.util.Arrays;

public class Msg {

    //  Size in bytes of the largest message that is still copied around
    //  rather than being reference-counted.
    
    public final static int more = 1;
    public final static int identity = 64;
    public final static int shared = 128;
    
    private final static byte type_min = 101;
    private final static byte type_vsm = 102;
    private final static byte type_lmsg = 103;
    private final static byte type_delimiter = 104;
    private final static byte type_max = 105;
    
    private byte type;
    private int flags;
    private int size;
    private byte[] header;
    private byte[] data;
    private ByteBuffer buf;
    
    public Msg() {
        init(type_vsm);
    }

    public Msg(boolean buffered) {
        if (buffered)
            init(type_lmsg);
        else
            init(type_vsm);
    }

    public Msg(int size) {
        init(type_vsm);
        size(size);
    }
    
    public Msg(int size, boolean buffered) {
        if (buffered)
            init(type_lmsg);
        else
            init(type_vsm);
        size(size);
    }

    
    public Msg(Msg m) {
        clone(m);
    }
    
    public Msg(byte[] src) {
        this(src, false);
    }
    
    public Msg(String src) {
        this(src.getBytes(), false);
    }
    
    public Msg(byte[] src, boolean copy) {
        this();
        if (src != null) {
            size = src.length;
            if (copy)
                data = Arrays.copyOf(src, src.length);
            else
                data = src;
        }
    }
    
    public Msg (ByteBuffer src) 
    {
        init (type_lmsg);
        buf = src.duplicate ();
        buf.rewind ();
        size = buf.remaining ();
    }
    
    public final boolean is_identity ()
    {
        return (flags & identity) == identity ;
    }

    public final boolean is_delimiter ()
    {
        return type == type_delimiter;
    }


    public final boolean check ()
    {
         return type >= type_min && type <= type_max;
    }

    private final void init(byte type_) {
        type = type_;
        flags = 0;
        size = 0;
        data = null;
        buf = null;
        header = null;
    }

    public final void size (int size_)
    {
        size = size_;
        if (type == type_lmsg) {
            flags = 0;
            
            buf = ByteBuffer.allocate(size_);
            data = null;
        }
        else {
            flags = 0;
            data = new byte[size_];
            buf = null;
        }
    }

    public final int flags ()
    {
        return flags;
    }
    
    public final boolean has_more ()
    {
        return (flags & Msg.more) > 0;
    }
    
    public final byte type ()
    {
        return type;
    }
    
    public final void set_flags (int flags_)
    {
        flags = flags | flags_;
    }


    
    public final void init_delimiter() {
        type = type_delimiter;
        flags = 0;
    }

    
    public final byte[] data ()
    {
        if (data == null && type == type_lmsg) 
            data = buf.array();
        return data;
    }
    
    public final ByteBuffer buf()
    {
        if (buf == null && type != type_lmsg)
            buf = ByteBuffer.wrap(data);
        return buf;
    }
    
    
    public final int size ()
    {
        return size;
    }
    

    public final int header_size ()
    {
        if (header == null) {
            if (size < 255)
                return 2;
            else
                return 10;
        }
        else if (header[0] == 0xff)
            return 10;
        else
            return 2;
    }
    
    public final byte[] header() {
        if (header == null) {
            if (size < 255) {
                header = new byte[2];
                header[0] = (byte)size;
                header[1] = (byte)flags;
            } else {
                header = new byte[10];
                ByteBuffer hbuf = ByteBuffer.wrap(header);

                hbuf.put((byte)0xff);
                hbuf.put((byte)flags);
                hbuf.putLong((long)size);
            }
        } 
        return header;
        
    }
    public final ByteBuffer header_buf()
    {
        return ByteBuffer.wrap(header());
    }

    public final void close ()
    {
        if (!check ()) {
            throw new IllegalStateException();
        }

        init(type_vsm);
    }



    @Override
    public String toString () {
        return super.toString() + "[" + type + "," + size + "," + flags + "]";
    }

    private void clone (Msg m) {
        type = m.type;
        flags = m.flags;
        size = m.size;
        buf = m.buf;
        data = m.data;
    }

    public final void reset_flags (int f) {
        flags = flags &~ f;
    }
    
    public final void put(byte[] src, int i) {
        
        if (src == null)
            return;

        System.arraycopy(src, 0, data, i, src.length);
    }
    
    public final void put(byte[] src, int i, int len_) {
        
        if (len_ == 0 || src == null)
            return;
        
        System.arraycopy(src, 0, data, i, len_);
    }

    public final boolean is_vsm() {
        return type == type_vsm;
    }

    
    public final void put(byte b) {
        data[0] = b;
    }

    public final void put(byte b, int i) {
        data[i] = b;
    }

    public final void put(String str, int i) {
        put(str.getBytes(), i);
    }

    public final void put(Msg data, int i) {
        put(data.data, i);
    }
    

}