Procházet zdrojové kódy

人脸上传,比对接口

wujiefeng před 1 rokem
rodič
revize
caf1252c09

+ 3 - 3
centers/AuthCenter/AuthServer/src/main/java/com/github/microservice/auth/server/core/service/local/UserFaceServiceImpl.java

@@ -95,15 +95,15 @@ public class UserFaceServiceImpl implements UserFaceService {
         //上传人脸库,获取向量
         String fileBase64 = userFaceUploadModel.getFile();
         byte[] decode = Base64.getDecoder().decode(fileBase64);
-        @Cleanup InputStream inputStream = new ByteArrayInputStream(decode);
+//        @Cleanup InputStream inputStream = new ByteArrayInputStream(decode);
 
-        HttpResponse<String> response = faceHelper.uploadFs(inputStream, key);
+        HttpResponse<String> response = faceHelper.uploadFs(new ByteArrayInputStream(decode), key);
         if (response.statusCode() != 200) {
             return ResultContent.build(ResultState.Fail, "人脸特征提取失败");
         }
 
         //图片入库
-        String fileId = gridFsTemplate.store(inputStream, UUID.randomUUID().toString(), Map.of("userId", userFaceUploadModel.getUserId())).toHexString();
+        String fileId = gridFsTemplate.store(new ByteArrayInputStream(decode), UUID.randomUUID().toString(), Map.of("userId", userFaceUploadModel.getUserId())).toHexString();
 
         //向量入库
         userFace.setUserId(user.getId());