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方法作為備選方案。