2019/08/02

src/main/java/com/izouma/walkchina/service/UserInfoService.java

@@ -198,6 +198,9 @@ public class UserInfoService implements UserDetailsService {
     }
 
     public void acceptInvite(Long userId, Long inviter, boolean recruit) {
+        if (userId.equals(inviter)) {
+            return;
+        }
         friendInfoService.saveFriend(userId, inviter);
         if (recruit) {
             teamService.recruit(inviter, userId);

src/main/java/com/izouma/walkchina/web/TeamController.java

@@ -25,6 +25,9 @@ public class TeamController {
     @PostMapping("/hire")
     public Result hire(@RequestParam("userId") Long userId) {
         UserInfo userInfo = SecurityUtils.getAuthenticatedUser();
+        if (userId.equals(userInfo.getId())) {
+            return Result.error("不能招募自己");
+        }
         teamService.hire(userInfo.getId(), userId);
         return Result.ok();
     }