Преглед на файлове

店铺列表新增排序字段

lyhzzz преди 3 години
родител
ревизия
e41f56ee4d

+ 3 - 2
src/main/java/com/cdf/business/LoginService.java

@@ -82,7 +82,8 @@ public class LoginService {
     }
 
     public void logout(String token) {
-        Integer userId = JwtUtil.getId(token);
-        userTokenService.removeById(userId);
+        LambdaQueryWrapper<UserToken> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(UserToken::getLoginToken,token);
+        userTokenService.remove(wrapper);
     }
 }

+ 7 - 5
src/main/java/com/cdf/entity/UserToken.java

@@ -1,9 +1,7 @@
 package com.cdf.entity;
 
-import com.baomidou.mybatisplus.annotation.TableField;
-import com.baomidou.mybatisplus.annotation.TableId;
-import com.baomidou.mybatisplus.annotation.TableLogic;
-import com.baomidou.mybatisplus.annotation.TableName;
+import com.baomidou.mybatisplus.annotation.*;
+
 import java.io.Serializable;
 import java.time.LocalDateTime;
 import lombok.Getter;
@@ -24,7 +22,11 @@ public class UserToken implements Serializable {
 
     private static final long serialVersionUID = 1L;
 
-    @TableId("user_id")
+
+    @TableId(value = "id", type = IdType.AUTO)
+    private Integer id;
+
+    @TableField("user_id")
     private Integer userId;
 
     @TableField("login_token")

+ 6 - 2
src/main/java/com/cdf/interceptor/BackUserInterceptor.java

@@ -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;
 		}