001/* ***** BEGIN LICENSE BLOCK *****
002 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
003 *
004 * The contents of this file are subject to the Mozilla Public License Version
005 * 1.1 (the "License"); you may not use this file except in compliance with
006 * the License. You may obtain a copy of the License at
007 * http://www.mozilla.org/MPL/
008 *
009 * Software distributed under the License is distributed on an "AS IS" basis,
010 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
011 * for the specific language governing rights and limitations under the
012 * License.
013 *
014 * The Original Code is part of dcm4che, an implementation of DICOM(TM) in
015 * Java(TM), hosted at https://github.com/gunterze/dcm4che.
016 *
017 * The Initial Developer of the Original Code is
018 * Agfa Healthcare.
019 * Portions created by the Initial Developer are Copyright (C) 2012
020 * the Initial Developer. All Rights Reserved.
021 *
022 * Contributor(s):
023 * See @authors listed below
024 *
025 * Alternatively, the contents of this file may be used under the terms of
026 * either the GNU General Public License Version 2 or later (the "GPL"), or
027 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
028 * in which case the provisions of the GPL or the LGPL are applicable instead
029 * of those above. If you wish to allow use of your version of this file only
030 * under the terms of either the GPL or the LGPL, and not to allow others to
031 * use your version of this file under the terms of the MPL, indicate your
032 * decision by deleting the provisions above and replace them with the notice
033 * and other provisions required by the GPL or the LGPL. If you do not delete
034 * the provisions above, a recipient may use your version of this file under
035 * the terms of any one of the MPL, the GPL or the LGPL.
036 *
037 * ***** END LICENSE BLOCK ***** */
038
039package org.dcm4che3.hl7;
040
041import java.io.BufferedOutputStream;
042import java.io.Closeable;
043import java.io.IOException;
044import java.net.Socket;
045
046import org.slf4j.Logger;
047import org.slf4j.LoggerFactory;
048
049/**
050 * @author Gunter Zeilinger <gunterze@gmail.com>
051 */
052public class MLLPConnection implements Closeable {
053
054    private static Logger LOG = LoggerFactory.getLogger(MLLPConnection.class);
055
056    private final Socket sock;
057    private final MLLPInputStream mllpIn;
058    private final MLLPOutputStream mllpOut;
059
060    public MLLPConnection(Socket sock) throws IOException {
061        this.sock = sock;
062        mllpIn = new MLLPInputStream(sock.getInputStream());
063        mllpOut = new MLLPOutputStream(sock.getOutputStream());
064    }
065
066    public MLLPConnection(Socket sock, int bufferSize) throws IOException {
067        this.sock = sock;
068        mllpIn = new MLLPInputStream(sock.getInputStream());
069        mllpOut = new MLLPOutputStream(new BufferedOutputStream(sock.getOutputStream(), bufferSize));
070    }
071
072    public final Socket getSocket() {
073        return sock;
074    }
075
076    public void writeMessage(byte[] b) throws IOException {
077       writeMessage(b, 0, b.length);
078    }
079
080    public void writeMessage(byte[] b, int off, int len) throws IOException {
081        log("{} << {}", b, off, len);
082        mllpOut.writeMessage(b, off, len);
083    }
084
085    public byte[] readMessage() throws IOException {
086        byte[] b = mllpIn.readMessage();
087        if (b != null)
088            log("{} >> {}", b, 0, b.length);
089        return b;
090    }
091
092    private void log(String format, byte[] b, int off, int len) {
093        if (!LOG.isInfoEnabled())
094            return;
095        int mshlen = 0;
096        while (mshlen < len && b[off + mshlen] != '\r')
097            mshlen++;
098        LOG.info(format, sock, new String(b, off, mshlen));
099        if (LOG.isDebugEnabled())
100            LOG.debug(format, sock, new String(b, off, len).replace('\r', '\n'));
101    }
102
103    @Override
104    public void close() throws IOException {
105        sock.close();
106    }
107}