springboot webflux 過濾器(使用RouterFunction實現(xiàn))

更新時間：2022年03月09日 15:04:38 作者：o_瓜田李下_o

這篇文章主要介紹了springboot webflux 過濾器(使用RouterFunction實現(xiàn))，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

webflux過濾器（RouterFunction實現(xiàn)）

示例

config 層

CustomRouterConfig

@Configuration
public class CustomRouterConfig {
?
? ? @Bean
? ? public RouterFunction<ServerResponse> initRouterFunction(){
? ? ? ? return RouterFunctions.route()
? ? ? ? ? ? ? ? .GET("/test/**",serverRequest -> {
? ? ? ? ? ? ? ? ? ? System.out.println("path："+serverRequest.exchange().getRequest().getPath().pathWithinApplication().value());
?
? ? ? ? ? ? ? ? ? ? return ServerResponse.ok().bodyValue("hello world");
? ? ? ? ? ? ? ? })
? ? ? ? ? ? ? ? .filter((serverRequest, handlerFunction) -> {
? ? ? ? ? ? ? ? ? ? System.out.println("custom filter");
?
? ? ? ? ? ? ? ? ? ? return handlerFunction.handle(serverRequest);
? ? ? ? ? ? ? ? })
? ? ? ? ? ? ? ? .build();
? ? }
}

使用測試

localhost:8080/test/text，控制臺輸出：

2020-06-21 15:18:08.005 INFO 16336 --- [ main] o.s.b.web.embedded.netty.NettyWebServer : Netty started on port(s): 8080
2020-06-21 15:18:08.018 INFO 16336 --- [ main] com.example.demo.DemoApplication : Started DemoApplication in 1.807 seconds (JVM running for 2.641)
custom filter
path：/test/text

RouterFunction的webflux

RouterFunction可以運行在servlet或netty上，所以我們需要將兩個容器間的不同點抽象出來。

整個開發(fā)過程有幾步：

1.HandleFunction，實現(xiàn)輸入ServerRequest，輸出ServerResponse

2.RouterFunction，把請求url和HandlerFunction對應(yīng)起來

3.把RouterFunction包裝成HttpHandler，交給容器Server處理。

代碼

實體類和倉庫不變

handler：

@Component
public class UserHandler {
? ? private final UserRepository repository;
?
? ? public UserHandler(UserRepository repository) {
? ? ? ? this.repository = repository;
? ? }
?
? ? public Mono<ServerResponse> getAllUser(ServerRequest request){
? ? ? ? return ServerResponse.ok().contentType(MediaType.APPLICATION_JSON)
? ? ? ? ? ? ? ? .body(repository.findAll() , User.class);
? ? }
? ? public Mono<ServerResponse> createUser(ServerRequest request){
? ? ? ? Mono<User> userMono = request.bodyToMono(User.class);
? ? ? ? return ServerResponse.ok().contentType(MediaType.APPLICATION_JSON)
? ? ? ? ? ? ? ? .body(repository.saveAll(userMono) , User.class);
? ? }
? ? public Mono<ServerResponse> deleteUserById(ServerRequest request){
? ? ? ? String id = request.pathVariable("id");
? ? ? ? return this.repository.findById(id)
? ? ? ? ? ? ? ? .flatMap(user -> this.repository.delete(user)
? ? ? ? ? ? ? ? ? ? ? ? .then(ServerResponse.ok().build()))
? ? ? ? ? ? ? ? .switchIfEmpty(ServerResponse.notFound().build());
? ? }
}

router：

@Configuration
public class AllRouters {
? ? @Bean
? ? RouterFunction<ServerResponse> userRouter(UserHandler handler){
? ? ? ? return RouterFunctions.nest(
? ? ? ? ? ? ? ? //相當(dāng)于requestMapping
? ? ? ? ? ? ? ? RequestPredicates.path("/user") ,
? ? ? ? ? ? ? ? RouterFunctions.route(RequestPredicates.GET("/") , handler::getAllUser)
? ? ? ? ? ? ? ? ? ? .andRoute(RequestPredicates.POST("/").and(RequestPredicates.accept(MediaType.APPLICATION_JSON)) , handler::createUser)
? ? ? ? ? ? ? ? ? ? .andRoute(RequestPredicates.DELETE("/{id}") , handler::deleteUserById));
?
? ? }
}

接下來看看routerFunction下的參數(shù)校驗

改造下代碼（這里只寫一個做例子）

public Mono<ServerResponse> createUser(ServerRequest request){
? ? ? ? Mono<User> userMono = request.bodyToMono(User.class);
? ? ? ? return userMono.flatMap(user -> {
? ? ? ? ? ? //在這里做校驗
? ? ? ? ? ? //xxx
? ? ? ? ? ? return ServerResponse.ok().contentType(MediaType.APPLICATION_JSON)
? ? ? ? ? ? ? ? ? ? .body(repository.saveAll(userMono) , User.class);
? ? ? ? });
? ? }

異常捕獲，用aop的方式：

@Component
@Order(-99)
public class ExceptionHandler implements WebExceptionHandler {
? ? @Override
? ? public Mono<Void> handle(ServerWebExchange serverWebExchange, Throwable throwable) {
? ? ? ? ServerHttpResponse response = serverWebExchange.getResponse();
? ? ? ? response.setStatusCode(HttpStatus.BAD_REQUEST);
? ? ? ? response.getHeaders().setContentType(MediaType.TEXT_PLAIN);
? ? ? ? String errorMsg = toStr(throwable);
? ? ? ? DataBuffer db = response.bufferFactory().wrap(errorMsg.getBytes());
? ? ? ? return response.writeWith(Mono.just(db));
? ? }
?
? ? private String toStr(Throwable throwable) {
? ? ? ? //已知異常，自定義異常，這里懶得寫了，就隨便找一個代替
? ? ? ? if (throwable instanceof NumberFormatException){
? ? ? ? ? ? NumberFormatException e = (NumberFormatException) throwable;
? ? ? ? ? ? return e.getMessage();
? ? ? ? }
? ? ? ? //未知異常
? ? ? ? else {
? ? ? ? ? ? throwable.printStackTrace();
? ? ? ? ? ? return throwable.toString();
? ? ? ? }
? ? }
}

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: