【黑马点评日记02】：Session+ThreadLocal实现短信登录

【用户请求】GET /user/order
    ↓
① 请求进入 Tomcat（携带 Cookie: JSESSIONID=ABC123）
    ↓
② Spring MVC 查找对应的拦截器
    ↓
③ 拦截器 preHandle 方法执行
    ↓
④ 从 Session 获取用户信息
    ↓
⑤ 判断是否已登录
    ↓
┌─────────────────┴─────────────────┐
↓                                   ↓
【已登录】                          【未登录】
↓                                   ↓
存入 ThreadLocal                    返回 401 状态码
↓                                   ↓
放行到 Controller                   拦截，不进入 Controller
↓                                   ↓
执行业务逻辑                         响应返回前端
↓
返回数据给前端

【请求到达】携带 Cookie: JSESSIONID=ABC123
    ↓
【Tomcat 根据 JSESSIONID 找到 Session】
    Session 内容: {user: User对象(张三)}
    ↓
【拦截器 preHandle】
    session.getAttribute("user") → 拿到 User(张三)
    UserHolder.setUser(张三)  ← 存入 ThreadLocal
    ↓
【Controller 方法】
    UserHolder.getUser() → 拿到 User(张三)  ← 从 ThreadLocal 取
    处理业务...
    ↓
【Service 层】
    UserHolder.getUser() → 仍然拿到 User(张三)  ← 同一个线程
    处理业务...
    ↓
【拦截器 afterCompletion】
    UserHolder.remove()  ← 清理 ThreadLocal，防止内存泄漏
    ↓
【响应返回给客户端】

/**
     * 发送手机验证码
     */
    @PostMapping("code")
    public Result sendCode(@RequestParam("phone") String phone, HttpSession session) {
        // 发送短信验证码并保存验证码

        return userService.sendCode(phone,session);
    }

/**
     * 发送验证码
     * @param phone
     * @param session
     * @return
     */
    public Result sendCode(String phone, HttpSession session) {
        //校验手机号
        if(RegexUtils.isPhoneInvalid(phone)){
            return Result.fail("手机号格式错误");
        }
        //生成验证码
        String code = RandomUtil.randomNumbers(6);
        //保存验证码
        session.setAttribute("code",code);
        //发送验证码（模拟）
        System.out.println("发送验证码成功："+code);
        return Result.ok();
    }

  /**
     * 登录功能
     * @param loginForm 登录参数，包含手机号、验证码；或者手机号、密码
     */
    @PostMapping("/login")
    public Result login(@RequestBody LoginFormDTO loginForm, HttpSession session){
        // 实现登录功能
        return userService.login(loginForm,session);
    }

 /**
     * 登录和注册功能
     * @param loginForm
     * @param session
     * @return
     */
    public Result login(LoginFormDTO loginForm, HttpSession session) {
        //校验手机号
        if(RegexUtils.isPhoneInvalid(loginForm.getPhone())){
            return Result.fail("手机号格式错误");
        }
        //校验验证码
        String cacheCode = (String)session.getAttribute("code");
        String code= loginForm.getCode();
        if(cacheCode==null||!cacheCode.equals(code)){
            return Result.fail("验证码错误");
        }
        //查询用户
        User user = query().eq("phone", loginForm.getPhone()).one();
        //判断用户是否存在
        if(user==null){
            //不存在，创建新用户并保存
           user= createUserWithPhone(loginForm.getPhone());
        }
        //保存用户
        session.setAttribute("user",user);


        return null;
    }

    private User createUserWithPhone(String phone) {
        //创建用户
        User user=new User();
        user.setPhone(phone);
        user.setNickName(RandomUtil.randomString(10));
        //保存用户
        save(user);
        return  user;
    }

query().eq("phone", loginForm.getPhone()).one();

save(user);

java

save(user);  // 把 user 对象插入到数据库

sql

INSERT INTO user (phone, nick_name, create_time, update_time) 
VALUES ('13800138000', 'user_abc123', '2024-01-15 10:30:00', '2024-01-15 10:30:00')

java

// 你的代码
private User createUserWithPhone(String phone) {
    User user = new User();
    user.setPhone(phone);
    user.setNickName("user_" + RandomUtil.randomString(8));
    user.setCreateTime(LocalDateTime.now());
    user.setUpdateTime(LocalDateTime.now());
    save(user);  // ← 这行代码把 user 存进数据库
    return user;
}

text

user 对象（内存中）
    ↓
save(user)
    ↓
MyBatis-Plus 自动生成 SQL
    ↓
INSERT INTO user (phone, nick_name, ...) VALUES (?, ?, ...)
    ↓
数据库执行插入
    ↓
数据库里多了一条用户记录

java

boolean success = save(user);  // true=插入成功，false=失败

java

// User 实体类
@Data
@TableName("user")  // 指定表名
public class User {
    @TableId(type = IdType.AUTO)  // 主键自增
    private Long id;
    
    private String phone;        // 对应数据库 phone 字段
    private String nickName;     // 对应数据库 nick_name 字段
    private LocalDateTime createTime;
    private LocalDateTime updateTime;
}

java

@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {
    
    public void demo() {
        // 插入
        save(user);                    // 插入一条
        saveBatch(userList);           // 批量插入
        
        // 删除
        removeById(1L);                // 根据 ID 删除
        removeByIds(Arrays.asList(1,2,3));  // 批量删除
        
        // 更新
        updateById(user);              // 根据 ID 更新
        update(user, updateWrapper);   // 条件更新
        
        // 查询
        getById(1L);                   // 根据 ID 查询
        list();                        // 查询所有
        listByIds(Arrays.asList(1,2)); // 批量查询
        count();                       // 查询总数
        
        // 条件查询（Lambda 方式）
        query().eq("phone", "13800138000").one();      // 等值查询单条
        query().eq("phone", phone).list();             // 等值查询多条
        lambdaQuery().eq(User::getPhone, phone).one(); // Lambda 方式（推荐）
    }
}

用户请求
    ↓
拦截器（preHandle）
    ↓
判断 Session 中是否有 user
    ↓
┌─────────────┴─────────────┐
↓                           ↓
有（已登录）                 无（未登录）
↓                           ↓
放行，进入 Controller       拦截，返回 401

  @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //获取session
        HttpSession session = request.getSession();
        //获取session的用户
        Object user = session.getAttribute("user");
        //判断用户是否存在
        //如果不存在，拦截
        if (user==null){
            response.setStatus(401);
            return false;
        }
        //若存在，则保存到ThreadLocal
        User loginUser = (User) user;
        UserDTO userDTO = new UserDTO();
        userDTO.setId(loginUser.getId());
        userDTO.setNickName(loginUser.getNickName());
        userDTO.setIcon(loginUser.getIcon());
        UserHolder.saveUser(userDTO);

        return true;
    }

@Configuration
public class MvcConfig implements WebMvcConfigurer {
   /**
     * 添加拦截器
     * @param registry
     */
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LoginInterceptor())
                .excludePathPatterns(
                        "/shop/**",
                        "/voucher/**",
                        "/shop-type/**",
                        "/upload/**",
                        "/blog/hot",
                        "/user/code",
                        "/user/login"
                );
        WebMvcConfigurer.super.addInterceptors(registry);
    }
}

   session.setAttribute("user", BeanUtil.copyProperties(user, UserDTO.class));

【登录时】
数据库 User 实体（包含 password、phone 等所有字段）
    ↓
转换为 UserDTO（只包含 id、nickName、icon 等公开字段）
    ↓
session.setAttribute("user", userDTO)  ← Session 中存的是 UserDTO
    ↓
【后续请求】
拦截器从 Session 取出
    ↓
UserDTO userDTO = (UserDTO) session.getAttribute("user")  ← 取出的是 UserDTO
    ↓
UserHolder.setUser(userDTO)  ← ThreadLocal 存的是 UserDTO
    ↓
Controller/Service 中使用
    ↓
UserDTO currentUser = UserHolder.getUser()  ← 拿到的也是 UserDTO
    ↓
返回给前端（只包含公开字段，安全）