36 lines
714 B
Go
36 lines
714 B
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"robot/config"
|
|
"robot/internal/ws"
|
|
)
|
|
|
|
// ModuleWebsocket Websocket客户端模块
|
|
type ModuleWebsocket struct {
|
|
manager *ws.Manager
|
|
}
|
|
|
|
func (p *ModuleWebsocket) init() error {
|
|
cfg := config.Get().Client
|
|
httpAddr := fmt.Sprintf("%s:%d", cfg.Http.Address, cfg.Http.Port)
|
|
if cfg.Http.Port == 0 {
|
|
httpAddr = cfg.Http.Address
|
|
}
|
|
websocketAddr := fmt.Sprintf("%s:%d", cfg.Websocket.Address, cfg.Websocket.Port)
|
|
if cfg.Websocket.Port == 0 {
|
|
websocketAddr = cfg.Websocket.Address
|
|
}
|
|
p.manager = ws.NewManager(httpAddr, websocketAddr)
|
|
return nil
|
|
}
|
|
|
|
func (p *ModuleWebsocket) start() error {
|
|
p.manager.Start()
|
|
return nil
|
|
}
|
|
|
|
func (p *ModuleWebsocket) stop() error {
|
|
return nil
|
|
}
|