Spring Boot中实现API限流与熔断

在Spring Boot中，我们可以使用一些现成的库来实现API限流和熔断功能。这里我们将介绍如何使用Guava RateLimiter实现限流，以及如何使用Hystrix实现熔断。

1. 使用Guava RateLimiter实现限流

首先，需要在项目中引入Guava依赖：

<dependency>
    <groupId>com.google.guava</groupId>
    <artifactId>guava</artifactId>
    <version>30.1-jre</version>
</dependency>

接下来，创建一个配置类，用于初始化RateLimiter：

import com.google.common.util.concurrent.RateLimiter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class RateLimiterConfig {

    @Bean
    public RateLimiter rateLimiter() {
        return RateLimiter.create(1); // 设置每秒最多处理1个请求
    }
}

现在，可以在Controller中使用@RateLimited注解来实现限流：

import com.google.common.util.concurrent.RateLimiter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class ApiController {

    @Autowired
    private RateLimiter rateLimiter;

    @GetMapping("/api")
    public String api() {
        if (!rateLimiter.tryAcquire()) {
            return "Too many requests, please try again later.";
        }
        return "Hello, world!";
    }
}

2. 使用Hystrix实现熔断

首先，需要在项目中引入Hystrix依赖：

<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-netflix-hystrix</artifactId>
</dependency>

接下来，创建一个配置类，用于启用Hystrix并定义熔断方法：

import com.netflix.hystrix.HystrixCommand;
import com.netflix.hystrix.HystrixCommandGroupKey;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class HystrixConfig {

    @Bean
    public HelloWorldCommand helloWorldCommand() {
        return new HelloWorldCommand();
    }
}

创建一个实现HystrixCommand接口的类：

import com.netflix.hystrix.HystrixCommand;
import com.netflix.hystrix.HystrixCommandGroupKey;

public class HelloWorldCommand extends HystrixCommand<String> {

    public HelloWorldCommand() {
        super(HystrixCommandGroupKey.Factory.asKey("ExampleGroup"));
    }

    @Override
    protected String run() {
        // 调用远程服务或执行业务逻辑
        return "Hello, world!";
    }

    @Override
    protected String getFallback() {
        // 返回熔断时的备选方案
        return "Service is unavailable, please try again later.";
    }
}

现在，可以在Controller中使用@HystrixCommand注解来实现熔断：

import com.netflix.hystrix.HystrixCommand;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class ApiController {

    @GetMapping("/api")
    @HystrixCommand(fallbackMethod = "helloWorldFallback")
    public String api() {
        return new HelloWorldCommand().execute();
    }

    public String helloWorldFallback() {
        return "Service is unavailable, please try again later.";
    }
}

这样，当HelloWorldCommand执行超时时，将会调用helloWorldFallback方法作为备选方案。