Package eu.stratosphere.nephele.jobgraph

Source Code of eu.stratosphere.nephele.jobgraph.JobGenericInputVertex

/***********************************************************************************************************************
* Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
*
* 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 eu.stratosphere.nephele.jobgraph;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;

import eu.stratosphere.configuration.IllegalConfigurationException;
import eu.stratosphere.core.io.StringRecord;
import eu.stratosphere.nephele.execution.librarycache.LibraryCacheManager;
import eu.stratosphere.nephele.template.AbstractInputTask;
import eu.stratosphere.nephele.template.AbstractInvokable;
import eu.stratosphere.util.StringUtils;

public class JobGenericInputVertex extends JobInputVertex
{
  /**
   * Class of input task.
   */
  protected Class<? extends AbstractInputTask<?>> inputClass = null;

  /**
   * Creates a new job input vertex with the specified name.
   *
   * @param name The name of the new job file input vertex.
   * @param id The ID of this vertex.
   * @param jobGraph The job graph this vertex belongs to.
   */
  public JobGenericInputVertex(String name, JobVertexID id, JobGraph jobGraph) {
    super(name, id, jobGraph);
  }

  /**
   * Creates a new job file input vertex with the specified name.
   *
   * @param name The name of the new job file input vertex.
   * @param jobGraph The job graph this vertex belongs to.
   */
  public JobGenericInputVertex(String name, JobGraph jobGraph) {
    super(name, null, jobGraph);
  }

  /**
   * Creates a new job file input vertex.
   *
   * @param jobGraph The job graph this vertex belongs to.
   */
  public JobGenericInputVertex(JobGraph jobGraph) {
    super(null, null, jobGraph);
  }

  /**
   * Sets the class of the vertex's input task.
   *
   * @param inputClass The class of the vertex's input task.
   */
  public void setInputClass(Class<? extends AbstractInputTask<?>> inputClass) {
    this.inputClass = inputClass;
  }

  /**
   * Returns the class of the vertex's input task.
   *
   * @return the class of the vertex's input task or <code>null</code> if no task has yet been set
   */
  public Class<? extends AbstractInputTask<?>> getInputClass() {
    return this.inputClass;
  }


  @SuppressWarnings("unchecked")
  @Override
  public void read(DataInput in) throws IOException
  {
    super.read(in);

    // Read class
    boolean isNotNull = in.readBoolean();
    if (isNotNull) {
      // Read the name of the class and try to instantiate the class object
      final ClassLoader cl = LibraryCacheManager.getClassLoader(this.getJobGraph().getJobID());
      if (cl == null) {
        throw new IOException("Cannot find class loader for vertex " + getID());
      }

      // Read the name of the expected class
      final String className = StringRecord.readString(in);

      try {
        this.inputClass = (Class<? extends AbstractInputTask<?>>) Class.forName(className, true, cl).asSubclass(AbstractInputTask.class);
      }
      catch (ClassNotFoundException cnfe) {
        throw new IOException("Class " + className + " not found in one of the supplied jar files: "
          + StringUtils.stringifyException(cnfe));
      }
      catch (ClassCastException ccex) {
        throw new IOException("Class " + className + " is not a subclass of "
          + AbstractInputTask.class.getName() + ": " + StringUtils.stringifyException(ccex));
      }
    }
  }


  @Override
  public void write(DataOutput out) throws IOException
  {
    super.write(out);

    // Write out the name of the class
    if (this.inputClass == null) {
      out.writeBoolean(false);
    } else {
      out.writeBoolean(true);
      StringRecord.writeString(out, this.inputClass.getName());
    }
  }


  @Override
  public void checkConfiguration(AbstractInvokable invokable) throws IllegalConfigurationException
  {
    // see if the task itself has a valid configuration
    // because this is user code running on the master, we embed it in a catch-all block
    try {
      invokable.checkConfiguration();
    }
    catch (IllegalConfigurationException icex) {
      throw icex; // simply forward
    }
    catch (Throwable t) {
      throw new IllegalConfigurationException("Checking the invokable's configuration caused an error: "
        + StringUtils.stringifyException(t));
    }
  }


  @Override
  public Class<? extends AbstractInvokable> getInvokableClass() {

    return this.inputClass;
  }


  @Override
  public int getMaximumNumberOfSubtasks(AbstractInvokable invokable)
  {
    return invokable.getMaximumNumberOfSubtasks();
  }


  @Override
  public int getMinimumNumberOfSubtasks(AbstractInvokable invokable) {

    return invokable.getMinimumNumberOfSubtasks();
  }
}
TOP

Related Classes of eu.stratosphere.nephele.jobgraph.JobGenericInputVertex

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.