Golang BitTorrent客户端与rain库的使用

rain是一个用Go语言编写的轻量级BitTorrent客户端库，它提供了完整的BitTorrent协议实现，可以用于构建自定义的BitTorrent客户端或集成到其他应用中。

rain库简介

rain库的主要特点包括：

• 纯Go实现，跨平台支持
• 支持DHT网络
• 支持磁力链接
• 支持加密连接
• 轻量级且易于集成

安装rain库

go get -u github.com/cenkalti/rain

基本使用示例

1. 最简单的下载示例

package main

import (
	"log"
	"time"

	"github.com/cenkalti/rain"
)

func main() {
	// 创建会话
	sess, err := rain.NewSession(rain.Config{
		DataDir:    "./downloads",
		Port:       6881,
		DisableDHT: false,
	})
	if err != nil {
		log.Fatal(err)
	}
	defer sess.Close()

	// 添加磁力链接
	torrent, err := sess.AddURI(
		"magnet:?xt=urn:btih:6a9759bffd5c0af65319979fb7832189f4f3c35d",
		nil,
	)
	if err != nil {
		log.Fatal(err)
	}

	// 等待下载完成
	for {
		stats := torrent.Stats()
		log.Printf("Progress: %.2f%%, Downloaded: %d, Uploaded: %d",
			stats.Progress*100, stats.BytesDownloaded, stats.BytesUploaded)

		if stats.Completed {
			log.Println("Download completed!")
			break
		}
		time.Sleep(time.Second)
	}
}

2. 更完整的客户端示例

package main

import (
	"fmt"
	"log"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/cenkalti/rain"
)

func main() {
	// 配置会话
	config := rain.Config{
		DataDir:           "./downloads",
		Port:              6881,
		DisableDHT:        false,
		DisableTCP:        false,
		DisableUTP:        false,
		DisableUpload:     false,
		DisableIPv6:       false,
		EncryptionPolicy:  rain.EncryptionPolicyPreferred,
		DownloadRateLimit: 0, // 0表示不限速
		UploadRateLimit:   0,
	}

	// 创建会话
	sess, err := rain.NewSession(config)
	if err != nil {
		log.Fatal(err)
	}
	defer sess.Close()

	// 处理命令行参数
	if len(os.Args) < 2 {
		log.Fatal("Usage: ./client <magnet-uri or torrent-file>")
	}
	source := os.Args[1]

	// 添加下载任务
	var torrent *rain.Torrent
	if _, err := os.Stat(source); err == nil {
		// 如果是文件
		torrent, err = sess.AddTorrentFile(source, nil)
	} else {
		// 如果是磁力链接
		torrent, err = sess.AddURI(source, nil)
	}
	if err != nil {
		log.Fatal(err)
	}

	// 打印基本信息
	fmt.Printf("Downloading: %s\n", torrent.Name())
	fmt.Printf("InfoHash: %s\n", torrent.InfoHash())

	// 设置信号处理，优雅退出
	sigCh := make(chan os.Signal, 1)
	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)

	// 进度监控循环
	ticker := time.NewTicker(3 * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-ticker.C:
			stats := torrent.Stats()
			fmt.Printf("\rProgress: %.2f%% | Peers: %d | ↓ %s/s | ↑ %s/s | Seeds: %d",
				stats.Progress*100,
				stats.ConnectedPeers,
				formatBytes(stats.SpeedDownload),
				formatBytes(stats.SpeedUpload),
				stats.AvailablePeers,
			)

			if stats.Completed {
				fmt.Println("\nDownload completed!")
				return
			}

		case <-sigCh:
			fmt.Println("\nShutting down...")
			return
		}
	}
}

// 格式化字节大小为易读格式
func formatBytes(bytes int64) string {
	const unit = 1024
	if bytes < unit {
		return fmt.Sprintf("%d B", bytes)
	}
	div, exp := int64(unit), 0
	for n := bytes / unit; n >= unit; n /= unit {
		div *= unit
		exp++
	}
	return fmt.Sprintf("%.1f %ciB", float64(bytes)/float64(div), "KMGTPE"[exp])
}

高级功能

1. 事件监听

// 添加事件监听器
sess.OnTorrentAdded = func(t *rain.Torrent) {
	log.Printf("Torrent added: %s", t.Name())
}

sess.OnTorrentRemoved = func(t *rain.Torrent) {
	log.Printf("Torrent removed: %s", t.Name())
}

sess.OnTorrentComplete = func(t *rain.Torrent) {
	log.Printf("Torrent completed: %s", t.Name())
}

2. 选择性下载文件

// 获取torrent文件列表
torrent, err := sess.AddURI(magnetURI, nil)
if err != nil {
    log.Fatal(err)
}

// 等待torrent信息加载完成
for torrent.Stats().Status != rain.StatusDownloadingMetadata {
    time.Sleep(time.Second)
}

// 获取文件列表并选择要下载的文件
files := torrent.Files()
for i, file := range files {
    fmt.Printf("%d: %s (%.2f MB)\n", i, file.Path, float64(file.Length)/1024/1024)
}

// 选择要下载的文件索引
var selected []int
// ... 根据用户输入设置selected

// 设置文件优先级
for i := range files {
    if contains(selected, i) {
        torrent.SetFilePriority(i, rain.PriorityNormal)
    } else {
        torrent.SetFilePriority(i, rain.PriorityNone)
    }
}

3. 保存和恢复会话

// 保存会话状态
err = sess.SaveSession()
if err != nil {
    log.Printf("Error saving session: %v", err)
}

// 在下次启动时自动恢复之前的下载任务
// 只需要在创建会话时指定相同的DataDir

注意事项

1. 端口设置：BitTorrent客户端通常使用6881-6889端口，某些网络可能会阻止这些端口。

2. DHT网络：DHT可以帮助在没有Tracker的情况下找到peer，但在某些网络环境下可能需要禁用。

3. 速率限制：合理设置上传/下载速率限制，避免占用过多带宽。

4. 磁盘空间：确保有足够的磁盘空间存放下载文件。

5. 版权问题：确保下载的内容不侵犯版权。

rain库提供了丰富的配置选项和API，可以根据需要构建功能完善的BitTorrent客户端。以上示例展示了基本用法，更多高级功能可以参考官方文档和源代码。