fusenapi/server/websocket/internal/logic/allocation_processing_factory.go
laodaming b8dbda6a4c fix
2023-08-30 17:46:47 +08:00

34 lines
771 B
Go

package logic
import (
"fusenapi/constants"
)
// 消息分发工厂
type allocationProcessorFactory interface {
allocationMessage(data []byte)
}
var mapAllocationProcessor = make(map[constants.Websocket]allocationProcessorFactory)
func (w *wsConnectItem) newAllocationProcessor(msgType constants.Websocket) allocationProcessorFactory {
if obj, ok := mapAllocationProcessor[msgType]; ok {
return obj
}
var obj allocationProcessorFactory
switch msgType {
//图片渲染
case constants.WEBSOCKET_RENDER_IMAGE:
obj = &renderProcessor{w}
//刷新重连请求恢复上次连接的标识
case constants.WEBSOCKET_REQUEST_REUSE_LAST_CONNECT:
obj = &reuseConnProcessor{w}
default:
}
if obj != nil {
mapAllocationProcessor[msgType] = obj
}
return obj
}