net.grinder.synchronisation.ClientBarrierGroups Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grinder-core Show documentation
The Grinder.
The newest version!
// Copyright (C) 2011 Philip Aston
// All rights reserved.
//
// This file is part of The Grinder software distribution. Refer to
// the file LICENSE which is part of The Grinder distribution for
// licensing details. The Grinder distribution is available on the
// Internet at http://grinder.sourceforge.net/
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
// COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
// HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
// OF THE POSSIBILITY OF SUCH DAMAGE.

package net.grinder.synchronisation;

import java.util.Set;

import net.grinder.communication.CommunicationException;
import net.grinder.communication.MessageDispatchRegistry;
import net.grinder.communication.Sender;
import net.grinder.communication.MessageDispatchRegistry.AbstractHandler;
import net.grinder.synchronisation.messages.AddBarrierMessage;
import net.grinder.synchronisation.messages.AddWaiterMessage;
import net.grinder.synchronisation.messages.BarrierIdentity;
import net.grinder.synchronisation.messages.CancelWaiterMessage;
import net.grinder.synchronisation.messages.OpenBarrierMessage;
import net.grinder.synchronisation.messages.RemoveBarriersMessage;


/**
 * {@link BarrierGroups} implementation which delegates to a remote instance.
 *
 * @author Philip Aston
 */
public class ClientBarrierGroups extends AbstractBarrierGroups {

  private final Sender m_sender;

  /**
   * Constructor.
   *
   * @param sender Used to send messages to the remote instance (the console).
   * @param messageDispatch Used to receive messages from the remote instance.
   */
  public ClientBarrierGroups(Sender sender,
                             MessageDispatchRegistry messageDispatch) {
    m_sender = sender;

    messageDispatch.set(
      OpenBarrierMessage.class,
      new AbstractHandler() {
        public void handle(OpenBarrierMessage message) {
          final BarrierGroupImplementation existingGroup =
            getExistingGroup(message.getName());

          if (existingGroup != null) {
            final Set removedWaiters =
              existingGroup.clearWaiters(message.getWaiters());
            existingGroup.fireAwaken(removedWaiters);
          }
        }
      });
  }

  /**
   * {@inheritDoc}
   */
  @Override
  protected BarrierGroupImplementation createBarrierGroup(String name) {

    /**
     * Extending {@link AbstractBarrierGroups} verifies the consistency of the
     * local partition of the barrier group.
     */
    return new BarrierGroupImplementation(name) {

      @Override
      public void addBarrier()
        throws CommunicationException {

        super.addBarrier();

        m_sender.send(new AddBarrierMessage(getName()));
      }

      @Override
      public void removeBarriers(long n) throws CommunicationException {

        super.removeBarriers(n);

        m_sender.send(new RemoveBarriersMessage(getName(), n));
      }

      @Override
      public void addWaiter(BarrierIdentity barrierIdentity)
        throws CommunicationException {

        super.addWaiter(barrierIdentity);

        m_sender.send(new AddWaiterMessage(getName(), barrierIdentity));
      }

      @Override
      public void cancelWaiter(BarrierIdentity barrierIdentity)
        throws CommunicationException {

        super.cancelWaiter(barrierIdentity);

        m_sender.send(new CancelWaiterMessage(getName(), barrierIdentity));
      }
    };
  }
}