From be0ef664d01c1b600573aa6c7c00350e9c960925 Mon Sep 17 00:00:00 2001
From: divenswu <divenswu@163.com>
Date: Tue, 22 Mar 2022 15:41:30 +0800
Subject: [PATCH] =?UTF-8?q?update:=E4=BF=AE=E5=A4=8Dbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/FaceSearchServiceImpl.java      | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/face-search-server/src/main/java/com/visual/face/search/server/service/impl/FaceSearchServiceImpl.java b/face-search-server/src/main/java/com/visual/face/search/server/service/impl/FaceSearchServiceImpl.java
index 8c61320..016d23b 100644
--- a/face-search-server/src/main/java/com/visual/face/search/server/service/impl/FaceSearchServiceImpl.java
+++ b/face-search-server/src/main/java/com/visual/face/search/server/service/impl/FaceSearchServiceImpl.java
@@ -86,9 +86,23 @@ public class FaceSearchServiceImpl extends BaseService implements FaceSearchServ
         }
         //结果和人数是否一致
         List<SearchResult> result = searchResponse.getResult();
-        if(result.size() != faceInfos.size()){
+        if(result.size() > 0 && result.size() != faceInfos.size()){
             throw new RuntimeException("search result error");
         }
+        //如数据库中没有任何样本的情况下，会出现异常，这里进行单独处理
+        if(result.size() == 0 && faceInfos.size() > 0){
+            List<FaceSearchRepVo> vos = new ArrayList<>();
+            for(int i=0; i<faceInfos.size(); i++) {
+                FaceInfo.FaceBox box = faceInfos.get(i).box;
+                FaceSearchRepVo vo = FaceSearchRepVo.build();
+                vo.setLocation(FaceLocation.build(box.leftTop.x, box.leftTop.y, box.width(), box.height()));
+                vo.setFaceScore((float) Math.floor(faceInfos.get(i).score * 1000000) / 10000);
+                List<SampleFaceVo> match = new ArrayList<>();
+                vo.setMatch(match);
+                vos.add(vo);
+            }
+            return vos;
+        }
         //获取关联数据ID
         boolean needFixFaceId = false;
         Set<Long> faceIds = new HashSet<>();