config.json supports wss

This commit is contained in:
源文雨 2022-12-11 15:38:01 +08:00
parent 34a3cf82e7
commit 8f009bb4ee

View File

@ -189,6 +189,7 @@ import (
type zbpcfg struct { type zbpcfg struct {
Z zero.Config `json:"zero"` Z zero.Config `json:"zero"`
W []*driver.WSClient `json:"ws"` W []*driver.WSClient `json:"ws"`
S []*driver.WSServer `json:"wss"`
} }
var config zbpcfg var config zbpcfg
@ -250,10 +251,13 @@ func init() {
if err != nil { if err != nil {
panic(err) 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 { for i, w := range config.W {
config.Z.Driver[i] = w config.Z.Driver[i] = w
} }
for i, s := range config.S {
config.Z.Driver[i+len(config.W)] = s
}
logrus.Infoln("[main] 从", *runcfg, "读取配置文件") logrus.Infoln("[main] 从", *runcfg, "读取配置文件")
return return
} }