关键代码
@Override
protected boolean onAccessDenied(ServletRequest request, ServletResponse response, Object mappedValue) throws Exception {
try {
if (response instanceof HttpServletResponse){
((HttpServletResponse) response).setStatus(org.springframework.http.HttpStatus.UNAUTHORIZED.value());
}
response.setContentType("application/json;charset=utf-8");
response.getWriter().write(JSON.toJSONString(Result.<String>build()
.withCode(HttpStatus.UNAUTHORIZED.value())
.withMessage("未实名认证")));
} catch (IOException e) {
return false;
}
return false;
}
shiro完整代码
package com.ddb.app.shiro.filter;
import com.alibaba.fastjson.JSON;
import com.ddb.app.shiro.param.JwtToken;
import com.ddb.app.utils.IpUtils;
import com.ddb.common.param.Result;
import com.ddb.common.user.LoginUser;
import com.ddb.common.utils.JwtHelper;
import com.ddb.common.utils.LoginUserUtils;
import org.apache.shiro.web.filter.authc.BasicHttpAuthenticationFilter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpStatus;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
/**
* @author Duhuafei
* @date 11/27/2018
*/
public class JwtFilter extends BasicHttpAuthenticationFilter {
private static final Logger logger = LoggerFactory.getLogger(JwtFilter.class);
/**
* 判断用户是否想要登入。
* 检测header里面是否包含Authorization字段即可
*/
@Override
protected boolean isLoginAttempt(ServletRequest request, ServletResponse response) {
String authorization = super.getAuthzHeader(request);
LoginUser loginUser = JwtHelper.checkAndGetUser(authorization);
if (null == loginUser) {
loginUser = new LoginUser();
loginUser.setAuthenticated(false);
}
HttpServletRequest req = (HttpServletRequest) request;
//防止ip地址为多个时,获取第一个ip地址
loginUser.setIp(IpUtils.getIpAddress(req));
loginUser.setUri(req.getRequestURI());
loginUser.setFrom(req.getHeader("From"));
LoginUserUtils.setUser(loginUser);
return loginUser.isAuthenticated();
}
/**
*
*/
@Override
protected boolean executeLogin(ServletRequest request, ServletResponse response) {
String authorization = super.getAuthzHeader(request);
JwtToken token = new JwtToken(authorization.replaceFirst("Bearer ", ""));
// 提交给realm进行登入,如果错误他会抛出异常并被捕获
getSubject(request, response).login(token);
logger.debug("登录成功");
// 如果没有抛出异常则代表登入成功,返回true
return true;
}
/**
* 这里我们详细说明下为什么最终返回的都是true,即允许访问
* 例如我们提供一个地址 GET /article
* 登入用户和游客看到的内容是不同的
* 如果在这里返回了false,请求会被直接拦截,用户看不到任何东西
* 所以我们在这里返回true,Controller中可以通过 subject.isAuthenticated() 来判断用户是否登入
* 如果有些资源只有登入用户才能访问,我们只需要在方法上面加上 @RequiresAuthentication 注解即可
* 但是这样做有一个缺点,就是不能够对GET,POST等请求进行分别过滤鉴权(因为我们重写了官方的方法),但实际上对应用影响不大
*/
@Override
protected boolean isAccessAllowed(ServletRequest request, ServletResponse response, Object mappedValue) {
if (isLoginAttempt(request, response)) {
try {
executeLogin(request, response);
} catch (Exception e) {
logger.error("登录错误:", e);
return false;
}
}
return true;
}
@Override
protected boolean onAccessDenied(ServletRequest request, ServletResponse response, Object mappedValue) throws Exception {
try {
if (response instanceof HttpServletResponse){
((HttpServletResponse) response).setStatus(org.springframework.http.HttpStatus.UNAUTHORIZED.value());
}
response.setContentType("application/json;charset=utf-8");
response.getWriter().write(JSON.toJSONString(Result.<String>build()
.withCode(HttpStatus.UNAUTHORIZED.value())
.withMessage("未实名认证")));
} catch (IOException e) {
return false;
}
return false;
}
/**
* 操作完成后清除登录信息
*/
@Override
public void afterCompletion(ServletRequest request, ServletResponse response, Exception exception) throws Exception {
logger.debug("清除用户信息");
super.afterCompletion(request, response, exception);
LoginUserUtils.clear();
}
}