caimei-mall-admin/src/main/java/com/caimei/modules/shiro/controller/ShiroController.java

package com.caimei.modules.shiro.controller;


import com.caimei.modules.shiro.entity.CmMallAdminUser;
import com.caimei.modules.shiro.service.ShiroService;
import com.caimei.utils.JsonModel;
import com.caimei.utils.MD5Util;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;


@RestController
@RequestMapping("/user")
public class ShiroController {

    @Autowired
    private ShiroService shiroService;


    /**
     * 登录
     */
    @GetMapping("/login")
    public JsonModel login(String account, String password, Integer organizeID) throws Exception {
        JsonModel jsonModel = JsonModel.newInstance();
        //用户信息
        CmMallAdminUser user = shiroService.findByUsername(account, organizeID);
        //账号不存在、密码错误
        if (user == null || !user.getPassword().equals(MD5Util.MD5(password))) {
            return jsonModel.error("账号或密码有误");
        } else {
            //生成token
            String token = null;
            try {
                token = shiroService.createToken(user);
            } catch (Exception e) {
                return jsonModel.error("生成token失败");
            }
            return jsonModel.success(token);
        }
    }

    @GetMapping("/info")
    public JsonModel info(String token) {
        String[] tokens = token.split("#,#");
        //用户信息
        CmMallAdminUser user = shiroService.findByUserId(Integer.valueOf(tokens[1]));
        return JsonModel.newInstance().success(user);
    }
}