From 247f19f6c3f17c2ac2f37b55c7d0550731f31ffe Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Thu, 25 Mar 2021 17:24:16 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/dev' into dev

---
 screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleMenuServiceImpl.java |   82 +++++++++++++++++++++++++++++++++++++++++
 1 files changed, 82 insertions(+), 0 deletions(-)

diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleMenuServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleMenuServiceImpl.java
index 0ce859d..2c7e1fc 100644
--- a/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleMenuServiceImpl.java
+++ b/screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleMenuServiceImpl.java
@@ -1,10 +1,20 @@
 package com.moral.api.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.moral.api.entity.ManageRoleMenu;
 import com.moral.api.mapper.ManageRoleMenuMapper;
 import com.moral.api.service.ManageRoleMenuService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.moral.constant.ResponseCodeEnum;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 /**
  * <p>
@@ -17,4 +27,76 @@
 @Service
 public class ManageRoleMenuServiceImpl extends ServiceImpl<ManageRoleMenuMapper, ManageRoleMenu> implements ManageRoleMenuService {
 
+    @Autowired(required = false)
+    private ManageRoleMenuMapper manageRoleMenuMapper;
+
+    @Override
+    @Transactional
+    public Map<String, Object> updateRoleMenu(List list, int id) {
+        Map<String,Object> resultMap = new HashMap<>();
+        if (list.size()==0){
+            resultMap.put("code",ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode());
+            resultMap.put("msg",ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg());
+            return resultMap;
+        }
+        QueryWrapper<ManageRoleMenu> wrapper = new QueryWrapper();
+        wrapper.eq("role_id",id);
+        wrapper.eq("is_delete",0);
+        List<Map> result = manageRoleMenuMapper.getDataByMenuIds(list,id);
+        List<Map> result1 = manageRoleMenuMapper.getDataByRoleId(id);
+        if (manageRoleMenuMapper.selectCount(wrapper)==0 || result.size()==0){
+            List<Map> insertList = new ArrayList<>();
+            for (Object temp:list) {
+                Map<String,Object> map = new HashMap<>();
+                map.put("role_id",id);
+                map.put("menu_id",Integer.parseInt(temp.toString()));
+                insertList.add(map);
+            }
+            manageRoleMenuMapper.insertBatch(insertList);
+        }
+        List updateList = new ArrayList();
+        for (Map manageRoleMenu:result1) {
+            int have = 1;
+            for (Object temp:list){
+                if (manageRoleMenu.get("menu_id").equals(Integer.parseInt(temp.toString()))){
+                    have = 2;
+                    continue;
+                }
+            }
+            if (have == 2){
+                continue;
+            }
+            updateList.add(manageRoleMenu.get("menu_id"));
+        }
+        List insertList = new ArrayList();
+        for (Object temp:list) {
+            int have = 1;
+            for (Map manageRoleMenu:result1){
+                if (manageRoleMenu.get("menu_id").equals(Integer.parseInt(temp.toString()))){
+                    have = 2;
+                    continue;
+                }
+            }
+            if (have == 2){
+                continue;
+            }
+            insertList.add(temp);
+        }
+        if (updateList.size()>0){
+            manageRoleMenuMapper.updateDeleteStateByRoleIdMenuIds(updateList,id);
+        }
+        if (insertList.size()>0){
+            List mapList = new ArrayList();
+            for (Object temp:insertList) {
+                Map<String,Integer> insertMap = new HashMap<>();
+                insertMap.put("role_id",id);
+                insertMap.put("menu_id",Integer.parseInt(temp.toString()));
+                mapList.add(insertMap);
+            }
+            manageRoleMenuMapper.insertBatch(mapList);
+        }
+        resultMap.put("code",ResponseCodeEnum.SUCCESS.getCode());
+        resultMap.put("msg",ResponseCodeEnum.SUCCESS.getMsg());
+        return resultMap;
+    }
 }

--
Gitblit v1.8.0