commit fe5140c67959dc9e8f2958a1cfb5d0d93a3ca0f8
author JinGe <22301009@bjtu.edu.cn> Fri Jun 06 20:07:42 2025 +0800
committer JinGe <22301009@bjtu.edu.cn> Fri Jun 06 20:07:42 2025 +0800
tree 3440014d46b0849292fdd64a850103c612f8029f
parent 750643d4e4147fd4af3bbad961dec80b2c1962e0

调试交互

Change-Id: Ib9d7b8432cb622c7e28c842ab4dc4e156fcd6414

src/main/java/com/example/myproject/service/UserService.java

diff --git a/src/main/java/com/example/myproject/service/UserService.java b/src/main/java/com/example/myproject/service/UserService.java
index 283a85c..de6bf8b 100644
--- a/src/main/java/com/example/myproject/service/UserService.java
+++ b/src/main/java/com/example/myproject/service/UserService.java
@@ -1,8 +1,10 @@
 package com.example.myproject.service;
-
+import cn.dev33.satoken.stp.StpUtil;
+import com.example.myproject.entity.FriendRelation;
 import com.example.myproject.entity.User;
 import com.example.myproject.entity.Users;
 import com.example.myproject.entity.UserInviteCode;
+import com.example.myproject.repository.FriendRelationRepository;
 import com.example.myproject.repository.UserRepository;
 import com.example.myproject.repository.UserInviteCodeRepository;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -27,6 +29,9 @@
     @Autowired
     private UserInviteCodeRepository userInviteCodeRepository;
 
+    @Autowired
+    private FriendRelationRepository friendRelationRepository;
+
     // 生成邀请码
     public Map<String, Object> generateInviteCode(Long userId) {
         // 获取用户信息
@@ -135,6 +140,17 @@
         // 保存用户信息
         userRepository.save(newUser);
 
+        FriendRelation newFriendRelation  = new FriendRelation();
+        newFriendRelation.setUserId(newUser.getUserId());
+        newFriendRelation.setCreateTime(new Date());
+        newFriendRelation.setFriendId(inviteUserId);
+
+        FriendRelation newFriendRelations  = new FriendRelation();
+        newFriendRelations.setUserId(inviteUserId);
+        newFriendRelations.setCreateTime(new Date());
+        newFriendRelations.setFriendId(newUser.getUserId());
+        friendRelationRepository.save(newFriendRelation);
+        friendRelationRepository.save(newFriendRelations);
         // 更新邀请码的使用状态
         inviteEntity.setIsUsed(true);
         userInviteCodeRepository.save(inviteEntity);
@@ -156,7 +172,8 @@
         if (!user.getPassword().equals(password)) {
             return "密码错误";
         }
-
+        StpUtil.login(user.getUserId());
+        
         // 登录成功
         return "登录成功";
     }