TcpMessageSender.java

/*******************************************************************************
 * Copyright (c) 2004-2013 Steve Flasby
 * All rights reserved.
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 * <ul>
 *     <li>Redistributions of source code must retain the above copyright notice,
 *         this list of conditions and the following disclaimer.</li>
 *     <li>Redistributions in binary form must reproduce the above copyright notice,
 *         this list of conditions and the following disclaimer in the documentation
 *         and/or other materials provided with the distribution.</li>
 * </ul>
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 ******************************************************************************/

package org.flasby.net;

import java.io.BufferedOutputStream;
import java.io.IOException;
import java.net.Socket;

/*
 * @author Steve Flasby
 * Jan 31, 2013
 */

/**
 * sends a message to a TCP server somewhere. It sends the message and then closes
 * any resources immediately so should only be used for infrequent requests.
 * 
 * @author steve
 *
 */
public class TcpMessageSender {
	private final Socket socket;

	public TcpMessageSender( SocketProperties socketProperties, byte[] message ) throws IOException {
		socket = new Socket(socketProperties.host, socketProperties.port);
		sendAndClose(message);
	}
	
	public TcpMessageSender( SocketProperties socketProperties, Messages messages ) throws IOException {
		socket = new Socket(socketProperties.host, socketProperties.port);
		sendAndClose(messages);
	}
	
	private void sendAndClose( Messages messages) throws IOException {
		byte[] message = messages.get();
		try ( BufferedOutputStream os = new BufferedOutputStream(socket.getOutputStream()) ) {
			do {
				os.write(message, 0, message.length);
			} while ( null != (message=messages.get()) );
			os.close();
		} finally {
			socket.close();
		}
	}
	
	private void sendAndClose( byte[] message ) throws IOException {
		try ( BufferedOutputStream os = new BufferedOutputStream(socket.getOutputStream()) ) {
			os.write(message, 0, message.length);
			os.close();
		} finally {
			socket.close();
		}
	}
}