SimpleMessageHandler.java

package org.handwerkszeug.riak.transport.rest.internal;

import org.handwerkszeug.riak._;
import org.handwerkszeug.riak.op.RiakResponseHandler;
import org.handwerkszeug.riak.transport.internal.CompletionSupport;
import org.handwerkszeug.riak.transport.internal.CountDownRiakFuture;
import org.handwerkszeug.riak.transport.internal.MessageHandler;
import org.handwerkszeug.riak.util.NettyUtil;
import org.jboss.netty.handler.codec.http.HttpResponse;
import org.jboss.netty.handler.codec.http.HttpResponseStatus;

/**
 * @author taichi
 */
public class SimpleMessageHandler implements MessageHandler {
	final String name;
	final RiakResponseHandler<_> users;
	final CompletionSupport support;

	public SimpleMessageHandler(String name, RiakResponseHandler<_> users,
			CompletionSupport support) {
		this.users = users;
		this.name = name;
		this.support = support;
	}

	@Override
	public boolean handle(Object receive, CountDownRiakFuture future)
			throws Exception {
		if (receive instanceof HttpResponse) {
			HttpResponse response = (HttpResponse) receive;
			HttpResponseStatus status = response.getStatus();
			if (NettyUtil.isError(status)) {
				this.users
						.onError(new RestErrorResponse(response));
				future.setFailure();
				return true;
			}
			if (NettyUtil.isSuccessful(status)) {
				this.users.handle(this.support.newResponse());
				future.setSuccess();
				return true;
			}
		}
		return false;
	}
}