提交 13b7f72f authored 作者: yangqi's avatar yangqi

1

上级 ac77b3fd
...@@ -58,8 +58,8 @@ public class FaceController { ...@@ -58,8 +58,8 @@ public class FaceController {
@NoAuth @NoAuth
@GetMapping("/addFaceImage") @GetMapping("/addFaceImage")
public ApiResponse<Object> addFaceImage(String dbName, String entityId, String url) throws Exception { public ApiResponse<Object> addFaceImage(String dbName, String entityId, String url) throws Exception {
String faceId = faceService.addFaceImage(dbName, entityId, url); faceService.addFaceImage(dbName, entityId, url);
return new ApiResponse<>(faceId); return new ApiResponse<>();
} }
@NoAuth @NoAuth
......
package cn.lefull.repository.mapper.face;
import cn.lefull.repository.model.face.AliFaceEntity;
import cn.lefull.repository.model.face.AliFaceImage;
import org.apache.ibatis.annotations.Mapper;
/**
* @author 杨奇
*/
@Mapper
public interface AliFaceImageMapper {
int addImage(AliFaceImage image);
}
package cn.lefull.repository.model.face;
import lombok.Data;
import java.time.LocalDateTime;
/**
* @author 杨奇
*/
@Data
public class AliFaceImage {
private Integer aliFaceImageId;
private Integer aliFaceEntityId;
private String faceId;
private String imageUrl;
private String ossUrl;
private Integer operatorEmployeeId;
private LocalDateTime createTime;
private LocalDateTime updateTime;
private Byte isDelete;
private LocalDateTime deleteTime;
}
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.lefull.repository.mapper.face.AliFaceImageMapper">
<insert id="addImage" keyProperty="aliFaceImageId" useGeneratedKeys="true" parameterType="cn.lefull.repository.model.face.AliFaceImage">
INSERT INTO ali_face_image (ali_face_entity_id, face_id, image_url, oss_url)
VALUES(#{aliFaceEntityId}, #{faceId}, #{imageUrl}, #{ossUrl});
</insert>
</mapper>
...@@ -39,7 +39,7 @@ public interface FaceService { ...@@ -39,7 +39,7 @@ public interface FaceService {
/** /**
* 添加人脸数据 * 添加人脸数据
*/ */
public String addFaceImage(String dbName, String entityId, String url) throws Exception; public void addFaceImage(String dbName, String entityId, String url) throws Exception;
/** /**
* 搜索人脸 * 搜索人脸
......
...@@ -4,9 +4,11 @@ import cn.lefull.common.sdk.ali.base.Face; ...@@ -4,9 +4,11 @@ import cn.lefull.common.sdk.ali.base.Face;
import cn.lefull.common.utils.Utils; import cn.lefull.common.utils.Utils;
import cn.lefull.repository.mapper.face.AliFaceDbMapper; import cn.lefull.repository.mapper.face.AliFaceDbMapper;
import cn.lefull.repository.mapper.face.AliFaceEntityMapper; import cn.lefull.repository.mapper.face.AliFaceEntityMapper;
import cn.lefull.repository.mapper.face.AliFaceImageMapper;
import cn.lefull.repository.mapper.test.TestMapper; import cn.lefull.repository.mapper.test.TestMapper;
import cn.lefull.repository.model.face.AliFaceDb; import cn.lefull.repository.model.face.AliFaceDb;
import cn.lefull.repository.model.face.AliFaceEntity; import cn.lefull.repository.model.face.AliFaceEntity;
import cn.lefull.repository.model.face.AliFaceImage;
import com.aliyun.facebody20191230.models.SearchFaceResponseBody; import com.aliyun.facebody20191230.models.SearchFaceResponseBody;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
...@@ -25,6 +27,8 @@ public class FaceServiceImpl implements FaceService{ ...@@ -25,6 +27,8 @@ public class FaceServiceImpl implements FaceService{
private AliFaceDbMapper dbMapper; private AliFaceDbMapper dbMapper;
@Autowired @Autowired
private AliFaceEntityMapper entityMapper; private AliFaceEntityMapper entityMapper;
@Autowired
private AliFaceImageMapper imageMapper;
@Override @Override
public List<String> getFaceDbList() throws Exception { public List<String> getFaceDbList() throws Exception {
...@@ -87,12 +91,32 @@ public class FaceServiceImpl implements FaceService{ ...@@ -87,12 +91,32 @@ public class FaceServiceImpl implements FaceService{
} }
@Override @Override
public String addFaceImage(String dbName, String entityId, String url) throws Exception { public void addFaceImage(String dbName, String entityId, String url) throws Exception {
Utils.checkParamsNotEmpty(dbName, "dbName"); Utils.checkParamsNotEmpty(dbName, "dbName");
Utils.checkParamsNotEmpty(entityId, "entityId"); Utils.checkParamsNotEmpty(entityId, "entityId");
Utils.checkParamsNotEmpty(url, "url"); Utils.checkParamsNotEmpty(url, "url");
url = sdk.uploadImageUrl(url);
return sdk.addFaceImage(dbName, entityId, url); AliFaceDb faceDb = new AliFaceDb();
faceDb.setDbName(dbName);
AliFaceDb db = dbMapper.getDbByName(faceDb);
if(db == null){
throw new Exception("未找到人脸数据库:" + dbName);
}
AliFaceEntity entityDb = new AliFaceEntity();
entityDb.setAliFaceDbId(db.getAliFaceDbId());
entityDb.setEntityId(entityId);
AliFaceEntity entity = entityMapper.getEntityByEntityId(entityDb);
if(entity == null){
throw new Exception("没找到样本:" + entityId);
}
String ossUrl = sdk.uploadImageUrl(url);
String faceId = sdk.addFaceImage(dbName, entityId, ossUrl);
AliFaceImage image = new AliFaceImage();
image.setAliFaceEntityId(entity.getAliFaceEntityId());
image.setFaceId(faceId);
image.setImageUrl(url);
image.setOssUrl(ossUrl);
imageMapper.addImage(image);
} }
@Override @Override
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论