org.jppf.server.node.remote.JPPFRemoteContainer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of jppf-server Show documentation
JPPF, the open source grid computing solution
The newest version!
/*
 * JPPF.
 * Copyright (C) 2005-2019 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.server.node.remote;

import java.util.List;
import java.util.concurrent.*;

import org.jppf.classloader.AbstractJPPFClassLoader;
import org.jppf.io.*;
import org.jppf.node.protocol.TaskBundle;
import org.jppf.server.node.*;
import org.jppf.utils.ExceptionUtils;
import org.slf4j.*;

/**
 * Instances of this class represent dynamic class loading, and serialization/deserialization, capabilities, associated
 * with a specific client application.

 * The application is identified through a unique uuid. This class effectively acts as a container for the classes of
 * a client application, a provides the methods to enable the transport, serialization and deserialization of these classes.
 * @author Laurent Cohen
 */
public class JPPFRemoteContainer extends JPPFContainer {
  /**
   * Logger for this class.
   */
  private static Logger log = LoggerFactory.getLogger(JPPFRemoteContainer.class);
  /**
   * Determines whether the debug level is enabled in the logging configuration, without the cost of a method call.
   */
  private static boolean debugEnabled = log.isDebugEnabled();
  /**
   * Determines whether the trace level is enabled in the logging configuration, without the cost of a method call.
   */
  private static boolean traceEnabled = log.isTraceEnabled();
  /**
   * The socket connection wrapper.
   */
  private RemoteNodeConnection nodeConnection;

  /**
   * Initialize this container with a specified application uuid.
   * @param node the node holding this container.
   * @param uuidPath the unique identifier of a submitting application.
   * @param classLoader the class loader for this container.
   * @param clientAccess whether the node has access to the client that submitted the job.
   * @throws Exception if an error occurs while initializing.
   */
  public JPPFRemoteContainer(final AbstractCommonNode node, final List uuidPath, final AbstractJPPFClassLoader classLoader, final boolean clientAccess) throws Exception {
    super(node, uuidPath, classLoader, clientAccess);
    this.nodeConnection = (RemoteNodeConnection) node.getNodeConnection();
  }

  /**
   * Deserialize a number of objects from a socket client.
   * @param list a list holding the resulting deserialized objects.
   * @param count the number of objects to deserialize.
   * @param executor the number of objects to deserialize.
   * @return the new position in the source data after deserialization.
   * @throws Throwable if an error occurs while deserializing.
   */
  @Override
  public int deserializeObjects(final Object[] list, final int count, final ExecutorService executor) throws Throwable {
    if (debugEnabled) log.debug("deserializing {} objects", count);
    final ClassLoader cl = Thread.currentThread().getContextClassLoader();
    try {
      Thread.currentThread().setContextClassLoader(classLoader);
      final CompletionService completionService = new ExecutorCompletionService<>(executor, new ArrayBlockingQueue>(count));
      final InputSource is = new SocketWrapperInputSource(nodeConnection.getChannel());
      for (int i = 0; i < count; i++) {
        final DataLocation dl = IOHelper.readData(is);
        if (traceEnabled) log.trace("i = {}, read data size = {}", i, (dl == null) ? -1 : dl.getSize());
        completionService.submit(new ObjectDeserializationTask(this, (TaskBundle) list[0], dl, i));
      }
      Throwable t = null;
      int throwableCount = 0;
      for (int i=0; i f = completionService.take();
        final ObjectDeserializationTask task = f.get();
        final Object o = task.getObject();
        if (o instanceof Throwable) {
          throwableCount++;
          if (t == null) t = (Throwable) o;
        }
        list[task.getIndex() + 1] = o;
      }
      if (t != null) {
        if (debugEnabled) log.debug("tasks deserialization resulted in {} errors, first throwable = {}", throwableCount, ExceptionUtils.getMessage(t));
        throw t;
      }
      return 0;
    } finally {
      Thread.currentThread().setContextClassLoader(cl);
    }
  }

  /**
   * The node connection wrapper.
   * @param nodeConnection the node connection to set.
   */
  public void setNodeConnection(final RemoteNodeConnection nodeConnection) {
    this.nodeConnection = nodeConnection;
  }
}