若依 1.0

ruoyi-gateway/src/main/java/com/ruoyi/gateway/config/RouterFunctionConfiguration.java

@@ -0,0 +1,38 @@
+package com.ruoyi.gateway.config;
+
+import org.springframework.beans.factory.annotation.Autowired;
+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 com.ruoyi.gateway.handler.HystrixFallbackHandler;
+import com.ruoyi.gateway.handler.ValidateCodeHandler;
+
+/**
+ * 路由配置信息
+ * 
+ * @author ruoyi
+ */
+@Configuration
+public class RouterFunctionConfiguration
+{
+    @Autowired
+    private HystrixFallbackHandler hystrixFallbackHandler;
+
+    @Autowired
+    private ValidateCodeHandler imageCodeHandler;
+
+    @SuppressWarnings("rawtypes")
+    @Bean
+    public RouterFunction routerFunction()
+    {
+        return RouterFunctions
+                .route(RequestPredicates.path("/fallback").and(RequestPredicates.accept(MediaType.TEXT_PLAIN)),
+                        hystrixFallbackHandler)
+                .andRoute(RequestPredicates.GET("/code").and(RequestPredicates.accept(MediaType.TEXT_PLAIN)),
+                        imageCodeHandler);
+    }
+
+}