luoyb
2021-06-03 ae8267320996082be9bae84aa5248ca76d735777
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
package cc.mrbird.febs.gateway.enhance.runner;
 
import cc.mrbird.febs.gateway.enhance.service.BlackListService;
import cc.mrbird.febs.gateway.enhance.service.RateLimitRuleService;
import cc.mrbird.febs.gateway.enhance.service.RouteEnhanceCacheService;
import lombok.RequiredArgsConstructor;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
 
/**
 * @author MrBird
 */
@RequiredArgsConstructor
public class FebsRouteEnhanceRunner implements ApplicationRunner {
 
    private final RouteEnhanceCacheService cacheService;
    private final BlackListService blackListService;
    private final RateLimitRuleService rateLimitRuleService;
 
    @Override
    public void run(ApplicationArguments args) {
        System.out.println("已开启网关增强功能:请求日志、黑名单&限流。");
        cacheService.saveAllBlackList(blackListService.findAll());
        cacheService.saveAllRateLimitRules(rateLimitRuleService.findAll());
    }
}