org.jppf.comm.recovery.ClientConnection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jppf-common Show documentation
JPPF, the open source grid computing solution
There is a newer version: 6.3-alpha
/*
 * JPPF.
 * Copyright (C) 2005-2015 JPPF Team.
 * http://www.jppf.org
 *
 * 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.jppf.comm.recovery;

import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;

import org.jppf.comm.socket.*;
import org.jppf.utils.*;
import org.slf4j.*;

/**
 * Client-side connection for the recovery mechanism.
 * @author Laurent Cohen
 */
public class ClientConnection extends AbstractRecoveryConnection
{
  /**
   * Logger for this class.
   */
  private static Logger log = LoggerFactory.getLogger(ClientConnection.class);
  /**
   * Determines whether the debug level is enabled in the log configuration, without the cost of a method call.
   */
  private static boolean debugEnabled = LoggingUtils.isDebugEnabled(log);
  /**
   * Used to synchronize access to the underlying socket from multiple threads.
   */
  private SocketInitializer socketInitializer;
  /**
   * The list of listeners to this object's events.
   */
  private final List listeners = new CopyOnWriteArrayList<>();
  /**
   * 
   */
  private String host = null;
  /**
   * 
   */
  private int port = -1;

  /**
   * Initialize this client connection with the specified uuid.
   * @param uuid the JPPF node or client uuid.
   */
  public ClientConnection(final String uuid)
  {
    this.uuid = uuid;
  }

  /**
   * Initialize this client connection with the specified uuid.
   * @param uuid the JPPF node or client uuid.
   * @param host the host ot which to connect.
   * @param port the port number to connect to on the host.
   */
  public ClientConnection(final String uuid, final String host, final int port)
  {
    this.uuid = uuid;
    this.host = host;
    this.port = port;
  }

  @Override
  public void run()
  {
    runThread = Thread.currentThread();
    try
    {
      configure();
      if (debugEnabled) log.debug("initializing recovery client connection " + socketWrapper);
      socketInitializer = new SocketInitializerImpl();
      socketInitializer.initializeSocket(socketWrapper);
      if (!socketInitializer.isSuccessful())
      {
        log.error("Could not initialize recovery client connection " + socketWrapper);
        close();
        return;
      }
      while (!isStopped())
      {
        String message = receiveMessage(maxRetries, socketReadTimeout);
        if ((message != null) && message.startsWith("handshake")) setInitialized(true);
        String response = "checked;" + uuid;
        sendMessage(response);
      }
    }
    catch (Exception e)
    {
      log.error(e.getMessage(), e);
      if (!(e instanceof InterruptedException)) fireClientConnectionEvent();
      close();
    }
    if (debugEnabled) log.debug(Thread.currentThread().getName() + " stopping");
  }

  /**
   * Configure this client connection from the JPPF properties.
   */
  private void configure()
  {
    if (debugEnabled) log.debug("configuring connection");
    TypedProperties config = JPPFConfiguration.getProperties();
    if (host == null) host = config.getString("jppf.server.host", "localhost");
    if (port < 0) port = config.getInt("jppf.recovery.server.port", 22222);
    maxRetries = config.getInt("jppf.recovery.max.retries", 2);
    socketReadTimeout = config.getInt("jppf.recovery.read.timeout", 60000);
    socketWrapper = new BootstrapSocketClient();
    socketWrapper.setHost(host);
    socketWrapper.setPort(port);
  }

  /**
   * Close this client and release any resources it is using.
   */
  @Override
  public void close()
  {
    setStopped(true);
    if (runThread != null) runThread.interrupt();
    try
    {
      if (debugEnabled) log.debug("closing connection");
      SocketWrapper tmp = socketWrapper;
      socketWrapper = null;
      if (tmp != null) tmp.close();
      if (socketInitializer != null) socketInitializer.close();
      socketInitializer = null;
      listeners.clear();
    }
    catch (Exception e)
    {
      log.error(e.getMessage(), e);
    }
  }

  /**
   * Add a listener to the list of listeners.
   * @param listener the listener to add.
   */
  public void addClientConnectionListener(final ClientConnectionListener listener)
  {
    if (listener == null) return;
    listeners.add(listener);
  }

  /**
   * Remove a listener from the list of listeners.
   * @param listener the listener to remove.
   */
  public void removeClientConnectionListener(final ClientConnectionListener listener)
  {
    if (listener == null) return;
    listeners.remove(listener);
  }

  /**
   * Notify all listeners that an event has occurred.
   */
  private void fireClientConnectionEvent()
  {
    ClientConnectionEvent event = new ClientConnectionEvent(this);
    for (ClientConnectionListener listener : listeners) listener.clientConnectionFailed(event);
  }
}