Java下載遠端伺服器檔案到本地(http協議和ssh2協議)

勝金發表於2021-01-15

  Java中java.io包為我們提供了輸入流和輸出流,對檔案的讀寫基本上都依賴於這些封裝好的關於流的類中來實現。前段時間遇到了以下兩種需求:

  1、與某系統對接,每天獲取最新的圖片並顯示在前端頁面。該系統提供的是一個http協議的圖片URL,本來獲取到該系統的圖片地址以後在HTML中顯示就可以了,但是該系統不太穩定,圖片URL經常不能使用,而且每天生成圖片不一定成功,

對自己系統的功能影響很大,emm。。。所以就考慮每天從該系統下載最新的圖片到本地更新儲存,沒有新圖片就繼續使用上次的圖片。

  2、公司演算法團隊的同事完成了一個視訊分析的檢測功能,會生成一些擷取的短視訊檔案,系統需要獲取並儲存這些視訊檔案。演算法執行在Linux系統,沒有搭建FTP伺服器,所以就需要系統從執行演算法的Linux系統複製檔案到系統本地儲存起來。

  這兩個需求實現起來都是大同小異,思路就是讀取檔案然後寫到指定路徑,只不過http協議的圖片地址可以直接讀取然後儲存,而Linux系統的檔案需要遠端連線到該伺服器然後再下載檔案,這就需要我們引入以下依賴:

<!--遠端讀取伺服器檔案-->
        <dependency>
            <groupId>ch.ethz.ganymed</groupId>
            <artifactId>ganymed-ssh2</artifactId>
            <version>262</version>
        </dependency>
        <dependency>
            <groupId>com.jcraft</groupId>
            <artifactId>jsch</artifactId>
            <version>0.1.53</version>
        </dependency>
        <dependency>
            <groupId>commons-net</groupId>
            <artifactId>commons-net</artifactId>
            <version>3.4</version>
        </dependency>

  廢話不說了,先上程式碼:

package com.XXX.utils;

import ch.ethz.ssh2.*;
import com.databus.Log;
import com.mysql.jdbc.StringUtils;

import java.io.*;
import java.net.*;

public class FileUtils {

    /*
    * 根據http路徑下載檔案儲存到指定路徑
    * urlString:檔案路徑
    * fileName:儲存到本地的檔名稱
    * filePath:本地要儲存的指定路徑
    * */
    public static boolean downloadFile(String urlString,String fileName,String filePath) {
        boolean bool = false;

        InputStream is = null;
        FileOutputStream os = null;
        try {
            Log.info("檔案路徑:" + urlString);
            // 構造URL
            java.net.URL url = new java.net.URL(urlString);
            // 開啟連線
            URLConnection con = url.openConnection();
            // 輸入流
            is = con.getInputStream();
            // 1K的資料緩衝
            byte[] bs = new byte[1024];
            // 讀取到的資料長度
            int len;
            //判斷指定目錄是否存在,不存在則先建立目錄
            File file = new File(filePath);
            if (!file.exists())
                file.mkdirs();
            //fileName如果不包含檔案字尾,則需要加上字尾,如:fileName + ".jpg";fileName + ".txt";
            os = new FileOutputStream(filePath + fileName, false);//false:覆蓋檔案,true:在原有檔案後追加
            // 開始讀取
            while ((len = is.read(bs)) != -1) {
                os.write(bs, 0, len);
            }

            bool = true;
            Log.info("檔案儲存成功:" + fileName);
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            // 完畢,關閉所有連結
            if (null != os){
                try {
                    os.flush();
                    os.close();
                }catch (IOException e){
                    e.printStackTrace();
                }
            }
            if (null != is){
                try {
                    is.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return bool;
    }

    //遠端下載伺服器檔案
    public static boolean copyFile(String ip,int port,String userName,String password,String sourceFile,String targetFile,String targetFileName){
        boolean bool = false;
        Connection conn = null;
        Session session = null;
        try {
            if (StringUtils.isNullOrEmpty(ip) || StringUtils.isNullOrEmpty(userName) || StringUtils.isNullOrEmpty(password) ||
                    StringUtils.isNullOrEmpty(sourceFile) || StringUtils.isNullOrEmpty(targetFile)){
                return bool;
            }
            conn = new Connection(ip,port);
            conn.connect();
            boolean isAuth = conn.authenticateWithPassword(userName,password);
            if (!isAuth){
                Log.info("演算法主機連線失敗");
                return bool;
            }
            //執行命令
            session = conn.openSession();

            //執行命令並列印執行結果
            session.execCommand("df -h");
            InputStream staout = new StreamGobbler(session.getStdout());
            BufferedReader br = new BufferedReader(new InputStreamReader(staout));
            String line = null;
            while ((line = br.readLine()) != null){
                System.out.println(line);
            }
            br.close();

            //下載檔案到本地
            SCPClient scpClient = conn.createSCPClient();
            SCPInputStream scpis = scpClient.get(sourceFile);

            //判斷指定目錄是否存在,不存在則先建立目錄
            File file = new File(targetFile);
            if (!file.exists())
                file.mkdirs();

            FileOutputStream fos = new FileOutputStream(targetFile + targetFileName);
            byte[] buffer = new byte[1024];
            int len = 0;
            while ((len = scpis.read(buffer)) != -1){
                fos.write(buffer,0,len);
            }
            fos.close();
            bool = true;
            //SFTP
            /*SFTPv3Client sftPClient = new SFTPv3Client(conn);
            sftPClient.createFile("");
            sftPClient.close();*/
        }catch (Exception e){
            e.printStackTrace();
            Log.info(e.getMessage());
            Log.info("儲存失敗:" + sourceFile);
        }finally {
            if (null != session){
                session.close();
            }
            if (null != conn) {
                conn.close();
            }
        }

        return bool;
    }


}

  第一個方法downloadFile(String urlString,String fileName,String filePath)與我們讀寫檔案沒什麼區別,我們主要說一下遠端讀取Linux伺服器檔案的方法:

  copyFile(String ip,int port,String userName,String password,String sourceFile,String targetFile,String targetFileName)

  可以看到,我們需要Linux伺服器的ip、ssh開放的埠號(一般預設是22)、伺服器使用者名稱和密碼,所以我們要確保Linux伺服器已經開放ssh連線,否則我們的程式根本就連不上伺服器,更不要說複製檔案了。

  其實ssh2連線遠端伺服器,就和我們用Xshell連線伺服器是一樣的,不但可以複製檔案,也可以執行Linux命令對Linux進行操作,看上面的一段程式碼:

//執行命令
            session = conn.openSession();

            //執行命令並列印執行結果
            session.execCommand("df -h");
            InputStream staout = new StreamGobbler(session.getStdout());
            BufferedReader br = new BufferedReader(new InputStreamReader(staout));
            String line = null;
            while ((line = br.readLine()) != null){
                System.out.println(line);
            }
            br.close();

  這段程式碼與複製檔案沒有關係,之所以保留就是要說明一下,我們執行了df -h d 命令(查詢伺服器磁碟使用情況),將會得到磁碟的具體使用情況,與下圖效果相同。所以我們可以用ssh2做很多事情,有興趣的童鞋可以多瞭解。

 

   另外在copyFile()方法的最後有一段註釋的程式碼:

//SFTP
            /*SFTPv3Client sftPClient = new SFTPv3Client(conn);
            sftPClient.createFile("files");
            sftPClient.close();*/

  我們也可以通過建立SFTP連線來遠端操作目錄和檔案,比如:建立、刪除目錄,建立、刪除檔案等。

  關於ssh2包的其他功能和用法不再引申,有興趣的童鞋歡迎在評論區交流。

 

相關文章