diff --git a/network/websocketserver.go b/network/websocketserver.go index 4c07ad2..2ce0c80 100644 --- a/network/websocketserver.go +++ b/network/websocketserver.go @@ -83,7 +83,7 @@ func (slf *WebsocketServer) Init(pattern string, port uint16, messageReciver IMe slf.mapClient = make(map[uint64]*WSClient) slf.messageReciver = messageReciver - http.HandleFunc("/ws", slf.wsHandler) + http.HandleFunc(slf.pattern, slf.wsHandler) } func (slf *WebsocketServer) startListen() { diff --git a/service/Service.go b/service/Service.go index ad58061..920f8e9 100644 --- a/service/Service.go +++ b/service/Service.go @@ -35,6 +35,7 @@ type IService interface { OnFetchService(iservice IService) error GetServiceType() int GetServiceName() string + SetServiceName(serviceName string) bool GetServiceId() int IsTimeOutTick(microSecond int64) bool @@ -94,6 +95,11 @@ func (slf *BaseService) GetServiceName() string { return slf.servicename } +func (slf *BaseService) SetServiceName(serviceName string) bool { + slf.servicename = serviceName + return true +} + func (slf *BaseService) GetStatus() int { return slf.Status }