琴房课注册流程处理

cooleshow-user/user-biz/src/main/java/com/yonge/cooleshow/biz/dal/service/impl/StudentServiceImpl.java

@@ -388,9 +388,24 @@ public class StudentServiceImpl extends ServiceImpl<StudentDao, Student> impleme
         }
         Long customerId = studentVo.getCustomerId();
         if (userId.equals(customerId)) {
-            Set<Long> studentIds = new HashSet<>();
-            studentIds.add(studentVo.getUserId());
-            imUserFriendService.saveUserFriend(userId, studentIds, EImUserFriendSourceForm.TEACHER);
+//            Set<Long> studentIds = new HashSet<>();
+//            studentIds.add(studentVo.getUserId());
+//            imUserFriendService.saveUserFriend(userId, studentIds, EImUserFriendSourceForm.TEACHER);
+            imUserFriendService.lambdaUpdate()
+                    .set(ImUserFriend::getSourceForm, EImUserFriendSourceForm.TEACHER)
+                    .eq(ImUserFriend::getUserId, userId)
+                    .eq(ImUserFriend::getClientType, ClientEnum.TEACHER)
+                    .eq(ImUserFriend::getFriendId, studentVo.getUserId())
+                    .eq(ImUserFriend::getFriendType, ClientEnum.STUDENT)
+                    .update();
+
+            imUserFriendService.lambdaUpdate()
+                    .set(ImUserFriend::getSourceForm, EImUserFriendSourceForm.TEACHER)
+                    .eq(ImUserFriend::getUserId, studentVo.getUserId())
+                    .eq(ImUserFriend::getClientType, ClientEnum.STUDENT)
+                    .eq(ImUserFriend::getFriendId, userId)
+                    .eq(ImUserFriend::getFriendType, ClientEnum.TEACHER)
+                    .update();
         }
         resMap.put("now", detail);
         return HttpResponseResult.succeed(resMap);