通用後臺管理系統(9)-編寫首頁接口、實現、控制器

接口

package com.sundablog.service.backend.index;

import java.util.List;

import com.sundablog.result.MenuResult;

/**
 * 首頁接口定義類
 * @ClassName:  IndexService   
 * @Description:TODO(這裏用一句話描述這個類的作用)   
 * @author: 噠噠 
 * @date:   2018年3月30日 下午5:28:31   
 *     
 * @Copyright: 2018 www.sundablog.com Inc. All rights reserved.
 */
public interface IndexService {
    /**
     * 獲取菜單
     * @Title: getMenuResult   
     * @Description: TODO(這裏用一句話描述這個方法的作用)   
     * @param: @param userId
     * @param: @return      
     * @return: List<MenuResult>      
     * @throws
     */
     List<MenuResult> getMenuResult(Integer userId);
}

實現

package com.sundablog.service.impl.backend.index;

import java.util.ArrayList;
import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.sundablog.pojo.AdminPermission;
import com.sundablog.pojo.AdminRolePermission;
import com.sundablog.pojo.AdminUserRole;
import com.sundablog.result.MenuResult;
import com.sundablog.service.backend.index.IndexService;
import com.sundablog.service.backend.system.upms.permissions.PermissionsService;
import com.sundablog.service.backend.system.upms.role.RoleService;
import com.sundablog.service.backend.system.upms.user.UserService;

/**
 * 首頁接口實現類
 * @ClassName:  IndexServiceImpl   
 * @Description:TODO(這裏用一句話描述這個類的作用)   
 * @author: 噠噠 
 * @date:   2018年3月30日 下午5:30:13   
 *     
 * @Copyright: 2018 www.sundablog.com Inc. All rights reserved.
 */
@Service
public class IndexServiceImpl implements IndexService {
    
    @Autowired
    private RoleService roleService;

    @Autowired
    private UserService userService;
    
    @Autowired
    private PermissionsService permissionsService;
    
    @Override
    public List<MenuResult> getMenuResult(Integer userId) {
        //用戶角色關聯的信息
        AdminUserRole userRole = userService.selectByUserId(userId);
        
        List<AdminRolePermission> adminRolePermissions = roleService.selectRolePermissionByRoleId(userRole.getRoleId());
        
        List<MenuResult> menuResults = new ArrayList<>();
        for (AdminRolePermission adminRolePermission : adminRolePermissions) {
            AdminPermission adminPermission = permissionsService.selectPermissionsByPermissionId(adminRolePermission.getPermissionId());
            if (adminPermission.getType() != 3) {
                MenuResult menuResult = new MenuResult();
                menuResult.setId(adminPermission.getPermissionId().toString());
                menuResult.setPid(adminPermission.getPid().toString());
                menuResult.setTitle(adminPermission.getName());
                menuResult.setIcon(adminPermission.getIcon());
                menuResult.setUrl(adminPermission.getUri());
                menuResult.setFont("larry-icon");
                menuResults.add(menuResult);
            }
        }
        List<MenuResult> trees = new ArrayList<MenuResult>(); 
        for (MenuResult menuResult : menuResults) {
            if ("0".equals(menuResult.getPid())) {  
                trees.add(findChildren(menuResult,menuResults));  
            }  
        }
        return trees;
    }
    
    /**
     * 遞歸查找子節點
     * 
     * @param treeNodes
     * @return
     */
    public static MenuResult findChildren(MenuResult treeNode, List<MenuResult> treeNodes) {
        for (MenuResult it : treeNodes) {
            if (treeNode.getId().equals(it.getPid())) {
                if (treeNode.getChildren() == null) {
                    treeNode.setChildren(new ArrayList<MenuResult>());
                }
                treeNode.getChildren().add(findChildren(it, treeNodes));
            }
        }
        return treeNode;
    }
    
}

控制器

package com.sundablog.controller.backend.login;

import java.io.IOException;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.DisabledAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import com.sundablog.pojo.AdminUser;
import com.sundablog.result.BaseResult;
import com.sundablog.service.backend.system.upms.user.UserService;
import com.sundablog.utlis.CaptchaUtil;

import com.sundablog.utlis.RedisUtil;

import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.digest.DigestUtil;

/**
 * 登錄控制器
 * @ClassName:  LoginController   
 * @Description:登錄控制器  
 * @author: 噠噠 
 * @date:   2018年3月18日 下午12:45:41   
 *     
 * @Copyright: 2018 www.sundablog.com Inc. All rights reserved.
 */
@Controller
public class LoginController {
    
    @Autowired
    private UserService userService;
    
    /**
     * 登入界面顯示
     * @Title: login   
     * @Description: TODO(這裏用一句話描述這個方法的作用)   
     * @param: @return      
     * @return: String      
     * @throws
     */
    @RequestMapping(value= {"/","/login"})
    public String login() {
        return "/login/login";
    }
    
    /**
     * 驗證碼   
     * @Title: code   
     * @Description: TODO(這裏用一句話描述這個方法的作用)   
     * @param: @param request
     * @param: @param response
     * @param: @param session
     * @param: @throws IOException      
     * @return: void      
     * @throws
     */
    @RequestMapping("/captcha")
    public void code(HttpServletRequest request, HttpServletResponse response) throws IOException {

        // 設置響應的類型格式爲圖片格式
        response.setContentType("image/jpeg");
        response.setHeader("Pragma", "no-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.setDateHeader("Expires", 0);
        // 自定義參數
        CaptchaUtil code = new CaptchaUtil(156, 38, 4, 4);
        request.getSession().setAttribute("validateCode", code.getCode());
        code.write(response.getOutputStream());
        System.err.println(request);
    
    }
    
    /**
     * 登錄
     * @Title: loginClick   
     * @Description: TODO(這裏用一句話描述這個方法的作用)   
     * @param: @param userName
     * @param: @param password
     * @param: @param verificationCode
     * @param: @return
     * @param: @throws DisabledAccountException      
     * @return: BaseResult      
     * @throws
     */
    @RequestMapping("/loginClick")
    @ResponseBody
    public BaseResult loginClick(String userName, String password, String verificationCode,HttpServletRequest request)
            throws DisabledAccountException {
        String captcha = (String)request.getSession().getAttribute("validateCode");
        if (StrUtil.isEmpty(verificationCode)) {
            return BaseResult.build(209, "驗證碼錯誤");
        } else {
            if (captcha.equals(verificationCode)) {
                /**
                 * 獲得當前用戶對象,狀態爲“未認證”
                 */
                Subject subject = SecurityUtils.getSubject();
                AdminUser adminUser = userService.selectAdminUserByUserName(userName);
                if (1 == adminUser.getLocked().intValue()) {
                    return BaseResult.build(202, "賬戶以及被鎖定");
                }
                AuthenticationToken token = new UsernamePasswordToken(userName,
                        DigestUtil.md5Hex(password + adminUser.getSalt()));// 創建用戶名密碼令牌對象
                
                try {
                    subject.login(token);
                    return BaseResult.ok();
                } catch (AuthenticationException e) {
                    return BaseResult.build(203, "用戶名密碼錯誤");
                }
                
            } else {
                //驗證碼錯誤
                return BaseResult.build(204, "驗證碼錯誤");
            }
        }
    }
    
    /**
     * 退出
     * @Title: quit   
     * @Description: TODO(這裏用一句話描述這個方法的作用)   
     * @param: @return      
     * @return: BaseResult      
     * @throws
     */
    @RequestMapping("/quit")
    @ResponseBody
    public BaseResult quit() {
        Subject subject = SecurityUtils.getSubject();
        try {
            subject.logout();
            return BaseResult.ok();
        } catch (Exception e) {
            return BaseResult.build(201, "退出失敗");
        }
    }
    
}