提交 7942e2d6 authored 作者: yangqi's avatar yangqi

1

上级 f6cf319b
package cn.lefull.repository.mapper.face; package cn.lefull.repository.mapper.face;
import cn.lefull.repository.model.face.AliFaceDb;
import cn.lefull.repository.model.face.AliFaceEntity; import cn.lefull.repository.model.face.AliFaceEntity;
import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Mapper;
...@@ -13,4 +12,6 @@ public interface AliFaceEntityMapper { ...@@ -13,4 +12,6 @@ public interface AliFaceEntityMapper {
AliFaceEntity getEntityByEntityId(AliFaceEntity entity); AliFaceEntity getEntityByEntityId(AliFaceEntity entity);
int addEntity(AliFaceEntity entity); int addEntity(AliFaceEntity entity);
int updateEntity(AliFaceEntity entity);
} }
...@@ -9,4 +9,7 @@ ...@@ -9,4 +9,7 @@
<insert id="addEntity" keyProperty="aliFaceEntityId" useGeneratedKeys="true" parameterType="cn.lefull.repository.model.face.AliFaceEntity"> <insert id="addEntity" keyProperty="aliFaceEntityId" useGeneratedKeys="true" parameterType="cn.lefull.repository.model.face.AliFaceEntity">
INSERT INTO ali_face_entity (ali_face_db_id, entity_id, labels) VALUES(#{aliFaceDbId}, #{entityId}, #{labels}); INSERT INTO ali_face_entity (ali_face_db_id, entity_id, labels) VALUES(#{aliFaceDbId}, #{entityId}, #{labels});
</insert> </insert>
<update id="updateEntity" keyProperty="aliFaceEntityId" useGeneratedKeys="true" parameterType="cn.lefull.repository.model.face.AliFaceEntity">
UPDATE ali_face_entity set labels = #{lables} where ali_face_entity_id = #{aliFaceEntiyId};
</update>
</mapper> </mapper>
...@@ -57,7 +57,12 @@ public class FaceServiceImpl implements FaceService{ ...@@ -57,7 +57,12 @@ public class FaceServiceImpl implements FaceService{
entity.setAliFaceDbId(db.getAliFaceDbId()); entity.setAliFaceDbId(db.getAliFaceDbId());
entity.setEntityId(entityId); entity.setEntityId(entityId);
entity.setLabels(labels); entity.setLabels(labels);
entityMapper.addEntity(entity); AliFaceEntity data = entityMapper.getEntityByEntityId(entity);
if(data == null){
entityMapper.addEntity(entity);
}else{
entityMapper.updateEntity(entity);
}
sdk.addFace(dbName, entityId, labels); sdk.addFace(dbName, entityId, labels);
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论