添加用户操作的MQ通知功能，包括新增、更新和删除功能实现。

alexyang
1 parent 436cb39c
Showing 1 changed file with 41 additions and 4 deletions
itcast-auth/itcast-auth-server/src/main/java/com/itheima/authority/controller/auth/UserController.java
 package com.itheima.authority.controller.auth;
  
+import cn.hutool.json.JSONObject;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.google.common.collect.Lists;
+import com.itheima.authority.biz.mq.RabbitMessageSender;
 import com.itheima.authority.biz.service.auth.RoleService;
 import com.itheima.authority.biz.service.auth.UserGroupService;
 import com.itheima.authority.biz.service.auth.UserGroupUserService;
@@ -34,10 +37,7 @@ import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
  
-import java.util.Collection;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
  
 import static com.itheima.tools.exception.code.ExceptionCode.BAD_REQUEST;
@@ -69,6 +69,8 @@ public class UserController extends BaseController {
     private StationService stationService;
     @Autowired
     private DozerUtils dozer;
+    @Autowired
+    private RabbitMessageSender rabbitMessageSender;
  
     @ApiOperation(value = "分页查询用户", notes = "分页查询用户")
     @ApiImplicitParams({
@@ -247,6 +249,13 @@ public class UserController extends BaseController {
     public R<User> save(@RequestBody @Validated UserSaveDTO data) {
         User user = dozer.map(data, User.class);
         userService.saveUser(user);
+
+        // 同步mq通知修改信息
+        JSONObject jsonObject = new JSONObject();
+        jsonObject.put("type", "USER");
+        jsonObject.put("operation", "ADD");
+        jsonObject.put("content", Lists.newArrayList(user));
+        rabbitMessageSender.send("itcast-auth-user", "add", jsonObject.toString(), 3);
         return success(user);
     }
  
@@ -256,6 +265,12 @@ public class UserController extends BaseController {
     public R<User> update(@RequestBody @Validated(SuperEntity.Update.class) UserUpdateDTO data) {
         User user = dozer.map(data, User.class);
         userService.updateUser(user);
+        // 同步mq通知修改信息
+        JSONObject jsonObject = new JSONObject();
+        jsonObject.put("type", "USER");
+        jsonObject.put("operation", "UPDATE");
+        jsonObject.put("content", Lists.newArrayList(user));
+        rabbitMessageSender.send("itcast-auth-user", "update", jsonObject.toString(), 3);
         return success(user);
     }
  
@@ -267,6 +282,15 @@ public class UserController extends BaseController {
         user.setId(data.getId());
         user.setRoles(data.getRoles());
         userService.updateUserRole(user);
+
+        // 同步mq通知修改信息
+        JSONObject jsonObject = new JSONObject();
+        jsonObject.put("type", "USER");
+        jsonObject.put("operation", "UPDATE");
+        user = this.userService.getById(data.getId());
+        user.setRoles(data.getRoles());
+        jsonObject.put("content", Lists.newArrayList(user));
+        rabbitMessageSender.send("itcast-auth-user", "update", jsonObject.toString(), 3);
         return success(user);
     }
  
@@ -289,7 +313,20 @@ public class UserController extends BaseController {
     @DeleteMapping
     @SysLog("删除用户")
     public R<Boolean> delete(@RequestParam("ids[]") List<Long> ids) {
+        // 查询待删除的组织
+        List<User> delList = new ArrayList<>();
+        for (Long id : ids) {
+            User user = this.userService.getById(id);
+            delList.add(user);
+        }
         userService.remove(ids);
+
+        // 同步mq通知修改信息
+        JSONObject jsonObject = new JSONObject();
+        jsonObject.put("type", "USER");
+        jsonObject.put("operation", "DEL");
+        jsonObject.put("content", delList);
+        rabbitMessageSender.send("itcast-auth-user", "del", jsonObject.toString(), 3);
         return success(true);
     }