/**
* Copyright 2007-2015, Kaazing Corporation. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.kaazing.k3po.driver.internal.behavior.handler.barrier;
import static java.lang.String.format;
import static org.kaazing.k3po.driver.internal.netty.channel.ChannelFutureListeners.chainedFuture;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;
import org.jboss.netty.channel.ChannelDownstreamHandler;
import org.jboss.netty.channel.ChannelEvent;
import org.jboss.netty.channel.ChannelFuture;
import org.jboss.netty.channel.ChannelFutureListener;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.kaazing.k3po.driver.internal.behavior.Barrier;
import org.kaazing.k3po.driver.internal.behavior.handler.prepare.PreparationEvent;
public class AwaitBarrierDownstreamHandler extends AbstractBarrierHandler implements ChannelDownstreamHandler {
private Queue<ChannelEvent> queue;
public AwaitBarrierDownstreamHandler(Barrier barrier) {
super(barrier);
}
@Override
public void prepareRequested(final ChannelHandlerContext ctx, PreparationEvent evt) {
super.prepareRequested(ctx, evt);
// when pipeline future complete, pay attention to barrier future
final ChannelFuture handlerFuture = getHandlerFuture();
ChannelFuture pipelineFuture = getPipelineFuture();
pipelineFuture.addListener(new ChannelFutureListener() {
@Override
public void operationComplete(final ChannelFuture f) throws Exception {
// If the pipeline was not complete successfully we dont want nor need to wait for the barrier
if (!f.isSuccess()) {
if (f.isCancelled()) {
handlerFuture.cancel();
} else {
handlerFuture.setFailure(f.getCause());
}
} else {
// when barrier future complete, trigger handler future
Barrier barrier = getBarrier();
ChannelFuture barrierFuture = barrier.getFuture();
barrierFuture.addListener(chainedFuture(handlerFuture));
}
}
});
// when handler future complete, flush queued channel events
queue = new ConcurrentLinkedQueue<>();
handlerFuture.addListener(new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) throws Exception {
if (future.isSuccess()) {
// TODO: review need for synchronized
synchronized (ctx) {
Queue<ChannelEvent> pending = queue;
queue = null;
for (ChannelEvent evt : pending) {
ctx.sendDownstream(evt);
}
}
}
}
});
}
@Override
public void handleDownstream(ChannelHandlerContext ctx, ChannelEvent evt) throws Exception {
ChannelFuture pipelineFuture = getPipelineFuture();
ChannelFuture handlerFuture = getHandlerFuture();
// while handler future not complete, queue channel events
if (queue != null && pipelineFuture.isDone() && !handlerFuture.isDone()) {
queue.add(evt);
}
else {
// TODO: review need for synchronized
synchronized (ctx) {
ctx.sendDownstream(evt);
}
}
}
@Override
protected StringBuilder describe(StringBuilder sb) {
return sb.append(format("write await %s", getBarrier()));
}
boolean hasQueuedChannelEvents() {
return queue != null && !queue.isEmpty();
}
}