|
@ -3,6 +3,7 @@ package org.dromara.sample.wayline.service.impl; |
|
|
import cn.hutool.core.convert.Convert; |
|
|
import cn.hutool.core.convert.Convert; |
|
|
import cn.hutool.core.util.ObjectUtil; |
|
|
import cn.hutool.core.util.ObjectUtil; |
|
|
import cn.hutool.core.util.StrUtil; |
|
|
import cn.hutool.core.util.StrUtil; |
|
|
|
|
|
import com.alibaba.fastjson.JSON; |
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
|
@ -15,8 +16,13 @@ import org.dromara.common.core.utils.StringUtils; |
|
|
import org.dromara.common.mybatis.core.page.PageQuery; |
|
|
import org.dromara.common.mybatis.core.page.PageQuery; |
|
|
import org.dromara.common.mybatis.core.page.TableDataInfo; |
|
|
import org.dromara.common.mybatis.core.page.TableDataInfo; |
|
|
import org.dromara.common.satoken.utils.LoginHelper; |
|
|
import org.dromara.common.satoken.utils.LoginHelper; |
|
|
|
|
|
import org.dromara.common.sdk.mqtt.CommonTopicRequest; |
|
|
import org.dromara.common.sdk.mqtt.MqttGatewayPublish; |
|
|
import org.dromara.common.sdk.mqtt.MqttGatewayPublish; |
|
|
import org.dromara.common.sdk.mqtt.TopicConst; |
|
|
import org.dromara.common.sdk.mqtt.TopicConst; |
|
|
|
|
|
import org.dromara.sample.media.mapper.IFileMapper; |
|
|
|
|
|
import org.dromara.sample.media.model.MediaFileEntity; |
|
|
|
|
|
import org.dromara.sample.media.service.IFileService; |
|
|
|
|
|
import org.dromara.sample.media.service.IMediaService; |
|
|
import org.dromara.sample.wayline.mapper.IAiCompareMapper; |
|
|
import org.dromara.sample.wayline.mapper.IAiCompareMapper; |
|
|
import org.dromara.sample.wayline.mapper.IWaylineJobMapper; |
|
|
import org.dromara.sample.wayline.mapper.IWaylineJobMapper; |
|
|
import org.dromara.sample.wayline.model.dto.AiCompareDTO; |
|
|
import org.dromara.sample.wayline.model.dto.AiCompareDTO; |
|
@ -48,6 +54,7 @@ public class AiCompareServiceImpl implements IAiCompareService { |
|
|
private final IAiCompareQueueService aiCompareQueueService; |
|
|
private final IAiCompareQueueService aiCompareQueueService; |
|
|
private final IAiCompareMapper aiCompareMapper; |
|
|
private final IAiCompareMapper aiCompareMapper; |
|
|
private final IWaylineJobMapper waylineJobMapper; |
|
|
private final IWaylineJobMapper waylineJobMapper; |
|
|
|
|
|
private final IFileService fileService; |
|
|
private final MqttGatewayPublish mqttGatewayPublish; |
|
|
private final MqttGatewayPublish mqttGatewayPublish; |
|
|
@Override |
|
|
@Override |
|
|
public TableDataInfo<AiCompareEntity> queryPageList(AiCompareDTO bo, PageQuery pageQuery) { |
|
|
public TableDataInfo<AiCompareEntity> queryPageList(AiCompareDTO bo, PageQuery pageQuery) { |
|
@ -113,18 +120,23 @@ public class AiCompareServiceImpl implements IAiCompareService { |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
String topic = TopicConst.IMAGE +TopicConst.COMPARE; |
|
|
String topic = TopicConst.IMAGE +TopicConst.COMPARE; |
|
|
|
|
|
List<AiCompareMediaFileDTO> mediaFileDTOS = listMediaFile(aiCompareEntity.getJobId()); |
|
|
// mqttGatewayPublish.publish(topic,byId,1);
|
|
|
mqttGatewayPublish.publish(topic, new CommonTopicRequest().setData(mediaFileDTOS),1); |
|
|
insertList.add(aiCompareEntity); |
|
|
insertList.add(aiCompareEntity); |
|
|
} |
|
|
} |
|
|
List<BatchResult> resultList = aiCompareMapper.insertOrUpdate(insertList, 10); |
|
|
List<BatchResult> resultList = aiCompareMapper.insertOrUpdate(insertList, 10); |
|
|
System.out.println(resultList); |
|
|
|
|
|
List<Long> list =compareDTOList.stream().map(e->Convert.toLong(e.get("queueId"))).collect(Collectors.toList()); |
|
|
List<Long> list =compareDTOList.stream().map(e->Convert.toLong(e.get("queueId"))).collect(Collectors.toList()); |
|
|
|
|
|
|
|
|
return aiCompareQueueService.updateStatus(list,AiCompareStatusConstants.QUEUW_TYPE_1); |
|
|
return aiCompareQueueService.updateStatus(list,AiCompareStatusConstants.QUEUW_TYPE_1); |
|
|
} |
|
|
} |
|
|
public List<Map<String,Object>> listMediaFile(AiCompareEntity aiCompareEntity){ |
|
|
public List<AiCompareMediaFileDTO> listMediaFile(String jobId){ |
|
|
return null; |
|
|
List<AiCompareMediaFileDTO> mediaFileDTOList = aiCompareMapper.listMediaFile(jobId); |
|
|
|
|
|
if(!mediaFileDTOList.isEmpty()){ |
|
|
|
|
|
mediaFileDTOList.stream().forEach(e->{ |
|
|
|
|
|
e.setFilePath(fileService.getObjectUrl(e.getWorkspaceId(),e.getFileId()).toString()); |
|
|
|
|
|
e.setMateFilePath(fileService.getObjectUrl(e.getMateWorkspaceId(),e.getMateFileId()).toString()); |
|
|
|
|
|
}); |
|
|
|
|
|
} |
|
|
|
|
|
return mediaFileDTOList; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -137,8 +149,12 @@ public class AiCompareServiceImpl implements IAiCompareService { |
|
|
if(StrUtil.isNotEmpty(templateId)){ |
|
|
if(StrUtil.isNotEmpty(templateId)){ |
|
|
aiCompareEntity.setTemplateId(templateId); |
|
|
aiCompareEntity.setTemplateId(templateId); |
|
|
} |
|
|
} |
|
|
|
|
|
int i = aiCompareMapper.updateById(aiCompareEntity); |
|
|
return aiCompareMapper.updateById(aiCompareEntity)>0; |
|
|
AiCompareEntity compareEntity = aiCompareMapper.selectById(id); |
|
|
|
|
|
String topic = TopicConst.IMAGE +TopicConst.COMPARE; |
|
|
|
|
|
List<AiCompareMediaFileDTO> mediaFileDTOS = listMediaFile(compareEntity.getJobId()); |
|
|
|
|
|
mqttGatewayPublish.publish(topic, new CommonTopicRequest().setData(mediaFileDTOS),1); |
|
|
|
|
|
return i>0; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
@Override |
|
|
@Override |
|
@ -149,7 +165,7 @@ public class AiCompareServiceImpl implements IAiCompareService { |
|
|
@Override |
|
|
@Override |
|
|
public List<AiCompareMediaFileDTO> queryById(Long id) { |
|
|
public List<AiCompareMediaFileDTO> queryById(Long id) { |
|
|
AiCompareEntity aiCompareEntity = aiCompareMapper.selectById(id); |
|
|
AiCompareEntity aiCompareEntity = aiCompareMapper.selectById(id); |
|
|
List<AiCompareMediaFileDTO> mediaFileDTOList = aiCompareMapper.listMediaFile(aiCompareEntity); |
|
|
List<AiCompareMediaFileDTO> mediaFileDTOList = aiCompareMapper.listMediaFile(aiCompareEntity.getJobId()); |
|
|
return mediaFileDTOList; |
|
|
return mediaFileDTOList; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
@ -158,4 +174,6 @@ public class AiCompareServiceImpl implements IAiCompareService { |
|
|
|
|
|
|
|
|
return true; |
|
|
return true; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|