Golang实现FTP客户端功能 - 使用goftp库

在Golang中实现FTP客户端功能，可以使用第三方库goftp，它实现了RFC 959定义的FTP协议规范。下面我将详细介绍如何使用这个库来实现常见的FTP操作。

安装goftp库

首先需要安装goftp库：

go get github.com/goftp/ftp

基本FTP操作示例

下面是一个完整的FTP客户端示例，包含连接、登录、上传、下载、列出目录等常见操作：

package main

import (
	"fmt"
	"io"
	"log"
	"os"
	"time"

	"github.com/goftp/ftp"
)

func main() {
	// FTP服务器配置
	config := &ftp.Config{
		User:               "username",      // FTP用户名
		Passwd:             "password",      // FTP密码
		Timeout:            10 * time.Second, // 超时时间
		DisableEPSV:        true,            // 禁用EPSV模式
		RemoteServerTime:   false,           // 不获取服务器时间
		ConnectionsPerHost: 10,              // 每个主机最大连接数
	}

	// 创建FTP客户端
	client, err := ftp.New(config)
	if err != nil {
		log.Fatal("创建FTP客户端失败:", err)
	}
	defer client.Close()

	// 连接到FTP服务器
	err = client.Connect("ftp.example.com:21")
	if err != nil {
		log.Fatal("连接FTP服务器失败:", err)
	}
	fmt.Println("成功连接到FTP服务器")

	// 登录FTP服务器
	err = client.Login()
	if err != nil {
		log.Fatal("登录FTP服务器失败:", err)
	}
	fmt.Println("成功登录FTP服务器")

	// 列出当前目录文件
	files, err := client.List(".")
	if err != nil {
		log.Fatal("列出目录失败:", err)
	}
	fmt.Println("当前目录内容:")
	for _, file := range files {
		fmt.Printf("%s\t%d\t%s\n", file.Name(), file.Size(), file.ModTime().Format("2006-01-02 15:04:05"))
	}

	// 上传文件
	err = uploadFile(client, "localfile.txt", "remotefile.txt")
	if err != nil {
		log.Fatal("上传文件失败:", err)
	}
	fmt.Println("文件上传成功")

	// 下载文件
	err = downloadFile(client, "remotefile.txt", "downloaded.txt")
	if err != nil {
		log.Fatal("下载文件失败:", err)
	}
	fmt.Println("文件下载成功")

	// 创建目录
	err = client.MakeDir("new_directory")
	if err != nil {
		log.Fatal("创建目录失败:", err)
	}
	fmt.Println("目录创建成功")

	// 删除文件
	err = client.Delete("remotefile.txt")
	if err != nil {
		log.Fatal("删除文件失败:", err)
	}
	fmt.Println("文件删除成功")
}

// 上传文件到FTP服务器
func uploadFile(client *ftp.Client, localPath, remotePath string) error {
	// 打开本地文件
	file, err := os.Open(localPath)
	if err != nil {
		return err
	}
	defer file.Close()

	// 创建FTP服务器上的文件
	err = client.Stor(remotePath, file)
	if err != nil {
		return err
	}

	return nil
}

// 从FTP服务器下载文件
func downloadFile(client *ftp.Client, remotePath, localPath string) error {
	// 创建本地文件
	file, err := os.Create(localPath)
	if err != nil {
		return err
	}
	defer file.Close()

	// 从FTP服务器获取文件
	resp, err := client.Retr(remotePath)
	if err != nil {
		return err
	}
	defer resp.Close()

	// 将文件内容写入本地文件
	_, err = io.Copy(file, resp)
	if err != nil {
		return err
	}

	return nil
}

高级功能实现

1. 断点续传

// 断点续传下载
func resumeDownload(client *ftp.Client, remotePath, localPath string) error {
	// 获取本地文件大小
	localFile, err := os.OpenFile(localPath, os.O_WRONLY|os.O_CREATE, 0666)
	if err != nil {
		return err
	}
	defer localFile.Close()

	fi, err := localFile.Stat()
	if err != nil {
		return err
	}
	offset := fi.Size()

	// 设置偏移量
	err = client.Rest(offset)
	if err != nil {
		return err
	}

	// 继续下载
	resp, err := client.Retr(remotePath)
	if err != nil {
		return err
	}
	defer resp.Close()

	// 定位到文件末尾
	_, err = localFile.Seek(0, io.SeekEnd)
	if err != nil {
		return err
	}

	_, err = io.Copy(localFile, resp)
	return err
}

2. 递归下载目录

// 递归下载目录
func downloadDir(client *ftp.Client, remoteDir, localDir string) error {
	// 创建本地目录
	err := os.MkdirAll(localDir, 0755)
	if err != nil {
		return err
	}

	// 列出远程目录内容
	entries, err := client.List(remoteDir)
	if err != nil {
		return err
	}

	for _, entry := range entries {
		remotePath := remoteDir + "/" + entry.Name()
		localPath := localDir + "/" + entry.Name()

		if entry.IsDir() {
			// 如果是目录，递归下载
			err = downloadDir(client, remotePath, localPath)
			if err != nil {
				return err
			}
		} else {
			// 如果是文件，直接下载
			err = downloadFile(client, remotePath, localPath)
			if err != nil {
				return err
			}
		}
	}

	return nil
}

3. 文件校验

// 比较本地文件和远程文件大小
func compareFileSize(client *ftp.Client, remotePath, localPath string) (bool, error) {
	// 获取远程文件大小
	remoteSize, err := client.Size(remotePath)
	if err != nil {
		return false, err
	}

	// 获取本地文件大小
	localFile, err := os.Stat(localPath)
	if err != nil {
		return false, err
	}

	return remoteSize == localFile.Size(), nil
}

注意事项

1. 安全性：FTP协议本身是不加密的，考虑使用FTPS或SFTP来传输敏感数据
2. 错误处理：所有FTP操作都可能失败，需要妥善处理错误
3. 资源释放：确保及时关闭文件和FTP连接
4. 并发控制：FTP服务器通常有并发连接限制
5. 超时设置：网络操作应设置合理的超时时间

替代方案

如果goftp不满足需求，还可以考虑以下替代库：

1. github.com/jlaffaye/ftp - 另一个流行的FTP客户端库
2. github.com/pkg/sftp - 用于SFTP协议
3. github.com/secsy/goftp - 支持TLS的FTP客户端

以上代码提供了基本的FTP操作实现，可以根据实际需求进行扩展和优化。