MapReduce剖析笔记之六:TaskTracker初始化任务并启动JVM过程

在上面一节我们分析了JobTracker调用JobQueueTaskScheduler进行任务分配,JobQueueTaskScheduler又调用JobInProgress按照一定顺序查找任务的流程,获得了任务之后,将任务封装为TaskTrackerAction数组返回的整个过程。TaskTracker通过心跳响应接收到了这个数组。本节我们继续分析,TaskTracker拿到了这个数组之后,如何对任务进行处理的。

 

1,TaskTracker在其方法offerService中,将得到的任务加入队列:

        TaskTrackerAction[] actions = heartbeatResponse.getActions();
。。。。。。。
        if (actions != null){ 
          for(TaskTrackerAction action: actions) {
            if (action instanceof LaunchTaskAction) {
              addToTaskQueue((LaunchTaskAction)action);
            } else if (action instanceof CommitTaskAction) {
              CommitTaskAction commitAction = (CommitTaskAction)action;
              if (!commitResponses.contains(commitAction.getTaskID())) {
                commitResponses.add(commitAction.getTaskID());
              }
            } else {
              addActionToCleanup(action);
            }
          }
        }

可见,返回的TaskTrackerAction对象有多种可能,最典型的是LaunchTaskAction,即启动一个任务;另外还有CommitTaskAction等。TaskTrackerAction是一个抽象类,其实现总共有5个,如下所示:

 

  public static enum ActionType {
    /** Launch a new task. */
    LAUNCH_TASK,
    
    /** Kill a task. */
    KILL_TASK,
    
    /** Kill any tasks of this job and cleanup. */
    KILL_JOB,
    
    /** Reinitialize the tasktracker. */
    REINIT_TRACKER,

    /** Ask a task to save its output. */
    COMMIT_TASK
  };

 

LaunchTaskAction表示启动任务,内部包含一个Task对象,即要启动的任务对象;

CommitTaskAction表示让一个正在执行中的任务提交其输出结果,其内部包含一个TaskAttemptID对象,对于某些Job来说,只有Map阶段而没有Reduce阶段,此时可以让其直接提交结果;

KillJobAction表示杀死属于该Job的任意任务,内部包含一个JobID;

KillTaskAction表示杀死一个任务,内部包含一个TaskAttemptID;

ReinitTrackerAction表示让TaskTracker重新初始化,无需带什么参数,只要随着心跳响应返回,对应的TaskTracker自然知道应该重新初始化。

我们对最典型的LaunchTaskAction进行分析,其内部的Task是一个抽象类,主要有MapTask和ReduceTask两种实现。在上面的代码中,addToTaskQueueLaunchTaskAction按照Map或Reduce类型分别加入到不同的队列:

  private void addToTaskQueue(LaunchTaskAction action) {
    if (action.getTask().isMapTask()) {
      mapLauncher.addToTaskQueue(action);
    } else {
      reduceLauncher.addToTaskQueue(action);
    }
  }

mapLauncherreduceLauncher都是TaskLauncher类型的对象,顾名思义用于Task的启动。TaskLauncher继承了Thread类,因此是一个线程类。其加入队列的代码为:

    public void addToTaskQueue(LaunchTaskAction action) {
      synchronized (tasksToLaunch) {
        TaskInProgress tip = registerTask(action, this);
        tasksToLaunch.add(tip);
        tasksToLaunch.notifyAll();
      }
    }

tasksToLaunch是一个任务链表:LinkedList<TaskInProgress>(),可见,我们再一次又看见了TaskInProgress,心跳响应时,双方交互的实际上是Task对象,自己内部维护的是TaskInProgress对象。通过registerTask这个方法,将其转换回来。其实其转换代码也简单:

  private TaskInProgress registerTask(LaunchTaskAction action, 
      TaskLauncher launcher) {
    Task t = action.getTask();
    LOG.info("LaunchTaskAction (registerTask): " + t.getTaskID() +
             " task's state:" + t.getState());
    TaskInProgress tip = new TaskInProgress(t, this.fConf, launcher);
    synchronized (this) {
      tasks.put(t.getTaskID(), tip);
      runningTasks.put(t.getTaskID(), tip);
      boolean isMap = t.isMapTask();
      if (isMap) {
        mapTotal++;
      } else {
        reduceTotal++;
      }
    }
    return tip;
  }

在上面的注册代码中,创建了一个TaskInProgress对象,之后加入到两个映射表中:

  Map<TaskAttemptID, TaskInProgress> tasks = new HashMap<TaskAttemptID, TaskInProgress>();
  /**
   * Map from taskId -> TaskInProgress.
   */
  Map<TaskAttemptID, TaskInProgress> runningTasks = null;

映射表的索引是目前的任务ID,值即为刚刚创建的TaskInProgress对象。

之后,加入到tasksToLaunch对象中,并执行notifyAll方法,唤醒TaskLauncher线程(该线程可能处于睡眠等待中)处理队列里面的任务。

接下来,另一个线程,即TaskLauncher的run方法会处理这个对象。心跳线程即TaskTracker的主线程返回,继续等待执行后续的心跳。

 

2,TaskLauncher的run方法接着执行:

run方法的前半部分是从队列中获取任务:

          synchronized (tasksToLaunch) {
            while (tasksToLaunch.isEmpty()) {
              tasksToLaunch.wait();
            }
            //get the TIP
            tip = tasksToLaunch.remove(0);
            task = tip.getTask();
            LOG.info("Trying to launch : " + tip.getTask().getTaskID() + 
                     " which needs " + task.getNumSlotsRequired() + " slots");
          }

如果为空,则调用wait进行等待;如果不为空,自然有人会通知自己(即notifyAll),之后,取出第一个任务并删除队列中的第一个元素。wait和notifyAll是JAVA任意一个对象都会具有的方法。

取出任务之后,需要判断是否有资源可用,使用IntWritable numFreeSlots进行判断,如果没有足够资源,则一直会等待:

            while (numFreeSlots.get() < task.getNumSlotsRequired()) {
。。。。。。。。。。。。。
              numFreeSlots.wait();
            }

而numFreeSlots的更新是TaskLauncher另一个方法中进行的:

    public void addFreeSlots(int numSlots) {
      synchronized (numFreeSlots) {
        numFreeSlots.set(numFreeSlots.get() + numSlots);
        assert (numFreeSlots.get() <= maxSlots);
        LOG.info("addFreeSlot : current free slots : " + numFreeSlots.get());
        numFreeSlots.notifyAll();
      }
    }

而该方法又被另一个方法调用:

    private synchronized void releaseSlot() {
      if (slotTaken) {
        if (launcher != null) {
          launcher.addFreeSlots(task.getNumSlotsRequired());
        }
        slotTaken = false;
      } else {
        // wake up the launcher. it may be waiting to block slots for this task.
        if (launcher != null) {
          launcher.notifySlots();
        }
      }
    }

releaseSlot表示释放资源,又是在kill和reportTaskFinished方法中调用:

    void reportTaskFinished(boolean commitPending) {
      if (!commitPending) {
        taskFinished();
        releaseSlot();
      }
      notifyTTAboutTaskCompletion();
    }

也就是说,当任务被杀掉或者执行完毕时,会释放资源,进而发出通告。

如果获得了足够的资源,则该变量会变化:

 numFreeSlots.set(numFreeSlots.get() - task.getNumSlotsRequired());

之后,如果任务可以执行(如果不可以执行还需要回收资源),进入startNewTask方法。在该方法里,会创建一个新线程进行任务的启动工作,上一个线程只是从队列里取出任务,判断资源是否充足等功能。启动了新线程后,它的使命结束了,返回去继续等待队列LinkedList<TaskInProgress>()中是否出现了新的TaskInProgress。

 

3,startNewTask创建的线程继续执行:

  void startNewTask(final TaskInProgress tip) throws InterruptedException {
    Thread launchThread = new Thread(new Runnable() {
      @Override
      public void run() {
        try {
          RunningJob rjob = localizeJob(tip);
          tip.getTask().setJobFile(rjob.getLocalizedJobConf().toString());
          launchTaskForJob(tip, new JobConf(rjob.getJobConf()), rjob); 
        } catch (Throwable e) {
。。。。。
          try {
            tip.kill(true);
            tip.cleanup(false, true);
          } catch (IOException ie2) {
。。。。。。
        }
      }
    });
    launchThread.start();
  }

该线程中,首先执行localizeJob方法。这个方法用于初始化Job的目录。首先,执行addTaskToJob方法,将任务加入到正在运行作业的对象中,runningJobs是TaskTracker中的一个变量,记录了当前这个TaskTracker上面运行了哪些Job。

  Map<JobID, RunningJob> runningJobs = new TreeMap<JobID, RunningJob>();

runningJobs里面保存了Job ID和RunningJob之间的对应关系,RunningJob是一个类,记录了当前正在初始化的任务的信息。主要包含以下信息:

    private JobID jobid; 
    private JobConf jobConf;
    private Path localizedJobConf;
    // keep this for later use
    volatile Set<TaskInProgress> tasks;
    //the 'localizing' and 'localized' fields have the following
    //state transitions (first entry is for 'localizing')
    //{false,false} -> {true,false} -> {false,true}
    volatile boolean localized;
    boolean localizing;
    boolean keepJobFiles;
    UserGroupInformation ugi;
    FetchStatus f;
    TaskDistributedCacheManager distCacheMgr;

即Job ID、Job Conf即配置参数、Job配置文件路径、该Job包含的任务集合(当前TaskTracker内的)以及一些用户权限等信息。

接下来需要对Job所在目录进行定位,使用initializeJob方法:

上一节我们在分析任务调度的时候,有一个obtainNewMapTaskCommon方法,该方法通过查找任务后,执行addRunningTaskToTIP方法,并调用addRunningTask方法,即创建MapTask或ReduceTask对象。此时会将任务的目录信息传进来,而最终一个任务的目录信息是在TaskInProgress的构造方法中传进来的。我们再回顾下Job的初始化过程,有以下代码:

    maps = new TaskInProgress[numMapTasks];
    for(int i=0; i < numMapTasks; ++i) {
      inputLength += splits[i].getInputDataLength();
      maps[i] = new TaskInProgress(jobId, jobFile, 
                                   splits[i], 
                                   jobtracker, conf, this, i, numSlotsPerMap);
    }

这里的jobFile即为任务所在的目录,实际上任务所在目录来源于作业所在目录,因此一个JobInProgress对象在创建的时候就会将目录信息记录下来,实际上,最终的文件目录是在Job初始化时的submitJobInternal方法传进来的。我们回顾以下关键代码:

        Path jobStagingArea = JobSubmissionFiles.getStagingDir(JobClient.this,
            jobCopy);
        JobID jobId = jobSubmitClient.getNewJobId();
        Path submitJobDir = new Path(jobStagingArea, jobId.toString());

这个目录即Staging目录,是根据一定规则自动生成的,最终位于HDFS的某个路径中。

我们回到initializeJob这个方法。因为Job提交时其实已经确定了目录,所以以下代码就是找到这个目录对应的文件系统:

    final JobID jobId = t.getJobID();
    final Path jobFile = new Path(t.getJobFile());
    final Configuration conf = getJobConf();
    FileSystem userFs = getFS(jobFile, jobId, conf);

这个文件系统就是HDFS中作业所在的Staging目录。Staging是集结地、驻留区的意思。用户在提交作业的时候,会将所需的jar文件、配置文件(conf、xml)、其它依赖的lib包、要处理数据的Split信息文件等全部上传到这个地方。

获得了这个目录后,接下来,TaskTracker的任务初始化过程会将上述处于HDFS中的文件拷贝到本地(如Linux)文件系统。使用localizeJobConfFile方法:

    FileStatus status = null;
    long jobFileSize = -1;
    try {
      status = userFs.getFileStatus(jobFile);
      jobFileSize = status.getLen();
    } catch(FileNotFoundException fe) {
      jobFileSize = -1;
    }
    Path localJobFile = lDirAlloc.getLocalPathForWrite(getPrivateDirJobConfFile(user,jobId.toString()), jobFileSize, fConf);

    // Download job.xml
    userFs.copyToLocalFile(jobFile, localJobFile);

lDirAlloc是一个LocalDirAllocator对象,主要用于本地目录的操作。getLocalPathForWrite方法:

  public Path getLocalPathForWrite(String pathStr, long size, 
                                   Configuration conf,
                                   boolean checkWrite) throws IOException {
    AllocatorPerContext context = obtainContext(contextCfgItemName);
    return context.getLocalPathForWrite(pathStr, size, conf, checkWrite);
  }

AllocatorPerContext对象通过 obtainContext方法获得。这里面涉及到一个参数contextCfgItemName,该参数是LocalDirAllocator对象在创建的时候传入的:该对象本身是TaskTracker的一个变量:

    this.localDirAllocator = new LocalDirAllocator("mapred.local.dir");

  public LocalDirAllocator(String contextCfgItemName) {
    this.contextCfgItemName = contextCfgItemName;
  }

之后调用 context.getLocalPathForWrite方法,首先执行confChanged方法。在该方法里,有以下代码:

      String newLocalDirs = conf.get(contextCfgItemName);
      if (!newLocalDirs.equals(savedLocalDirs)) {
        String[] localDirs = conf.getStrings(contextCfgItemName);
        localFS = FileSystem.getLocal(conf);

此处,通过调用文件系统相关方法,会创建"mapred.local.dir"所指定的本地目录。

创建好了本地目录后,执行文件拷贝操作copyToLocalFile:

  public void copyToLocalFile(boolean delSrc, Path src, Path dst)
    throws IOException {
    FileUtil.copy(this, src, getLocal(getConf()), dst, delSrc, getConf());
  }

即job.xml文件被从HDFS中拷贝到TaskTracker所在的本地文件系统。该目录由配置文件"mapred.local.dir"指定。

另外,在这之前还有一些拷贝关于用户操作权限等文件:localizeJobTokenFile(与localizeJobConfFile过程类似)。

接下来对拷贝到本地的配置文件进行解析,得到JobConf对象:

final JobConf localJobConf = new JobConf(localJobFile);

并将这个信息加入到缓存中,因为localJobConf是根据某个任务拷贝过来的job.xml解析的,如果后面需要再用,则可以使用该缓存信息。TaskTracker里面维护了一个TrackerDistributedCacheManager distributedCacheManager对象,该对象里面维护了一个任务缓存:

  private Map<JobID, TaskDistributedCacheManager> jobArchives =
    Collections.synchronizedMap(
        new HashMap<JobID, TaskDistributedCacheManager>());

以及一个TaskController taskController。

首先将上面解析的任务配置参数加入到缓存中:

  public TaskDistributedCacheManager newTaskDistributedCacheManager(JobID jobId,
                                 Configuration taskConf) throws IOException {
    TaskDistributedCacheManager result = new TaskDistributedCacheManager(this, taskConf);
    jobArchives.put(jobId, result);
    return result;
  }

TrackerDistributedCacheManager以及TaskDistributedCacheManager 有什么意义呢?每个TaskTracker创建时会创建一个TrackerDistributedCacheManager对象,其主要作用是管理该机器上所有任务的Cache文件。而当任务在初始化时,会由TrackerDistributedCacheManager产生一个TaskDistributedCacheManager对象(如上面代码示例),用于管理本任务的Cache文件。有什么好处呢?比如,一个TaskTracker上可能会有一个Job的多个Task,这样,某个TaskDistributedCacheManager已经获取处理了某些Cache文件,则其他任务无需再处理。这里所谓的Cache文件,指的是从HDFS文件系统拷贝到本地文件系统的那些文件,本地的Linux系统相对于HDFS来说可以看做一种缓存,因为不能每次要访问某个参数,或者一个机器上的所有任务要访问Job信息都去HDFS中去获取和解析吧,这就是Distributed Cache分布式缓存的含义,因为有很多机器会拷贝这个文件。对于某个TaskTracker而言,通过拷贝到本地,不仅加速了后面的处理时间,也为其他任务做出了贡献。只是需要注意的是,如何维护本地的缓存信息和HDFS中的一致是需要解决的问题。Hadoop里设计了一个CacheFile类,里面包含了时间戳等信息,只要比较一下该时间戳以及HDFS中文件时间戳是否一致即可。

获得了缓存对象后,本地的Job配置会有一些更新,更新后写回到本地的配置文件:

          // Set some config values
          localJobConf.set(JobConf.MAPRED_LOCAL_DIR_PROPERTY,
              getJobConf().get(JobConf.MAPRED_LOCAL_DIR_PROPERTY));
          if (conf.get("slave.host.name") != null) {
            localJobConf.set("slave.host.name", conf.get("slave.host.name"));
          }
          resetNumTasksPerJvm(localJobConf);
          localJobConf.setUser(t.getUser());

          // write back the config (this config will have the updates that the
          // distributed cache manager makes as well)
          JobLocalizer.writeLocalJobFile(localJobFile, localJobConf);

接下来进入到taskController的initializeJob方法,进行后续的初始化。TaskController是一个抽象类,其实现是DefaultTaskController和LinuxTaskController。DefaultTaskController是默认的任务控制器,LinuxTaskController在Linux系统上使用,在用户权限设置等方面有些不同,可能是利用Linux的用户管理等等进行任务管理,以后有时间再研究,现在先来看默认的DefaultTaskController。

其initializeJob方法声明如下,可以看出,该方法从TaskTracker的私有空间拷贝证书文件到Job的私有空间;创建Job工作目录,下载job.jar文件并解压得到要执行的任务的.class文件,并更新配置。之后,创建一个JobConf配置对象,设置分布式缓存和用户日志目录等

  /**
   * This routine initializes the local file system for running a job.
   * Details:
   * <ul>
   * <li>Copies the credentials file from the TaskTracker's private space to
   * the job's private space </li>
   * <li>Creates the job work directory and set 
   * {@link TaskTracker#JOB_LOCAL_DIR} in the configuration</li>
   * <li>Downloads the job.jar, unjars it, and updates the configuration to 
   * reflect the localized path of the job.jar</li>
   * <li>Creates a base JobConf in the job's private space</li>
   * <li>Sets up the distributed cache</li>
   * <li>Sets up the user logs directory for the job</li>
   * </ul>
   * This method must be invoked in the access control context of the job owner 
   * user. This is because the distributed cache is also setup here and the 
   * access to the hdfs files requires authentication tokens in case where 
   * security is enabled.
   * @param user the user in question (the job owner)
   * @param jobid the ID of the job in question
   * @param credentials the path to the credentials file that the TaskTracker
   * downloaded
   * @param jobConf the path to the job configuration file that the TaskTracker
   * downloaded
   * @param taskTracker the connection to the task tracker
   * @throws IOException
   * @throws InterruptedException
   */
  @Override
  public void initializeJob(String user, String jobid, 
                            Path credentials, Path jobConf, 
                            TaskUmbilicalProtocol taskTracker,
                            InetSocketAddress ttAddr
                            ) 

其输入参数主要有:用户、JobID、证书目录、Job配置文件目录、一个TaskUmbilicalProtocol对象,以及TaskTracker的IP地址信息,这里的TaskUmbilicalProtocol对象需要细说一下,TaskUmbilicalProtocol是和InterTrackerProtocol等等类似的一个RPC接口,我们前面分析过,再来总结一下:

1)JobClient和JobTracker之间存在一个RPC接口JobSubmissionProtocol,JobClient调用这个接口,JobTracker实现了这个接口。

2)JobTracker和TaskTracker之间存在一个RPC接口InterTrackerProtocol,TaskTracker调用这个接口,JobTracker实现了这个接口。

TaskUmbilicalProtocol也类似,不过是TaskTracker实现的,那么,谁来掉用呢?

Umbilical大致是脐带的意思,直观含义是连接胎儿和母体之间的重要通道。TaskTracker要启动一个Map或Reduce任务的时候,是通过启动另外一个JAVA虚拟机去运行这个任务,用户提交作业时,就是JobClient和JobTracker在打交道,JobTracker对这个作业初始化的过程会创建HDFS目录,将jar文件、分析处理的文件(即FileInputFormat.addInputPath里面指定的),生成Split信息文件等统统放到这个目录。在某个TaskTracker与自己心跳时,从作业队列挑选一些合适的任务,让TaskTracker运行,两者之间交互了Task的信息,TaskTracker接收到这个信息后,去HDFS目录里面将Job文件拷贝到本地,同时jar文件也拷贝下来,进行解压以后怎么运行呢?TaskTracker会启动一个JAVA虚拟机,这个JAVA虚拟机当然也是一个进程,不过称为TaskTracker进程的子进程,这个子进程和TaskTracker进程之间也需要交互一些信息,比如汇报状态等等,此时就导致了TaskUmbilicalProtocol这个RPC接口的出现。我们看看他有哪些方法:

  JvmTask getTask(JvmContext context) throws IOException;

  boolean statusUpdate(TaskAttemptID taskId, TaskStatus taskStatus,
      JvmContext jvmContext) throws IOException, InterruptedException;
  
  void reportDiagnosticInfo(TaskAttemptID taskid, String trace,
      JvmContext jvmContext) throws IOException;
 
  void reportNextRecordRange(TaskAttemptID taskid, SortedRanges.Range range,
      JvmContext jvmContext) throws IOException;

  boolean ping(TaskAttemptID taskid, JvmContext jvmContext) throws IOException;

  void done(TaskAttemptID taskid, JvmContext jvmContext) throws IOException;
  
  void commitPending(TaskAttemptID taskId, TaskStatus taskStatus,
      JvmContext jvmContext) throws IOException, InterruptedException;  

  boolean canCommit(TaskAttemptID taskid, JvmContext jvmContext) throws IOException;

  void shuffleError(TaskAttemptID taskId, String message, JvmContext jvmContext) throws IOException;
  
  void fsError(TaskAttemptID taskId, String message, JvmContext jvmContext) throws IOException;

  void fatalError(TaskAttemptID taskId, String message, JvmContext jvmContext)  throws IOException;

  MapTaskCompletionEventsUpdate getMapCompletionEvents(JobID jobId, 
                                                       int fromIndex, 
                                                       int maxLocs,
                                                       TaskAttemptID id,
                                                       JvmContext jvmContext)  throws IOException;

  void updatePrivateDistributedCacheSizes(org.apache.hadoop.mapreduce.JobID jobId,long[] sizes) throws IOException;

可见,主要是状态更新、错误信息上报、周期地ping、任务完成通知等等。

所以,现在可以理解,上面initializeJob方法中传入的TaskUmbilicalProtocol对象其实就是TaskTracker.this。

我们接下里分析initializeJob方法。

这个方法的主要功能上面已经进行了简单描述,来看一下细节。首先需要做的就是将TaskTracker从HDFS拷贝过来的文件进行二次拷贝,拷贝到作业所在的目录。这里使用了JobLocalizer类,这个类会创建以下目录:

/**
 * Internal class responsible for initializing the job, not intended for users.
 * Creates the following hierarchy:
 *   <li>$mapred.local.dir/taskTracker/$user</li>
 *   <li>$mapred.local.dir/taskTracker/$user/jobcache</li>
 *   <li>$mapred.local.dir/taskTracker/$user/jobcache/$jobid/work</li>
 *   <li>$mapred.local.dir/taskTracker/$user/jobcache/$jobid/jars</li>
 *   <li>$mapred.local.dir/taskTracker/$user/jobcache/$jobid/jars/job.jar</li>
 *   <li>$mapred.local.dir/taskTracker/$user/jobcache/$jobid/job.xml</li>
 *   <li>$mapred.local.dir/taskTracker/$user/jobcache/$jobid/jobToken</li>
 *   <li>$mapred.local.dir/taskTracker/$user/distcache</li>
 */

主要的代码是:

    FileSystem localFs = FileSystem.getLocal(getConf());
    JobLocalizer localizer = new JobLocalizer((JobConf)getConf(), user, jobid);
    localizer.createLocalDirs();
    localizer.createUserDirs();
    localizer.createJobDirs();
    JobConf jConf = new JobConf(jobConf);
    localizer.createWorkDir(jConf);
    //copy the credential file
    Path localJobTokenFile = lDirAlloc.getLocalPathForWrite(
        TaskTracker.getLocalJobTokenFile(user, jobid), getConf());
    FileUtil.copy(localFs, credentials, localFs, localJobTokenFile, false, getConf());
    //setup the user logs dir
    localizer.initializeJobLogDir();
    // Download the job.jar for this job from the system FS
    // setup the distributed cache
    // write job acls
    // write localized config
    localizer.localizeJobFiles(JobID.forName(jobid), jConf, localJobTokenFile, taskTracker);

这里比较重要的最后一行:localizeJobFiles,jar文件就是在这里处理的,进去看看。

执行的第一行代码就是localizeJobJarFile,前面已经把job.xml等文件拷贝到本地了,jar文件还没拷贝,此处就是做这个工作的。

主要代码为:

      // Here we check for five times the size of jarFileSize to accommodate for
      // unjarring the jar file in the jars directory
      Path localJarFile = lDirAlloc.getLocalPathForWrite(JARDST, 5 * jarFileSize, ttConf);
      //Download job.jar
      userFs.copyToLocalFile(jarFilePath, localJarFile);
      localJobConf.setJar(localJarFile.toString());
      // Also un-jar the job.jar files. We un-jar it so that classes inside
      // sub-directories, for e.g., lib/, classes/ are available on class-path
      RunJar.unJar(new File(localJarFile.toString()),
          new File(localJarFile.getParent().toString()));
      FileUtil.chmod(localJarFile.getParent().toString(), "ugo+rx", true);

在获得了本地的jar目录后,从HDFS中进行拷贝,并利用RunJar的unJar方法进行解压。JAVA里有JarFile这个类用于JAR打包和解压。顺便说一下,jar文件采用的就是ZIP格式,JarFile其实就是利用ZIP的解压方法,而JAVA中本身也有ZIP的解压方法,ZIP利用的是LZ系列字典压缩算法以及Huffman、游程编码等算法的融合。JarFile继承于java.util.zip.ZipFile这个类。

至此,TaskController的initializeJob方法结束,返回TaskTracker的initializeJob方法,并继续返回localizeJob方法。RunningJob这个对象初始化完毕,改变下面的值:

 rjob.localized = true;

并向其他线程执行通告:

      synchronized (rjob) {
        if (rjob.localizing) {
          rjob.localizing = false;
          rjob.notifyAll();
        }
      }

    synchronized (runningJobs) {
      runningJobs.notify(); //notify the fetcher thread
    }

localizeJob方法执行完毕,之后返回startNewTask方法。接下来执行launchTaskForJob方法:

  protected void launchTaskForJob(TaskInProgress tip, JobConf jobConf,
                                RunningJob rjob) throws IOException {
    synchronized (tip) {
      jobConf.set(JobConf.MAPRED_LOCAL_DIR_PROPERTY,
                  localStorage.getDirsString());
      tip.setJobConf(jobConf);
      tip.setUGI(rjob.ugi);
      tip.launchTask(rjob);
    }
  }

首先是设置一些任务参数,之后调用launchTask方法,在launchTask方法里,因为前面涉及的都是Job相关的配置信息,到了此时,Task本身的配置信息需要进行设置:

  /**
   * Localize the given JobConf to be specific for this task.
   */
  public void localizeConfiguration(JobConf conf) throws IOException {
    conf.set("mapred.tip.id", taskId.getTaskID().toString()); 
    conf.set("mapred.task.id", taskId.toString());
    conf.setBoolean("mapred.task.is.map", isMapTask());
    conf.setInt("mapred.task.partition", partition);
    conf.set("mapred.job.id", taskId.getJobID().toString());
  }

比如Task ID、是否是Map任务等等。

设置Task的状态为运行状态:

        if (this.taskStatus.getRunState() == TaskStatus.State.UNASSIGNED) {
          this.taskStatus.setRunState(TaskStatus.State.RUNNING);
        }

接下来会创建一个TaskRunner对象,TaskRunner 对象本身是TaskInProgress中的一个变量,其创建则是在MapTask或ReduceTask中实现的。

  @Override
  public TaskRunner createRunner(TaskTracker tracker, 
                                 TaskTracker.TaskInProgress tip,
                                 TaskTracker.RunningJob rjob
                                 ) throws IOException {
    return new MapTaskRunner(tip, tracker, this.conf, rjob);
  }

TaskRunner是一个抽象类,主要有MapTaskRunner和ReduceTaskRunner两种实现。该抽象类继承于Thread,是一个线程类,并且在TaskRunner中实现了run方法。MapTaskRunner在创建过程中,利用父类构造方法,将一系列参数传进去:

  public TaskRunner(TaskTracker.TaskInProgress tip, TaskTracker tracker, 
                    JobConf conf, TaskTracker.RunningJob rjob
                    ) throws IOException {
    this.tip = tip;
    this.t = tip.getTask();
    this.tracker = tracker;
    this.conf = conf;
    this.mapOutputFile = new MapOutputFile();
    this.mapOutputFile.setConf(conf);
    this.jvmManager = tracker.getJvmManagerInstance();
    this.localdirs = conf.getLocalDirs();
    taskDistributedCacheManager = rjob.distCacheMgr;
  }

比如Map任务之后的目录等,这里还有一个jvmManager对象,该对象是TaskTracker中的一个对象,每个TaskTracker一个,在TaskTracker初始化的时候会创建出来:

  public JvmManager(TaskTracker tracker) {
    mapJvmManager = new JvmManagerForType(tracker.getMaxCurrentMapTasks(), true, tracker);
    reduceJvmManager = new JvmManagerForType(tracker.getMaxCurrentReduceTasks(), false, tracker);
  }

其中:JvmManagerForType的构造方法如下,第一个参数是要管理的JAVA虚拟机的个数,而tracker.getMaxCurrentMapTasks()获得的就是为本TaskTracker配置的最大Map槽数。

    public JvmManagerForType(int maxJvms, boolean isMap, 
        TaskTracker tracker) {
      this.maxJvms = maxJvms;
      this.isMap = isMap;
      this.tracker = tracker;
      sleeptimeBeforeSigkill =
        tracker.getJobConf().getLong(DELAY_BEFORE_KILL_KEY,
                                     DEFAULT_SLEEPTIME_BEFORE_SIGKILL);
    }

创建好了TaskRunner之后,即启动新线程:

        setTaskRunner(task.createRunner(TaskTracker.this, this, rjob));
        this.runner.start();

launchTask这个方法启动了TaskRunner之后,则返回launchTaskForJob方法,继续返回startNewTask方法。对tasksToLaunch中的其他TaskInProgress方法重复执行上面的初始化操作。startNewTask创建的局部线程已经结束。如果有新的任务到达,TaskLauncher会启动新的线程来执行上面的工作,下面进入到TaskRunner的run方法执行。

 

4,TaskRunner的run方法接力执行:

这个方法为JVM的启动提供命令行生成、环境变量、类库路径等信息。首先创建Task工作目录,注意上面的JobLocalizer类创建了一系列Job目录,这里创建的是Task相关的目录。

      final File workDir =
      new File(new Path(localdirs[rand.nextInt(localdirs.length)], 
          TaskTracker.getTaskWorkDir(t.getUser(), taskid.getJobID().toString(), 
          taskid.toString(),
          t.isTaskCleanupTask())).toString());
      
      String user = tip.getUGI().getUserName();

之后调用getClassPaths方法获取解压后的.class文件路径

  private static List<String> getClassPaths(JobConf conf, File workDir,
      TaskDistributedCacheManager taskDistributedCacheManager)
      throws IOException {
    // Accumulates class paths for child.
    List<String> classPaths = new ArrayList<String>();
    boolean userClassesTakesPrecedence =  conf.getBoolean(MAPREDUCE_USER_CLASSPATH_FIRST,false);
    if (!userClassesTakesPrecedence) {
      // start with same classpath as parent process
      appendSystemClasspaths(classPaths);
    }
    // include the user specified classpath
    appendJobJarClasspaths(conf.getJar(), classPaths);
    // Distributed cache paths
    classPaths.addAll(taskDistributedCacheManager.getClassPaths());
    // Include the working dir too
    classPaths.add(workDir.toString());
    if (userClassesTakesPrecedence) {
      // parent process's classpath is added last
      appendSystemClasspaths(classPaths);
    }
    return classPaths;
  }

总之,一个JAVA程序运行需要很多库目录,比如本地目录、JRE目录等,首先加入系统目录:

  private static void appendSystemClasspaths(List<String> classPaths) {
    for (String c : System.getProperty("java.class.path").split(
        SYSTEM_PATH_SEPARATOR)) {
      classPaths.add(c);
    }
  }

appendJobJarClasspaths方法加入到Job的jar文件所在的本地目录(已经从HDFS中拷贝下来了)。

之后,创建JAVA虚拟机运行所需要的参数:

      //  Build exec child JVM args.
      Vector<String> vargs = getVMArgs(taskid, workDir, classPaths, logSize);

说白了,就是产生一堆“java xxx -D ...”等等这个命令,比如一些参数示例,虚拟机内存占多大等等:

    //  <property>
    //    <name>mapred.map.child.java.opts</name>
    //    <value>-Xmx 512M -verbose:gc -Xloggc:/tmp/@taskid@.gc \
    //           -Dcom.sun.management.jmxremote.authenticate=false \
    //           -Dcom.sun.management.jmxremote.ssl=false \
    //    </value>
    //  </property>

首先找到JAVA虚拟机那个可执行文件在哪个目录(安装目录/bin),然后加进去,这自然是第一步。

    File jvm =                                  // use same jvm as parent
      new File(new File(System.getProperty("java.home"), "bin"), "java");
  vargs.add(jvm.toString());

另外,就是一些库文件目录等等,比如:

    Path childTmpDir = createChildTmpDir(workDir, conf, false);
    vargs.add("-Djava.io.tmpdir=" + childTmpDir);

    // Add classpath.
    vargs.add("-classpath");
    String classPath = StringUtils.join(SYSTEM_PATH_SEPARATOR, classPaths);
    vargs.add(classPath);

关键的是把要执行的class的类名写上:

vargs.add(Child.class.getName());  // main of Child

可以看出,执行的是Child这个类,这个类有一个Main方法。我们后面再看。

获得了JAVA运行参数后,继续获取环境变量:

      Map<String, String> env = new HashMap<String, String>();
      errorInfo = getVMEnvironment(errorInfo, user, workDir, conf, env, taskid,
                                   logSize);

 比如LD_LIBRARY_PATH等等这些环境变量。

之后,执行launchJvmAndWait方法:

  void launchJvmAndWait(List <String> setup, Vector<String> vargs, File stdout,
      File stderr, long logSize, File workDir)
      throws InterruptedException, IOException {
    jvmManager.launchJvm(this, jvmManager.constructJvmEnv(setup, vargs, stdout,
        stderr, logSize, workDir, conf));
    synchronized (lock) {
      while (!done) {
        lock.wait();
      }
    }
  }

里面的核心方法是jvmManager.launchJvm

  public void launchJvm(TaskRunner t, JvmEnv env
                        ) throws IOException, InterruptedException {
    if (t.getTask().isMapTask()) {
      mapJvmManager.reapJvm(t, env);
    } else {
      reduceJvmManager.reapJvm(t, env);
    }
  }

mapJvmManagerreduceJvmManager都是一个JvmManagerForType对象,前面提过,在TaskTracker初始化的时候会进行创建。执行JvmManagerForType的reapJvm方法,该方法中,有一个变量spawnNewJvm记录了是否应该启动一个新的JVM。spawn是产卵的意思,该标志的含义很明显,如果不满足一些条件,那么是不能启动虚拟机的,那么有哪些条件呢?

JvmManagerForType对象里面有几个映射表:

    //Mapping from the JVM IDs to running Tasks
    Map <JVMId,TaskRunner> jvmToRunningTask = new HashMap<JVMId, TaskRunner>();
    //Mapping from the tasks to JVM IDs
    Map <TaskRunner,JVMId> runningTaskToJvm = new HashMap<TaskRunner, JVMId>();
    //Mapping from the JVM IDs to Reduce JVM processes
    Map <JVMId, JvmRunner> jvmIdToRunner =  new HashMap<JVMId, JvmRunner>();
    //Mapping from the JVM IDs to process IDs
    Map <JVMId, String> jvmIdToPid = new HashMap<JVMId, String>();

jvmToRunningTask记录了JVMID和TaskRunner对象的关系;runningTaskToJvm则反过来记录;jvmIdToRunner记录了JVMID和JvmRunner的关系。注意这里有几个不同的对象:TaskRunner代表了要运行的任务;JvmRunner代表了要启动的虚拟机。

在判断spawnNewJvm变量时,首先获得当前已经运行的JVM:

int numJvmsSpawned = jvmIdToRunner.size();

如果numJvmsSpawned < maxJvms,那么表明目前启动的JVM还没达到最大值,可以启动:spawnNewJvm=true;

如果numJvmsSpawned >= maxJvms,则表明目前的数量已经至少等于最大可允许的JVM数量(=配置的Map或Reduce Slot数目),这种情况下按理来说可能不能启动了,不过也不一定。这里会对每一个JVM进行逐一判断,找找看,看哪个JVM适合处理这个任务。这是什么逻辑呢?有两种可能,一种可能是利用已经启动的JVM,判断代码如下:

          //look for a free JVM for this job; if one exists then just break
          if (jId.equals(jobId) && !jvmRunner.isBusy() && !jvmRunner.ranAll()){
            setRunningTaskForJvm(jvmRunner.jvmId, t); //reserve the JVM
            LOG.info("No new JVM spawned for jobId/taskid: " + 
                     jobId+"/"+t.getTask().getTaskID() +
                     ". Attempting to reuse: " + jvmRunner.jvmId);
            return;
          }

如果一个JVM内的任务与要启动任务的Job ID一样(jId.equals(jobId)),且已经不忙了,即处于空闲状态(!jvmRunner.isBusy()),并且所有任务没有执行完毕(!jvmRunner.ranAll()),此时则将该任务(TaskRunner代表了要运行的任务)直接让该虚拟机执行,执行setRunningTaskForJvm方法:

    synchronized public void setRunningTaskForJvm(JVMId jvmId, TaskRunner t) {
      jvmToRunningTask.put(jvmId, t);
      runningTaskToJvm.put(t,jvmId);
      jvmIdToRunner.get(jvmId).setBusy(true);
    }

除了上面这种情况,还有一种情况就是看看能不能杀掉一个JVM,判断代码如下:

          if ((jId.equals(jobId) && jvmRunner.ranAll()) ||
              (!jId.equals(jobId) && !jvmRunner.isBusy())) {
            runnerToKill = jvmRunner;
            spawnNewJvm = true;
          }

即两者Job ID相同且JVM已经执行完毕了,此时可以杀掉这个JVM来执行新任务;或者两者Job ID不同且那个JVM处于空闲状态。这说明,一个新的任务并不是一定要启动一个新的JVM,可能会利用已有的JVM继续运行。

如果spawnNewJvm=true,则表示可以启动新的JVM:

      if (spawnNewJvm) {
        if (runnerToKill != null) {
          LOG.info("Killing JVM: " + runnerToKill.jvmId);
          killJvmRunner(runnerToKill);
        }
        spawnNewJvm(jobId, env, t);
        return;
      }

接下来我们看看spawnNewJvm这个方法,其代码较简单:

    private void spawnNewJvm(JobID jobId, JvmEnv env,  
        TaskRunner t) {
      JvmRunner jvmRunner = new JvmRunner(env, jobId, t.getTask());
      jvmIdToRunner.put(jvmRunner.jvmId, jvmRunner);
      jvmRunner.setDaemon(true);
      jvmRunner.setName("JVM Runner " + jvmRunner.jvmId + " spawned.");
      setRunningTaskForJvm(jvmRunner.jvmId, t);
      LOG.info(jvmRunner.getName());
      jvmRunner.start();
    }

首先创建一个JvmRunner对象,表示要运行的JVM,将该JVM记录下来与JVMID与JVMRunner的关系。JVMID是一个JobID加上随机生成的整数组合成的一个对象。因为JvmRunner对象是一个线程类,所以最后一句启动了一个新线程:JVMRunner。此时,TaskRunner线程类的工作结束。

 

5,JVMRunner的run方法继续执行,其run方法也很简单,就一句话:

 

      @Override
      public void run() {
        try {
          runChild(env);
        } catch (InterruptedException ie) {
          return;
        } catch (IOException e) {
。。。。。。。。。。。
        } catch (Throwable e) {
。。。。。。。。。
        } finally {
          jvmFinished();
        }
      }

 

在方法runChild里,关键的一句是:

                exitCode = tracker.getTaskController().launchTask(user,
                    jvmId.jobId.toString(), taskAttemptIdStr, env.setup,
                    env.vargs, env.workDir, env.stdout.toString(),
                    env.stderr.toString());

于是进入DefaultTaskController的launchTask方法,该方法的作用是创建所有Task所需的目录,启动子JVM。

里面关键的代码是:

      // get the JVM command line.
      String cmdLine = 
        TaskLog.buildCommandLine(setup, jvmArguments,
            new File(stdout), new File(stderr), logSize, true);

      // write the command to a file in the
      // task specific cache directory
      // TODO copy to user dir
      Path p = new Path(allocator.getLocalPathForWrite(
          TaskTracker.getPrivateDirTaskScriptLocation(user, jobId, attemptId),
          getConf()), COMMAND_FILE);

      String commandFile = writeCommand(cmdLine, rawFs, p);
      rawFs.setPermission(p, TaskController.TASK_LAUNCH_SCRIPT_PERMISSION);

cmdLin即生成要启动JVM的命令行字符串;第二句话是创建一个文件COMMAND_FILE,其值是COMMAND_FILE = "taskjvm.sh",即创建一个脚本文件,之后将命令行字符串写入该脚本文件,并设置权限。

最后,做好了这些准备工作后,执行脚本:

      shExec = new ShellCommandExecutor(new String[]{
          "bash", commandFile},
          currentWorkDirectory);
      shExec.execute();

ShellCommandExecutor是一个脚本命令的执行类,execute方法将执行以下代码:

    /** Execute the shell command. */
    public void execute() throws IOException {
      this.run();    
    }

  /** check to see if a command needs to be executed and execute if needed */
  protected void run() throws IOException {
    if (lastTime + interval > System.currentTimeMillis())
      return;
    exitCode = 0; // reset for next run
    runCommand();
  }

因此,进入runCommand方法,该方法使用了JAVA中提供的一个类:java.lang.ProcessBuilder,这个类用于创建操作系统进程。

首先设置环境变量等:

    if (environment != null) {
      builder.environment().putAll(this.environment);
    }
    if (dir != null) {
      builder.directory(this.dir);
    }

然后到了启动JVM的最关键一步,启动进程:

    process = builder.start();

等待进程运行:

      // wait for the process to finish and check the exit code
      exitCode  = process.waitFor();

接下来就是新进程的运行,启动的是Child这个类的Main方法:

  public static void main(String[] args) throws Throwable {
    LOG.debug("Child starting");
。。。。。。

这个执行较长,我们留作下一节分析。

 

总结一下,本节主要分析了JobTracker将任务调度给TaskTracker后,主要执行了5步操作,对应于5个线程:

1)TaskTracker调用addToTaskQueue加入TaskLauncher(mapLauncher和reduceLauncher)的队列。在这个过程中,会创建对应的TaskInProgress对象加入到队列List<TaskInProgress> tasksToLaunch中,然后返回(这一部分操作是由TaskTracker的心跳线程执行的);

2)TaskLauncher这个线程类的run会不停地等待新的TaskInProgress,如果出现新的任务,进入startNewTask方法,startNewTask会创建一个新线程,TaskLauncher这个线程类返回去等待新的TaskInProgress任务;

3)startNewTask中的局部线程:从HDFS拷贝相应文件至本地等操作,该线程还会创建一个TaskRunner线程类,接着由其run方法执行;

4)TaskRunner线程类的run方法,负责获取JVM所需的参数信息,又启动JvmRunner这个线程类的run方法;

5)JvmRunner线程类的run方法,负责将JVM启动所需的命令写入一个脚本文件,执行该脚本文件,在另外一个进程中执行了Child的Main方法。

至于该Main进行了什么操作,留作后续分析。

 

posted on 2014-08-30 18:52  esingchan  阅读(1860)  评论(0编辑  收藏  举报