net.grinder.console.communication.server.DispatchClientCommands 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) 2006 - 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.console.communication.server;

import net.grinder.communication.Message;
import net.grinder.communication.MessageDispatchRegistry;
import net.grinder.communication.MessageDispatchRegistry.AbstractBlockingHandler;
import net.grinder.console.communication.ProcessControl;
import net.grinder.console.communication.server.messages.GetNumberOfAgentsMessage;
import net.grinder.console.communication.server.messages.ResetRecordingMessage;
import net.grinder.console.communication.server.messages.ResetWorkerProcessesMessage;
import net.grinder.console.communication.server.messages.ResultMessage;
import net.grinder.console.communication.server.messages.StartRecordingMessage;
import net.grinder.console.communication.server.messages.StartWorkerProcessesMessage;
import net.grinder.console.communication.server.messages.StopAgentAndWorkerProcessesMessage;
import net.grinder.console.communication.server.messages.StopRecordingMessage;
import net.grinder.console.communication.server.messages.SuccessMessage;
import net.grinder.console.model.SampleModel;
import net.grinder.console.model.SampleModelViews;



/**
 * DispatchClientCommands.
 *
 * @author Philip Aston
 */
public class DispatchClientCommands {

  private final SampleModel m_model;
  private final SampleModelViews m_sampleModelViews;
  private final ProcessControl m_processControl;


  /**
   * Constructor for DispatchClientCommands.
   *
   * @param model The model.
   * @param sampleModelViews Console sample model views.
   * @param processControl Process control interface.
   */
  public DispatchClientCommands(SampleModel model,
                                SampleModelViews sampleModelViews,
                                ProcessControl processControl) {
    m_model = model;
    m_sampleModelViews = sampleModelViews;
    m_processControl = processControl;
  }

  /**
   * Registers message handlers with a dispatcher.
   *
   * @param messageDispatcher The dispatcher.
   */
  public void registerMessageHandlers(
    MessageDispatchRegistry messageDispatcher) {

    messageDispatcher.set(
      StartRecordingMessage.class,
      new AbstractBlockingHandler() {
        public Message blockingSend(StartRecordingMessage message) {
          m_model.start();
          return new SuccessMessage();
        }
      });

    messageDispatcher.set(
      StopRecordingMessage.class,
      new AbstractBlockingHandler() {
        public Message blockingSend(StopRecordingMessage message) {
          m_model.stop();
          return new SuccessMessage();
        }
      });

    messageDispatcher.set(
      ResetRecordingMessage.class,
      new AbstractBlockingHandler() {
        public Message blockingSend(ResetRecordingMessage message) {
          m_model.reset();
          m_sampleModelViews.resetStatisticsViews();
          return new SuccessMessage();
        }
      });

    messageDispatcher.set(
      GetNumberOfAgentsMessage.class,
      new AbstractBlockingHandler() {
        public Message blockingSend(GetNumberOfAgentsMessage message) {
          return new ResultMessage(m_processControl.getNumberOfLiveAgents());
        }
      });

    messageDispatcher.set(
      StopAgentAndWorkerProcessesMessage.class,
      new AbstractBlockingHandler() {
        public Message
          blockingSend(StopAgentAndWorkerProcessesMessage message) {
            m_processControl.stopAgentAndWorkerProcesses();
            return new SuccessMessage();
        }
      });

    messageDispatcher.set(
      StartWorkerProcessesMessage.class,
      new AbstractBlockingHandler() {
        public Message blockingSend(StartWorkerProcessesMessage message) {
          m_processControl.startWorkerProcesses(message.getProperties());
          return new SuccessMessage();
        }
      });

    messageDispatcher.set(
      ResetWorkerProcessesMessage.class,
      new AbstractBlockingHandler() {
        public Message blockingSend(ResetWorkerProcessesMessage message) {
          m_processControl.resetWorkerProcesses();
          return new SuccessMessage();
        }
      });
  }
}