Spring gateway 修改響應參數(shù)

Spring Cloud Gateway 作為微服務(wù)的最前沿氮发,可以實現(xiàn)限流、鑒權(quán)等等操作冗懦,本文將以修改統(tǒng)一格式的返回參數(shù)作為講解

新建GlobalFilter的實現(xiàn)類

Spring 修改前的默認返回參數(shù)如下

{
    "status": 404,
    "error":"",
    "message": "NOT FUND",
    "path": "/test",
    "timestamp": 1565334087971
}
@Slf4j
@Component
public class OpenGatewayFilter implements GlobalFilter, Ordered {
    /** 將 List 數(shù)據(jù)以""分隔進行拼接 **/
    private static Joiner joiner = Joiner.on("");

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        ServerHttpResponse response = exchange.getResponse();
        DataBufferFactory bufferFactory = response.bufferFactory();

        ServerHttpResponseDecorator decorator = new ServerHttpResponseDecorator(response) {
            @Override
            public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
               if (body instanceof Flux) {
                Flux<? extends DataBuffer> fluxBody = (Flux<? extends DataBuffer>)body;
                return super.writeWith(fluxBody.buffer().map(dataBuffers -> {
                    List<String> list = Lists.newArrayList();
                    // gateway 針對返回參數(shù)過長的情況下會分段返回爽冕,使用如下方式接受返回參數(shù)則可避免
                    dataBuffers.forEach(dataBuffer -> {
                        // probably should reuse buffers
                        byte[] content = new byte[dataBuffer.readableByteCount()];
                        dataBuffer.read(content);
                        // 釋放掉內(nèi)存
                        DataBufferUtils.release(dataBuffer);

                        list.add(new String(content, StandardCharsets.UTF_8));
                    });
                    // 將多次返回的參數(shù)拼接起來
                    String responseData = joiner.join(list);

                    // 重置返回參數(shù)
                    String result = response(responseData);
                    byte[] uppedContent =
                        new String(result.getBytes(StandardCharsets.UTF_8), StandardCharsets.UTF_8).getBytes();

                    // 修改后的返回參數(shù)應該重置長度,否則如果修改后的參數(shù)長度超出原始參數(shù)長度時會導致客戶端接收到的參數(shù)丟失一部分
                    response.getHeaders().setContentLength(uppedContent.length);

                    return bufferFactory.wrap(uppedContent);
                }));
             }
                return super.writeWith(body);
            }
        };

        return chain.filter(exchange.mutate().response(decorator).build());
    }

    @Override
    public int getOrder() {
        // 必須<=-1
        return -2;
    }

    private String response(String result) {
        try {
            JSONObject json = JSONObject.fromObject(result);
           if (json.containsKey("error")) {
                json.clear();
                json.put("code", json.get("status"));
                json.put("msg", json.getString("message"));
                json.put("payload","");
                json.put("timestamp", System.currentTimeMillis());
                result = json.toString();
            }
        } catch (Exception e) {
            log.warn("轉(zhuǎn)換JSON異常:{}", result);
        }

        return result;
    }
}

新建配置類

覆蓋默認的配置

@Configuration
@EnableConfigurationProperties({ServerProperties.class, ResourceProperties.class})
public class ErrorHandlerConfiguration {

    private final ServerProperties serverProperties;

    private final ApplicationContext applicationContext;

    private final ResourceProperties resourceProperties;

    private final List<ViewResolver> viewResolvers;

    private final ServerCodecConfigurer serverCodecConfigurer;

    public ErrorHandlerConfiguration(ServerProperties serverProperties, ResourceProperties resourceProperties,
        ObjectProvider<List<ViewResolver>> viewResolversProvider, ServerCodecConfigurer serverCodecConfigurer,
        ApplicationContext applicationContext) {
        this.serverProperties = serverProperties;
        this.applicationContext = applicationContext;
        this.resourceProperties = resourceProperties;
        this.viewResolvers = viewResolversProvider.getIfAvailable(Collections::emptyList);
        this.serverCodecConfigurer = serverCodecConfigurer;
    }

    @Bean
    @Order(Ordered.HIGHEST_PRECEDENCE)
    public ErrorWebExceptionHandler errorWebExceptionHandler(ErrorAttributes errorAttributes) {
        GatewayExceptionHandler exceptionHandler = new GatewayExceptionHandler(errorAttributes, this.resourceProperties,
            this.serverProperties.getError(), this.applicationContext);
        exceptionHandler.setViewResolvers(this.viewResolvers);
        exceptionHandler.setMessageWriters(this.serverCodecConfigurer.getWriters());
        exceptionHandler.setMessageReaders(this.serverCodecConfigurer.getReaders());
        return exceptionHandler;
    }

}

修改后的返回結(jié)果

{
    "code": 101,
    "msg": "請求方式不被支持",
    "payload": "",
    "timestamp": 1565334087971
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末披蕉,一起剝皮案震驚了整個濱河市颈畸,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌没讲,老刑警劉巖眯娱,帶你破解...
    沈念sama閱讀 219,539評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異爬凑,居然都是意外死亡徙缴,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,594評論 3 396
  • 文/潘曉璐 我一進店門嘁信,熙熙樓的掌柜王于貴愁眉苦臉地迎上來于样,“玉大人,你說我怎么就攤上這事吱抚“儆睿” “怎么了?”我有些...
    開封第一講書人閱讀 165,871評論 0 356
  • 文/不壞的土叔 我叫張陵秘豹,是天一觀的道長。 經(jīng)常有香客問我昌粤,道長既绕,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,963評論 1 295
  • 正文 為了忘掉前任涮坐,我火速辦了婚禮凄贩,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘袱讹。我一直安慰自己疲扎,他們只是感情好昵时,可當我...
    茶點故事閱讀 67,984評論 6 393
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著椒丧,像睡著了一般壹甥。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上壶熏,一...
    開封第一講書人閱讀 51,763評論 1 307
  • 那天句柠,我揣著相機與錄音,去河邊找鬼棒假。 笑死溯职,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的帽哑。 我是一名探鬼主播谜酒,決...
    沈念sama閱讀 40,468評論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼妻枕!你這毒婦竟也來了甚带?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,357評論 0 276
  • 序言:老撾萬榮一對情侶失蹤佳头,失蹤者是張志新(化名)和其女友劉穎鹰贵,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體康嘉,經(jīng)...
    沈念sama閱讀 45,850評論 1 317
  • 正文 獨居荒郊野嶺守林人離奇死亡碉输,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,002評論 3 338
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了亭珍。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片敷钾。...
    茶點故事閱讀 40,144評論 1 351
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖肄梨,靈堂內(nèi)的尸體忽然破棺而出阻荒,到底是詐尸還是另有隱情,我是刑警寧澤众羡,帶...
    沈念sama閱讀 35,823評論 5 346
  • 正文 年R本政府宣布侨赡,位于F島的核電站,受9級特大地震影響粱侣,放射性物質(zhì)發(fā)生泄漏羊壹。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,483評論 3 331
  • 文/蒙蒙 一齐婴、第九天 我趴在偏房一處隱蔽的房頂上張望油猫。 院中可真熱鬧,春花似錦柠偶、人聲如沸情妖。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,026評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽毡证。三九已至电爹,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間情竹,已是汗流浹背藐不。 一陣腳步聲響...
    開封第一講書人閱讀 33,150評論 1 272
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留秦效,地道東北人雏蛮。 一個月前我還...
    沈念sama閱讀 48,415評論 3 373
  • 正文 我出身青樓,卻偏偏與公主長得像阱州,于是被迫代替她去往敵國和親挑秉。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 45,092評論 2 355

推薦閱讀更多精彩內(nèi)容