/*
 * Copyright 2012-2013 Continuuity,Inc. All Rights Reserved.
 */
package com.continuuity.data2.queue;

import com.continuuity.common.queue.QueueName;
import com.continuuity.data2.transaction.Transaction;
import com.continuuity.data2.transaction.TransactionAware;
import com.google.common.collect.ImmutableList;

import java.io.IOException;
import java.util.Collection;

/**
 * A {@link TransactionAware} {@link Queue2Consumer} that forwards all queue operations to another
 * {@link Queue2Consumer} and optionally forward transaction operations if the target {@link Queue2Consumer}
 * is also {@link TransactionAware}.
 */
public abstract class ForwardingQueue2Consumer implements Queue2Consumer, TransactionAware {

  protected final Queue2Consumer consumer;
  protected final TransactionAware txAware;

  public ForwardingQueue2Consumer(Queue2Consumer consumer) {
    this.consumer = consumer;
    if (consumer instanceof TransactionAware) {
      txAware = (TransactionAware) consumer;
    } else {
      txAware = null;
    }
  }

  @Override
  public String getName() {
    return getClass().getSimpleName() + "(" + txAware + getName() + ")";
  }

  @Override
  public QueueName getQueueName() {
    return consumer.getQueueName();
  }

  @Override
  public ConsumerConfig getConfig() {
    return consumer.getConfig();
  }

  @Override
  public DequeueResult dequeue() throws IOException {
    return consumer.dequeue();
  }

  @Override
  public DequeueResult dequeue(int maxBatchSize) throws IOException {
    return consumer.dequeue(maxBatchSize);
  }

  @Override
  public void startTx(Transaction tx) {
    if (txAware != null) {
      txAware.startTx(tx);
    }
  }

  @Override
  public Collection<byte[]> getTxChanges() {
    return (txAware != null) ? txAware.getTxChanges() : ImmutableList.<byte[]>of();
  }

  @Override
  public boolean commitTx() throws Exception {
    return (txAware == null) || txAware.commitTx();
  }

  @Override
  public void postTxCommit() {
    if (txAware != null) {
      txAware.postTxCommit();
    }
  }

  @Override
  public boolean rollbackTx() throws Exception {
    return (txAware == null) || txAware.rollbackTx();
  }
}

@@ -1124,19 +1124,17 @@
  txAware
- + 
+.
 getName(

fix recurse call for queue name