fusenapi/server/websocket/internal/logic/ws_render_image_logic.go
laodaming 36aad6a088 fix
2023-07-26 15:15:37 +08:00

45 lines
1.1 KiB
Go

package logic
import (
"encoding/json"
"fusenapi/server/websocket/internal/types"
"github.com/zeromicro/go-zero/core/logx"
)
// 渲染请求数据处理发送云渲染服务处理
func (w *wsConnectItem) SendToCloudRender(data []byte) {
var renderImageData []types.RenderImageReqMsg
if err := json.Unmarshal(data, &renderImageData); err != nil {
logx.Error("invalid format of websocket render image message", err)
return
}
logx.Info("收到请求云渲染图片数据:", renderImageData)
//把需要渲染的图片任务加进去
for _, v := range renderImageData {
key := w.getRenderImageMapKey(v.ProductId, v.SizeId, v.TemplateId)
w.renderImageTaskCtlChan <- renderImageControlChanItem{
Option: 1,
Key: key,
}
}
}
// 操作连接中渲染任务的增加/删除
func (w *wsConnectItem) operationRenderTask() {
for {
select {
case <-w.closeChan:
return
case data := <-w.renderImageTaskCtlChan:
switch data.Option {
case 0: //删除任务
delete(w.renderImageTask, data.Key)
case 1: //新增任务
w.renderImageTask[data.Key] = struct{}{}
default:
}
}
}
}