【原】Java程序調用遠程Shell腳本

此程序的目的是執行遠程機器上的Shell腳本。

【環境參數】
遠程機器IP：192.168.234.123
用戶名：root
密碼：root
Shell腳本的路徑：/home/IFileGenTool/BakProvisionAndOccurEntrance.sh

【具體步驟】
1、在遠程機器上，准備Shell腳本。
[root@localhost IFileGenTool]# vim ./load_data.sh

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

 

2、導入需要依賴的jar包。
Java遠程調用Shell腳本這個程序需要ganymed-ssh2-build210.jar包。
下載地址：http://www.ganymed.ethz.ch/ssh2/
為了調試方便，可以將\ganymed-ssh2-build210\src下的代碼直接拷貝到我們的工程里，
此源碼的好處就是沒有依賴很多其他的包，拷貝過來干干凈凈。

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

3、編寫RemoteShellExecutor工具類。

import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;

import org.apache.commons.io.IOUtils;

import ch.ethz.ssh2.ChannelCondition;
import ch.ethz.ssh2.Connection;
import ch.ethz.ssh2.Session;
import ch.ethz.ssh2.StreamGobbler;

public class RemoteShellExecutor {
     
     private Connection conn;
     /** 遠程機器IP */
     private String ip;
     /** 用戶名 */
     private String osUsername;
     /** 密碼 */
     private String password;
     private String charset = Charset.defaultCharset().toString();

     private static final int TIME_OUT = 1000 * 5 * 60;

     /**
      * 構造函數
      * @param ip
      * @param usr
      * @param pasword
      */
     public RemoteShellExecutor(String ip, String usr, String pasword) {
          this.ip = ip;
         this.osUsername = usr;
         this.password = pasword;
     }


     /**
     * 登錄
     * @return
     * @throws IOException
     */
     private boolean login() throws IOException {
         conn = new Connection(ip);
         conn.connect();
         return conn.authenticateWithPassword(osUsername, password);
     }

     /**
     * 執行腳本
     * 
     * @param cmds
     * @return
     * @throws Exception
     */
     public int exec(String cmds) throws Exception {
         InputStream stdOut = null;
         InputStream stdErr = null;
         String outStr = "";
         String outErr = "";
         int ret = -1;
         try {
         if (login()) {
             // Open a new {@link Session} on this connection
             Session session = conn.openSession();
             // Execute a command on the remote machine.
             session.execCommand(cmds);
             
             stdOut = new StreamGobbler(session.getStdout());
             outStr = processStream(stdOut, charset);
             
             stdErr = new StreamGobbler(session.getStderr());
             outErr = processStream(stdErr, charset);
             
             session.waitForCondition(ChannelCondition.EXIT_STATUS, TIME_OUT);
             
             System.out.println("outStr=" + outStr);
             System.out.println("outErr=" + outErr);
             
             ret = session.getExitStatus();
         } else {
             throw new Exception("登錄遠程機器失敗" + ip); // 自定義異常類 實現略
         }
         } finally {
             if (conn != null) {
                 conn.close();
             }
             IOUtils.closeQuietly(stdOut);
             IOUtils.closeQuietly(stdErr);
         }
         return ret;
     }

     /**
     * @param in
     * @param charset
     * @return
     * @throws IOException
     * @throws UnsupportedEncodingException
     */
     private String processStream(InputStream in, String charset) throws Exception {
         byte[] buf = new byte[1024];
         StringBuilder sb = new StringBuilder();
         while (in.read(buf) != -1) {
             sb.append(new String(buf, charset));
         }
         return sb.toString();
     }

    public static void main(String args[]) throws Exception {
        RemoteShellExecutor executor = new RemoteShellExecutor("192.168.234.123", "root", "beebank");
        // 執行myTest.sh 參數為java Know dummy
        System.out.println(executor.exec("/home/IFileGenTool /load_data.sh t_users myDataBase01"));
    }
}

 

4、Java程序調用遠程Shell

private void backupAndRestoreData(String originalTableName) throws Exception {
    
    //1. 調用遠程Shell腳本，對生產庫中數據進行導出和導入備份。                
    LogUtil.getLogger().info("######    1. 開始對數據庫中數據利用進行導出和導入備份   ######");
    RemoteShellExecutor executor = new RemoteShellExecutor("192.168.234.123", "root", "root");
    // 執行myTest.sh 參數為java Know dummy
    System.out.println(executor.exec("/home/IFileGenTool/load_data.sh"));
}

 

5、運行結果

備份數據成功。

 

6、說明：
0 // getExitStatus方法的返回值

注：一般情況下shell腳本正常執行完畢，getExitStatus方法返回0。
此方法通過遠程命令取得Exit Code/status。但並不是每個server設計時都會返回這個值，如果沒有則會返回null。
在調用getExitStatus時，要先調用WaitForCondition方法，通過ChannelCondition.java接口的定義可以看到每個條件的具體含義。見以下代碼：
ChannelCondition.java的源代碼

參考連接：http://yu.you163.blog.163.com/blog/static/339877742012117101039968/