快捷導(dǎo)航

SpringBoot實現(xiàn)token登錄的示例代碼

更新時間：2022年03月25日 10:27:56 作者：C'z?x

在進行登錄驗證時，我們需要session或cookie會話進行驗證，當我們脫離瀏覽器用app等向服務(wù)端發(fā)請求就沒有session和cookie機制，這時我們就需要使用token令牌進行登錄驗證，本文就詳細的介紹一下，感興趣的可以了解一下

為什么引入token機制?

在進行登錄驗證時，我們需要session或cookie會話進行驗證，客戶端包括瀏覽器、app、微信小程序、公眾號，只有瀏覽器有session和cookie機制，當我們脫離瀏覽器用app等向服務(wù)端發(fā)請求就沒有session和cookie機制，這時我們就需要使用token令牌進行登錄驗證。

代碼實現(xiàn)

先建個util包，并創(chuàng)建TokenUtil類用于生成token

TokenUtil類代碼

package com.qcby.util;
 
import com.qcby.entity.User;
 
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
 
public class TokenUtil {
    /**
     * 創(chuàng)建map用于存儲所有的令牌
     *
     * token  -  User
     */
    private static Map<String, User> tokenMap=new HashMap<>();
 
    /**
     * 生成token，存儲token-user對應(yīng)關(guān)系
     * 返回token令牌
     * @param user
     * @return
     */
    public static String generateToken(User user){
        //生成唯一不重復(fù)的字符串
        String token = UUID.randomUUID().toString();
        tokenMap.put(token,user);
        return token;
    }
 
    /**
     * 驗證token是否合法
     * @param token
     * @return
     */
    public static boolean verify(String token){
        return tokenMap.containsKey(token);
    }
 
    /**
     * 根據(jù)token獲取用戶信息
     * @param token
     * @return
     */
    public static User getUser(String token){
        return tokenMap.get(token);
    }
}

LoginInterceptor類代碼

package com.qcby.interceptor;
 
import com.qcby.util.TokenUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
 
public class LoginInterceptor  implements HandlerInterceptor {
 
   
    //Controller邏輯執(zhí)行之前
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        System.out.println("preHandle....");
        String uri = request.getRequestURI();
        System.out.println("當前路徑"+uri);
 
        /**
         * HandlerMethod=>Controller中標注@RequestMapping的方法
         *  需要配置靜態(tài)資源不攔截時，添加這塊邏輯  => 前后端分離項目
         */
        if (!(handler instanceof HandlerMethod)) {
            return true;
        }
 
        String token=request.getHeader("qcby-token");
 
        if (!TokenUtil.verify(token)) {
            // 未登錄跳轉(zhuǎn)到登錄界面
            throw  new RuntimeException("no login!");
        } else {
            return true;
        }
    }
 
    //Controller邏輯執(zhí)行完畢但是視圖解析器還未進行解析之前
    @Override
    public void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) throws Exception {
        System.out.println("postHandle....");
    }
 
    //Controller邏輯和視圖解析器執(zhí)行完畢
    @Override
    public void afterCompletion(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, Exception e) throws Exception {
        System.out.println("afterCompletion....");
    }
}

實現(xiàn)類

@RestController
@RequestMapping("user")
public class UserController { 
    @Autowired
    private UserService userService;
    @ApiOperation("用戶登錄接口")
    @RequestMapping(value="login",method = {RequestMethod.GET,RequestMethod.POST})
    public Map<String,Object>login(User user){
        Map<String,Object> map=new HashMap<>();
        map.put("code",0);
        if(StringUtils.isEmpty(user.getUsername())||StringUtils.isEmpty(user.getPassword())){
            map.put("msg","用戶或密碼為空！");
            return map;
        }
        QueryWrapper<User> queryWrapper=new QueryWrapper<>();
        queryWrapper.eq("username",user.getUsername())
                .eq("password",user.getPassword());
        User user1=userService.getOne(queryWrapper);
        if(user1!=null){
 
            String token= TokenUtil.generateToken(user1);
            map.put("cod",1);
            map.put("data",user1);
            map.put("token",token);
        }else {
            map.put("msg","用戶名或密碼錯誤!");
        }
        return map;
    }
}

攔截器類可以參考這篇文章，這個token驗證就是在攔截器上改動的。

SpringBoot攔截器實現(xiàn)登錄攔截

需要有postman驗證