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包的其他功能和用法不再引申,有興趣的童鞋歡迎在評論區交流。