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 {
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
}