前言
有一个奇怪的需求,就是将本地的内容直接提交到远程服务器,并且需要针对文件夹进行处理。所以,这里就直接采用JSch
来实现。在这里,感谢秀发浓密的程序猿的这篇博客,给了很大启发。
思路
既然是直接读取文件夹,我们就不考虑浏览器不能访问本地文件的问题了。直接就是Java
处理,然后直接读取文件提交服务器。
当然,介于上传过程中是Windows
上传到Linux
,如果只传图片倒还没什么问题,传文件可就造了老罪了。还好我现在只有图片需求。
但也并不能掉以轻心,因为,就算是图片,名字也是中文的。
所以直接用UUID
避免了上传乱码,等上传结束了再mv
到目标文件夹中。
上代码
private static final String TMPDIR = "/data/ftp";
public static void sshSftpUpload(String ip, String user, String psw, int port, String localDirFileName, String destDir, String fileName) throws Exception {// 这些需要后期关闭的资源需要单独拉出来// 而由于OutputStream与InputStream在初始化时需要单独Override其中的write方法// 所以没办法直接用`try-catch-resource`语法糖给直接new出来// 所以单独声明,就干脆全放一起了Channel channel = null, channelShell = null;Session session = null;ChannelSftp sftp = null;String tmpFileName = "";OutputStream outStream = null;InputStream inStream = null;try {JSch jsch = new JSch();// 如果端口非法,使用默认端口if (port <= 0)session = jsch.getSession(user, ip);elsesession = jsch.getSession(user, ip, port);// 如果服务器连接不上,则抛出异常// 在真正返回值的地方再处理异常,单凭返回值没办法细致处理问题if (session == null)throw new Exception("session is null");// 设置主机令牌session.setPassword(psw);//设置第一次登陆的时候提示,可选值:(ask | yes | no)session.setConfig("StrictHostKeyChecking", "no");//设置登陆超时时间(30s)session.connect(30 * 1000);//创建sftp通信通道channel = session.openChannel("sftp");channel.connect(1000);// 强制转换,否则Channel类无法使用cd方法sftp = (ChannelSftp) channel;// 进入临时目录文件夹sftp.cd(TMPDIR);// 以下代码实现从本地上传一个文件到服务器// 上传时中文可能乱码,采用UUID规避tmpFileName = UUID.randomUUID().toString().replace("-", "").toLowerCase() + "." + fileName.substring(fileName.lastIndexOf('.') + 1);outStream = sftp.put(tmpFileName);// Files的构造函数是私有的,经典的单例模式,在项目中表现更佳inStream = Files.newInputStream(new File(localDirFileName).toPath());// 一如既往的上传操作byte[] b = new byte[1024];int n;while ((n = inStream.read(b)) != -1)outStream.write(b, 0, n);// 需要注意的是,你完全可以在这里调用flush,就像这样// outStream.flush();// 但flush往往会造成系统级IO操作// 所以个人不愿意在这里flush,而是在finally中flush// 传输完成 将文件转移到正式目录 - 注:exec不起作用channelShell = session.openChannel("shell");// 需要放置回车符模拟Shell的回车操作channelShell.setInputStream(new ByteArrayInputStream(("cp " + TMPDIR + "/" + tmpFileName + " " + destDir + "/" + fileName + " \n").getBytes(StandardCharsets.UTF_8)));// 将输出重定向到控制台channelShell.setOutputStream(System.out);channelShell.connect(5 * 1000);} catch (Exception e) {throw new Exception(e.getMessage());} finally {// 结束后关闭资源,有的是close,有的是flush,有的是disconnectif (outStream != null) {outStream.flush();outStream.close();}if (inStream != null) inStream.close();if (sftp != null) {// 删除临时文件sftp.rm(tmpFileName);sftp.disconnect();}if (session != null) session.disconnect();if (channel != null) channel.disconnect();if (channelShell != null) channelShell.disconnect();}
}
虽然finally
很长,但是起码交付没啥问题。
主打一个面向测试用例编程了。