SpringBoot 并发登录人数控制的实现方法

下面我来为你详细讲解“SpringBoot 并发登录人数控制的实现方法”的完整攻略。

1. 前言

在实际开发过程中，我们经常需要加入并发登录人数控制的功能。SpringBoot 作为目前最流行的 JavaWeb 框架之一，其内置的 Spring Security 在实现登录控制方面有很大的优势。同时，SpringBoot 还提供了一些自定义实现方式，用于满足更加实用的业务需求。

本文将通过 SpringBoot 自带的限流机制、Redis 分布式锁等方案，实现一个并发登录人数控制的完整示例，读者可根据自身业务需求来选择相应的实现方案。

2. SpringBoot 自带的限流机制实现并发登录人数控制

SpringBoot 自带的限流机制，可以通过 application.yml 配置文件进行配置，具体配置方式如下所示：

spring:
  security:
    sessions:
      # 允许的最大并发登录数
      maximumSessions: 1
      # 控制同用户名的登录是否允许
      maxSessionsPreventsLogin: true

在上述配置中，maximumSessions 代表允许的最大并发登录数，maxSessionsPreventsLogin 则表示控制同用户名的登录是否允许。

通过配置上述内容以后，Spring Security 就会自动对登录进行控制，只有在已登录用户退出系统或登录超时后，才能下一个用户登录。

3. Redis 分布式锁实现并发登录人数控制

除了 SpringBoot 自带的限流机制，还可以利用 Redis 分布式锁来实现并发登录人数控制的功能。

3.1 实现流程

• 用户登录：通过 Redis 获取分布式锁；
• 执行登录操作：判断当前登录人数是否超出最大限制，如未超出则登录成功，否则登录失败；
• 释放分布式锁。

3.2 示例代码

@Service
public class LoginService {
    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    /**
     * 登录操作
     */
    public boolean login(String username) {
        // 获取分布式锁
        RLock lock = redisTemplate.getLock("login_lock_" + username);
        try {
            // 加锁并设置锁过期时间（避免死锁）
            boolean isLock = lock.tryLock(100, 10, TimeUnit.SECONDS);
            if (!isLock) {
                throw new BusinessException(ResultCode.LOGIN_CONCURRENT_ERROR);
            }
            // 查询当前在线用户数是否超过最大限制（例如 100）
            List<Object> onlineUsers = redisTemplate.opsForList().range("online_users", 0, -1);
            if (onlineUsers != null && onlineUsers.size() >= 100) {
                throw new BusinessException(ResultCode.LOGIN_CONCURRENT_ERROR);
            }
            // 将当前用户添加到在线用户列表中
            redisTemplate.opsForList().rightPush("online_users", username);
            return true;
        } catch (Exception e) {
            throw new BusinessException(ResultCode.LOGIN_EXCEPTION);
        } finally {
            // 释放锁
            lock.unlock();
        }
    }

    /**
     * 退出登录操作
     */
    public void logout(String username) {
        // 获取分布式锁
        RLock lock = redisTemplate.getLock("login_lock_" + username);
        try {
            // 加锁并设置锁过期时间（避免死锁）
            boolean isLock = lock.tryLock(100, 10, TimeUnit.SECONDS);
            if (isLock) {
                // 从在线用户列表中移除当前用户
                redisTemplate.opsForList().remove("online_users", 0, username);
            }
        } catch (InterruptedException e) {
            throw new BusinessException(ResultCode.LOGIN_EXCEPTION);
        } finally {
            // 释放锁
            lock.unlock();
        }
    }
}

在上述示例代码中，我们使用 Redis 分布式锁作为并发登录人数控制的方案。通过 RedisTemplate 来获取 Redis 分布式锁，然后在登录时，查询当前在线用户数是否超过最大限制，如果未超出限制，则将该用户加入在线用户列表中；在退出时，从该列表中移除该用户。这里需要注意的是，需要将分布式锁的过期时间设置得合理，避免出现死锁的情况。

总结

本篇文章通过 SpringBoot 的内置机制和 Redis 分布式锁两种方案，分别演示了如何实现并发登录人数控制的功能。其中 SpringBoot 的限流机制相对简单易懂，但只支持单机限流；而 Redis 分布式锁则支持分布式限流，并且可以很方便地和其他 Redis 分布式解决方案结合使用。读者可根据自身业务需求和技术实力选择相应的实现方案。