使用Go语言实现BitTorrent客户端功能的torrent库

Go语言有一个非常流行的BitTorrent客户端库叫做torrent，由anacrolix开发。下面我将介绍如何使用这个库实现基本的BitTorrent客户端功能。

安装torrent库

首先需要安装这个库：

go get github.com/anacrolix/torrent

基本使用示例

1. 简单的BT客户端实现

package main

import (
	"context"
	"fmt"
	"log"
	"os"
	"time"

	"github.com/anacrolix/torrent"
	"github.com/anacrolix/torrent/metainfo"
)

func main() {
	// 创建客户端配置
	cfg := torrent.NewDefaultClientConfig()
	cfg.ListenPort = 0 // 随机端口
	cfg.DataDir = "./downloads" // 下载目录

	// 创建客户端
	client, err := torrent.NewClient(cfg)
	if err != nil {
		log.Fatalf("Error creating client: %v", err)
	}
	defer client.Close()

	// 添加种子文件或磁力链接
	var t *torrent.Torrent
	if len(os.Args) < 2 {
		log.Fatal("请提供种子文件路径或磁力链接")
	}

	arg := os.Args[1]
	if _, err := os.Stat(arg); err == nil {
		// 如果是文件路径，解析种子文件
		metaInfo, err := metainfo.LoadFromFile(arg)
		if err != nil {
			log.Fatal(err)
		}
		t, err = client.AddTorrent(metaInfo)
		if err != nil {
			log.Fatal(err)
		}
	} else {
		// 否则当作磁力链接处理
		t, err = client.AddMagnet(arg)
		if err != nil {
			log.Fatal(err)
		}
	}

	// 获取种子信息
	<-t.GotInfo()
	fmt.Println("开始下载:", t.Name())

	// 下载所有文件
	t.DownloadAll()

	// 打印下载进度
	ticker := time.NewTicker(5 * time.Second)
	defer ticker.Stop()

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Minute)
	defer cancel()

	for {
		select {
		case <-ticker.C:
			// 打印下载进度
			stats := t.Stats()
			fmt.Printf("进度: %.2f%%, 下载速度: %s/s, 上传速度: %s/s\n",
				float64(t.BytesCompleted())/float64(t.Info().TotalLength())*100,
				humanizeBytes(stats.BytesReadUsefulData.Int64()),
				humanizeBytes(stats.BytesWrittenData.Int64()))
			
			// 检查是否完成
			if t.BytesCompleted() == t.Info().TotalLength() {
				fmt.Println("下载完成!")
				return
			}

		case <-ctx.Done():
			fmt.Println("下载超时")
			return
		}
	}
}

// 辅助函数：将字节数转换为易读格式
func humanizeBytes(b int64) string {
	const unit = 1024
	if b < unit {
		return fmt.Sprintf("%d B", b)
	}
	div, exp := int64(unit), 0
	for n := b / unit; n >= unit; n /= unit {
		div *= unit
		exp++
	}
	return fmt.Sprintf("%.1f %ciB", float64(b)/float64(div), "KMGTPE"[exp])
}

2. 高级功能示例

package main

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

	"github.com/anacrolix/torrent"
	"github.com/anacrolix/torrent/metainfo"
)

func main() {
	// 更详细的客户端配置
	cfg := torrent.NewDefaultClientConfig()
	cfg.ListenPort = 50007 // 固定端口
	cfg.DataDir = "./downloads"
	cfg.Debug = true // 开启调试日志
	cfg.Seed = true  // 下载完成后继续做种
	cfg.UploadRateLimiter = rate.NewLimiter(rate.Limit(100*1024), 100*1024) // 限制上传速度100KB/s

	client, err := torrent.NewClient(cfg)
	if err != nil {
		log.Fatal(err)
	}
	defer client.Close()

	// 添加种子
	metaInfo, err := metainfo.LoadFromFile("example.torrent")
	if err != nil {
		log.Fatal(err)
	}

	t, err := client.AddTorrent(metaInfo)
	if err != nil {
		log.Fatal(err)
	}

	<-t.GotInfo()

	// 选择性地下载文件
	for _, file := range t.Files() {
		if file.Path() == "important.txt" {
			file.Download()
			fmt.Println("开始下载:", file.Path())
		}
	}

	// 监控下载状态
	for {
		time.Sleep(5 * time.Second)
		stats := t.Stats()
		fmt.Printf("连接数: %d (活跃: %d)\n", stats.TotalPeers, stats.ActivePeers)
		fmt.Printf("下载: %s, 上传: %s\n",
			humanizeBytes(stats.BytesReadUsefulData.Int64()),
			humanizeBytes(stats.BytesWrittenData.Int64()))
	}
}

关键功能说明

1. 添加种子/磁力链接:

   • AddTorrent() - 通过种子文件添加
   • AddMagnet() - 通过磁力链接添加

2. 文件选择下载:

   • 可以遍历t.Files()选择特定文件下载

3. 下载控制:

   • DownloadAll() - 下载所有文件
   • file.Download() - 下载特定文件
   • t.StopDataDownload() - 停止下载

4. 统计信息:

   • t.Stats() - 获取统计信息
   • t.BytesCompleted() - 已下载字节数
   • t.Info().TotalLength() - 总字节数

5. 做种设置:

   • 通过cfg.Seed = true开启下载完成后自动做种

注意事项

1. 防火墙需要开放监听端口
2. DHT和PEX功能默认开启
3. 可以通过配置限制上传/下载速度
4. 在生产环境中应考虑更完善的错误处理和日志记录

这个库功能强大，支持DHT、PEX、加密协议等高级功能，可以构建完整的BitTorrent客户端。以上示例展示了基本用法，你可以根据需求进一步扩展。