コード例 #1
0
 public static void main(String[] args) {
   ExecuteResult result = ExecuteCmdUtils.executeCmd("cd /home/machen && python hello.py");
   System.out.println(result.getExitCode());
   System.out.println(result.getReturnString());
   System.out.println(result.getStartTime().toString());
   System.out.println(result.getEndTime().toString());
 }
コード例 #2
0
  public static ExecuteResult executeCmd(String cmd, String tempShellFilePath) {
    ExecuteResult result = new ExecuteResult();
    PrintWriter out = null;
    int exitValue = -1;
    File shellFile =
        new File(
            tempShellFilePath); // 将这个命令写入shell,然后执行sh temp.sh,这样就可以避免不能执行cd /home/mac && python
    // test.py的问题了
    boolean fileChanged = true;
    ByteArrayOutputStream stdout = null;
    try {
      if (shellFile.exists()) {
        String content = FileUtils.readFileToString(shellFile, charCode);
        if (cmd.equals(content)) {
          fileChanged = false;
        }
      }
      if (fileChanged) {
        out =
            new PrintWriter(
                new OutputStreamWriter(new FileOutputStream(tempShellFilePath), charCode));
        out.write(cmd);
        out.flush();
        out.close();
      }

      CommandLine commandLine = CommandLine.parse("sh " + tempShellFilePath); // 执行那个temp的shell文件
      DefaultExecutor executor = new DefaultExecutor();
      // add return string ,stdout
      stdout = new ByteArrayOutputStream(); // 先放入ByteArrayOutputStream内部的toByteArray()方法返回的byte[]中
      ExecuteStreamHandler stream =
          new PumpStreamHandler(stdout, stdout); // 这样就把标准输出和标准错误输出都定向到stream里了,以供内存里返回给远程
      executor.setStreamHandler(stream);
      result.setOriginalCmd(cmd);
      result.setRealReplacedCmd(ReplaceRealCmdUtils.replaceCmdFromOriginalToReal(cmd));
      result.setStartTime(new Date());
      exitValue = executor.execute(commandLine); // 得到运行返回"结果码",0 -- 成功, 其他 -- 失败s
    } catch (Exception e) {
      e.printStackTrace();
      // System.out.println("GOD,ERROR!");
    } finally {
      result.setReturnString(stdout.toString()); // 这一句话就将 运行后的返回结果字符串放入了result的ReturnString里
      result.setExitCode(exitValue); //
      result.setSuccess(exitValue == ParamCommons.SUCCESS_EXIT_CODE ? true : false);
      result.setEndTime(new Date());
      result.setExecuteTempFilePath(tempShellFilePath);
      if (null != stdout) {
        try {
          stdout.close();

        } catch (IOException e) {
          // TODO Auto-generated catch block
          e.printStackTrace();
        }
      }
      ;
    }
    return result;
  }