在Spring Boot應(yīng)用中,我們經(jīng)常需要對HttpServletRequest中的參數(shù)進行處理,以滿足特定的業(yè)務(wù)需求或增強系統(tǒng)的安全性。本文介紹了一個實用的工具類——ParameterRequestWrapper,通過該類,我們可以更方便地對請求參數(shù)進行定制化處理。
在實際項目中,我們經(jīng)常會遇到對請求參數(shù)進行特殊處理的需求,例如去除空字符串參數(shù)、處理JSON格式的POST請求等。為了更靈活地處理這些情況,我們創(chuàng)建了ParameterRequestWrapper類,它繼承自HttpServletRequestWrapper,并在構(gòu)造方法中對請求參數(shù)進行了初始化和定制化處理。
在處理頁面提交的參數(shù)時,有時會出現(xiàn)空字符串參數(shù),如果不進行處理,這些空字符串可能會對SQL查詢等操作產(chǎn)生不良影響。為了解決這個問題,我們在modifyParameterValues方法中去除了頁面提交的空字符串參數(shù),確保參數(shù)的有效性。
public void modifyParameterValues() { // 去除頁面提交出現(xiàn)的空字符串參數(shù),防止SQL查詢默認(rèn)帶進去,當(dāng)作參數(shù),造成頁面列表不展示 params.entrySet().removeIf(e -> "".equals(e.getValue()[0])); Set<String> set = params.keySet(); for (String key : set) { String[] values = params.get(key); values[0] = values[0].trim(); params.put(key, values); }}
對于POST請求,特別是JSON格式的POST請求,我們通過重寫getInputStream方法,將請求中的JSON參數(shù)進行處理,去除兩端空格,并輸出日志,方便調(diào)試。
@Overridepublic ServletInputStream getInputStream() throws IOException { // 非json類型,直接返回 if (!super.getHeader(HttpHeaders.CONTENT_TYPE).equalsIgnoreCase(MediaType.APPLICATION_JSON_VALUE)) { return super.getInputStream(); } // 為空,直接返回 String json = IOUtils.toString(super.getInputStream(), StandardCharsets.UTF_8); if (StringUtils.isEmpty(json)) { return super.getInputStream(); } log.info("去除POST請求數(shù)據(jù)兩端的空格前參數(shù):" + json); Map<String, Object> map = StringJsonUtils.jsonStringToMap(json); log.info("去除POST請求數(shù)據(jù)兩端的空格后參數(shù):" + JSON.toJSONString(map)); ByteArrayInputStream bis = new ByteArrayInputStream(JSON.toJSONString(map).getBytes(StandardCharsets.UTF_8)); return new MyServletInputStream(bis);}
要使用ParameterRequestWrapper,只需在Controller中將HttpServletRequest替換為該類的實例即可,如下所示:
@PostMapping("/examplepublic ResponseEntity<String> example(ParameterRequestWrapper request) { // 處理定制化的HttpServletRequest,可以直接使用request.getParameter等方法 // ... return ResponseEntity.ok("Success");}
通過這樣的替換,我們就可以在請求到達(dá)Controller之前對參數(shù)進行定制化處理,更好地適應(yīng)項目的需求。
ParameterRequestWrapper是一個簡單而實用的工具類,通過對HttpServletRequest的定制化處理,能夠滿足一些特殊場景下的需求。在實際項目中,根據(jù)具體情況,我們可以進一步擴展該類,以滿足更復(fù)雜的參數(shù)處理需求。希望這個工具類能夠幫助到開發(fā)者更方便地處理請求參數(shù),提升項目的可維護性和靈活性。
本文鏈接:http://www.www897cc.com/showinfo-26-35331-0.html優(yōu)化Spring Boot中HttpServletRequest參數(shù)處理的實用工具類
聲明:本網(wǎng)頁內(nèi)容旨在傳播知識,若有侵權(quán)等問題請及時與本網(wǎng)聯(lián)系,我們將在第一時間刪除處理。郵件:2376512515@qq.com