text: /*
   * Tigase XMPP Server - The instant messaging server
   * Copyright (C) 2004 Tigase, Inc. (office@tigase.com)
   *
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, version 3 of the License.
   *
   * This program 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 Affero General Public License for more details.
   *
   * You should have received a copy of the GNU Affero General Public License
   * along with this program. Look for COPYING file in the top folder.
   * If not, see http://www.gnu.org/licenses/.
   */
  package tigase.io;
  
  import tigase.stats.StatisticsList;
  import tigase.util.IOListener;
  import tigase.util.ZLibWrapper;
  
  import java.io.IOException;
  import java.nio.ByteBuffer;
  import java.nio.channels.SocketChannel;
  import java.util.logging.Level;
  import java.util.logging.Logger;
  
  /**
   * Created: Jul 29, 2009 11:58:02 AM
   *
   * @author <a href="mailto:artur.hefczyc@tigase.org">Artur Hefczyc</a>
  */
  public class ZLibIO
  		implements IOInterface {
  
  	public static final String ZLIB_CAPS = "zlib-caps";
  
  	private static Logger log = Logger.getLogger(ZLibIO.class.getName());
  
  	private IOInterface io = null;
  	private ZLibWrapper zlib = null;
  
  	public ZLibIO(final IOInterface ioi, final int level) {
  		this.io = ioi;
  		zlib = new ZLibWrapper();
  	}
  
  	@Override
  	public int bytesRead() {
  		return io.bytesRead();
  	}
  
  	@Override
  	public long getBytesSent(boolean reset) {
  		return io.getBytesSent(reset);
  	}
  
  	@Override
  	public long getTotalBytesSent() {
  		return io.getTotalBytesSent();
  	}
  
  	@Override
  	public long getBytesReceived(boolean reset) {
  		return io.getBytesReceived(reset);
  	}
  
  	@Override
  	public long getTotalBytesReceived() {
  		return io.getTotalBytesReceived();
  	}
  
  	@Override
  	public long getBuffOverflow(boolean reset) {
  		return io.getBuffOverflow(reset);
  	}
  
  	@Override
  	public long getTotalBuffOverflow() {
  		return io.getTotalBuffOverflow();
  	}
  
  	@Override
  	public boolean checkCapabilities(String caps) {
  		return caps.contains(ZLIB_CAPS) || io.checkCapabilities(caps);
  	}
  
  	@Override
  	public int getInputPacketSize() throws IOException {
  		return io.getInputPacketSize();
  	}
  
  	@Override
  	public SocketChannel getSocketChannel() {
  		return io.getSocketChannel();
  	}
  
  	@Override
  	public void getStatistics(StatisticsList list, boolean reset) {
  		if (io != null) {
  			io.getStatistics(list, reset);
  		}
  
  		if (zlib != null) {
  			list.add("zlibio", "Average compression rate", zlib.averageCompressionRate(), Level.FINE);
  			list.add("zlibio", "Average decompression rate", zlib.averageDecompressionRate(), Level.FINE);
  		}
  	}
  
  	@Override
  	public boolean isConnected() {
  		return io.isConnected();
  	}
  
  	@Override
  	public boolean isRemoteAddress(String addr) {
  		return io.isRemoteAddress(addr);
  	}
  
  	@Override
  	public ByteBuffer read(ByteBuffer buff) throws IOException {
  		ByteBuffer tmpBuffer = io.read(buff);
  
  		if (io.bytesRead() > 0) {
  			ByteBuffer decompressed_buff = zlib.decompress(tmpBuffer);
  
  			// The buffer is reused to it needs to be cleared before it can be
  			// used again.
  			tmpBuffer.clear();
  
  			// System.out.println("Decompression rate: " + zlib.lastDecompressionRate());
  			return decompressed_buff;
  		}
  
  		return null;
  	}
  
  	@Override
  	public void stop() throws IOException {
  		if (log.isLoggable(Level.FINEST)) {
  			log.finest("Stop called..." + toString());
  		}
  
  		io.stop();
  		zlib.end();
  	}
  
  	@Override
  	public String toString() {
  		return "ZLIB: " + io.toString();
  	}
  
  	@Override
  	public boolean waitingToSend() {
  		return io.waitingToSend();
  	}
  
  	@Override
  	public int waitingToSendSize() {
  		return io.waitingToSendSize();
  	}
  
  	@Override
  	public int write(ByteBuffer buff) throws IOException {
  		if (buff == null) {
  			return io.write(null);
  		}
  
  		if (log.isLoggable(Level.FINER)) {
  			log.log(Level.FINER, "ZLIB - Writing data, remaining: {0}", buff.remaining());
  		}
  
  		ByteBuffer compressed_buff = zlib.compress(buff);
  
  		// System.out.println("Compression rate: " + zlib.lastCompressionRate());
  		return io.write(compressed_buff);
  	}
  
  	@Override
  	public void setLogId(String logId) {
  		io.setLogId(logId);
  	}
  
  	public void setIOListener(IOListener listener) {
  		zlib.setIOListener(listener);
  	}
  }
  
  