From 029feca7eeb4f6989108c4f813e435460e6b3d90 Mon Sep 17 00:00:00 2001
From: qydysky <qydysky@foxmail.com>
Date: Mon, 9 May 2022 22:20:49 +0800
Subject: [PATCH] fix

---
 Reply/stream.go | 6 ++++--
 demo/go.mod     | 2 +-
 demo/go.sum     | 2 ++
 go.mod          | 2 +-
 go.sum          | 2 ++
 5 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/Reply/stream.go b/Reply/stream.go
index d5b3f0a..e038410 100644
--- a/Reply/stream.go
+++ b/Reply/stream.go
@@ -368,6 +368,9 @@ func (t *M4SStream) saveStream() {
 
 			// 存在待下载切片
 			if len(download_seq) != 0 {
+				// 设置限制计划
+				download_limit.Plan(int64(len(download_seq)))
+
 				// 下载切片
 				for _, v := range download_seq {
 					// 已下载但还未移除的切片
@@ -421,8 +424,7 @@ func (t *M4SStream) saveStream() {
 				}
 
 				// 等待队列下载完成
-				download_limit.None()
-				download_limit.UnNone()
+				download_limit.PlanDone()
 			}
 
 			// 传递已下载切片
diff --git a/demo/go.mod b/demo/go.mod
index 5638e44..0be9cf9 100644
--- a/demo/go.mod
+++ b/demo/go.mod
@@ -4,7 +4,7 @@ go 1.14
 
 require (
 	github.com/qydysky/bili_danmu v0.5.9
-	github.com/qydysky/part v0.7.1 // indirect
+	github.com/qydysky/part v0.8.0 // indirect
 	github.com/stretchr/testify v1.7.1 // indirect
 )
 
diff --git a/demo/go.sum b/demo/go.sum
index eebf966..20f6b3b 100644
--- a/demo/go.sum
+++ b/demo/go.sum
@@ -42,6 +42,8 @@ github.com/qydysky/part v0.7.0 h1:n03o9jSEewzzydwGZLbwSjHPO8k5JUJqevo4ScuyEoY=
 github.com/qydysky/part v0.7.0/go.mod h1:43opuciW71sZvOR67kye50jgMDSDrn/t6+LefNdlXPg=
 github.com/qydysky/part v0.7.1 h1:ddbtNCE594hbcQYlnen3xBpJXRA835IZghZpL7LKZjA=
 github.com/qydysky/part v0.7.1/go.mod h1:43opuciW71sZvOR67kye50jgMDSDrn/t6+LefNdlXPg=
+github.com/qydysky/part v0.8.0 h1:D+vXHW65XS2GOSg9lwFAXRWbS3ymjHKaZW+OpRisenY=
+github.com/qydysky/part v0.8.0/go.mod h1:43opuciW71sZvOR67kye50jgMDSDrn/t6+LefNdlXPg=
 github.com/shirou/gopsutil v3.20.12+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
 github.com/shirou/gopsutil v3.21.5+incompatible h1:OloQyEerMi7JUrXiNzy8wQ5XN+baemxSl12QgIzt0jc=
 github.com/shirou/gopsutil v3.21.5+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
diff --git a/go.mod b/go.mod
index 1223f0b..b8da272 100644
--- a/go.mod
+++ b/go.mod
@@ -12,7 +12,7 @@ require (
 	github.com/mdp/qrterminal/v3 v3.0.0
 	github.com/miekg/dns v1.1.42 // indirect
 	github.com/mitchellh/mapstructure v1.4.1 // indirect
-	github.com/qydysky/part v0.7.1
+	github.com/qydysky/part v0.8.0
 	github.com/shirou/gopsutil v3.21.5+incompatible // indirect
 	github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e
 	github.com/skratchdot/open-golang v0.0.0-20200116055534-eef842397966
diff --git a/go.sum b/go.sum
index b24cfb5..c336bf1 100644
--- a/go.sum
+++ b/go.sum
@@ -96,6 +96,8 @@ github.com/qydysky/part v0.7.0 h1:n03o9jSEewzzydwGZLbwSjHPO8k5JUJqevo4ScuyEoY=
 github.com/qydysky/part v0.7.0/go.mod h1:43opuciW71sZvOR67kye50jgMDSDrn/t6+LefNdlXPg=
 github.com/qydysky/part v0.7.1 h1:ddbtNCE594hbcQYlnen3xBpJXRA835IZghZpL7LKZjA=
 github.com/qydysky/part v0.7.1/go.mod h1:43opuciW71sZvOR67kye50jgMDSDrn/t6+LefNdlXPg=
+github.com/qydysky/part v0.8.0 h1:D+vXHW65XS2GOSg9lwFAXRWbS3ymjHKaZW+OpRisenY=
+github.com/qydysky/part v0.8.0/go.mod h1:43opuciW71sZvOR67kye50jgMDSDrn/t6+LefNdlXPg=
 github.com/qydysky/part/msgq v0.0.0-20201213120821-f36e49c32bba h1:1ew9dRpc0Rux0WkWeT/4AE15ynYWmL2D7onJEJIFOB8=
 github.com/qydysky/part/msgq v0.0.0-20201213120821-f36e49c32bba/go.mod h1:w32TkJNVtTJd4LOS09cq+4uYG6itcN2vsqw+slp44Rg=
 github.com/shirou/gopsutil v2.20.7+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
-- 
2.39.2