[update]添加service 层代码

DougLee · DougLee · commit 1cf8ff431940 · 2018-04-03T17:27:12.000+08:00
diff --git a/spring-boot-shiro/src/main/java/com/example/springbootshiro/dao/MenuMapper.java b/spring-boot-shiro/src/main/java/com/example/springbootshiro/dao/MenuMapper.java
@@ -13,4 +13,10 @@ public interface MenuMapper extends MyMapper<MenuInfo> {
     List<MenuInfo> findUserPermissions(String userName);
 
     List<MenuInfo> findUserMenus(String userName);
+
+    /**
+     * 删除父节点, 子节点变成顶级节点
+     * @param menuIds
+     */
+    void changToTop(List<String> menuIds);
 }
diff --git a/spring-boot-shiro/src/main/java/com/example/springbootshiro/service/impl/MenuServiceImpl.java b/spring-boot-shiro/src/main/java/com/example/springbootshiro/service/impl/MenuServiceImpl.java
@@ -5,14 +5,18 @@
 import com.example.springbootshiro.domain.Tree;
 import com.example.springbootshiro.service.BaseService;
 import com.example.springbootshiro.service.MenuService;
+import com.example.springbootshiro.service.RoleMenuService;
 import com.example.springbootshiro.util.StringUtils;
+import com.example.springbootshiro.util.TreeUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
 import tk.mybatis.mapper.entity.Example;
 
 import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Date;
 import java.util.List;
 
 /**
@@ -25,6 +29,9 @@ public class MenuServiceImpl extends BaseService<MenuInfo> implements MenuServic
     @Autowired
     private MenuMapper menuMapper;
 
+    @Autowired
+    private RoleMenuService roleMenuService;
+
     @Override
     public List<MenuInfo> findUserPermissions(String userName) {
         return this.menuMapper.findUserPermissions(userName);
@@ -62,42 +69,93 @@ public Tree<MenuInfo> getMenuButtonTree() {
             tree.setText(menuInfo.getMenuName());
             trees.add(tree);
         }
-        Tree<MenuInfo> t = null;
+        Tree<MenuInfo> t = TreeUtils.build(trees);
         return t;
     }
 
     @Override
     public Tree<MenuInfo> getMenuTree() {
-        return null;
+        List<Tree<MenuInfo>> trees = new ArrayList<>();
+        Example example = new Example(MenuInfo.class);
+        example.createCriteria().andCondition("type=", 0);
+        example.setOrderByClause("create_time");
+        List<MenuInfo> menus = this.menuMapper.selectByExample(example);
+        for (MenuInfo menuInfo: menus){
+            Tree<MenuInfo> tree = new Tree<>();
+            tree.setId(menuInfo.getMenuId().toString());
+            tree.setParentId(menuInfo.getParentId().toString());
+            tree.setText(menuInfo.getMenuName());
+
+            trees.add(tree);
+        }
+
+        Tree<MenuInfo> t = TreeUtils.build(trees);
+        return t;
     }
 
     @Override
     public Tree<MenuInfo> getUserMenu(String userName) {
-        return null;
+
+        List<Tree<MenuInfo>> trees = new ArrayList<>();
+        List<MenuInfo> menus = findUserMenus(userName);
+
+        for (MenuInfo menu : menus) {
+            Tree<MenuInfo> tree = new Tree<MenuInfo>();
+            tree.setId(menu.getMenuId().toString());
+            tree.setParentId(menu.getParentId().toString());
+            tree.setText(menu.getMenuName());
+            tree.setUrl(menu.getUrl());
+            trees.add(tree);
+        }
+        Tree<MenuInfo> t = TreeUtils.build(trees);
+        return t;
     }
 
     @Override
     public MenuInfo findById(Long menuId) {
-        return null;
+        return selectByKey(menuId);
     }
 
     @Override
     public MenuInfo findByNameAndType(String menuName, String type) {
-        return null;
+        Example example = new Example(MenuInfo.class);
+        example.createCriteria().andCondition("lower(menu_name)=", menuName.toLowerCase())
+                .andEqualTo("type", Long.valueOf(type));
+        List<MenuInfo> list = menuMapper.selectByExample(example);
+        if (list.size() == 0) {
+            return null;
+        }else {
+            return list.get(0);
+        }
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void addMenu(MenuInfo menu) {
-
+        menu.setCreateTime(new Date());
+        if (menu.getParentId() == null) {
+            menu.setParentId(0L);
+        }
+        this.save(menu);
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void updateMenu(MenuInfo menu) {
+        menu.setModifyTime(new Date());
+        if (menu.getParentId() == null) {
+            menu.setParentId(0L);
+        }
+        this.updateNotNull(menu);
 
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void deleteMeuns(String menuIds) {
-
+        List<String> list = Arrays.asList(menuIds.split(","));
+        batchDelete(list, "menuId", MenuInfo.class);
+        roleMenuService.deleteRoleMenusByMenuId(menuIds);
+        menuMapper.changToTop(list);
     }
 }
diff --git a/spring-boot-shiro/src/main/java/com/example/springbootshiro/service/impl/RoleMenuServiceImpl.java b/spring-boot-shiro/src/main/java/com/example/springbootshiro/service/impl/RoleMenuServiceImpl.java
@@ -1,12 +1,16 @@
 package com.example.springbootshiro.service.impl;
 
 import com.example.springbootshiro.domain.RoleMenu;
+import com.example.springbootshiro.domain.UserRole;
 import com.example.springbootshiro.service.BaseService;
 import com.example.springbootshiro.service.RoleMenuService;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.util.Arrays;
+import java.util.List;
+
 /**
  * Created by Douglee on 2018/4/2.
  */
@@ -15,12 +19,17 @@
 public class RoleMenuServiceImpl extends BaseService<RoleMenu> implements RoleMenuService {
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void deleteRoleMenusByRoleId(String roleIds) {
-
+        List<String> list = Arrays.asList(roleIds.split(","));
+        this.batchDelete(list, "roleId", RoleMenu.class);
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void deleteRoleMenusByMenuId(String menuIds) {
+        List<String> list = Arrays.asList(menuIds.split(","));
+        this.batchDelete(list, "menuId", RoleMenu.class);
 
     }
 }
diff --git a/spring-boot-shiro/src/main/java/com/example/springbootshiro/service/impl/RoleServiceImpl.java b/spring-boot-shiro/src/main/java/com/example/springbootshiro/service/impl/RoleServiceImpl.java
@@ -1,14 +1,23 @@
 package com.example.springbootshiro.service.impl;
 
 import com.example.springbootshiro.dao.RoleMapper;
+import com.example.springbootshiro.dao.RoleMenuMapper;
 import com.example.springbootshiro.domain.RoleInfo;
+import com.example.springbootshiro.domain.RoleMenu;
+import com.example.springbootshiro.domain.UserRole;
 import com.example.springbootshiro.service.BaseService;
+import com.example.springbootshiro.service.RoleMenuService;
 import com.example.springbootshiro.service.RoleService;
+import com.example.springbootshiro.service.UserRoleService;
+import com.example.springbootshiro.util.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
+import tk.mybatis.mapper.entity.Example;
 
+import java.util.Arrays;
+import java.util.Date;
 import java.util.List;
 
 /**
@@ -21,6 +30,14 @@ public class RoleServiceImpl extends BaseService<RoleInfo> implements RoleServic
     @Autowired
     private RoleMapper roleMapper;
 
+    @Autowired
+    private RoleMenuMapper roleMenuMapper;
+
+    @Autowired
+    private UserRoleService userRoleService;
+
+    @Autowired
+    private RoleMenuService roleMenuService;
 
     @Override
     public List<RoleInfo> findUserRole(String userName) {
@@ -29,26 +46,67 @@ public List<RoleInfo> findUserRole(String userName) {
 
     @Override
     public List<RoleInfo> findAllRole(RoleInfo roleInfo) {
-        return null;
+        Example example = new Example(RoleInfo.class);
+        if (!StringUtils.isNullorEmpty(roleInfo.getRoleName())){
+            example.createCriteria().andCondition("role_name=", roleInfo.getRoleName());
+        }
+        example.setOrderByClause("create_time");
+        return selectByExample(example);
     }
 
     @Override
     public RoleInfo findByName(String roleName) {
-        return null;
+        Example example = new Example(RoleInfo.class);
+        example.createCriteria().andCondition("lower(role_name)=", roleName.toLowerCase());
+        List<RoleInfo> list = this.roleMapper.selectByExample(example);
+        if (list.size() == 0){
+            return null;
+        }else {
+            return list.get(0);
+        }
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void addRole(RoleInfo roleInfo, Long[] menuIds) {
+        roleInfo.setCreateTime(new Date());
+        this.save(roleInfo);
+        for (Long menuId: menuIds){
+            RoleMenu roleMenu = new RoleMenu();
+
+            roleMenu.setMenuId(menuId);
+            roleMenu.setRoleId(roleInfo.getRoleId());
 
+            this.roleMenuMapper.insert(roleMenu);
+        }
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void updateRole(RoleInfo roleInfo, Long[] menuIds) {
+        roleInfo.setModifyTime(new Date());
+        this.roleMapper.updateByPrimaryKeySelective(roleInfo);
+        Example example = new Example(RoleMenu.class);
+        example.createCriteria().andCondition("role_id=", roleInfo.getRoleId());
+        this.roleMenuMapper.deleteByExample(example);
 
+        for (Long menuId: menuIds){
+            RoleMenu roleMenu = new RoleMenu();
+
+            roleMenu.setMenuId(menuId);
+            roleMenu.setRoleId(roleInfo.getRoleId());
+            this.roleMenuMapper.insert(roleMenu);
+        }
     }
 
     @Override
+    @Transactional(propagation = Propagation.REQUIRED, readOnly = false)
     public void deleteRoles(String roleIds) {
+        List<String> list = Arrays.asList(roleIds.split(","));
+        this.batchDelete(list, "roleId", RoleInfo.class);
+
+        this.roleMenuService.deleteRoleMenusByRoleId(roleIds);
+        this.userRoleService.deleteUserRoleByRoleId(roleIds);
 
     }
 }
diff --git a/spring-boot-shiro/src/main/java/com/example/springbootshiro/util/TreeUtils.java b/spring-boot-shiro/src/main/java/com/example/springbootshiro/util/TreeUtils.java
@@ -12,20 +12,20 @@
  */
 public class TreeUtils {
 
-    public static <T> Tree<T> build(List<Tree<T>> nodes){
+    public static <T> Tree<T> build(List<Tree<T>> nodes) {
         if (nodes == null) {
             return null;
         }
 
         List<Tree<T>> topNodes = new ArrayList<>();
-        for (Tree<T> children: nodes){
+        for (Tree<T> children : nodes) {
             String pid = children.getParentId();
             if (pid == null || "0".equals(pid)) {
                 topNodes.add(children);
                 continue;
             }
 
-            for (Tree<T> parent: nodes){
+            for (Tree<T> parent : nodes) {
                 String id = parent.getId();
                 if (id != null && id.equals(pid)) {
                     parent.getChildren().add(children);
@@ -48,6 +48,33 @@ public static <T> Tree<T> build(List<Tree<T>> nodes){
         root.setState(state);
 
         return root;
+    }
+
+    public static <T> List<Tree<T>> buildList(List<Tree<T>> nodes, String idParam) {
+        if (nodes == null) {
+            return null;
+        }
+
+        List<Tree<T>> topNodes = new ArrayList<>();
+        for (Tree<T> children : nodes) {
+            String pid = children.getParentId();
+            if (pid == null || idParam.equals(pid)) {
+                topNodes.add(children);
+                continue;
+            }
+            for (Tree<T> parent : nodes) {
+                String id = parent.getId();
+                if (id != null && id.equals(pid)) {
+                    parent.getChildren().add(children);
+                    children.setHasParent(true);
+                    parent.setHasChildren(true);
+                    continue;
+                }
+            }
 
+        }
+        return topNodes;
     }
+
+
 }
