123

ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserService.java

@@ -45,4 +45,7 @@ public interface UserService extends IService<User> {
 
     void refreshByAge();
 
+    void unBindInviteUser(Long userId);
+
+    void bindInviteUser(Long userId, String inviteUsercode);
 }

ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserServiceImpl.java

@@ -5,6 +5,7 @@ import com.alibaba.fastjson2.JSON;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.cai.auth.CaiLoginManager;
 import com.ruoyi.cai.domain.*;
 import com.ruoyi.cai.dto.admin.vo.UserAdminVo;
 import com.ruoyi.cai.dto.app.query.index.GreetQuery;
@@ -14,6 +15,7 @@ import com.ruoyi.cai.dto.app.vo.user.OnlineStatusVo;
 import com.ruoyi.cai.dto.app.vo.user.UserInfoVo;
 import com.ruoyi.cai.dto.app.vo.user.UserListVo;
 import com.ruoyi.cai.enums.GenderEnum;
+import com.ruoyi.cai.manager.CurrentUserManager;
 import com.ruoyi.cai.mapper.UserMapper;
 import com.ruoyi.cai.notice.YunxinHttpService;
 import com.ruoyi.cai.service.*;
@@ -78,6 +80,8 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
     private AnchorApplyService anchorApplyService;
     @Autowired
     private RoomService roomService;
+    @Autowired
+    private CaiLoginManager caiLoginManager;
 
     @Override
     public User getByUsername(String username) {
@@ -316,4 +320,42 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
         }
     }
 
+    @Autowired
+    private UnionUserService unionUserService;
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void unBindInviteUser(Long userId){
+        User user = this.getById(userId);
+        if(user == null){
+            throw new ServiceException("用户不存在");
+        }
+        this.update(Wrappers.lambdaUpdate(User.class)
+                .eq(User::getId,userId)
+                .set(User::getInviteId,null));
+        userInviteService.remove(Wrappers.lambdaQuery(UserInvite.class)
+                .eq(UserInvite::getUserId,userId));
+        // 退出工会
+        unionUserService.remove(Wrappers.lambdaQuery(UnionUser.class)
+                .eq(UnionUser::getUserId,userId));
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void bindInviteUser(Long userId, String inviteUsercode){
+        User user = this.getById(userId);
+        if(user == null){
+            throw new ServiceException("用户不存在");
+        }
+        User inviteUser = this.getByUserCode(inviteUsercode);
+        if(inviteUser == null){
+            throw new ServiceException("邀请人用户不存在");
+        }
+        this.update(Wrappers.lambdaUpdate(User.class)
+                .eq(User::getId,userId)
+                .set(User::getInviteId,inviteUser.getId()));
+        user = this.getById(userId);
+        caiLoginManager.dealInviteId(user,false);
+    }
+
 }