|
@@ -2,6 +2,7 @@ package com.cdf.interceptor;
|
|
|
|
|
|
|
|
|
import cn.hutool.jwt.JWTUtil;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
import com.cdf.common.ResultCode;
|
|
|
import com.cdf.common.ResultData;
|
|
|
import com.cdf.entity.User;
|
|
@@ -18,6 +19,7 @@ import org.springframework.web.servlet.HandlerInterceptor;
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
import java.io.IOException;
|
|
|
+import java.util.List;
|
|
|
|
|
|
@Component
|
|
|
public class BackUserInterceptor implements HandlerInterceptor {
|
|
@@ -49,8 +51,10 @@ public class BackUserInterceptor implements HandlerInterceptor {
|
|
|
public boolean checkUser(String token,HttpServletRequest request, HttpServletResponse response){
|
|
|
Integer userId = JwtUtil.getId(token);
|
|
|
User user = userService.getById(userId);
|
|
|
- UserToken userToken = userTokenService.getById(userId);
|
|
|
- if(userToken == null ){
|
|
|
+ LambdaQueryWrapper<UserToken> wrapper = new LambdaQueryWrapper<>();
|
|
|
+ wrapper.eq(UserToken::getLoginToken,token);
|
|
|
+ List<UserToken> list = userTokenService.list(wrapper);
|
|
|
+ if(list == null || list.size() <= 0){
|
|
|
this.needLogin(request, response);
|
|
|
return false;
|
|
|
}
|