diff --git a/server/websocket/internal/logic/ws_allocation_processing_factory.go b/server/websocket/internal/logic/ws_allocation_processing_factory.go index 9ba718cc..7fedddc1 100644 --- a/server/websocket/internal/logic/ws_allocation_processing_factory.go +++ b/server/websocket/internal/logic/ws_allocation_processing_factory.go @@ -10,20 +10,14 @@ type allocationProcessorFactory interface { allocationMessage(w *wsConnectItem, data []byte) } -var mapAllocationProcessor = make(map[constants.Websocket]allocationProcessorFactory) +// 要添加分发在这里 +var mapAllocationProcessor = map[constants.Websocket]allocationProcessorFactory{ + constants.WEBSOCKET_RENDER_IMAGE: &renderProcessor{}, //渲染工厂 +} func (w *wsConnectItem) newAllocationProcessor(msgType constants.Websocket) allocationProcessorFactory { if val, ok := mapAllocationProcessor[msgType]; ok { return val } - var obj allocationProcessorFactory - switch msgType { - //图片渲染 - case constants.WEBSOCKET_RENDER_IMAGE: - obj = &renderProcessor{} - default: - return nil - } - mapAllocationProcessor[msgType] = obj - return obj + return nil }