HessianRPCReplyEncoder.java

package org.rzo.netty.ahessian.rpc.message;

import java.io.OutputStream;

import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.ChannelPipelineCoverage;
import org.jboss.netty.channel.ChannelStateEvent;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelHandler;
import org.rzo.netty.ahessian.Constants;
import org.rzo.netty.ahessian.io.OutputStreamEncoder;
import org.rzo.netty.ahessian.rpc.io.Hessian2Output;

/**
 * writes an invocation reply message to an output stream
 */
@ChannelPipelineCoverage("one")
public class HessianRPCReplyEncoder extends SimpleChannelHandler
	{
	volatile Hessian2Output hOut = null;

		/* (non-Javadoc)
		 * @see org.jboss.netty.channel.SimpleChannelDownstreamHandler#writeRequested(org.jboss.netty.channel.ChannelHandlerContext, org.jboss.netty.channel.MessageEvent)
		 */
		synchronized public void writeRequested(ChannelHandlerContext ctx, MessageEvent e) throws Exception
		{
 			Object msg = e.getMessage();
 			if (msg instanceof FlushRequestMessage)
 			{
 				hOut.flush(e.getFuture());
 				e.getFuture().await(5000);
 				return;
 			}

			try
			{
//				if (e.getMessage() instanceof Integer)
//				{
//					hOut.flush();
//					return;
//				}
			HessianRPCReplyMessage message = (HessianRPCReplyMessage) e.getMessage();
			//Constants.ahessianLogger.warn("encode reply for #"+message.getHeaders().get(Constants.CALL_ID_STRING));

			hOut.resetReferences();
			hOut.writeReply(message);
			//hOut.flush();
			//e.getFuture().setSuccess();
			}
			catch (Exception ex)
			{
				Constants.ahessianLogger.warn("", ex);
				e.getFuture().setFailure(ex);
			}
		}
		
		public void channelConnected(ChannelHandlerContext ctx, ChannelStateEvent e) throws Exception
		{
			if (hOut == null)
			{
				OutputStream out = OutputStreamEncoder.getOutputStream(ctx);
				hOut = new Hessian2Output(out);
			}
			else
				hOut.reset();
				ctx.sendUpstream(e);
		}


}