Просмотр исходного кода

通过接口 获取AI整理的个人信息

zrd 3 месяцев назад
Родитель
Сommit
4d14328138

+ 7 - 0
yudao-module-member/yudao-module-member-biz/src/main/java/cn/iocoder/yudao/module/member/controller/app/user/AppUserInformationController.java

@@ -64,6 +64,13 @@ public class AppUserInformationController {
         return success(BeanUtils.toBean(userInformation, UserInformationRespVO.class));
     }
     
+    @GetMapping("/init")
+    @Operation(summary = "获得用户信息")
+    public CommonResult<String> init() {
+        userInformationService.init();
+        
+        return success("");
+    }
     @GetMapping("/getAi")
     @Operation(summary = "获得用户信息")
     public CommonResult<String> getAi() {

+ 30 - 9
yudao-module-member/yudao-module-member-biz/src/main/java/cn/iocoder/yudao/module/member/service/userinformation/UserInformationServiceImpl.java

@@ -88,33 +88,54 @@ public class UserInformationServiceImpl implements UserInformationService {
     public UserInformationDO getAi() {
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         
-        return userInformation(userId, false);
+        return userInformationAi(userId);
     }
     
+    private UserInformationDO userInformationAi(Long userId) {
+        UserInformationDO userInformation = userInformationMapper.selectOne(UserInformationDO::getUserId,
+                SecurityFrameworkUtils.getLoginUserId());
+        
+        if (userInformation != null) {
+            
+            
+            return userInformation;
+        } else {
+            UserInformationDO insertUserInformation = new UserInformationDO();
+            insertUserInformation = userAiInformation(userId);
+            insertUserInformation.setUserId(SecurityFrameworkUtils.getLoginUserId());
+            
+            
+            insertUserInformation.setUserId(userId);
+            userInformationMapper.insert(insertUserInformation);
+//            AiFrameworkUtils.getHtml("个人征信HTML", userInformation.getCreditInformation());
+            return insertUserInformation;
+        }
+    }
     private UserInformationDO userInformation(Long userId, Boolean updateFlag) {
         UserInformationDO userInformation = userInformationMapper.selectOne(UserInformationDO::getUserId,
                 SecurityFrameworkUtils.getLoginUserId());
         
         if (userInformation != null) {
-            Long id = userInformation.getId();
-            if (updateFlag) {
-                userInformation = userAiInformation(userId);
-                userInformation.setId(id);
+            
+            UserInformationDO tdo = userAiInformation(userId);
+            userInformation.setCreditInformation(tdo.getCreditInformation());
+            userInformation.setBusinessLicenseInformation(tdo.getBusinessLicenseInformation());
                 userInformation.setUserHtml(AiFrameworkUtils.getHtml("个人征信HTML",
                         "个人征信" + userInformation.getCreditInformation()
                                 + "营业执照" + userInformation.getCreditInformation()));
                 userInformationMapper.updateById(userInformation);
-            }
             return userInformation;
         } else {
             UserInformationDO insertUserInformation = new UserInformationDO();
-            insertUserInformation = userAiInformation(userId);
             insertUserInformation.setUserId(SecurityFrameworkUtils.getLoginUserId());
-            if (updateFlag) {
+            UserInformationDO tdo = userAiInformation(userId);
+            insertUserInformation.setCreditInformation(tdo.getCreditInformation());
+            insertUserInformation.setBusinessLicenseInformation(tdo.getBusinessLicenseInformation());
                 insertUserInformation.setUserHtml(AiFrameworkUtils.getHtml("个人征信HTML",
                         "个人征信" + insertUserInformation.getCreditInformation()
                                 + "营业执照" + insertUserInformation.getCreditInformation()));
-            }
+            
+            insertUserInformation.setUserId(userId);
             userInformationMapper.insert(insertUserInformation);
 //            AiFrameworkUtils.getHtml("个人征信HTML", userInformation.getCreditInformation());
             return insertUserInformation;