添加函数调用方式

diff --git a/bili_danmu.go b/bili_danmu.go
index 46f43bb88c9eb8e74dba4423635a555c02756b68..9b2cd7011da12495e544aee8378b6392d6033843 100644 (file)
--- 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()
                }