提交 d73a0cfa authored 作者: yangqi's avatar yangqi

1

上级 ad115701
......@@ -16,4 +16,10 @@ public class Utils {
throw new Exception(name + "未设置");
}
}
public static void checkParamsNotEmpty(String[] str, String name) throws Exception {
if(str == null){
throw new Exception(name + "未设置");
}
}
}
......@@ -41,6 +41,8 @@ public interface FaceService {
*/
void addFaceImage(String dbName, String entityId, String url) throws Exception;
void addFaceImageList(String dbName, String entityId, String[] urlList) throws Exception;
/**
* 搜索人脸
*/
......
......@@ -122,6 +122,35 @@ public class FaceServiceImpl implements FaceService{
imageMapper.addImage(image);
}
@Override
public void addFaceImageList(String dbName, String entityId, String[] urlList) throws Exception {
Utils.checkParamsNotEmpty(dbName, "dbName");
Utils.checkParamsNotEmpty(entityId, "entityId");
Utils.checkParamsNotEmpty(urlList, "urlList");
AliFaceDb db = dbMapper.getDbByName(dbName);
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);
}
for(String url : urlList){
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
public boolean searchFace(String dbName, String url) throws Exception {
Utils.checkParamsNotEmpty(dbName, "dbName");
......
......@@ -160,6 +160,7 @@ public class EquipmentServiceImpl implements EquipmentService {
String entityId = this.getEntityId(id);
String labels = this.getLabels(identity);
faceService.addFace(dbName, entityId, labels);
}
private String getLabels(Byte identity)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论