Springboot中WebFlux实现http重定向到https

水深无声 2023-02-26 14:29 8阅读 0赞

`Spring WebFlux`是一个新兴的技术，`Spring`团队把宝都压在响应式`Reactive`上了，于是推出了全新的`Web`实现。本文不讨论响应式编程，而是通过实例讲解`Springboot WebFlux`如何把`http`重定向到`https`。

![spring mvc and webflux venn][]

# 搭建WebFlux项目 #

引入`WebFlux`的依赖如下：

<dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-webflux</artifactId>
    </dependency>

实现`Controller`，与之前略有不同，返回值为`Mono<T>`：

@RestController
    public class HelloController {
        @GetMapping("/hello")
        public Mono<String> hello() {
            return Mono.just("hello,world!!!");
        }
    }

配置文件与普通的`Springboot`项目没什么差别，配置了两个端口，并配置`SSL`相关参数：

server.port=443
    http.port=80
    
    server.ssl.enabled=true
    server.ssl.key-store-type=jks
    server.ssl.key-store=classpath:xxx.jks
    server.ssl.key-store-password=xxx
    server.ssl.key-alias=xxx

重定向实现

主要做了两件事：

（1）在有https的情况下，启动另一个http服务，主要通过NettyReactiveWebServerFactory来生成一个Web服务。

（2）把http重定向到https，这里做了路径判断，加了一个简单的过滤函数。通过提供一个新的HttpHandler来实现重定向。

package application;
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.boot.web.embedded.netty.NettyReactiveWebServerFactory;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.http.HttpStatus;
    import org.springframework.http.server.reactive.HttpHandler;
    import reactor.core.publisher.Mono;
    import javax.annotation.PostConstruct;
    import java.net.URI;
    import java.net.URISyntaxException;
    
    @Configuration
    public class HttpToHttpsConfig {
    
        @Value("${server.port}")
        private int httpsPort;
    
        @Value("${http.port}")
        private int httpPort;
    
    
        @Autowired
        private HttpHandler httpHandler;
    
        @PostConstruct
        public void startRedirectServer() {
            NettyReactiveWebServerFactory factory = new NettyReactiveWebServerFactory(httpPort);
            factory.getWebServer(
                    (request, response) -> {
                        URI uri = request.getURI();
                        URI httpsUri;
                        try {
                            if (isNeedRedirect(uri.getPath())) {
                                httpsUri = new URI("https",
                                        uri.getUserInfo(),
                                        uri.getHost(),
                                        httpsPort,
                                        uri.getPath(),
                                        uri.getQuery(),
                                        uri.getFragment());
                                response.setStatusCode(HttpStatus.MOVED_PERMANENTLY);
                                response.getHeaders().setLocation(httpsUri);
                                return response.setComplete();
                            } else {
                                return httpHandler.handle(request, response);
                            }
    
                        } catch (URISyntaxException e) {
                            return Mono.error(e);
                        }
                    }
            ).start();
    
        }
    
        private boolean isNeedRedirect(String path) {
            return !path.startsWith("/actuator");
        }
    }

[spring mvc and webflux venn]: /images/20230209/fa0fd50a142c42d3be87193b7a317fd1.png