java登录限流实现

一、问题描述

某个系统的登录接口在被刷。现要建立一个防刷/限流机制，根据登录 IP，30 分钟之内，只能发起 30 次登录请求。如果超过该限制，则整个 IP 限制登录请求 30 分钟。

二、设计思路

这道题主要是设计两个 Map：

1、第一个 Map，记录每个 IP 及其登录的时间。题目要求，30 分钟之内只能登录 30 次，所以 Map 的 key 为 IP，value 可以设计一个队列，队列长度 30，队列元素为每次的登录时间。

2、第二个 Map，记录禁止登录的 IP 及禁止开始时间。Map 的 key 为 IP，value 为时间。

三、限流器实现代码

import java.time.LocalDateTime;
import java.util.HashMap;
import java.util.LinkedList;

public class LimitCacheTool {

	private static volatile LimitCacheTool instance;
    // 记录登录的ip地址及每次登录时间
    private static HashMap<String, LinkedList<LocalDateTime>> loginMap = new HashMap<>();
    // 记录禁止登录的ip地址及禁止开始时间
    private static HashMap<String, LocalDateTime> forbiddenMap = new HashMap<>();
    // 构造器私有化，不能在类的外部随意创建对象
    private LimitCacheTool() {
    }
    // 提供一个全局的访问点来获得这个"唯一"的对象
    public static LimitCacheTool getInstance() {
        if (instance == null) {
            synchronized (LimitCacheTool.class) {
                if (instance == null) {
                    instance = new LimitCacheTool();
                }
            }
        }
        return instance;
    }

    public static HashMap<String, LinkedList<LocalDateTime>> getLoginMap() {
        return loginMap;
    }
    public static HashMap<String, LocalDateTime> getForbiddenMap() {
        return forbiddenMap;
    }
	
}

四、限流实现逻辑

/**
	 * @author:JackRen
	 * @Description:防刷/限流机制
	 * @date:2021年2月26日上午10:35:18
	 * @param ip
	 */
	public void doLimitLogin(String ip) {
		String result = "";
		LimitCacheTool limitCache = LimitCacheTool.getInstance();
        LinkedList<LocalDateTime> queue = null;

        // 先判断ip地址是否禁止登录
        LocalDateTime forbiddenTime = limitCache.getForbiddenMap().get(ip);
        if (forbiddenTime != null) {
            Long after = ChronoUnit.MINUTES.between(forbiddenTime, LocalDateTime.now());
            if (after <= 30) {
                result = "当前时间=" + LocalDateTime.now() + " 上次禁止登录时间= " + forbiddenTime + " 距上次被禁时间没有超过30分钟";
                setRespAttr("msg", result);
                return;
            } else {
                limitCache.getForbiddenMap().clear();
            }
        }

        // 如果是首次登录，则创建队列
        if (limitCache.getLoginMap().get(ip) == null) {
            queue = new LinkedList<>();
        } else {
            queue = limitCache.getLoginMap().get(ip);
        }

        // 登录次数达到登录次数上限
        if (queue.size() == 30) {
            // 当前时间和队列中最早的登录时间比较 是否小于30分钟
            LocalDateTime now = LocalDateTime.now();
            LocalDateTime firstLoginTime = queue.poll();
            Long duration = ChronoUnit.MINUTES.between(firstLoginTime, now);
            if (duration <= 30) {
                result = "30分钟内登录超过30次，不允许登录,30分钟后再登录";
                // 禁止该IP登录
                limitCache.getLoginMap().clear();
                limitCache.getForbiddenMap().put(ip, now);
                setRespAttr("msg", result);
                return;
            }
        }
        queue.offer(LocalDateTime.now());
        limitCache.getLoginMap().put(ip, queue);

        result = ip + " 登录时间=" + queue.getLast() + "  队列长度=" + queue.size();
        setRespAttr("msg", result);
        return;
	}