package com.iplatform.recvideo.util;
|
|
import com.iplatform.model.po.Rc_video_t1;
|
import com.iplatform.recvideo.Constants;
|
import com.walker.infrastructure.utils.JsonUtils;
|
import com.walker.infrastructure.utils.NumberGenerator;
|
import com.walker.infrastructure.utils.StringUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.web.client.RestTemplate;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
public class PythonInvokeUtils {
|
|
private static final transient Logger logger = LoggerFactory.getLogger(PythonInvokeUtils.class);
|
|
public static List<Rc_video_t1> acquireImageSearchResult(String videoId, String imgPath
|
, String topN, String url, RestTemplate restTemplate) throws Exception{
|
SearchRequest request = new SearchRequest();
|
request.setImg_path(imgPath);
|
request.setTop_n(topN);
|
|
ResponseEntity<String> entity = restTemplate.postForEntity(url, request, String.class);
|
if(entity != null){
|
String jsonData = entity.getBody();
|
if(StringUtils.isEmpty(jsonData)){
|
return null;
|
}
|
|
List<Rc_video_t1> data = new ArrayList<>();
|
|
try {
|
List<Map<String, Object>> list = JsonUtils.jsonStringToObject(jsonData, List.class);
|
for(Map<String, Object> obj : list){
|
data.add(transferToVideoT1(obj, videoId, imgPath));
|
}
|
logger.debug("++++++ 请求返回 Rc_video_t1: " + data.size());
|
return data;
|
|
} catch (Exception e) {
|
logger.error("解析json结果错误:" + jsonData, e);
|
throw e;
|
}
|
}
|
return null;
|
}
|
|
/**
|
* 从文件路径中，截取文件ID，文件用id命名。
|
* @param videoPath
|
* @return
|
*/
|
public static final String getFileNameWithoutSuffix(String videoPath, String suffix){
|
String[] array = videoPath.split("/");
|
if(array == null || array.length == 0){
|
logger.error("视频名称截取id错误:" + videoPath);
|
return null;
|
}
|
String fileName = array[array.length-1];
|
// String[] idValue = fileName.split(".");
|
return fileName.replaceAll(suffix, StringUtils.EMPTY_STRING);
|
}
|
|
private static Rc_video_t1 transferToVideoT1(Map<String, Object> map, String videoId, String imgPath){
|
String videoPath = map.get("video").toString();
|
String distance = map.get("dis").toString();
|
String srcImageId = getFileNameWithoutSuffix(imgPath, Constants.IMAGE_SUFFIX);
|
|
Rc_video_t1 textBlock = new Rc_video_t1();
|
textBlock.setSim_video_id(getFileNameWithoutSuffix(videoPath, Constants.VIDEO_SUFFIX));
|
textBlock.setDistance(Double.parseDouble(distance));
|
textBlock.setSrc_video_id(videoId);
|
textBlock.setSrc_img(srcImageId);
|
textBlock.setId(NumberGenerator.getLongSequenceId());
|
return textBlock;
|
}
|
}
|
