zoukankan      html  css  js  c++  java
  • 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进行了什么操作,留作后续分析。

  • 相关阅读:
    Load Balancing 折半枚举大法好啊
    Big String 块状数组(或者说平方分割)
    K-th Number 线段树(归并树)+二分查找
    D. Powerful array 莫队算法或者说块状数组 其实都是有点优化的暴力
    CSU OJ PID=1514: Packs 超大背包问题,折半枚举+二分查找。
    运行时Runtime的API
    UIView的API
    UIControl的API
    UIScrollView的API
    使用KVO键值监听
  • 原文地址:https://www.cnblogs.com/esingchan/p/3946616.html
Copyright © 2011-2022 走看看