Rumah > Java > javaTutorial > Bagaimanakah SpringBoot menggunakan RateLimiter untuk mengehadkan arus melalui AOP?

Bagaimanakah SpringBoot menggunakan RateLimiter untuk mengehadkan arus melalui AOP?

PHPz
Lepaskan: 2023-05-10 23:46:20
ke hadapan
1744 orang telah melayarinya

Gunakan RateLimiter untuk mengehadkan arus melalui AOP

1 Perkenalkan dependensi

<!-- guava 限流 -->
<dependency>
     <groupId>com.google.guava</groupId>
     <artifactId>guava</artifactId>
     <version>25.1-jre</version>
</dependency>
Salin selepas log masuk

2 anotasi tersuai

@Target({ElementType.PARAMETER, ElementType.METHOD})    
@Retention(RetentionPolicy.RUNTIME)    
@Documented    
public  @interface ServiceLimit { 
     String description()  default "";
}
Salin selepas log masuk

3 4. Gunakan

@Component
@Scope
@Aspect
public class LimitAspect {
    每秒只发出5个令牌,此处是单进程服务的限流,内部采用令牌捅算法实现
    private static   RateLimiter rateLimiter = RateLimiter.create(5.0);
    
    //Service层切点  限流
    @Pointcut("@annotation(com.itstyle.seckill.common.aop.ServiceLimit)")  
    public void ServiceAspect() {
        
    }
    
    @Around("ServiceAspect()")
    public  Object around(ProceedingJoinPoint joinPoint) { 
        Boolean flag = rateLimiter.tryAcquire();
        Object obj = null;
        try {
            if(flag){
                obj = joinPoint.proceed();
            }
        } catch (Throwable e) {
            e.printStackTrace();
        } 
        return obj;
    } 
}
Salin selepas log masuk

Had semasa SpringBoot

Algoritma asas had semasa

Badi token dan baldi bocor

    Pelaksanaan algoritma baldi bocor selalunya bergantung pada baris gilir Jika permintaan tiba dan baris gilir tidak penuh, ia terus dimasukkan ke dalam baris gilir, dan kemudian pemproses mengeluarkan permintaan daripada ketua baris gilir. frekuensi tetap untuk pemprosesan. Jika volum permintaan adalah besar, baris gilir akan penuh dan permintaan baharu akan dibuang.
  • Algoritma baldi token ialah baldi yang menyimpan token dengan kapasiti tetap dan token ditambahkan pada baldi pada kadar tetap. Terdapat had maksimum untuk bilangan token yang disimpan dalam baldi Setelah melebihi, ia akan dibuang atau ditolak. Apabila permintaan trafik atau rangkaian tiba, setiap permintaan mesti mendapatkan token Jika ia boleh diperolehi, ia akan diproses terus dan token akan dipadamkan daripada baldi token. Jika ia tidak dapat diperoleh, permintaan akan dihadkan aliran dan sama ada dibuang terus atau menunggu dalam penimbal.
Perbandingan antara baldi token dan baldi bocor

    Timba token menambah token pada baldi pada kadar tetap , sama ada permintaan diproses bergantung pada sama ada terdapat cukup token dalam baldi Apabila bilangan token berkurangan kepada sifar, permintaan aliran keluar baldi bocor pada kadar tetap yang tetap, dan kadar permintaan masuk adalah sewenang-wenangnya permintaan masuk terkumpul Apabila kapasiti baldi bocor tercapai, permintaan masuk baharu akan ditolak selagi ada token, ia boleh diproses Sokongan Ambil 3 token dan 4 token pada satu masa; baldi bocor mengehadkan kadar aliran keluar yang berterusan, iaitu, kadar aliran keluar adalah nilai tetap, contohnya, ia sentiasa mengalir keluar pada kadar 1, tetapi ia tidak boleh menjadi 1 sekali dan 2 kali seterusnya melancarkan kadar aliran masuk pecah; adalah untuk melancarkan kadar aliran keluar;
  • Penghad Kadar Jambu
  • 1 Kebergantungan

    @Override
    @ServiceLimit
    @Transactional
        public Result startSeckil(long seckillId,long userId) {
            //todo 操作
        }
    Salin selepas log masuk
  • Kod sampel

    <dependency>
        <groupId>com.google.guava</groupId>
        <artifactId>guava</artifactId>
        <version>28.1-jre</version>
        <optional>true</optional>
    </dependency>
    Salin selepas log masuk
3 Ujian

@Slf4j
@Configuration
public class RequestInterceptor implements HandlerInterceptor {
    // 根据字符串分不同的令牌桶, 每天自动清理缓存
    private static LoadingCache<String, RateLimiter> cachesRateLimiter = CacheBuilder.newBuilder()
            .maximumSize(1000)  //设置缓存个数
            /**
             * expireAfterWrite是在指定项在一定时间内没有创建/覆盖时,会移除该key,下次取的时候从loading中取
             * expireAfterAccess是指定项在一定时间内没有读写,会移除该key,下次取的时候从loading中取
             * refreshAfterWrite是在指定时间内没有被创建/覆盖,则指定时间过后,再次访问时,会去刷新该缓存,在新值没有到来之前,始终返回旧值
             * 跟expire的区别是,指定时间过后,expire是remove该key,下次访问是同步去获取返回新值;
             * 而refresh则是指定时间后,不会remove该key,下次访问会触发刷新,新值没有回来时返回旧值
             */
            .expireAfterAccess(1, TimeUnit.HOURS)
            .build(new CacheLoader<String, RateLimiter>() {
                @Override
                public RateLimiter load(String key) throws Exception {
                    // 新的字符串初始化 (限流每秒2个令牌响应)
                    return RateLimiter.create(2);
                }
            });
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        log.info("request请求地址path[{}] uri[{}]", request.getServletPath(), request.getRequestURI());
        try {
            String str = "hello";
            // 令牌桶
            RateLimiter rateLimiter = cachesRateLimiter.get(str);
            if (!rateLimiter.tryAcquire()) {
                System.out.println("too many requests.");
                return false;
            }
        } catch (Exception e) {
            // 解决拦截器的异常,全局异常处理器捕获不到的问题
            request.setAttribute("exception", e);
            request.getRequestDispatcher("/error").forward(request, response);
        }
        return true;
    }
}
Salin selepas log masuk

http://localhost:8080/user/

Jika tiada kelas hasil, anda boleh mengembalikan rentetan sesuka hati

4. Keputusan ujian

Lain-lain

Buat

RateLimiter menyediakan dua kaedah kilang:

Bagaimanakah SpringBoot menggunakan RateLimiter untuk mengehadkan arus melalui AOP?Satu adalah letusan lancar Mengehadkan arus

@RestController
@RequestMapping(value = "user")
public class UserController {
    @GetMapping
    public Result test2(){
        System.out.println("1111");
        return new Result(true,200,"");
    }
}
Salin selepas log masuk

Satu adalah menghadkan arus prapanas lancar

RateLimiter r = RateLimiter.create(5); //项目启动,直接允许5个令牌
Salin selepas log masuk

Keburukan

RateLimiter hanya boleh digunakan untuk pengehadan arus mesin tunggal, jika anda mahukan kluster Untuk pengehadan semasa, anda perlu memperkenalkan redis atau perisian tengah sentinel sumber terbuka Alibaba.

rreeee

Atas ialah kandungan terperinci Bagaimanakah SpringBoot menggunakan RateLimiter untuk mengehadkan arus melalui AOP?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Label berkaitan:
sumber:yisu.com
Kenyataan Laman Web ini
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn
Tutorial Popular
Lagi>
Muat turun terkini
Lagi>
kesan web
Kod sumber laman web
Bahan laman web
Templat hujung hadapan