111 lines
3.3 KiB
Java
111 lines
3.3 KiB
Java
package com.jero.config;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import com.jero.handler.HystrixFallbackHandler;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.web.reactive.function.server.RequestPredicates;
|
|
import org.springframework.web.reactive.function.server.RouterFunction;
|
|
import org.springframework.web.reactive.function.server.RouterFunctions;
|
|
import org.springframework.web.reactive.function.server.ServerResponse;
|
|
|
|
import javax.annotation.Resource;
|
|
|
|
import static org.springframework.web.reactive.function.server.RequestPredicates.GET;
|
|
import static org.springframework.web.reactive.function.server.RouterFunctions.route;
|
|
import static org.springframework.web.reactive.function.server.ServerResponse.ok;
|
|
|
|
/**
|
|
* @author scott
|
|
* @date 2020/05/26
|
|
* 路由配置信息
|
|
*/
|
|
@Slf4j
|
|
@Configuration
|
|
public class GatewayRoutersConfiguration {
|
|
|
|
public static final long DEFAULT_TIMEOUT = 30000;
|
|
|
|
public static String SERVER_ADDR;
|
|
|
|
public static String NAMESPACE;
|
|
|
|
public static String DATA_ID;
|
|
|
|
public static String ROUTE_GROUP;
|
|
|
|
/**
|
|
* 路由配置文件数据获取方式yml,nacos,database
|
|
*/
|
|
public static String DATA_TYPE;
|
|
|
|
@Value("${spring.cloud.nacos.discovery.server-addr}")
|
|
public void setServerAddr(String serverAddr) {
|
|
synchronized (GatewayRoutersConfiguration.class){
|
|
if(SERVER_ADDR == null){
|
|
SERVER_ADDR = serverAddr;
|
|
}
|
|
}
|
|
}
|
|
|
|
@Value("${spring.cloud.nacos.discovery.namespace}")
|
|
public void setNamespace(String namespace) {
|
|
synchronized (GatewayRoutersConfiguration.class){
|
|
if(NAMESPACE == null){
|
|
NAMESPACE = namespace;
|
|
}
|
|
}
|
|
}
|
|
|
|
@Value("${jero.route.config.data-id:#{null}}")
|
|
public void setRouteDataId(String dataId) {
|
|
synchronized (GatewayRoutersConfiguration.class){
|
|
if(DATA_ID == null){
|
|
DATA_ID = dataId + ".json";
|
|
}
|
|
}
|
|
}
|
|
|
|
@Value("${jero.route.config.group:DEFAULT_GROUP:#{null}}")
|
|
public void setRouteGroup(String routeGroup) {
|
|
synchronized (GatewayRoutersConfiguration.class){
|
|
if(ROUTE_GROUP == null){
|
|
ROUTE_GROUP = routeGroup;
|
|
}
|
|
}
|
|
}
|
|
|
|
@Value("${jero.route.config.data-type}")
|
|
public void setDataType(String dataType) {
|
|
synchronized (GatewayRoutersConfiguration.class){
|
|
if(DATA_TYPE == null){
|
|
DATA_TYPE = dataType;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
* 路由断言
|
|
* @return
|
|
*/
|
|
@Bean
|
|
public RouterFunction<ServerResponse> routerFunction() {
|
|
return RouterFunctions.route(
|
|
RequestPredicates.path("/globalFallback").and(RequestPredicates.accept(MediaType.TEXT_PLAIN)), hystrixFallbackHandler);
|
|
|
|
}
|
|
|
|
@Bean
|
|
public RouterFunction<ServerResponse> indexRouter(@Value("classpath:/META-INF/resources/doc.html") final org.springframework.core.io.Resource indexHtml) {
|
|
return route(GET("/"), request -> ok().contentType(MediaType.TEXT_HTML).syncBody(indexHtml));
|
|
}
|
|
|
|
@Resource
|
|
private HystrixFallbackHandler hystrixFallbackHandler;
|
|
|
|
}
|