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-11-03 16:13:53 +08:00
parent 8759180068
commit 43e51b076f
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
server/websocket/internal/logic/rendernotifylogic.go
View File

@ -62,6 +62,10 @@ func (l *RenderNotifyLogic) RenderNotify(req *types.RenderNotifyReq, userinfo *a
unityRenderBeginTime := info.RenderBeginTime unityRenderBeginTime := info.RenderBeginTime
//获取连接 //获取连接
value, wsConnectOk := mapConnPool.Load(wid) value, wsConnectOk := mapConnPool.Load(wid)
var ws wsConnectItem
if wsConnectOk {
ws = value.(wsConnectItem)
}
if req.Code == 0 { //渲染成功 if req.Code == 0 { //渲染成功
//上传文件 //上传文件
var upload = file.Upload{ var upload = file.Upload{
@ -82,17 +86,14 @@ func (l *RenderNotifyLogic) RenderNotify(req *types.RenderNotifyReq, userinfo *a
}) })
if err != nil { if err != nil {
logx.Error("渲染回调上传文件失败：", err) logx.Error("渲染回调上传文件失败：", err)
if wsConnectOk { if ws.conn != nil {
ws := value.(wsConnectItem)
//发送错误信息给前端 //发送错误信息给前端
ws.renderErrResponse(requestId, info.TemplateTag, info.TaskId, "unity图片上传错误", 0, 0, 0, 0, 0, 0, 0) ws.renderErrResponse(requestId, info.TemplateTag, info.TaskId, "unity图片上传错误", 0, 0, 0, 0, 0, 0, 0)
} }
return resp.SetStatusWithMessage(basic.CodeFileUploadErr, "failed to upload render resource image") return resp.SetStatusWithMessage(basic.CodeFileUploadErr, "failed to upload render resource image")
} }
uploadUnityRenderImageTakesTime := time.Now().UTC().UnixMilli() - unityRenderEndTime uploadUnityRenderImageTakesTime := time.Now().UTC().UnixMilli() - unityRenderEndTime
if wsConnectOk { if ws.conn != nil {
//断言连接
ws := value.(wsConnectItem)
//发送到出口 //发送到出口
ws.sendRenderResultData(websocket_data.RenderImageRspMsg{ ws.sendRenderResultData(websocket_data.RenderImageRspMsg{
RequestId: requestId, RequestId: requestId,
@ -109,9 +110,7 @@ func (l *RenderNotifyLogic) RenderNotify(req *types.RenderNotifyReq, userinfo *a
return resp.SetStatusWithMessage(basic.CodeOK, "success：but websocket connect not found") return resp.SetStatusWithMessage(basic.CodeOK, "success：but websocket connect not found")
} }
//渲染失败走下面 //渲染失败走下面
if wsConnectOk { if ws.conn != nil {
//断言连接
ws := value.(wsConnectItem)
//发送错误信息给前端 //发送错误信息给前端
ws.renderErrResponse(requestId, info.TemplateTag, info.TaskId, "unity云渲染错误:"+req.Msg, 0, 0, 0, 0, 0, 0, 0) ws.renderErrResponse(requestId, info.TemplateTag, info.TaskId, "unity云渲染错误:"+req.Msg, 0, 0, 0, 0, 0, 0, 0)
logx.Info("渲染失败且发送了失败信息") logx.Info("渲染失败且发送了失败信息")