check CHANGELOG.md

CHANGELOG.md

@@ -3,4 +3,5 @@
 
 更新記錄：
 
-- 修復直播資訊沒有更新的問題
+- 新增 flag (`--port` 和 `--release`)
+- Production Mode 时将不再 `heartbeat reply` 洗屏

README.md

@@ -158,10 +158,11 @@ wss://blive.chu77.xyz/ws?id=abc
 运行参数(非必要)
 
 ```bash
-./biligo-live-ws 端口
+./biligo-live-ws --port 端口 --release
 ```
 
-端口: 不填则 8080
+- `port`: 不填则 8080
+- `release`: 添加此参数即等同设置环境参数中 `GIN_MODE` 为 `release` (即 `production mode`)
 
 ## 鸣谢
 

main.go

@@ -1,17 +1,29 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"github.com/eric2788/biligo-live-ws/controller/subscribe"
 	ws "github.com/eric2788/biligo-live-ws/controller/websocket"
+	"github.com/eric2788/biligo-live-ws/services/blive"
 	"github.com/gin-gonic/gin"
 	"log"
 	"os"
 )
 
+var release = flag.Bool("release", os.Getenv("GIN_MODE") == "release", "set release mode")
+var port = flag.Int("port", 8080, "set the websocket port")
+
 func main() {
 
-	log.Printf("biligo-live-ws version %v", "0.1.6")
+	flag.Parse()
+
+	log.Printf("biligo-live-ws version %v", "0.1.7")
+
+	if *release {
+		gin.SetMode(gin.ReleaseMode)
+		blive.Debug = false
+	}
 
 	router := gin.Default()
 
@@ -24,11 +36,9 @@ func main() {
 	subscribe.Register(router.Group("subscribe"))
 	ws.Register(router.Group("ws"))
 
-	port := ":8080"
+	port := fmt.Sprintf(":%d", *port)
 
-	if len(os.Args) > 1 {
-		port = fmt.Sprintf(":%v", os.Args[1])
-	}
+	log.Printf("使用端口 %s\n", port)
 
 	if err := router.Run(port); err != nil {
 		log.Fatal(err)

services/blive/live_server.go

@@ -14,6 +14,8 @@ import (
 var listening = set.NewSet()
 var excepted = set.NewSet()
 
+var Debug = true
+
 func LaunchLiveServer(room int64, handle func(data *LiveInfo, msg biligo.Msg)) (context.CancelFunc, error) {
 
 	liveInfo, err := GetLiveInfo(room, false) // 獲取直播資訊
@@ -22,7 +24,7 @@ func LaunchLiveServer(room int64, handle func(data *LiveInfo, msg biligo.Msg)) (
 		return nil, err
 	}
 
-	live := biligo.NewLive(true, 30*time.Second, 0, func(err error) {
+	live := biligo.NewLive(Debug, 30*time.Second, 0, func(err error) {
 		log.Fatal(err)
 	})