golang sftp上传文件

Go语言（Golang）是一种现代的、高性能的编程语言，它具有卓越的并发性能和简单易用的语法。在Go语言中，我们可以使用sftp协议来实现文件上传功能。sftp（Secure File Transfer Protocol）是一种安全的文件传输协议，它基于SSH协议进行数据传输，可以保证数据的机密性和完整性。

创建SFTP连接

在Go语言中，我们可以使用第三方库"github.com/pkg/sftp"来实现sftp上传文件的功能。首先，我们需要通过SSH客户端建立一个与远程服务器的连接，并通过该连接获取sftp客户端。

下面是使用Go语言创建sftp连接的示例代码：

import (
	"fmt"
	"golang.org/x/crypto/ssh"
	"github.com/pkg/sftp"
)

func createSFTPConnection() (*sftp.Client, error) {
    // 连接SSH服务器
    sshConfig := &ssh.ClientConfig{
        User: "username",
        Auth: []ssh.AuthMethod{
            ssh.Password("password"),
        },
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
    }
    
    sshClient, err := ssh.Dial("tcp", "hostname:22", sshConfig)
    if err != nil {
        return nil, fmt.Errorf("failed to connect to SSH server: %v", err)
    }
    
    // 创建sftp客户端
    sftpClient, err := sftp.NewClient(sshClient)
    if err != nil {
        return nil, fmt.Errorf("failed to create SFTP client: %v", err)
    }
    
    return sftpClient, nil
}

上传文件

创建sftp连接后，我们就可以使用sftp客户端的Put方法来上传文件了。Put方法接受两个参数，第一个参数是本地文件的路径，第二个参数是远程服务器上保存文件的路径。

下面是使用Go语言上传文件的示例代码：

func uploadFile(sftpClient *sftp.Client, localFilePath string, remoteFilePath string) error {
    // 打开本地文件
    localFile, err := os.Open(localFilePath)
    if err != nil {
        return fmt.Errorf("failed to open local file: %v", err)
    }
    defer localFile.Close()
    
    // 创建远程文件
    remoteFile, err := sftpClient.Create(remoteFilePath)
    if err != nil {
        return fmt.Errorf("failed to create remote file: %v", err)
    }
    defer remoteFile.Close()
    
    // 将本地文件内容复制到远程文件
    _, err = io.Copy(remoteFile, localFile)
    if err != nil {
        return fmt.Errorf("failed to copy local file to remote file: %v", err)
    }
    
    return nil
}

完整示例代码

下面是一个完整的示例代码，演示了如何使用Go语言实现sftp上传文件的功能：

import (
	"fmt"
	"github.com/pkg/sftp"
	"golang.org/x/crypto/ssh"
)

func main() {
    // 创建sftp连接
    sftpClient, err := createSFTPConnection()
    if err != nil {
        fmt.Printf("failed to create SFTP connection: %v\n", err)
        return
    }
    defer sftpClient.Close()
    
    // 上传文件
    localFilePath := "/path/to/local/file.txt"
    remoteFilePath := "/path/to/remote/file.txt"
    err = uploadFile(sftpClient, localFilePath, remoteFilePath)
    if err != nil {
        fmt.Printf("failed to upload file: %v\n", err)
        return
    }
    
    fmt.Println("file uploaded successfully")
}

func createSFTPConnection() (*sftp.Client, error) {
    // 连接SSH服务器
    sshConfig := &ssh.ClientConfig{
        User: "username",
        Auth: []ssh.AuthMethod{
            ssh.Password("password"),
        },
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
    }
    
    sshClient, err := ssh.Dial("tcp", "hostname:22", sshConfig)
    if err != nil {
        return nil, fmt.Errorf("failed to connect to SSH server: %v", err)
    }
    
    // 创建sftp客户端
    sftpClient, err := sftp.NewClient(sshClient)
    if err != nil {
        return nil, fmt.Errorf("failed to create SFTP client: %v", err)
    }
    
    return sftpClient, nil
}

func uploadFile(sftpClient *sftp.Client, localFilePath string, remoteFilePath string) error {
    // 打开本地文件
    localFile, err := os.Open(localFilePath)
    if err != nil {
        return fmt.Errorf("failed to open local file: %v", err)
    }
    defer localFile.Close()
    
    // 创建远程文件
    remoteFile, err := sftpClient.Create(remoteFilePath)
    if err != nil {
        return fmt.Errorf("failed to create remote file: %v", err)
    }
    defer remoteFile.Close()
    
    // 将本地文件内容复制到远程文件
    _, err = io.Copy(remoteFile, localFile)
    if err != nil {
        return fmt.Errorf("failed to copy local file to remote file: %v", err)
    }
    
    return nil
}

通过以上代码，我们可以实现在Go语言中使用sftp协议上传文件的功能。使用sftp协议可以保证数据的安全传输，同时Go语言的高性能和简单易用的语法也使得文件上传变得更加便捷。