Java如何實現上傳檔案到伺服器指定目錄
阿新 • • 發佈:2020-04-17
前言需求
使用freemarker生成的靜態檔案,統一儲存在某個伺服器上。本來一開始打算使用ftp實現的,奈何老連線不上,改用jsch。畢竟有現成的就很舒服,在此介紹給大家。
具體實現
引入的pom
<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.55</version> </dependency>
建立實體類
public class ResultEntity { private String code; private String message; private File file; public ResultEntity(){} public ResultEntity(String code,String message,File file) { super(); this.code = code; this.message = message; this.file = file; } public String getCode() { return code; } public void setCode(String code) { this.code = code; } public String getMessage() { return message; } public void setMessage(String message) { this.message = message; } public File getFile() { return file; } public void setFile(File file) { this.file = file; } }
public class ScpConnectEntity { private String userName; private String passWord; private String url; private String targetPath; public String getTargetPath() { return targetPath; } public void setTargetPath(String targetPath) { this.targetPath = targetPath; } public String getUserName() { return userName; } public void setUserName(String userName) { this.userName = userName; } public String getPassWord() { return passWord; } public void setPassWord(String passWord) { this.passWord = passWord; } public String getUrl() { return url; } public void setUrl(String url) { this.url = url; } }
建立檔案上傳工具類
@Configuration
@Configuration public class FileUploadUtil { @Value("${remoteServer.url}") private String url; @Value("${remoteServer.password}") private String passWord; @Value("${remoteServer.username}") private String userName; @Async public ResultEntity uploadFile(File file,String targetPath,String remoteFileName) throws Exception{ ScpConnectEntity scpConnectEntity=new ScpConnectEntity(); scpConnectEntity.setTargetPath(targetPath); scpConnectEntity.setUrl(url); scpConnectEntity.setPassWord(passWord); scpConnectEntity.setUserName(userName); String code = null; String message = null; try { if (file == null || !file.exists()) { throw new IllegalArgumentException("請確保上傳檔案不為空且存在!"); } if(remoteFileName==null || "".equals(remoteFileName.trim())){ throw new IllegalArgumentException("遠端伺服器新建檔名不能為空!"); } remoteUploadFile(scpConnectEntity,file,remoteFileName); code = "ok"; message = remoteFileName; } catch (IllegalArgumentException e) { code = "Exception"; message = e.getMessage(); } catch (JSchException e) { code = "Exception"; message = e.getMessage(); } catch (IOException e) { code = "Exception"; message = e.getMessage(); } catch (Exception e) { throw e; } catch (Error e) { code = "Error"; message = e.getMessage(); } return new ResultEntity(code,message,null); } private void remoteUploadFile(ScpConnectEntity scpConnectEntity,File file,String remoteFileName) throws JSchException,IOException { Connection connection = null; ch.ethz.ssh2.Session session = null; SCPOutputStream scpo = null; FileInputStream fis = null; try { createDir(scpConnectEntity); }catch (JSchException e) { throw e; } try { connection = new Connection(scpConnectEntity.getUrl()); connection.connect(); if(!connection.authenticateWithPassword(scpConnectEntity.getUserName(),scpConnectEntity.getPassWord())){ throw new RuntimeException("SSH連線伺服器失敗"); } session = connection.openSession(); SCPClient scpClient = connection.createSCPClient(); scpo = scpClient.put(remoteFileName,file.length(),scpConnectEntity.getTargetPath(),"0666"); fis = new FileInputStream(file); byte[] buf = new byte[1024]; int hasMore = fis.read(buf); while(hasMore != -1){ scpo.write(buf); hasMore = fis.read(buf); } } catch (IOException e) { throw new IOException("SSH上傳檔案至伺服器出錯"+e.getMessage()); }finally { if(null != fis){ try { fis.close(); } catch (IOException e) { e.printStackTrace(); } } if(null != scpo){ try { scpo.flush(); // scpo.close(); } catch (IOException e) { e.printStackTrace(); } } if(null != session){ session.close(); } if(null != connection){ connection.close(); } } } private boolean createDir(ScpConnectEntity scpConnectEntity ) throws JSchException { JSch jsch = new JSch(); com.jcraft.jsch.Session sshSession = null; Channel channel= null; try { sshSession = jsch.getSession(scpConnectEntity.getUserName(),scpConnectEntity.getUrl(),22); sshSession.setPassword(scpConnectEntity.getPassWord()); sshSession.setConfig("StrictHostKeyChecking","no"); sshSession.connect(); channel = sshSession.openChannel("sftp"); channel.connect(); } catch (JSchException e) { e.printStackTrace(); throw new JSchException("SFTP連線伺服器失敗"+e.getMessage()); } ChannelSftp channelSftp=(ChannelSftp) channel; if (isDirExist(scpConnectEntity.getTargetPath(),channelSftp)) { channel.disconnect(); channelSftp.disconnect(); sshSession.disconnect(); return true; }else { String pathArry[] = scpConnectEntity.getTargetPath().split("/"); StringBuffer filePath=new StringBuffer("/"); for (String path : pathArry) { if (path.equals("")) { continue; } filePath.append(path + "/"); try { if (isDirExist(filePath.toString(),channelSftp)) { channelSftp.cd(filePath.toString()); } else { // 建立目錄 channelSftp.mkdir(filePath.toString()); // 進入並設定為當前目錄 channelSftp.cd(filePath.toString()); } } catch (SftpException e) { e.printStackTrace(); throw new JSchException("SFTP無法正常操作伺服器"+e.getMessage()); } } } channel.disconnect(); channelSftp.disconnect(); sshSession.disconnect(); return true; } private boolean isDirExist(String directory,ChannelSftp channelSftp) { boolean isDirExistFlag = false; try { SftpATTRS sftpATTRS = channelSftp.lstat(directory); isDirExistFlag = true; return sftpATTRS.isDir(); } catch (Exception e) { if (e.getMessage().toLowerCase().equals("no such file")) { isDirExistFlag = false; } } return isDirExistFlag; } }
屬性我都寫在Spring的配置檔案裡面了。將這個類託管給spring容器。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。