fix no config panic bug (#332)

hr3lxphr6j · Jan 13, 2023 · 45f1f2e · 45f1f2e
1 parent 18c21d4
commit 45f1f2e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 13 deletions.
diff --git a/src/cmd/bililive/internal/flag/flag.go b/src/cmd/bililive/internal/flag/flag.go
@@ -32,20 +32,20 @@ func init() {
 
 // GenConfigFromFlags generates configuration by parsing command line parameters.
 func GenConfigFromFlags() *configs.Config {
-	cfg := &configs.Config{
-		RPC: configs.RPC{
-			Enable: *RPC,
-			Bind:   *RPCBind,
-		},
-		Debug:      *Debug,
-		Interval:   *Interval,
-		OutPutPath: *Output,
-		OutputTmpl: *OutputFileTmpl,
-		LiveRooms:  configs.NewLiveRoomsWithStrings(*Input),
-		Feature: configs.Feature{
-			UseNativeFlvParser: *NativeFlvParser,
-		},
+	cfg := configs.NewConfig()
+	cfg.RPC = configs.RPC{
+		Enable: *RPC,
+		Bind:   *RPCBind,
 	}
+	cfg.Debug = *Debug
+	cfg.Interval = *Interval
+	cfg.OutPutPath = *Output
+	cfg.OutputTmpl = *OutputFileTmpl
+	cfg.LiveRooms = configs.NewLiveRoomsWithStrings(*Input)
+	cfg.Feature = configs.Feature{
+		UseNativeFlvParser: *NativeFlvParser,
+	}
+
 	if SplitStrategies != nil && len(*SplitStrategies) > 0 {
 		for _, s := range *SplitStrategies {
 			// TODO: not hard code

diff --git a/src/configs/config.go b/src/configs/config.go
@@ -129,6 +129,12 @@ var defaultConfig = Config{
 	TimeoutInUs: 60000000,
 }
 
+func NewConfig() *Config {
+	config := new(Config)
+	config.liveRoomIndexCache = map[string]int{}
+	return config
+}
+
 // Verify will return an error when this config has problem.
 func (c *Config) Verify() error {
 	if c == nil {