spring cloud gateway获取response body

yefengyu · Oct 27, 2019 · 7256332 · 7256332
1 parent 85ff2e6
commit 7256332
Show file tree

Hide file tree

Showing 16 changed files with 303 additions and 17 deletions.
diff --git a/...lter/FluidControlGlobalGatewayFilter.java → ...obal/FluidControlGlobalGatewayFilter.java b/...lter/FluidControlGlobalGatewayFilter.java → ...obal/FluidControlGlobalGatewayFilter.java
@@ -1,4 +1,4 @@
-package com.yefengyu.gateway.globalFilter;
+package com.yefengyu.gateway.global;
 
 import io.github.bucket4j.Bandwidth;
 import io.github.bucket4j.Bucket;

diff --git a/...obalFilter/GlobalGatewayFilterConfig.java → ...way/global/GlobalGatewayFilterConfig.java b/...obalFilter/GlobalGatewayFilterConfig.java → ...way/global/GlobalGatewayFilterConfig.java
@@ -1,4 +1,4 @@
-package com.yefengyu.gateway.globalFilter;
+package com.yefengyu.gateway.global;
 
 import org.springframework.cloud.gateway.filter.GlobalFilter;
 import org.springframework.context.annotation.Bean;

diff --git a/...ateway/localFilter/AuthGatewayFilter.java → ...ngyu/gateway/local/AuthGatewayFilter.java b/...ateway/localFilter/AuthGatewayFilter.java → ...ngyu/gateway/local/AuthGatewayFilter.java
@@ -1,4 +1,4 @@
-package com.yefengyu.gateway.localFilter;
+package com.yefengyu.gateway.local;
 
 import com.yefengyu.gateway.utitls.AuthUtil;
 import org.springframework.cloud.gateway.filter.GatewayFilter;

diff --git a/...localFilter/AuthGatewayFilterFactory.java → ...teway/local/AuthGatewayFilterFactory.java b/...localFilter/AuthGatewayFilterFactory.java → ...teway/local/AuthGatewayFilterFactory.java
@@ -1,4 +1,4 @@
-package com.yefengyu.gateway.localFilter;
+package com.yefengyu.gateway.local;
 
 import org.springframework.cloud.gateway.filter.GatewayFilter;
 import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;

diff --git a/...lFilter/IPForbidGatewayFilterFactory.java → ...y/local/IPForbidGatewayFilterFactory.java b/...lFilter/IPForbidGatewayFilterFactory.java → ...y/local/IPForbidGatewayFilterFactory.java
@@ -1,4 +1,4 @@
-package com.yefengyu.gateway.localFilter;
+package com.yefengyu.gateway.local;
 
 import org.springframework.cloud.gateway.filter.GatewayFilter;
 import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;

diff --git a/gateway/src/main/java/com/yefengyu/gateway/local/ResponseBodyGatewayFilterFactory.java b/gateway/src/main/java/com/yefengyu/gateway/local/ResponseBodyGatewayFilterFactory.java
@@ -0,0 +1,102 @@
+package com.yefengyu.gateway.local;
+
+
+import org.reactivestreams.Publisher;
+import org.springframework.cloud.gateway.support.CachedBodyOutputMessage;
+import org.springframework.stereotype.Component;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+import org.springframework.cloud.gateway.filter.GatewayFilter;
+import org.springframework.cloud.gateway.filter.GatewayFilterChain;
+import org.springframework.cloud.gateway.filter.NettyWriteResponseFilter;
+import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;
+import org.springframework.cloud.gateway.support.BodyInserterContext;
+import org.springframework.core.Ordered;
+import org.springframework.core.io.buffer.DataBuffer;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.server.reactive.ServerHttpResponse;
+import org.springframework.http.server.reactive.ServerHttpResponseDecorator;
+import org.springframework.web.reactive.function.BodyInserter;
+import org.springframework.web.reactive.function.BodyInserters;
+import org.springframework.web.reactive.function.client.ClientResponse;
+import org.springframework.web.server.ServerWebExchange;
+
+import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR;
+
+@Component
+public class ResponseBodyGatewayFilterFactory extends AbstractGatewayFilterFactory<Object> {
+
+    @Override
+    public GatewayFilter apply(Object config) {
+
+        return new ModifyResponseGatewayFilter();
+    }
+
+
+    public class ModifyResponseGatewayFilter implements GatewayFilter, Ordered {
+
+        @Override
+        public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
+            return chain.filter(exchange.mutate().response(decorate(exchange)).build());
+        }
+
+        @SuppressWarnings("unchecked")
+        ServerHttpResponse decorate(ServerWebExchange exchange) {
+            return new ServerHttpResponseDecorator(exchange.getResponse()) {
+
+                @Override
+                public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
+
+                    Class inClass = String.class;
+                    Class outClass = String.class;
+
+                    String originalResponseContentType = exchange
+                            .getAttribute(ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR);
+                    HttpHeaders httpHeaders = new HttpHeaders();
+
+                    httpHeaders.add(HttpHeaders.CONTENT_TYPE,
+                            originalResponseContentType);
+
+                    ClientResponse clientResponse = ClientResponse
+                            .create(exchange.getResponse().getStatusCode())
+                            .headers(headers -> headers.putAll(httpHeaders))
+                            .body(Flux.from(body)).build();
+
+                    Mono modifiedBody = clientResponse.bodyToMono(inClass)
+                            .flatMap(originalBody -> {
+                                //TODO:此次可以对返回的body进行操作
+                                System.out.println(originalBody);
+                                return Mono.just(originalBody);
+                            });
+
+                    BodyInserter bodyInserter = BodyInserters.fromPublisher(modifiedBody,
+                            outClass);
+                    CachedBodyOutputMessage outputMessage = new CachedBodyOutputMessage(
+                            exchange, exchange.getResponse().getHeaders());
+                    return bodyInserter.insert(outputMessage, new BodyInserterContext())
+                            .then(Mono.defer(() -> {
+                                Flux<DataBuffer> messageBody = outputMessage.getBody();
+                                HttpHeaders headers = getDelegate().getHeaders();
+                                if (!headers.containsKey(HttpHeaders.TRANSFER_ENCODING)) {
+                                    messageBody = messageBody.doOnNext(data -> headers
+                                            .setContentLength(data.readableByteCount()));
+                                }
+                                return getDelegate().writeWith(messageBody);
+                            }));
+                }
+
+                @Override
+                public Mono<Void> writeAndFlushWith(
+                        Publisher<? extends Publisher<? extends DataBuffer>> body) {
+                    return writeWith(Flux.from(body).flatMapSequential(p -> p));
+                }
+            };
+        }
+
+        @Override
+        public int getOrder() {
+            return NettyWriteResponseFilter.WRITE_RESPONSE_FILTER_ORDER - 1;
+        }
+    }
+}
diff --git a/gateway/src/main/resources/application.yml b/gateway/src/main/resources/application.yml
@@ -15,6 +15,7 @@ spring:
             - StripPrefix=1
             - Auth
             - IPForbid=0:0:0:0:0:0:0:1
+            - ResponseBody
 my-load-balanced-service:
   ribbon:
     listOfServers: localhost:1001,localhost:1002,localhost:1003

diff --git a/provider1001/src/main/java/com/yefengyu/cloud/controller/HealthController.java b/provider1001/src/main/java/com/yefengyu/cloud/controller/HealthController.java
@@ -1,15 +1,13 @@
 package com.yefengyu.cloud.controller;
 
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 
 @RestController
 public class HealthController
 {
     @GetMapping("/heath")
-    @ResponseBody
     public String heath()
     {
         return "ok";

diff --git a/provider1001/src/main/java/com/yefengyu/cloud/controller/HelloWorldController.java b/provider1001/src/main/java/com/yefengyu/cloud/controller/HelloWorldController.java
@@ -1,15 +1,13 @@
 package com.yefengyu.cloud.controller;
 
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 
 @RestController
 public class HelloWorldController
 {
     @GetMapping("/hello")
-    @ResponseBody
     public String hello()
     {
         return "hello spring cloud, 1001";

diff --git a/provider1001/src/main/java/com/yefengyu/cloud/controller/PersonController.java b/provider1001/src/main/java/com/yefengyu/cloud/controller/PersonController.java
@@ -0,0 +1,65 @@
+package com.yefengyu.cloud.controller;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@RestController
+public class PersonController {
+
+    @GetMapping("/persons")
+    public List<Person> listPerson(){
+        List<Person> personList = new ArrayList<>();
+        for(int i = 0; i < 100; i++){
+            Person person = new Person();
+            person.setId(i);
+            person.setName("王五" + i);
+            person.setAddress("北京" + i);
+            personList.add(person);
+        }
+        return personList;
+    }
+
+    public static class Person{
+        private Integer id;
+        private String name;
+        private String address;
+
+        public Integer getId() {
+            return id;
+        }
+
+        public void setId(Integer id) {
+            this.id = id;
+        }
+
+        public String getName() {
+            return name;
+        }
+
+        public void setName(String name) {
+            this.name = name;
+        }
+
+        public String getAddress() {
+            return address;
+        }
+
+        public void setAddress(String address) {
+            this.address = address;
+        }
+
+        @Override
+        public String toString() {
+            return "Person{" +
+                    "id=" + id +
+                    ", name='" + name + '\'' +
+                    ", address='" + address + '\'' +
+                    '}';
+        }
+    }
+}
+
+
diff --git a/provider1002/src/main/java/cloud/controller/HealthController.java b/provider1002/src/main/java/cloud/controller/HealthController.java
@@ -1,15 +1,13 @@
 package cloud.controller;
 
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 
 @RestController
 public class HealthController
 {
     @GetMapping("/heath")
-    @ResponseBody
     public String heath()
     {
         return "ok";

diff --git a/provider1002/src/main/java/cloud/controller/HelloWorldController.java b/provider1002/src/main/java/cloud/controller/HelloWorldController.java
@@ -1,15 +1,13 @@
 package cloud.controller;
 
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 
 @RestController
 public class HelloWorldController
 {
     @GetMapping("/hello")
-    @ResponseBody
     public String hello()
     {
         return "hello spring cloud, 1002";

diff --git a/provider1002/src/main/java/cloud/controller/PersonController.java b/provider1002/src/main/java/cloud/controller/PersonController.java
@@ -0,0 +1,65 @@
+package cloud.controller;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@RestController
+public class PersonController {
+
+    @GetMapping("/persons")
+    public List<Person> listPerson(){
+        List<Person> personList = new ArrayList<>();
+        for(int i = 0; i < 100; i++){
+            Person person = new Person();
+            person.setId(i);
+            person.setName("王五" + i);
+            person.setAddress("北京" + i);
+            personList.add(person);
+        }
+        return personList;
+    }
+
+    public static class Person{
+        private Integer id;
+        private String name;
+        private String address;
+
+        public Integer getId() {
+            return id;
+        }
+
+        public void setId(Integer id) {
+            this.id = id;
+        }
+
+        public String getName() {
+            return name;
+        }
+
+        public void setName(String name) {
+            this.name = name;
+        }
+
+        public String getAddress() {
+            return address;
+        }
+
+        public void setAddress(String address) {
+            this.address = address;
+        }
+
+        @Override
+        public String toString() {
+            return "Person{" +
+                    "id=" + id +
+                    ", name='" + name + '\'' +
+                    ", address='" + address + '\'' +
+                    '}';
+        }
+    }
+}
+
+
diff --git a/provider1003/src/main/java/cloud/controller/HealthController.java b/provider1003/src/main/java/cloud/controller/HealthController.java
@@ -1,15 +1,13 @@
 package cloud.controller;
 
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 
 @RestController
 public class HealthController
 {
     @GetMapping("/heath")
-    @ResponseBody
     public String heath()
     {
         return "ok";

diff --git a/provider1003/src/main/java/cloud/controller/HelloWorldController.java b/provider1003/src/main/java/cloud/controller/HelloWorldController.java
@@ -1,15 +1,13 @@
 package cloud.controller;
 
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
 
 @RestController
 public class HelloWorldController
 {
     @GetMapping("/hello")
-    @ResponseBody
     public String hello()
     {
         return "hello spring cloud, 1003";