| 提交 | 用户 | 时间 | ||
| e7c126 | 1 | package com.iailab.module.infra.framework.file.core.client.sftp; |
| H | 2 | |
| 3 | import cn.hutool.core.io.FileUtil; | |
| 4 | import cn.hutool.extra.ssh.Sftp; | |
| 5 | import com.iailab.framework.common.util.io.FileUtils; | |
| 6 | import com.iailab.module.infra.framework.file.core.client.AbstractFileClient; | |
| 7 | ||
| 8 | import java.io.File; | |
| 9 | ||
| 10 | /** | |
| 11 | * Sftp 文件客户端 | |
| 12 | * | |
| 13 | * @author iailab | |
| 14 | */ | |
| 15 | public class SftpFileClient extends AbstractFileClient<SftpFileClientConfig> { | |
| 16 | ||
| 17 | private Sftp sftp; | |
| 18 | ||
| 19 | public SftpFileClient(Long id, SftpFileClientConfig config) { | |
| 20 | super(id, config); | |
| 21 | } | |
| 22 | ||
| 23 | @Override | |
| 24 | protected void doInit() { | |
| 25 | // 补全风格。例如说 Linux 是 /，Windows 是 \ | |
| 26 | if (!config.getBasePath().endsWith(File.separator)) { | |
| 27 | config.setBasePath(config.getBasePath() + File.separator); | |
| 28 | } | |
| 29 | // 初始化 Ftp 对象 | |
| 30 | this.sftp = new Sftp(config.getHost(), config.getPort(), config.getUsername(), config.getPassword()); | |
| 31 | } | |
| 32 | ||
| 33 | @Override | |
| 34 | public String upload(byte[] content, String path, String type) { | |
| 35 | // 执行写入 | |
| 36 | String filePath = getFilePath(path); | |
| 37 | File file = FileUtils.createTempFile(content); | |
| 38 | sftp.upload(filePath, file); | |
| 39 | // 拼接返回路径 | |
| 40 | return super.formatFileUrl(config.getDomain(), path); | |
| 41 | } | |
| 42 | ||
| 43 | @Override | |
| 44 | public void delete(String path) { | |
| 45 | String filePath = getFilePath(path); | |
| 46 | sftp.delFile(filePath); | |
| 47 | } | |
| 48 | ||
| 49 | @Override | |
| 50 | public byte[] getContent(String path) { | |
| 51 | String filePath = getFilePath(path); | |
| 52 | File destFile = FileUtils.createTempFile(); | |
| 53 | sftp.download(filePath, destFile); | |
| 54 | return FileUtil.readBytes(destFile); | |
| 55 | } | |
| 56 | ||
| 57 | private String getFilePath(String path) { | |
| 58 | return config.getBasePath() + path; | |
| 59 | } | |
| 60 | ||
| 61 | } | |
