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

1

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