package com.basic.analy.controller;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.basic.analy.config.Result;
|
import com.basic.analy.entity.FaceResults;
|
import com.basic.analy.service.FaceSdkTool;
|
import com.basic.analy.service.impl.BbPersonBaseServiceImpl;
|
import com.basic.analy.service.impl.SearchServiceImpl;
|
import com.basic.analy.utils.ImageUtils;
|
import com.cloud.common.model.FileInfos;
|
import com.cloud.common.utils.FastDFSUtil;
|
import org.apache.log4j.Logger;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.data.redis.core.ListOperations;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.http.HttpStatus;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
import org.springframework.web.bind.annotation.RestController;
|
import java.text.ParseException;
|
import java.text.SimpleDateFormat;
|
import java.util.*;
|
import java.util.concurrent.TimeUnit;
|
|
@RestController
|
@RequestMapping("/esSearch")
|
public class SearchController {
|
|
@Autowired
|
private SearchServiceImpl searchService;
|
@Value("${es_index_persons}")
|
private String PERSON_PIC_INDEX = "";
|
@Autowired
|
private RedisTemplate redisTemplate;
|
@Autowired
|
private BbPersonBaseServiceImpl baseServiceImpl;
|
@Autowired
|
private ImageUtils imageUtils;
|
@Autowired
|
private FastDFSUtil fastDFSUtil;
|
|
private Logger log = Logger.getLogger(SearchController.class);
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
|
SimpleDateFormat sdf1 = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss:SSS");
|
/**
|
* 供检索模块儿切图使用
|
*
|
* @param jsonMap
|
* @return
|
* @throws Exception
|
*/
|
@RequestMapping("/extractFace")
|
@ResponseBody
|
public Result extractFace(@RequestBody Map jsonMap)
|
throws Exception{
|
log.info("调用extractFace");
|
Object obj = jsonMap.get("fileInfo");
|
String s = JSONObject.toJSONString(obj);
|
log.info("fileInfo:"+s);
|
FileInfos fileInfo = JSONObject.parseObject(s,FileInfos.class);
|
Object obj1 = jsonMap.get("picQuality");
|
String s1 = JSONObject.toJSONString(obj1);
|
Integer picQuality = JSONObject.parseObject(s1,Integer.class);
|
Map manyFaceMap = new HashMap();
|
try {
|
log.info("路径是"+fileInfo.getPath());
|
byte[] imgdata = fastDFSUtil.downloadFDFS(fileInfo.getPath());
|
log.info("imageData为"+imgdata);
|
ArrayList<FaceResults> faceResults = FaceSdkTool.extractFace(imgdata);
|
if(faceResults.size()==0){
|
log.info("此图中无人脸");
|
Map noFaceMap = new HashMap();
|
noFaceMap.put("fileName",fileInfo.getName());
|
noFaceMap.put("path",fileInfo.getPath());
|
noFaceMap.put("reason","未在您上传的图片中检测到人脸，请您核查");
|
noFaceMap.put("list",null);
|
return Result.custom("比对结束", HttpStatus.OK.value(),true,noFaceMap);
|
}else {
|
log.info("切出"+faceResults.size()+"张人脸");
|
manyFaceMap.put("fileName", fileInfo.getName());
|
manyFaceMap.put("path", fileInfo.getPath());
|
manyFaceMap.put("reason", null);
|
List<FaceResults> list = new ArrayList<>();
|
for(FaceResults face: faceResults){
|
log.info("图片质量为"+face.getSocre());
|
if(face.getSocre()*100 > picQuality){
|
list.add(face);
|
}
|
}
|
manyFaceMap.put("list", list);
|
}
|
|
} catch (Exception e) {
|
log.info("切图异常");
|
e.printStackTrace();
|
}
|
return Result.custom("切图结束", HttpStatus.OK.value(),true,manyFaceMap);
|
}
|
|
/**
|
* 供检索模块儿比对使用
|
*
|
* @param jsonMap
|
* @return
|
* @throws Exception
|
*/
|
@RequestMapping("/findLikerPics")
|
@ResponseBody
|
public Result findLikerPics(@RequestBody Map jsonMap)
|
throws Exception{
|
|
// 比对条件
|
Map param = (Map)jsonMap.get("params");
|
|
String filePath = (String) param.get("path");
|
List<Map<String, Object>> currentList = new ArrayList<Map<String, Object>>(); // 比对结果全部集合
|
Map resultMap = new HashMap();
|
try {
|
|
long startTime = System.currentTimeMillis();
|
log.info("fastdfs文件路径为"+filePath);
|
byte[] imgData = fastDFSUtil.downloadFDFS(filePath);
|
ArrayList<FaceResults> faceResults = FaceSdkTool.extractFace(imgData);
|
|
byte[] imageData = faceResults.get(0).getFeature();
|
int liker = Integer.valueOf(param.get("liker")+"");
|
|
// EsThreadUtils.setCurrentList(currentList);
|
String FetchSource = "FaceFeature,content,personIsHub,personPicUrl,picAddress,picMaxUrl,picSmUrl";
|
log.info("起止时间为"+param.get("startDate")+"===="+param.get("endDate"));
|
String startDate = param.get("startDate")!=null?sdf.format(sdf.parse((String) param.get("startDate"))):null;
|
String endDate = param.get("endDate")!=null?sdf.format(sdf.parse((String) param.get("endDate"))):null;
|
|
currentList = searchService.searchScrollData(PERSON_PIC_INDEX,FetchSource,startDate,endDate,"picDate", liker, imageData);
|
|
long endTime = System.currentTimeMillis();
|
// currentList = EsThreadUtils.getCurrentList();
|
log.info("总时间"+(endTime-startTime)+"currentList:"+currentList.size());
|
|
// 排序
|
long endTime1 = System.currentTimeMillis();
|
Collections.sort(currentList, new Comparator<Map<String,Object>>() {
|
public int compare(Map<String,Object> map1, Map<String,Object> map2) {
|
int i = (int) ((Integer)map2.get("likePer") - (Integer)map1.get("likePer"));
|
if(i == 0){
|
|
try {
|
if((sdf1.parse(map2.get("picDate").toString())).getTime() > (sdf1.parse(map1.get("picDate").toString())).getTime()){
|
return 1;
|
}else if((sdf1.parse(map2.get("picDate").toString())).getTime() < (sdf1.parse(map1.get("picDate").toString())).getTime()){
|
return -1;
|
}else{
|
return 0;
|
}
|
} catch (ParseException e) {
|
e.printStackTrace();
|
return 1; //抛出异常的话默认大于
|
}
|
}else if(i>0){
|
return 1;
|
}else {
|
return -1;
|
}
|
}
|
});
|
log.info("排序时间 "+(System.currentTimeMillis()-endTime1));
|
ListOperations opsForList = redisTemplate.opsForList();
|
//把es比对数据放进redis
|
if(currentList != null && currentList.size() > 0){
|
String esKey = filePath+"_esList";
|
redisTemplate.delete(esKey);
|
opsForList.rightPushAll(esKey,currentList);
|
redisTemplate.expire(esKey,60*10,TimeUnit.SECONDS);
|
}
|
|
|
//比对人员底库中的人脸数据
|
List<Map> baseList = baseServiceImpl.selectLikerPerson(liker, imageData);
|
if(baseList != null && baseList.size() > 0) {
|
String baseKey = filePath + "_baseList";
|
redisTemplate.delete(baseKey);
|
log.info("baseList.size:"+baseList.size());
|
opsForList.rightPushAll(baseKey, baseList);
|
redisTemplate.expire(baseKey, 60 * 10, TimeUnit.SECONDS);
|
}
|
|
}catch (Exception e) {
|
e.printStackTrace();
|
log.info("比对异常--"+e.getMessage());
|
return Result.custom("比对异常", HttpStatus.OK.value(),false,null);
|
}
|
|
return Result.custom("比对结束", HttpStatus.OK.value(),true,resultMap);
|
}
|
|
/**
|
* 供布控模块儿切图使用
|
*
|
* @param jsonMap
|
* @return
|
* @throws Exception
|
*/
|
@RequestMapping("/extractFaceForControl")
|
@ResponseBody
|
public Result extractFaceForRetrieve(@RequestBody Map jsonMap)
|
throws Exception{
|
Object obj = jsonMap.get("fileList");
|
String s = JSONObject.toJSONString(obj);
|
List list = JSONObject.parseObject(s,List.class);
|
List<FileInfos> imgUrlList = new ArrayList<>();
|
list.forEach(fileInfo->{
|
FileInfos file = JSONObject.parseObject(((JSONObject)fileInfo).toJSONString(),FileInfos.class);
|
if(file != null){
|
byte[] imgdata = new byte[0];
|
try {
|
imgdata = fastDFSUtil.downloadFDFS(file.getPath());
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
log.info("路径是"+file.getPath());
|
log.info("imageData为"+imgdata);
|
ArrayList<FaceResults> faceResults = FaceSdkTool.extractFace(imgdata);
|
if(faceResults.size() > 0){
|
log.info("切出"+faceResults.size()+"张人脸");
|
faceResults.forEach(faceResult ->{
|
FileInfos fileUpload = imageUtils.cutPhotoFromFast(file,faceResult.left,faceResult.top,faceResult.width,faceResult.height);
|
imgUrlList.add(fileUpload);
|
});
|
}
|
}
|
});
|
return Result.custom("切图结束", HttpStatus.OK.value(),true,imgUrlList);
|
}
|
|
}
|
