ZLibStreamCompressor.java 2.22 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
package org.jivesoftware.openfire.net;

import java.io.BufferedWriter;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.UnsupportedEncodingException;
import java.io.Writer;

import org.apache.mina.filter.CompressionFilter;

import com.jcraft.jzlib.JZlib;
import com.jcraft.jzlib.ZInputStream;
import com.jcraft.jzlib.ZOutputStream;

/**
 * Implements ZLib Stream Compression.
 * 
 * Note that much of this code was copy/pasted from various other Openfire
 * classes. These classes always assumed the ZLIB algorithm, if compression was
 * enabled.
 * 
 * @author Guus der Kinderen, guus@nimbuzz.com
 */
public class ZLibStreamCompressor implements StreamCompressor
{
	/*
	 * (non-Javadoc)
	 * 
	 * @see org.jivesoftware.openfire.net.StreamCompressor#getMethodIdentifier()
	 */
	public String getMethodIdentifier()
	{
		return "zlib";
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.jivesoftware.openfire.net.StreamCompressor#getWriter(java.io.OutputStream)
	 */
	public Writer getWriter(OutputStream outputStream)
			throws UnsupportedEncodingException
	{
		// TODO: move traffic counting outside this class!
		final ZOutputStream out = new ZOutputStream(
			ServerTrafficCounter.wrapOutputStream(outputStream),
			JZlib.Z_BEST_COMPRESSION);
		out.setFlushMode(JZlib.Z_PARTIAL_FLUSH);

		return new BufferedWriter(new OutputStreamWriter(out,
			SocketConnection.CHARSET));
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.jivesoftware.openfire.net.StreamCompressor#getReader(java.io.InputStream)
	 */
	public Reader getReader(InputStream inputStream)
			throws UnsupportedEncodingException
	{
		// TODO: move traffic counting outside this class!
		final ZInputStream in = new ZInputStream(
			ServerTrafficCounter.wrapInputStream(inputStream));
		in.setFlushMode(JZlib.Z_PARTIAL_FLUSH);
		return new InputStreamReader(in, SocketConnection.CHARSET);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.jivesoftware.openfire.net.StreamCompressor#getCompressingIOFilter()
	 */
	public CompressionFilter getCompressingIOFilter()
	{
		return new CompressionFilter(CompressionFilter.COMPRESSION_MAX);
	}
}