From df23f054644067ba7c44614185f54fa2314fb195 Mon Sep 17 00:00:00 2001
From: qydysky <qydysky@foxmail.com>
Date: Mon, 5 Oct 2020 10:05:27 +0800
Subject: [PATCH] =?utf8?q?=E6=B7=BB=E5=8A=A0=E5=87=BD=E6=95=B0=E8=B0=83?=
 =?utf8?q?=E7=94=A8=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 bili_danmu.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/bili_danmu.go b/bili_danmu.go
index 46f43bb..9b2cd70 100644
--- a/bili_danmu.go
+++ b/bili_danmu.go
@@ -15,7 +15,7 @@ import (
 
 var danmulog = p.Logf().New().Open("danmu.log").Base(-1, "bili_danmu.go").Level(c.LogLevel)
 
-func Demo() {
+func Demo(roomid ...int) {
 
 	danmulog.Base(-1, "测试")
 	defer danmulog.Base(0)
@@ -30,6 +30,10 @@ func Demo() {
 		flag.Parse()
 	
 		var room = *groomid
+		if room == 0 && len(roomid) != 0 {
+			room = roomid[0]
+		}
+
 		fmt.Printf("输入房间号: ")
 		if room == 0 {
 			_, err := fmt.Scanln(&room)
@@ -113,6 +117,7 @@ func Demo() {
 			p.Sys().Timeoutf(1)
 		}
 
+		close(interrupt)
 		{//附加功能 直播流
 			reply.Saveflv_wait()
 		}
-- 
2.39.2