org.xlightweb.SimpleMessageWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xlightweb
xLightweb is a lightweight, high performance, scalable web network library
There is a newer version: 2.13.2
/*
 *  Copyright (c) xlightweb.org, 2008. All rights reserved.
 *
 *  This library 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 2.1 of the License, or (at your option) any later version.
 *
 *  This library 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 library; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Please refer to the LGPL license at: http://www.gnu.org/copyleft/lesser.txt
 * The latest copy of this software may be found on http://www.xlightweb.org/
 */
package org.xlightweb;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.xsocket.DataConverter;
import org.xsocket.connection.IConnection.FlushMode;




/**
 * Simple message writer 
 *  
 * @author [email protected]
 */
final class SimpleMessageWriter extends AbstractBodyWriter {

	private static final Logger LOG = Logger.getLogger(SimpleMessageWriter.class.getName());
	
	private boolean isHeaderWritten = false;
		
	private int written = 0;

	private IHttpHeader header = null;

	
	/**
	 * constructor
	 *  
	 * @param httpConnection   the http connection
	 * @param header           the header
	 * @throws IOException if an exception occurs
	 */
	public SimpleMessageWriter(AbstractHttpConnection httpConnection, IHttpHeader header) throws IOException {
		super(httpConnection);
		
		this.header = header;
	}
	
	
	
	/**
	 * {@inheritDoc}
	 */
	public void flush(ByteBuffer[] bodyData, FlushMode flushMode) throws IOException {

		AbstractHttpConnection con = getConnection();
		
		if (con != null) {
			
			try {
				con.setFlushmode(flushMode);
				
				int dataWritten = 0;
				
				if (!isHeaderWritten) {
					isHeaderWritten = true;
					
					con.write(header.toString() + "\r\n");			
					dataWritten += writeBody(con, bodyData);
					
				} else {
					dataWritten = writeBody(con,bodyData);
				}
		
				if (dataWritten > 0) {
					con.flush();
				}
				
			} catch (IOException ioe) {
				if (LOG.isLoggable(Level.FINE)) {
					LOG.fine("error occured by flushing bound data sink. Destroying connection. reason " + DataConverter.toString(ioe));
				}
				
				destroy();
				throw ioe;
			}
		}
	}
	
	
	private int writeBody(AbstractHttpConnection con, ByteBuffer[] bodyData) throws IOException {
		int dataWritten = 0;
		
		if (bodyData != null) {
			dataWritten = (int) con.write(bodyData);
			written += dataWritten;
		}
		
		return dataWritten;
	}
	
	
	/**
	 * {@inheritDoc}
	 */
	public void close() {
		if (getConnection() == null) {
			return;
		}
		
		destroy();
	}
}