实现持久化登录

Author: wang-qa

README.md

@@ -20,18 +20,22 @@
 > git commit -m "备注" # 提交到本地仓库
 > git push origin master # 推送到名为origin的远程仓库的master分支
 > git pull origin master # 拉取名为origin的远程仓库的master分支
+> git commit --amend --no-edit # 提交 > 追加文件到上次本地提交 不需要改备注
+> git push -f origin master # 本地强制提交
 > ```
 
 ## 进度
 > 1. `GitHub`登录之调用 `authorize`
 > 2. `GitHub`登录之获取 用户信息`code``token``UserName`
 > 3. 配置文件`application.properties`新增配置参数 GitHub.client 信息
 > 4. 使用 `MyBaits` 链接数据库并插入数据
+> 5. 实现持久化登录 服务器下发 user_token 并在数据库查询
 >
 
 ## 数据库脚本
 
 ```sql
+table For H2
 -- ----------------------------
 -- Table structure for User 
 -- ----------------------------
@@ -47,3 +51,4 @@ CREATE TABLE "PUBLIC"."USER"(
 
 ```
 
+

src/main/java/life/majiang/community/controller/AuthorizeController.java

@@ -1,6 +1,6 @@
 package life.majiang.community.controller;
 
-/*
+/**
  * 获取code
  */
 
@@ -15,7 +15,8 @@
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 
-import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletResponse;
 import java.util.UUID;
 
 @Controller // 把类作为路由Api的承载者
@@ -33,38 +34,45 @@ public class AuthorizeController {
     @Value("${github.client.uri}")
     private String clientUri;
 
-    @Autowired // 把 mapper 对象放入容器内
+    @Autowired // 把 Usermapper 对象放入容器内
     private UserMapper userMapper;
 
     @GetMapping("/callback") // 登录成功后返回到登录页
     // 接收返回后的参数 code, state
     public String callback(@RequestParam(name = "code") String code,
                            @RequestParam(name = "state") String state,
-                           HttpServletRequest request) {
+                           // HttpServletRequest request,//不需要，可删除
+                           HttpServletResponse response) {
         AccessTokenDTO accessTokenDTO = new AccessTokenDTO();
         accessTokenDTO.setClient_id(clientID);
         accessTokenDTO.setClient_secret(clientSecret);
         accessTokenDTO.setCode(code);
         accessTokenDTO.setRedirect_uri(clientUri);
         accessTokenDTO.setState(state);
         String accessToken = githubProvider.getAccess_token(accessTokenDTO);
-        GithubUser githubUser = githubProvider.getUser(accessToken);
+        GithubUser githubUser = githubProvider.getUser(accessToken); // 使用 GitHub 登录
 //        System.out.println("用户昵称 >>> " + user.getName()); // 输出 User昵称
 
-//        登录跳转
+        /** 登录跳转 */
         if (githubUser != null) {
+            /** 使用 GitHub 登录成功 */
             User user = new User();
-            user.setToken(UUID.randomUUID().toString()); // 生成唯一的标识码
+            // 获取用户信息生成>>>用户 token
+            String token = UUID.randomUUID().toString();
+            /** token 放入 User 对象中并存储到数据库 */
+            user.setToken(token); // 生成唯一的标识码
             user.setName(githubUser.getName());
             user.setAccountID(String.valueOf(githubUser.getId()));
             user.setGmtCreate(System.currentTimeMillis());
             user.setGmtModified(user.getGmtCreate());
             System.out.println("--->>> usermapper >>>--- " + userMapper);
             System.out.println("--->>> insertuser >>>--- " + user);
             userMapper.insert(user);
-
-            // 登录成功 >>> 写 cookies 和 session
-            request.getSession().setAttribute("user", githubUser); // session存入 user对象
+            /** Cookie 写入 */
+            // 自动写入  服务器生成 Token 放入 Cookie
+            response.addCookie(new Cookie("token", token));
+            // 手动写入  登录成功 >>> 写 cookies 和 session
+//            request.getSession().setAttribute("user", githubUser); // session存入 user对象
             return "redirect:/";
         } else {
             // 登录失败 >>> 重新登录 原因
@@ -73,6 +81,3 @@ public String callback(@RequestParam(name = "code") String code,
     }
 }
 
-
-
-

src/main/java/life/majiang/community/controller/IndexController.java

@@ -4,13 +4,42 @@
  * 定义自己的 index
  */
 
+import life.majiang.community.mapper.UserMapper;
+import life.majiang.community.model.User;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.GetMapping;
 
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletRequest;
+
 @Controller
 public class IndexController {
+
+    @Autowired // 注入 UserMapper
+    private UserMapper userMapper;
+
+    /**
+     * 访问首页的时候 循环看所有的Cookie
+     * 找到 Cookie 为 token 的值
+     * 用这个值去数据库中查找是否存在
+     * 如果有 把 user放入 session
+     * 前端通过页面数据判断是否登录
+     */
     @GetMapping("/")
-    public String index(){
+    public String index(HttpServletRequest request) {
+        Cookie[] cookies = request.getCookies(); // 获取用户 cookie
+        for (Cookie cookie : cookies) {
+            if (cookie.getName().equals("token")) { // 检查 cookies_key是否为 token
+                String token = cookie.getValue();
+                User user = userMapper.findByToken(token);
+                // 如果user != null 写入session
+                if (user != null) {
+                    request.getSession().setAttribute("user", user);
+                }
+                break; // 命中后结束循环
+            }
+        }
         return "index";
     }
 }

src/main/java/life/majiang/community/mapper/UserMapper.java

@@ -3,12 +3,18 @@
 import life.majiang.community.model.User;
 import org.apache.ibatis.annotations.Insert;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+import org.apache.ibatis.annotations.Select;
 
 /*
  * 引入数据库
+ * 类会自动放入 不是的需要引入 @Param 注解
  */
 @Mapper
 public interface UserMapper {
     @Insert("insert into user (name, account_ID, token, gmt_Create, gmt_Modified) values (#{name}, #{accountID}, #{token}, #{gmtCreate}, #{gmtModified})")
     void insert(User user);
+
+    @Select("select * from USER where token = #{token}")
+    User findByToken(@Param("token") String token);
 }

src/main/java/life/majiang/community/model/User.java

@@ -1,7 +1,9 @@
 package life.majiang.community.model;
 
-// 数据库表格式
-/*create table USER
+
+/**
+ * 数据库User表脚本
+    create table USER
         (
         ID           INT auto_increment,
         ACCOUNT_ID   VARCHAR,
@@ -11,7 +13,9 @@ TOKEN        CHAR(36),
         GMT_MODIFIED BIGINT,
         constraint TABLE_NAME_PK
         primary key (ID)
-        );*/
+        );
+*/
+
 public class User {
     private Integer id; // 用户自增id
     private String name; // username

src/main/resources/application.properties

@@ -1,4 +1,4 @@
-# 更改server端口
+# server_web¶Ë¿Ú
 server.port=8080
 # GitHubÅäÖÃ
 github.client.id=639f4c8d6441c6136b44