Springboot修改post請求接口入?yún)⒒蛑匦沦x值方式

更新時(shí)間：2025年05月12日 09:17:11 作者：小目標(biāo)青年

這篇文章主要介紹了Springboot修改post請求接口入?yún)⒒蛑匦沦x值方式,具有很好的參考價(jià)值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

前言

很久之前寫過一篇就是自動填充接口參數(shù)的，利用的 HandlerMethodArgumentResolver 自定義注解

Springboot Controller接口默認(rèn)自動填充業(yè)務(wù)實(shí)體參數(shù)值

現(xiàn)在這一篇也差不多，達(dá)到的目的就是重新去給post請求的參數(shù)賦值、修改啥的。

不多說，開始。

首先知道一點(diǎn)，post請求的參數(shù)，一旦取出來就沒了，因?yàn)槭橇鲾?shù)據(jù)。

所以老樣子，我們寫多一個(gè)HttpServletRequestWrapper ：

BodyReaderHttpServletRequestWrapper

import org.apache.commons.lang3.StringUtils;
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.Enumeration;
import java.util.Map;
import java.util.Vector;

/**
 * @Author : JCccc
 * @CreateTime : 2020/3/27
 * @Description :
 **/
public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {

    private final byte[] body;

    /**
     * 所有參數(shù)的集合
     */
    private Map<String, String[]> parameterMap;


    public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        BufferedReader reader = request.getReader();
        body = readBytes(reader);
        parameterMap = request.getParameterMap();
    }


    @Override
    public BufferedReader getReader() throws IOException {

        ServletInputStream inputStream = getInputStream();

        if (null == inputStream) {
            return null;
        }

        return new BufferedReader(new InputStreamReader(inputStream));
    }

    @Override
    public Enumeration<String> getParameterNames() {
        Vector<String> vector = new Vector<>(parameterMap.keySet());
        return vector.elements();
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {

        if (body == null) {
            return null;
        }

        final ByteArrayInputStream bais = new ByteArrayInputStream(body);
        return new ServletInputStream() {

            @Override
            public boolean isFinished() {
                return false;
            }

            @Override
            public boolean isReady() {
                return false;
            }

            @Override
            public void setReadListener(ReadListener listener) {

            }

            @Override
            public int read() throws IOException {
                return bais.read();
            }
        };
    }

    /**
     * 通過BufferedReader和字符編碼集轉(zhuǎn)換成byte數(shù)組
     *
     * @param br
     * @return
     * @throws IOException
     */
    private byte[] readBytes(BufferedReader br) throws IOException {
        String str;
        StringBuilder retStr = new StringBuilder();
        while ((str = br.readLine()) != null) {
            retStr.append(str);
        }
        if (StringUtils.isNotBlank(retStr.toString())) {
            return retStr.toString().getBytes(StandardCharsets.UTF_8);
        }
        return null;
    }
}

然后是我們這一期的核心，

DefaultDataFillHandlerMethodArgumentResolver

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.speediance.admin.auth.ShiroUtils;
import com.speediance.admin.dto.req.UserDefaultDataFill;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.BeanWrapper;
import org.springframework.beans.BeanWrapperImpl;
import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

import javax.servlet.http.HttpServletRequest;
import java.io.BufferedReader;
import java.util.Date;
import java.util.Objects;

@Component
public class DefaultDataFillHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.hasParameterAnnotation(UserDefaultDataFill.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
                                  NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        HttpServletRequest request = webRequest.getNativeRequest( HttpServletRequest.class );
        BodyReaderHttpServletRequestWrapper myRequestWrapper = new BodyReaderHttpServletRequestWrapper((HttpServletRequest) request);
        // 從Request的包裝類中讀取數(shù)據(jù)
        BufferedReader reader = myRequestWrapper.getReader();
        StringBuilder sb = new StringBuilder();
        String line;
        while ((line = reader.readLine()) != null) {
            sb.append(line);
        }
        reader.close();
        JSONObject jsonObject = JSON.parseObject(sb.toString());
        Object bean = BeanUtils.instantiateClass(parameter.getParameterType());
        BeanWrapper beanWrapper = new BeanWrapperImpl(bean);
        //重新賦值
        for (String key : jsonObject.keySet()) {
            beanWrapper.setPropertyValue(key,jsonObject.get(key));
        }
        Long userId = //比如從數(shù)據(jù)庫查出來 或者從token解析出來;
   
 
        //舉例子給如入?yún)⒆詣犹畛?更新操作人
        beanWrapper.setPropertyValue("updateUserId", userId);
      
        return bean;
    }


}

然后就是重寫一個(gè)自定義注解，相當(dāng)于需要填充數(shù)據(jù)或者進(jìn)入到參數(shù)修改，我們采用它標(biāo)注：

UserDefaultDataFill

import java.lang.annotation.*;


@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface UserDefaultDataFill {

}

然后就OK了，看看怎么用：

比如說一個(gè)更新接口，我們想默認(rèn)填充上入?yún)⒗锩娴?修改用戶ID （簡單舉例，其余業(yè)務(wù)要修改值，賦值自己套用就可以）