Command.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.fs.shell;
  19. import java.io.*;
  20. import org.apache.hadoop.conf.Configuration;
  21. import org.apache.hadoop.conf.Configured;
  22. import org.apache.hadoop.fs.FileStatus;
  23. import org.apache.hadoop.fs.FileSystem;
  24. import org.apache.hadoop.fs.Path;
  25. import org.apache.hadoop.ipc.RemoteException;
  26. /**
  27.  * An abstract class for the execution of a file system command
  28.  */
  29. abstract public class Command extends Configured {
  30.   protected String[] args;
  31.   
  32.   /** Constructor */
  33.   protected Command(Configuration conf) {
  34.     super(conf);
  35.   }
  36.   
  37.   /** Return the command's name excluding the leading character - */
  38.   abstract public String getCommandName();
  39.   
  40.   /** 
  41.    * Execute the command on the input path
  42.    * 
  43.    * @param path the input path
  44.    * @throws IOException if any error occurs
  45.    */
  46.   abstract protected void run(Path path) throws IOException;
  47.   
  48.   /** 
  49.    * For each source path, execute the command
  50.    * 
  51.    * @return 0 if it runs successfully; -1 if it fails
  52.    */
  53.   public int runAll() {
  54.     int exitCode = 0;
  55.     for (String src : args) {
  56.       try {
  57.         Path srcPath = new Path(src);
  58.         FileSystem fs = srcPath.getFileSystem(getConf());
  59.         FileStatus[] statuses = fs.globStatus(srcPath);
  60.         if (statuses == null) {
  61.           System.err.println("Can not find listing for " + src);
  62.           exitCode = -1;
  63.         } else {
  64.           for(FileStatus s : statuses) {
  65.             run(s.getPath());
  66.           }
  67.         }
  68.       } catch (RemoteException re) {
  69.         exitCode = -1;
  70.         String content = re.getLocalizedMessage();
  71.         int eol = content.indexOf('n');
  72.         if (eol>=0) {
  73.           content = content.substring(0, eol);
  74.         }
  75.         System.err.println(getCommandName() + ": " + content);
  76.       } catch (IOException e) {
  77.         exitCode = -1;
  78.         System.err.println(getCommandName() + ": " + e.getLocalizedMessage());
  79.       }
  80.     }
  81.     return exitCode;
  82.   }
  83. }