Java替換RequestBody和RequestParam參數的屬性

来源:https://www.cnblogs.com/ancold/archive/2023/10/24/17785227.html
-Advertisement-
Play Games

Java替換RequstBody和RequestParam參數的屬性 本文主要講解在Java環境中如何替換RequestBody和RequestParam參數中的屬性 背景 近期由於接手的老項目中存在所有介面中新增一個加密串來給介面做一個加密效果(項目歷史原因，不方便上Jwt授權這套)，所以就研究了 ...


Java替換RequstBody和RequestParam參數的屬性

本文主要講解在Java環境中如何替換RequestBody和RequestParam參數中的屬性

背景

近期由於接手的老項目中存在所有介面中新增一個加密串來給介面做一個加密效果(項目歷史原因，不方便上Jwt授權這套)，所以就研究了一下Http請求鏈路，發現可以通過 javax.servlet.Filter去實現

替換RequestParam參數

首先通過繼續HttpServletRequestWrapper來達到獲取和替換RequestParam中的參數信息，接下來我們通過javax.servlet.Filter去獲取ServletRequest中參數的信息，並且定義對應規則，來實現替換參數
代碼示例：

package com.simplemessage.cloudpayservice.infrastructure.config.http;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.Vector;

/**
 * @CreateAt: 2023/10/24 12:13
 * @ModifyAt: 2023/10/24 12:13
 * @Version 1.0
 */
public class MyRequestWrapper  extends HttpServletRequestWrapper {

    private Map params = new HashMap<>();
    public MyRequestWrapper(HttpServletRequest request, Map newParams) {
        super(request);
        if(request.getParameterMap() != null){
            this.params.putAll(request.getParameterMap());
        }
        if(newParams != null){
            this.params.putAll(newParams);
        }
    }

    /**
     * 獲取參數
     * @return
     */
    @Override
    public Map getParameterMap() {
        return params;
    }

    @Override
    public Enumeration getParameterNames() {
        Vector l = new Vector(params.keySet());
        return l.elements();
    }


    @Override
    public String[] getParameterValues(String name) {
        Object v = params.get(name);
        if (v == null) {
            return null;
        } else if (v instanceof String[]) {
            return (String[]) v;
        } else if (v instanceof String) {
            return new String[]{(String) v};
        } else {
            return new String[]{v.toString()};
        }
    }

    /**
     * 根據參數的key獲取參數
     * @param name
     * @return
     */
    @Override
    public String getParameter(String name) {
        Object v = params.get(name);
        if (v == null) {
            return null;
        } else if (v instanceof String[]) {
            String[] strArr = (String[]) v;
            if (strArr.length > 0) {
                return strArr[0];
            } else {
                return null;
            }
        } else if (v instanceof String) {
            return (String) v;
        } else {
            return v.toString();
        }
    }
}

package com.simplemessage.cloudpayservice.infrastructure.config.http;

import com.fasterxml.jackson.core.io.JsonEOFException;
import lombok.extern.slf4j.Slf4j;
import org.apache.catalina.connector.RequestFacade;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

/**
 * @CreateAt: 2023/10/24 12:16
 * @ModifyAt: 2023/10/24 12:16
 * @Version 1.0
 */
@Slf4j
@Component
@WebFilter(filterName = "replaceGetRequestFilter", urlPatterns = {"/*"})
public class ReplaceGetRequestFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) {
    }

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        long start = System.currentTimeMillis();
        //獲取HttpServletRequest對象
        HttpServletRequest httpServletRequest = (HttpServletRequest) request;
        //判斷當前是否為Get請求
        if ("GET".equalsIgnoreCase(httpServletRequest.getMethod())) {
        	// 獲取參數信息
            String param= request.getParameter("param");
            //判斷參數是否為空，為空則放行
            if (StringUtils.isEmpty(param)) {
                chain.doFilter(request, response);
                return;
            } else {
                Map<String, String[]> newParameterMap = new HashMap<>();
                // 替換參數(自定義規則)
                String newParama="test";
                newParameterMap.put("param", newParama);
                // 實現參數替換
                MyRequestWrapper myRequestWrapper = new MyRequestWrapper(httpServletRequest, newParameterMap);
                chain.doFilter(myRequestWrapper, response);
            }

        } else {
            try {
                chain.doFilter(request, response);
            } catch (HttpMessageNotReadableException httpMessageNotReadableException) {
                log.error(((RequestFacade) request).getRequestURI() + ", " + httpMessageNotReadableException.getMessage());
            } catch (JsonEOFException jsonEOFException) {
                log.error(((RequestFacade) request).getRequestURI() + ", " + jsonEOFException.getMessage());
            }
        }
        long end = System.currentTimeMillis();
        log.info("{} 介面耗時：{} ms", httpServletRequest.getRequestURI(), (end - start));
    }

    @Override
    public void destroy() {
    }
}

替換RequestBody參數

主要思路就是通過獲取Post中請求的輸入流信息，解析輸入流信息，按照對應的規則進行替換參數信息，最後將對應的流信息包裝進行返回
代碼示例：

package com.simplemessage.cloudpayservice.infrastructure.config.http;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.core.MethodParameter;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpInputMessage;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.lang.Nullable;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.RequestBodyAdvice;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Type;
import java.util.List;

/**
 * @version 1.0
 * @createAt: 2023/10/24 12:23:23
 * @modifyAt: 2023/10/24 12:23:23
 */
@RestControllerAdvice
@Slf4j
public class DecryptRequestBodyHandler implements RequestBodyAdvice {
    
    /**
     * 該方法用於判斷當前請求，是否要執行beforeBodyRead方法
     * methodParameter方法的參數對象
     * type方法的參數類型
     * aClass 將會使用到的Http消息轉換器類類型
     * 註意：此判斷方法，會在beforeBodyRead 和 afterBodyRead方法前都觸發一次。
     * @return 返回true則會執行beforeBodyRead
     */
    @Override
    public boolean supports(MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
        return true;
    }

    /**
     * 在Http消息轉換器執轉換，之前執行
     * inputMessage 客戶端請求的信息
     * parameter 參數信息
     * targetType 參數類型
     * converterType Http消息轉換器類類型
     *
     * @return 返回 一個自定義的HttpInputMessage
     */
    @Override
    public HttpInputMessage beforeBodyRead(HttpInputMessage inputMessage, MethodParameter parameter, Type targetType, Class<? extends HttpMessageConverter<?>> converterType) throws IOException {
        // 如果body是空內容直接返回原來的請求
        if (inputMessage.getBody().available() <= 0) {
            return inputMessage;
        }
        // 請求中的header信息
        HttpHeaders headers = inputMessage.getHeaders();     
      
        // 將輸入流讀出來，註意 body 裡面的流只能讀一次
        ByteArrayOutputStream requestBodyDataByte = new ByteArrayOutputStream();
        try {
        	//複製流信息
            IOUtils.copy(inputMessage.getBody(), requestBodyDataByte);
        } catch (Exception e) {
            log.error("參數流拷貝失敗: ", e.toString());
            return inputMessage;
        }
        ByteArrayOutputStream requestBodyDataByteNew = null;
        try {
            JSONObject body = JSON.parseObject(requestBodyDataByte.toByteArray(), JSONObject.class);
            if (ObjectUtils.isEmpty(body)) {
                return inputMessage;
            }
            //自定義規則西悉尼
            if (body.containsKey("param")) {
                String custId = body.getString("param"); 
                String newParam="";              
                body.put("custId", newParam);
                requestBodyDataByteNew = new ByteArrayOutputStream();
                //拷貝流信息
                IOUtils.copy(new ByteArrayInputStream(body.toJSONString().getBytes()), requestBodyDataByteNew);
            }
        } catch (Throwable e) {
            log.error("流轉換異常 ", e.toString());
        }
        // 如果上述發生異常，仍然使用原來的請求內容
        requestBodyDataByte = requestBodyDataByteNew != null ? requestBodyDataByteNew : requestBodyDataByte;
        InputStream rawInputStream = new ByteArrayInputStream(requestBodyDataByte.toByteArray());
        inputMessage.getHeaders().set(HttpHeaders.CONTENT_LENGTH, String.valueOf(rawInputStream.available()));
        return new HttpInputMessage() {
            @Override
            public HttpHeaders getHeaders() {
                return inputMessage.getHeaders();
            }

            @Override
            public InputStream getBody() throws IOException {
                return rawInputStream;
            }
        };
    }

    /**
     * 在Http消息轉換器執轉換，之後執行
     * body 轉換後的對象
     * inputMessage 客戶端的請求數據
     * parameter handler方法的參數類型
     * targetType handler方法的參數類型
     * converterType 使用的Http消息轉換器類類型
     *
     * @return 返回一個新的對象
     */
    @Override
    public Object afterBodyRead(Object body, HttpInputMessage inputMessage, MethodParameter parameter, Type targetType, Class<? extends HttpMessageConverter<?>> converterType) {
        return body;
    }

    /**
     * 參數與afterBodyRead相同，不過這個方法body為空的情況
     */
    @Override
    public Object handleEmptyBody(@Nullable Object body, HttpInputMessage inputMessage, MethodParameter parameter, Type targetType, Class<? extends HttpMessageConverter<?>> converterType) {
        return body;
    }
}

如有哪裡講得不是很明白或是有錯誤，歡迎指正
如您喜歡的話不妨點個贊收藏一下吧

您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
  • Telegram 正式引入國產小程式技術
    Telegram 宣佈為其開發者提供了一項“能夠在 App 中運行迷你應用”的新功能（ 迷你應用即 Mini App，下文中以“小程式”代替）。 ...
  • 設計模式（九）裝飾
    一、定義 動態地給一個對象增加一些額外的職責。就擴展功能而言，裝飾模式提供了一種比使用子類更加靈活的替代方案。裝飾模式是一種結構型模式。 二、描述 包含以下三個角色：1、Component（抽象構件）：它是具體構件和抽象裝飾類的父類，聲明瞭在具體構件中實現的業務方法，它的引入可以是客戶端以一致的方式 ...
  • 架構師必備的37項技能清單
    幾年前，我被問到“你是如何變成一名架構師的？”。基於這個話題，我們討論了很多，比如必要的技術、經驗以及所需要的知識儲備等。這一次討論促使我開始思考要成為一名架構師應該具備和學習的東西有哪些，成為一個優秀的架構師應該具備哪些能力和做哪些事情。為此我查閱資料，走訪各位大佬，當然也結合自己的經歷，最終我輸 ...
  • 淺談Python非同步編程
    1. 非同步編程概述 非同步編程是一種編程範式，用於處理那些需要等待I/O操作完成或者耗時任務的情況。在傳統的同步編程中，代碼會按照順序逐行執行，直到遇到一個耗時操作，它會阻塞程式的執行直到操作完成。這種阻塞式的模型在某些場景下效率低下，因為代碼在等待操作完成時無法執行其他任務。 非同步編程通過使用非阻塞 ...
  • Flutter/Dart第16天：Dart類方法
    類方法就是為對象提供一些數據操作的函數，也就是類方法是操作對象的函數。那麼在Dart中，類方法到底有哪些特別之處呢…… ...
  • frp實現內網穿透
    目錄1. 什麼是內網穿透2. 實現背景3. 實現步驟3.1 安裝3.1.1 下載3.1.2 安裝服務端3.1.3 安裝客戶端4. Nginx 配置5. 官方文檔 1. 什麼是內網穿透 內網穿透（Port Forwarding）是一種網路技術，允許你將外部網路中的請求轉發到內部網路中的特定電腦或設備 ...
  • Flask後端開發(一)-基礎知識和前期準備
    本文主要介紹了python flask後端開發項目中的相關知識，包括flask項目結構、flask部署方式、前後端對接、後端介面測試工具。 ...
  • 21.4 Python 使用GeoIP2地圖定位
    GeoIP2是一種IP地址定位庫，它允許開發人員根據`IP`地址查找有關位置和地理位置的信息。它使用`MaxMind`公司的IP地址資料庫，並提供一個方便的Python API。GeoIP2可以用於許多不同的應用程式，例如網站分析、廣告定位和身份驗證。GeoIP2提供了許多不同的信息，例如國家、城市... ...
一周排行
    -Advertisement-
    Play Games
  • 移動開發（一）：使用.NET MAUI開發第一個安卓APP
    移動開發（一）：使用.NET MAUI開發第一個安卓APP 對於工作多年的C#程式員來說，近來想嘗試開發一款安卓APP，考慮了很久最終選擇使用.NET MAUI這個微軟官方的框架來嘗試體驗開發安卓APP，畢竟是使用Visual Studio開發工具，使用起來也比較的順手，結合微軟官方的教程進行了安卓 ...
  • wpf ToggleButton選中效果和一個登錄界面
    前言 QuestPDF 是一個開源 .NET 庫，用於生成 PDF 文檔。使用了C# Fluent API方式可簡化開發、減少錯誤並提高工作效率。利用它可以輕鬆生成 PDF 報告、發票、導出文件等。 項目介紹 QuestPDF 是一個革命性的開源 .NET 庫，它徹底改變了我們生成 PDF 文檔的方 ...
  • 彈幕樹洞項目功能新增篇
    項目地址 項目後端地址： https://github.com/ZyPLJ/ZYTteeHole 項目前端頁面地址： ZyPLJ/TreeHoleVue (github.com) https://github.com/ZyPLJ/TreeHoleVue 目前項目測試訪問地址： http://tree ...
  • 第27篇 sqlserver2022詳細安裝步驟
    話不多說，直接開乾 一.下載 1.官方鏈接下載： https://www.microsoft.com/zh-cn/sql-server/sql-server-downloads 2.在下載目錄中找到下麵這個小的安裝包 SQL2022-SSEI-Dev.exe，運行開始下載SQL server； 二. ...
  • .NET 開源高性能 MQTT 類庫
    前言 隨著物聯網（IoT）技術的迅猛發展，MQTT（消息隊列遙測傳輸）協議憑藉其輕量級和高效性，已成為眾多物聯網應用的首選通信標準。 MQTTnet 作為一個高性能的 .NET 開源庫，為 .NET 平臺上的 MQTT 客戶端與伺服器開發提供了強大的支持。 本文將全面介紹 MQTTnet 的核心功能 ...
  • Serilog文檔翻譯系列（六） - 可用的接收器、增強器、格式化輸出
    Serilog支持多種接收器用於日誌存儲，增強器用於添加屬性，LogContext管理動態屬性，支持多種輸出格式包括純文本、JSON及ExpressionTemplate。還提供了自定義格式化選項，適用於不同需求。 ...
  • 警惕 Visual Studio 屬性求值副作用導致邏輯不符合預期
    目錄簡介獲取 HTML 文檔解析 HTML 文檔測試參考文章 簡介 動態內容網站使用 JavaScript 腳本動態檢索和渲染數據，爬取信息時需要模擬瀏覽器行為，否則獲取到的源碼基本是空的。 本文使用的爬取步驟如下： 使用 Selenium 獲取渲染後的 HTML 文檔 使用 HtmlAgility ...
  • [使用目前最新版]HybridCLR6.9.0+YooAsset2.2.4實現純C# Unity熱更新方案 (一)
    1.前言 什麼是熱更新 游戲或者軟體更新時，無需重新下載客戶端進行安裝，而是在應用程式啟動的情況下，在內部進行資源或者代碼更新 Unity目前常用熱更新解決方案 HybridCLR，Xlua，ILRuntime等 Unity目前常用資源管理解決方案 AssetBundles，Addressable， ...
  • 在 ASP.NET Core Web API 中使用操作篩選器統一處理通用操作
    本文章主要是在C# ASP.NET Core Web API框架實現向手機發送驗證碼簡訊功能。這裡我選擇是一個互億無線簡訊驗證碼平臺，其實像阿裡雲，騰訊雲上面也可以。 首先我們先去 互億無線 https://www.ihuyi.com/api/sms.html 去註冊一個賬號 註冊完成賬號後，它會送 ...
  • 第28篇 如何.net中實現高效可靠數據同步api
    通過以下方式可以高效，並保證數據同步的可靠性 1.API設計 使用RESTful設計，確保API端點明確，並使用適當的HTTP方法（如POST用於創建，PUT用於更新）。 設計清晰的請求和響應模型，以確保客戶端能夠理解預期格式。 2.數據驗證 在伺服器端進行嚴格的數據驗證，確保接收到的數據符合預期格 ...
所有分類