| 提交 | 用户 | 时间 | ||
| e7c126 | 1 | package com.iailab.module.infra.framework.file.core.client.ftp; |
| H | 2 | |
| 3 | import cn.hutool.core.io.FileUtil; | |
| 4 | import cn.hutool.core.util.CharsetUtil; | |
| 5 | import cn.hutool.core.util.StrUtil; | |
| 6 | import cn.hutool.extra.ftp.Ftp; | |
| 7 | import cn.hutool.extra.ftp.FtpException; | |
| 8 | import cn.hutool.extra.ftp.FtpMode; | |
| 9 | import com.iailab.module.infra.framework.file.core.client.AbstractFileClient; | |
| 10 | ||
| 11 | import java.io.ByteArrayInputStream; | |
| 12 | import java.io.ByteArrayOutputStream; | |
| 13 | ||
| 14 | /** | |
| 15 | * Ftp 文件客户端 | |
| 16 | * | |
| 17 | * @author iailab | |
| 18 | */ | |
| 19 | public class FtpFileClient extends AbstractFileClient<FtpFileClientConfig> { | |
| 20 | ||
| 21 | private Ftp ftp; | |
| 22 | ||
| 23 | public FtpFileClient(Long id, FtpFileClientConfig config) { | |
| 24 | super(id, config); | |
| 25 | } | |
| 26 | ||
| 27 | @Override | |
| 28 | protected void doInit() { | |
| 29 | // 把配置的 \ 替换成 /, 如果路径配置 \a\test, 替换成 /a/test, 替换方法已经处理 null 情况 | |
| 30 | config.setBasePath(StrUtil.replace(config.getBasePath(), StrUtil.BACKSLASH, StrUtil.SLASH)); | |
| 31 | // ftp的路径是 / 结尾 | |
| 32 | if (!config.getBasePath().endsWith(StrUtil.SLASH)) { | |
| 33 | config.setBasePath(config.getBasePath() + StrUtil.SLASH); | |
| 34 | } | |
| 35 | // 初始化 Ftp 对象 | |
| 36 | this.ftp = new Ftp(config.getHost(), config.getPort(), config.getUsername(), config.getPassword(), | |
| 37 | CharsetUtil.CHARSET_UTF_8, null, null, FtpMode.valueOf(config.getMode())); | |
| 38 | } | |
| 39 | ||
| 40 | @Override | |
| 41 | public String upload(byte[] content, String path, String type) { | |
| 42 | // 执行写入 | |
| 43 | String filePath = getFilePath(path); | |
| 44 | String fileName = FileUtil.getName(filePath); | |
| 45 | String dir = StrUtil.removeSuffix(filePath, fileName); | |
| 46 | ftp.reconnectIfTimeout(); | |
| 47 | boolean success = ftp.upload(dir, fileName, new ByteArrayInputStream(content)); | |
| 48 | if (!success) { | |
| 49 | throw new FtpException(StrUtil.format("上传文件到目标目录 ({}) 失败", filePath)); | |
| 50 | } | |
| 51 | // 拼接返回路径 | |
| 52 | return super.formatFileUrl(config.getDomain(), path); | |
| 53 | } | |
| 54 | ||
| 55 | @Override | |
| 56 | public void delete(String path) { | |
| 57 | String filePath = getFilePath(path); | |
| 58 | ftp.reconnectIfTimeout(); | |
| 59 | ftp.delFile(filePath); | |
| 60 | } | |
| 61 | ||
| 62 | @Override | |
| 63 | public byte[] getContent(String path) { | |
| 64 | String filePath = getFilePath(path); | |
| 65 | String fileName = FileUtil.getName(filePath); | |
| 66 | String dir = StrUtil.removeSuffix(filePath, fileName); | |
| 67 | ByteArrayOutputStream out = new ByteArrayOutputStream(); | |
| 68 | ftp.reconnectIfTimeout(); | |
| 69 | ftp.download(dir, fileName, out); | |
| 70 | return out.toByteArray(); | |
| 71 | } | |
| 72 | ||
| 73 | private String getFilePath(String path) { | |
| 74 | return config.getBasePath() + path; | |
| 75 | } | |
| 76 | ||
| 77 | } | |
