关于Java限流功能的简单实现

 更新时间:2023年07月07日 11:06:36   作者:Blue92120  
这篇文章主要介绍了关于Java限流功能的简单实现,在Java中,限流是一种常见的技术手段,用于控制系统的访问速率,以保护系统免受过载和滥用,需要的朋友可以参考下

限流

在Java中,限流是一种常见的技术手段,用于控制系统的访问速率,以保护系统免受过载和滥用。

以下是一些常见的Java限流实现方法:

1.计数器限流

这是一种简单而常见的限流方法。在该方法中,我们可以使用计数器来记录每个时间窗口内的请求数量,并与预设的阈值进行比较。如果请求数量超过阈值,就拒绝进一步的请求。

使用时,我们可以在关键代码路径上调用allowRequest()方法,并根据返回值决定是否允许请求继续进行。

public class CounterLimiter {
    private int limit;
    private AtomicInteger counter;
    public CounterLimiter(int limit) {
        this.limit = limit;
        this.counter = new AtomicInteger(0);
    }
    public boolean allowRequest() {
        int currentCount = counter.incrementAndGet();
        if (currentCount > limit) {
            counter.decrementAndGet();
            return false;
        }
        return true;
    }
}

2.令牌桶限流

令牌桶算法是一种基于队列的限流算法。在令牌桶中,我们可以定义一个固定大小的令牌桶,该桶以固定速率生成令牌。每当一个请求到达时,它需要获取一个令牌才能执行。如果令牌桶中没有可用的令牌,请求将被暂时阻塞或丢弃。

public class TokenBucketLimiter {
    private int capacity;
    private int tokens;
    private long lastRefillTimestamp;
    private double refillRate;
    private Object lock = new Object();
    public TokenBucketLimiter(int capacity, double refillRate) {
        this.capacity = capacity;
        this.tokens = capacity;
        this.refillRate = refillRate;
        this.lastRefillTimestamp = System.currentTimeMillis();
    }
    public boolean allowRequest() {
        synchronized (lock) {
            refillTokens();
            if (tokens > 0) {
                tokens--;
                return true;
            }
            return false;
        }
    }
    private void refillTokens() {
        long currentTime = System.currentTimeMillis();
        double elapsedTime = (currentTime - lastRefillTimestamp) / 1000.0;
        int newTokens = (int) (elapsedTime * refillRate);
        if (newTokens > 0) {
            tokens = Math.min(tokens + newTokens, capacity);
            lastRefillTimestamp = currentTime;
        }
    }
}

在这个示例中,capacity表示令牌桶的最大容量,refillRate表示每秒生成的令牌数量。使用时,我们可以在关键代码路径上调用allowRequest()方法,并根据返回值决定是否允许请求继续进行。

3.漏桶限流

public class LeakyBucketLimiter {
    private int capacity;
    private int availableTokens;
    private long lastLeakTimestamp;
    private double leakRate;
    private Object lock = new Object();
    public LeakyBucketLimiter(int capacity, double leakRate) {
        this.capacity = capacity;
        this.availableTokens = 0;
        this.leakRate = leakRate;
        this.lastLeakTimestamp = System.currentTimeMillis();
    }
    public boolean allowRequest() {
        synchronized (lock) {
            leakTokens();
            if (availableTokens > 0) {
                availableTokens--;
                return true;
            }
            return false;
        }
    }
    private void leakTokens() {
        long currentTime = System.currentTimeMillis();
        double elapsedTime = (currentTime - lastLeakTimestamp) / 1000.0;
        int tokensToLeak = (int) (elapsedTime * leakRate);
        if (tokensToLeak > 0) {
            if (tokensToLeak >= availableTokens) {
                availableTokens = 0;
            } else {
                availableTokens -= tokensToLeak;
            }
            lastLeakTimestamp = currentTime;
        }
    }
}

在这个示例中,capacity表示桶的最大容量,leakRate表示每秒流出的水滴数量。使用时,我们可以在关键代码路径上调用allowRequest()方法,并根据返回值决定是否允许请求继续进行。

这些是一些常见的Java限流实现方法,我们可以根据具体的需求选择适合我们的场景的方法。此外,还有其他更复杂和高级的限流算法可用,例如漏斗算法、SmoothWarmingUp算法等。

到此这篇关于关于Java限流功能的简单实现的文章就介绍到这了,更多相关Java限流实现内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Java中的登录技术保姆级详细教程

    Java中的登录技术保姆级详细教程

    这篇文章主要介绍了Java中登录技术保姆级详细教程的相关资料,在Java中我们可以使用各种技术和框架来实现这些功能,文中通过代码介绍的非常详细,需要的朋友可以参考下
    2025-06-06
  • Java并发编程之CountDownLatch原理详解

    Java并发编程之CountDownLatch原理详解

    这篇文章主要介绍了Java并发编程之CountDownLatch原理详解,CountDownLatch类中使用了一个继承自AQS的共享锁Sync对象,构造CountDownLatch对象时会将传入的线程数值设为AQS的state值,需要的朋友可以参考下
    2023-12-12
  • springboot Jackson全局配置过程

    springboot Jackson全局配置过程

    SpringBoot Jackson 全局配置详解,介绍如何针对单个对象属性、配置文件及自定义 ObjectMapper 进行序列化规则的配置
    2026-05-05
  • spring boot项目打包成war在tomcat运行的全步骤

    spring boot项目打包成war在tomcat运行的全步骤

    这篇文章主要给大家介绍了关于spring boot项目打包成war在tomcat运行的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用spring boot具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧
    2019-04-04
  • Maven的安装和环境变量配置过程

    Maven的安装和环境变量配置过程

    文章介绍了如何下载、安装和配置Maven,包括环境变量配置和阿里云镜像的可选配置,此外,还展示了如何在IntelliJ IDEA中新建一个Maven项目
    2024-11-11
  • spring 自动注入AutowiredAnnotationBeanPostProcessor源码解析

    spring 自动注入AutowiredAnnotationBeanPostProcessor源码解析

    这篇文章主要介绍了spring自动注入AutowiredAnnotationBeanPostProcessor源码解析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-03-03
  • Java HashMap从源码到核心机制实现原理深度解析

    Java HashMap从源码到核心机制实现原理深度解析

    HashMap是 Java 集合框架中最常用的数据结构之一,基于哈希表(Hash Table)实现,下面这篇文章主要介绍了Java HashMap从源码到核心机制实现原理的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下
    2026-01-01
  • 使用Java编写GUI对话框的教程

    使用Java编写GUI对话框的教程

    这篇文章主要介绍了使用Java编写GUI对话框的教程,是Java图形化编程中的基础知识,需要的朋友可以参考下
    2015-10-10
  • Spring实战之Bean的后处理器操作示例

    Spring实战之Bean的后处理器操作示例

    这篇文章主要介绍了Spring实战之Bean的后处理器操作,结合实例形式详细分析了Bean的后处理器相关配置、操作方法及使用注意事项,需要的朋友可以参考下
    2019-12-12
  • java面向对象继承与多态介绍

    java面向对象继承与多态介绍

    大家好,本篇文章主要讲的是java面向对象继承与多态介绍,感兴趣的同学赶快来看一看吧,对你有帮助的话记得收藏一下
    2022-01-01

最新评论