This commit is contained in:
laodaming
2023-08-09 15:29:27 +08:00
parent 2d0d410c40
commit fcf7c909a3
3 changed files with 28 additions and 16 deletions

View File

@@ -48,6 +48,27 @@ func (m *MqConsumerRenderAssemble) Run(ctx context.Context, data []byte) error {
if !ok {
return errors.New("allmodels is nil!!")
}
rabbitmq := initalize.RabbitMqHandle{}
//查询有没有缓存的资源
resource, err := allmodels.FsResource.FindOneById(ctx, parseInfo.TaskId)
if err != nil {
if !errors.Is(err, gorm.ErrRecordNotFound) {
logx.Error("failed to find render resource:", err)
return err
}
} else {
//有数据则直接返回
cacheData := websocket_data.RenderImageNotify{
TaskId: parseInfo.TaskId,
Image: *resource.ResourceUrl,
}
d, _ := json.Marshal(cacheData)
if err = rabbitmq.SendMsg(constants.RABBIT_MQ_RENDER_RESULT_DATA, d); err != nil {
logx.Error("failed to send cache render resource to queue:RABBIT_MQ_RENDER_RESULT_DATA")
return err
}
return nil
}
timeSearchBegin := time.Now().UnixMilli()
//获取模板
templateInfo, err := allmodels.FsProductTemplateV2.FindOneByProductIdTagIdWithSizeTable(ctx, parseInfo.RenderData.ProductId, fmt.Sprintf("%d", parseInfo.RenderData.TemplateTagId))
@@ -250,7 +271,6 @@ func (m *MqConsumerRenderAssemble) Run(ctx context.Context, data []byte) error {
"folder": "", //todo 千人千面需要使用
}
b, _ := json.Marshal(sendData)
rabbitmq := initalize.RabbitMqHandle{}
if err = rabbitmq.SendMsg(constants.RABBIT_MQ_TO_UNITY, b); err != nil {
logx.Error("发送渲染组装数据到rabbitmq失败", err)
return err