From 1836f7863882ceae0e5c246556196be48bd12384 Mon Sep 17 00:00:00 2001
From: Jinquan_Ou <Jinquan@gdut.com>
Date: 星期六, 08 四月 2023 00:13:27 +0800
Subject: [PATCH] 111

---
 zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java |   54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 54 insertions(+), 0 deletions(-)

diff --git a/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java b/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
index 21a2b81..9fa02f2 100644
--- a/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
+++ b/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
@@ -6,7 +6,9 @@
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.SysUser;
 import com.ruoyi.common.utils.MapUtils;
+import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.domain.ZInfoUser;
 import com.ruoyi.domain.ZfEvent;
@@ -19,8 +21,10 @@
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -102,6 +106,7 @@
                     UserRelationDto userRelationDto = new UserRelationDto();
                     BeanUtils.copyProperties(otherUser,userRelationDto);
                     userRelationDto.setRelation(relation.getRelation());
+                    userRelationDto.setId(relation.getId());
                     otherUserList.add(userRelationDto);
                     return relation;
                 }
@@ -109,4 +114,53 @@
 
         return AjaxResult.success(otherUserList);
     }
+
+    /**
+     * 妫�鏌ヤ紶鍏ュ鏂圭殑id鏄惁瀛樺湪
+     * @param zfRelation
+     */
+    private void checkRelation(ZfRelation zfRelation){
+
+        LambdaQueryWrapper<ZInfoUser> lqw = new LambdaQueryWrapper<>();
+        lqw.eq(ZInfoUser::getUserId, zfRelation.getOtherId());
+
+        ZInfoUser zInfoUser2 = getOne(lqw);
+        if (zInfoUser2==null){
+            throw new RuntimeException("瀵规柟鐨刬d涓嶅瓨鍦�");
+        }
+    }
+
+    @Override
+    @Transactional
+    public AjaxResult addRelation(ZfRelation zfRelation) {
+        checkRelation(zfRelation);
+
+        //鑾峰彇鑷繁鐨刬d骞朵紶鍏�
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        Long userId = user.getUserId();
+        zfRelation.setMyId(userId);
+
+        zfRelationService.save(zfRelation);
+        return AjaxResult.success();
+    }
+
+    @Override
+    @Transactional
+    public AjaxResult updateRelation(ZfRelation zfRelation) {
+        checkRelation(zfRelation);
+        zfRelationService.updateById(zfRelation);
+        return AjaxResult.success();
+    }
+
+    @Override
+    public AjaxResult deleteRelation(Long[] ids) {
+        boolean flag = zfRelationService.removeByIds(Arrays.asList(ids));
+        if (flag){
+            return AjaxResult.success();
+        }else {
+            return AjaxResult.error();
+        }
+    }
+
+
 }

--
Gitblit v1.9.1