对于一个小白来说,就比如我,之前学习spring 的aop,属实是有点蒙圈,不知道这玩意能干嘛,记录日志这是我唯一想到的,然后就是注解的运用,也是一知半解。
在前几个项目中,我的登录判断都是通过全局配置,判断请求中是否有token,这样虽好,但是也不好,因为把一些不必要的接口也做了登录判断,后面也是在一些博客中看到了注解的方式实现登录判断,这里就记录一下
首先是定义一个注解:Login
import java.lang.annotation.*;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Login {
// 这里我是使用的CLIENT和ADMIN来作为权限符的,也可以自己换一下
String issuer() default "CLIENT";
}
然后就是通过AOP去对注解进行判断
import com.cl.blog.annotation.Login;
import com.cl.blog.common.Const;
import com.cl.blog.exception.WebException;
import com.cl.blog.util.JwtUtil;
import io.jsonwebtoken.Claims;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@Component
@Slf4j
public class AuthInterceptor implements HandlerInterceptor {
@Resource
JwtUtil jwtUtil;
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
Login annotation;
// 判断当前请求是否为方法,如果是,就对其进行注解判断
if (handler instanceof HandlerMethod) {
annotation = ((HandlerMethod) handler).getMethodAnnotation(Login.class);
} else {
return true;
}
if (annotation == null) {
return true;
}
String token = request.getHeader("token");
String issuer = annotation.issuer();
System.out.println("--->"+token);
// 判断token是否合法
if (!StringUtils.hasText(token)) {
throw new WebException("请先登录");
}
if (!StringUtils.hasText(issuer)) {
throw new WebException("请先登录");
}
// 判断token内容是否存在,且是否超时
Claims claim = jwtUtil.getClaimByToken(token, Const.GRADE_USER);
if (claim == null) {
claim = jwtUtil.getClaimByToken(token, Const.GRADE_ADMIN);
}
if (claim == null || jwtUtil.isTokenExpired(claim.getExpiration())) {
throw new WebException("请先登录");
}
if (!annotation.issuer().equals(claim.getIssuer()) && (annotation.issuer().equals(Const.GRADE_ADMIN))) {
throw new WebException("抱歉,权限不足");
}
// 将用户信息存储到session
request.getSession().setAttribute(Const.USER_KEY, Long.parseLong(claim.getSubject()));
return true;
}
}
当然,我这里也做了一个权限的判断,例如哪些是客户端能访问的,也就是普通用户访问的,哪些是管理员登录访问的,当然有些接口可以共用,就可以把参数改为数组的形式,这里就不做修改了
这里还用到了一个工具类,我也贴出来了
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.Date;
@Slf4j
@Data
@Component
@ConfigurationProperties("jwt")
public class JwtUtil {
private String secret;
private long expire;
/**
* 生成token
*/
public String generateToken(long userId,String issuer){
Date date = new Date();
// 过期时间
Date expireDate = new Date(date.getTime() + expire * 1000);
return Jwts.builder()
.setHeaderParam("typ","JWT")
.setSubject(userId + "")
.setIssuedAt(date)
.setIssuer(issuer)
.setExpiration(expireDate)
.signWith(SignatureAlgorithm.HS512,secret)
.compact();
}
/**
* 解析token
* @param token
* @return
*/
public Claims getClaimByToken(String token,String issuer){
try {
Claims claims = Jwts.parser()
.setSigningKey(secret)
.parseClaimsJws(token)
.getBody();
if (!issuer.equals(claims.getIssuer())|| issuer==null) {
return null;
}
return claims;
}catch (Exception e){
log.debug("validDate is token error------>" + e);
return null;
}
}
/**
* 判断token是否过期
* @param expiration
* @return true 过期
*/
public boolean isTokenExpired(Date expiration){
return expiration.before(new Date());
}
}
今天分享就这些了!!!