当前位置：首页 > article >正文

使用Java通过SSH协议在两个远程服务器之间传输文件

article 2024/11/16 11:48:05

使用Java通过SSH协议在两个远程服务器之间传输文件是一项常见的任务，特别是在需要自动化文件备份、同步或迁移的情况下。JSch库为Java开发者提供了一个方便的方式来实现这一功能。以下是从Codekru网站获取的信息摘要，并结合相关内容，展示如何使用JSch库实现从一个远程服务器向另一个远程服务器传输文件。

准备工作

首先，确保您的项目中已经包含了JSch库的依赖。如果您使用Maven作为构建工具，可以在pom.xml文件中添加如下依赖：

<dependency>
    <groupId>com.jcraft</groupId>
    <artifactId>jsch</artifactId>
    <version>0.1.55</version>
</dependency>

连接到远程服务器

在传输文件之前，需要先分别建立与源服务器和目标服务器的SSH连接。以下是一个简单的连接示例：

import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpException;

public class FileTransferBetweenServers {

    public static void main(String[] args) {
        try {
            JSch jsch = new JSch();
            
            // 源服务器的连接信息
            String sourceHost = "127.0.0.1"; // 源服务器地址
            String sourceUser = "codekru1"; // 源服务器用户名
            String sourcePassword = "codekru1"; // 源服务器密码
            int sourcePort = 22; // SSH端口，默认为22
            
            // 目标服务器的连接信息
            String targetHost = "127.0.0.2"; // 目标服务器地址
            String targetUser = "codekru2"; // 目标服务器用户名
            String targetPassword = "codekru2"; // 目标服务器密码
            int targetPort = 22; // SSH端口，默认为22
            
            // 建立与源服务器的连接
            Session sourceSession = connect(sourceUser, sourcePassword, sourceHost, sourcePort, null);
            
            // 建立与目标服务器的连接
            Session targetSession = connect(targetUser, targetPassword, targetHost, targetPort, null);
            
            // 打开SFTP通道
            ChannelSftp sourceChannel = (ChannelSftp) sourceSession.openChannel("sftp");
            sourceChannel.connect();
            
            ChannelSftp targetChannel = (ChannelSftp) targetSession.openChannel("sftp");
            targetChannel.connect();
            
            // 文件路径
            String sourceFilePath = "/usr/workspace/codekru.txt"; // 源文件路径
            String targetFilePath = "/usr/workspace/codekru.txt"; // 目标文件路径
            
            // 开始传输文件
            transferFile(sourceChannel, targetChannel, sourceFilePath, targetFilePath);
            
            // 关闭SFTP通道
            sourceChannel.disconnect();
            targetChannel.disconnect();
            
            // 断开SSH连接
            sourceSession.disconnect();
            targetSession.disconnect();
            
            System.out.println("File transfer successful!");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    public static Session connect(String username, String password, String host, int port, String FILE_KEY_NAME) {
        Session session = null;
        try {
            JSch jsch = new JSch();
            // 如果有密钥，则使用密钥连接；否则，使用密码连接
            if (FILE_KEY_NAME != null && !FILE_KEY_NAME.equals("")) {
                jsch.addIdentity(FILE_KEY_NAME);
                session = jsch.getSession(username, host, port);
            } else {
                session = jsch.getSession(username, host, port);
                session.setConfig("PreferredAuthentications", "publickey,keyboard-interactive,password");
                session.setPassword(password);
            }
            session.setConfig("StrictHostKeyChecking", "no");
            session.setTimeout(15000);
            session.connect();
            System.out.println("Connected to: " + host);
        } catch (Exception e) {
            e.printStackTrace();
            if (session != null) {
                session.disconnect();
            }
        }
        return session;
    }
    
    public static void transferFile(ChannelSftp sourceChannel, ChannelSftp targetChannel, String sourceFilePath, String targetFilePath) throws Exception {
        // 获取源文件的输入流
        try (InputStream inputStream = sourceChannel.get(sourceFilePath)) {
            // 获取目标文件的输出流
            try (OutputStream outputStream = targetChannel.put(targetFilePath)) {
                byte[] buffer = new byte[1024];
                int bytesRead;
                while ((bytesRead = inputStream.read(buffer)) != -1) {
                    outputStream.write(buffer, 0, bytesRead);
                }
            }
        }
    }
}