All Downloads are FREE. Search and download functionalities are using the official Maven repository.

quickfix.mina.message.FIXMessageEncoder Maven / Gradle / Ivy

The newest version!
/*******************************************************************************
 * Copyright (c) quickfixengine.org  All rights reserved.
 *
 * This file is part of the QuickFIX FIX Engine
 *
 * This file may be distributed under the terms of the quickfixengine.org
 * license as defined by quickfixengine.org and appearing in the file
 * LICENSE included in the packaging of this file.
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING
 * THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A
 * PARTICULAR PURPOSE.
 *
 * See http://www.quickfixengine.org/LICENSE for licensing information.
 *
 * Contact [email protected] if any conditions of this licensing
 * are not clear to you.
 ******************************************************************************/

package quickfix.mina.message;

import java.io.UnsupportedEncodingException;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecException;
import org.apache.mina.filter.codec.ProtocolEncoderOutput;
import org.apache.mina.filter.codec.demux.MessageEncoder;
import org.quickfixj.CharsetSupport;

import quickfix.Message;

/**
 * Encodes a Message object or message string as a byte array to be
 * transmitted on MINA connection.
 */
public class FIXMessageEncoder implements MessageEncoder {

    private static final Set> TYPES =
            new HashSet<>(Arrays.>asList(Message.class, String.class));
    private final String charsetEncoding;

    public FIXMessageEncoder() {
        charsetEncoding = CharsetSupport.getCharset();
    }

    public static Set> getMessageTypes() {
        return TYPES;
    }

    private byte[] toBytes(String str) throws ProtocolCodecException {
        try {
            return str.getBytes(charsetEncoding);
        } catch (UnsupportedEncodingException e) {
            throw new ProtocolCodecException(e);
        }
    }

    @Override
    public void encode(IoSession session, Object message, ProtocolEncoderOutput out)
            throws ProtocolCodecException {
        // get message bytes
        byte[] bytes;
        if (message instanceof String) {
            bytes = toBytes((String) message);
        } else if (message instanceof Message) {
            bytes = toBytes(message.toString());
        } else {
            throw new ProtocolCodecException("Invalid FIX message object type: "
                    + message.getClass());
        }
        // write bytes to buffer and output it
        IoBuffer buffer = IoBuffer.allocate(bytes.length);
        buffer.put(bytes);
        buffer.flip();
        out.write(buffer);
    }
}