Bagaimanakah SpringBoot menggunakan RateLimiter untuk mengehadkan arus melalui AOP?

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>

2 anotasi tersuai

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

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;
    } 
}

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 操作
      }

• Kod sampel

  <dependency>
      <groupId>com.google.guava</groupId>
      <artifactId>guava</artifactId>
      <version>28.1-jre</version>
      <optional>true</optional>
  </dependency>

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;
    }
}

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:

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,"");
    }
}

Satu adalah menghadkan arus prapanas lancar

RateLimiter r = RateLimiter.create(5); //项目启动，直接允许5个令牌

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!