【原】Java程序调用远程Shell脚本

1 #!/bin/sh
2 source /etc/profile
3 dbName=$1
4 tableName=$2
5 echo [`date +‘%Y-%m-%d %H:%M:%S‘`]‘ start loading data...‘
6 mysql -uroot -p123456 -P3306 ${dbName} -e "LOAD DATA LOCAL INFILE ‘/home/IFileGenTo    ol_ZhangJian/bak_data/t_acc_subject_provision_data_bak.txt‘ INTO TABLE ${tableName}     FIELDS TERMINATED BY ‘;‘"
7 echo [`date +‘%Y-%m-%d %H:%M:%S‘`]‘ end loading data...‘
8 exit
9 EOF

1 <dependency>
2     <groupId>org.jvnet.hudson</groupId>
3     <artifactId>ganymed-ssh2</artifactId>
4     <version>build210-hudson-1</version>
5 </dependency>

  1 import java.io.IOException;
  2 import java.io.InputStream;
  3 import java.io.UnsupportedEncodingException;
  4 import java.nio.charset.Charset;
  5 
  6 import org.apache.commons.io.IOUtils;
  7 
  8 import ch.ethz.ssh2.ChannelCondition;
  9 import ch.ethz.ssh2.Connection;
 10 import ch.ethz.ssh2.Session;
 11 import ch.ethz.ssh2.StreamGobbler;
 12 
 13 public class RemoteShellExecutor {
 14      
 15      private Connection conn;
 16      /** 远程机器IP */
 17      private String ip;
 18      /** 用户名 */
 19      private String osUsername;
 20      /** 密码 */
 21      private String password;
 22      private String charset = Charset.defaultCharset().toString();
 23 
 24      private static final int TIME_OUT = 1000 * 5 * 60;
 25 
 26      /**
 27       * 构造函数
 28       * @param ip
 29       * @param usr
 30       * @param pasword
 31       */
 32      public RemoteShellExecutor(String ip, String usr, String pasword) {
 33           this.ip = ip;
 34          this.osUsername = usr;
 35          this.password = pasword;
 36      }
 37 
 38 
 39      /**
 40      * 登录
 41      * @return
 42      * @throws IOException
 43      */
 44      private boolean login() throws IOException {
 45          conn = new Connection(ip);
 46          conn.connect();
 47          return conn.authenticateWithPassword(osUsername, password);
 48      }
 49 
 50      /**
 51      * 执行脚本
 52      * 
 53      * @param cmds
 54      * @return
 55      * @throws Exception
 56      */
 57      public int exec(String cmds) throws Exception {
 58          InputStream stdOut = null;
 59          InputStream stdErr = null;
 60          String outStr = "";
 61          String outErr = "";
 62          int ret = -1;
 63          try {
 64          if (login()) {
 65              // Open a new {@link Session} on this connection
 66              Session session = conn.openSession();
 67              // Execute a command on the remote machine.
 68              session.execCommand(cmds);
 69              
 70              stdOut = new StreamGobbler(session.getStdout());
 71              outStr = processStream(stdOut, charset);
 72              
 73              stdErr = new StreamGobbler(session.getStderr());
 74              outErr = processStream(stdErr, charset);
 75              
 76              session.waitForCondition(ChannelCondition.EXIT_STATUS, TIME_OUT);
 77              
 78              System.out.println("outStr=" + outStr);
 79              System.out.println("outErr=" + outErr);
 80              
 81              ret = session.getExitStatus();
 82          } else {
 83              throw new Exception("登录远程机器失败" + ip); // 自定义异常类 实现略
 84          }
 85          } finally {
 86              if (conn != null) {
 87                  conn.close();
 88              }
 89              IOUtils.closeQuietly(stdOut);
 90              IOUtils.closeQuietly(stdErr);
 91          }
 92          return ret;
 93      }
 94 
 95      /**
 96      * @param in
 97      * @param charset
 98      * @return
 99      * @throws IOException
100      * @throws UnsupportedEncodingException
101      */
102      private String processStream(InputStream in, String charset) throws Exception {
103          byte[] buf = new byte[1024];
104          StringBuilder sb = new StringBuilder();
105          while (in.read(buf) != -1) {
106              sb.append(new String(buf, charset));
107          }
108          return sb.toString();
109      }
110 
111     public static void main(String args[]) throws Exception {
112         RemoteShellExecutor executor = new RemoteShellExecutor("192.168.234.123", "root", "beebank");
113         // 执行myTest.sh 参数为java Know dummy
114         System.out.println(executor.exec("/home/IFileGenTool /load_data.sh t_users myDataBase01"));
115     }
116 }