feat: 命令行工具

download/base/constants.go

@@ -12,6 +12,16 @@ const (
 	DownloadStatusDone
 )
 
+type EventKey string
+
+const (
+	EventKeyStart    = "start"
+	EventKeyPause    = "pause"
+	EventKeyProgress = "progress"
+	EventKeyError    = "error"
+	EventKeyDone     = "done"
+)
+
 const (
 	HttpCodeOK             = 200
 	HttpCodePartialContent = 206

download/base/controller.go

@@ -142,3 +142,15 @@ func (d *DialerWarp) Dial(network, addr string) (c net.Conn, err error) {
 	}
 	return &ConnWarp{conn: conn}, nil
 }
+
+// 获取任务中各个文件的已下载字节数
+type Progress []int64
+
+// TotalDownloaded 获取任务总下载字节数
+func (p Progress) TotalDownloaded() int64 {
+	total := int64(0)
+	for _, d := range p {
+		total += d
+	}
+	return total
+}

download/base/fetcher.go

@@ -11,6 +11,8 @@ type Fetcher interface {
 	Pause() (err error)
 	Continue() (err error)
 
+	// 获取任务下载进度
+	Progress() Progress
 	// 该方法会一直阻塞，直到任务下载结束
 	Wait() (err error)
 }

download/base/timer.go

@@ -0,0 +1,25 @@
+package base
+
+import "time"
+
+// 计时器
+type Timer struct {
+	t    int64
+	used int64
+}
+
+func (t *Timer) Start() {
+	t.t = time.Now().UnixNano()
+}
+
+func (t *Timer) Pause() {
+	t.used += time.Now().UnixNano() - t.t
+}
+
+func (t *Timer) Continue() {
+	t.t = time.Now().UnixNano()
+}
+
+func (t *Timer) Used() int64 {
+	return (time.Now().UnixNano() - t.t) + t.used
+}