fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
laodaming 2023-08-21 11:25:56 +08:00
parent 0990644b4a
commit feae3ada33
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/websocket/internal/logic/ws_render_image_logic.go
View File

@ -34,9 +34,9 @@ func (w *wsConnectItem) sendToRenderChan(data []byte) {
select { select {
case <-w.closeChan: //已经关闭 case <-w.closeChan: //已经关闭
return return
case w.renderProperty.renderChan <- data: case w.renderProperty.renderChan <- data: //发入到缓冲池
return return
case <-time.After(time.Second * 3): case <-time.After(time.Second * 3): //三秒没进入缓冲池就丢弃
return return
} }
} }
@ -296,15 +296,17 @@ func (w *wsConnectItem) assembleRenderData(taskId string, info websocket_data.Re
func (w *wsConnectItem) operationRenderTask() { func (w *wsConnectItem) operationRenderTask() {
for { for {
select { select {
case <-w.closeChan: case <-w.closeChan: //整个链接关闭
return return
case data := <-w.renderProperty.renderImageTaskCtlChan: case data := <-w.renderProperty.renderImageTaskCtlChan: //处理任务
switch data.Option { switch data.Option {
case 0: //删除任务 case 0: //删除任务
delete(w.renderProperty.renderImageTask, data.TaskId) delete(w.renderProperty.renderImageTask, data.TaskId)
case 1: //新增任务 case 1: //新增任务
w.renderProperty.renderImageTask[data.TaskId] = data.RenderId w.renderProperty.renderImageTask[data.TaskId] = data.RenderId
} }
case <-time.After(time.Second * 3): //三秒阻塞则丢弃
return
} }
} }
} }