✨ config.json supports wss

FloatTech · Dec 11, 2022 · 8f009bb · 8f009bb
1 parent 34a3cf8
commit 8f009bb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -189,6 +189,7 @@ import (
 type zbpcfg struct {
 	Z zero.Config        `json:"zero"`
 	W []*driver.WSClient `json:"ws"`
+	S []*driver.WSServer `json:"wss"`
 }
 
 var config zbpcfg
@@ -250,10 +251,13 @@ func init() {
 		if err != nil {
 			panic(err)
 		}
-		config.Z.Driver = make([]zero.Driver, len(config.W))
+		config.Z.Driver = make([]zero.Driver, len(config.W)+len(config.S))
 		for i, w := range config.W {
 			config.Z.Driver[i] = w
 		}
+		for i, s := range config.S {
+			config.Z.Driver[i+len(config.W)] = s
+		}
 		logrus.Infoln("[main] 从", *runcfg, "读取配置文件")
 		return
 	}