1. 程式人生 > 實用技巧 >SpringBoot(四)——http請求工具類

SpringBoot(四)——http請求工具類

記錄一下單元測試中使用的http工具類,一般只需要留意路徑、請求體、請求頭。至少目前遇到的是這樣。

GetMapping沒有請求體,傳參的話直接用用?和&連線寫在請求路徑裡,例如"?id=10086&age=22"

PostMapping有請求體,以JSON格式傳遞的

1.路徑

例如,[ http://localhost:8080 ],這是基本路徑,一般在公共接口裡定義。後續的RequestMapping和GetMapping等具體路徑在測試方法裡自行拼接。

2.請求頭一般攜帶金鑰,一般在公共接口裡定義,其他測試類直接實現獲取請求頭。


import java.util.HashMap;
import java.util.Map;
public interface TestBase {
    //String IP = "10.8.45.120";
    String IP = "localhost";
    int PORT = 8103;
    String BAST_PATH = "http://" + IP + ":" + (PORT);

    //預設的金鑰
    String AUTHORIZATION = "Basic c2FiZXI6c2FV0";
    String SCC_AUTH = "bearer eyJ0eXAiOiJKc29uV2ViVG9rZ";

    
default Map<String, String> getHead() { Map<String, String> head = new HashMap<>(); head.put("authorization", AUTHORIZATION); head.put("scc-auth", SCC_AUTH); return head; } }

3.封裝get請求和post請求於一個工具類

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.MapUtils;
import org.apache.http.HttpEntity; import org.apache.http.client.config.RequestConfig; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpGet; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClients; import org.apache.http.util.EntityUtils; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.web.client.RestTemplate; import java.io.IOException; import java.util.Map; /** * http 請求工具類 */ @Slf4j public final class HttpRequestUtils { //設定字元編碼 private static final String CHARSET = "UTF-8"; private static RequestConfig defaultRequestConfig = RequestConfig .custom() //設定等待資料超時時間 .setSocketTimeout(30000) //設定連線超時時間 .setConnectTimeout(30000) //設定從連線池獲取連線的等待超時時間 .setConnectionRequestTimeout(30000) //.setStaleConnectionCheckEnabled(true) .build(); //釋放資源,httpResponse為響應流,httpClient為請求客戶端 private static void release(CloseableHttpResponse httpResponse, CloseableHttpClient httpClient) throws IOException { if (httpResponse != null) { httpResponse.close(); } if (httpClient != null) { httpClient.close(); } } //get請求帶引數、帶請求頭 public static String get(String urlWithParams, Map<String, String> header) throws IOException { CloseableHttpClient httpClient = HttpClients.createDefault(); HttpGet httpget = new HttpGet(urlWithParams); if (!MapUtils.isEmpty(header)) { header.forEach(httpget::addHeader); } CloseableHttpResponse response = httpClient.execute(httpget); HttpEntity entity = response.getEntity(); String result = EntityUtils.toString(entity, CHARSET); httpget.releaseConnection(); release(response, httpClient); return result; } public static String get(String urlWithParams) throws IOException { return get(urlWithParams, null); } //傳送post請求,帶json請求體和請求頭 public static String postJson(String url, String json, Map<String, String> headersMap) { RestTemplate restTemplate = new RestTemplate(); HttpHeaders headers = new HttpHeaders(); headers.setContentType(MediaType.APPLICATION_JSON); for (Map.Entry<String, String> entry : headersMap.entrySet()) { headers.add(entry.getKey(),entry.getValue()); } org.springframework.http.HttpEntity<String> request = new org.springframework.http.HttpEntity<>(json, headers); ResponseEntity<String> response = restTemplate.postForEntity(url, request, String.class); return response.getBody(); } }

4.單元測試

import org.junit.jupiter.api.Test;

import java.io.IOException;

public class GoodsCategoryMapperTest implements TestBase{
    private static final String PRE = BAST_PATH + "/goodsCategory";

    @Test
    public void test1() throws IOException {
        String url = "/getGoodsCategory?category_id=29";
        String s = HttpRequestUtils.get(PRE + url, getHead());
        System.out.println(s);
    }

    @Test
    public void test2() throws IOException {
        String url = "/getGoodsCategoryListInfo?page_index=1&page_size=10";
        String params = "{\n" +
            "    \"brandId\":\"1000000005\"\n" +
            "}";
        String s = HttpRequestUtils.postJson(PRE + url, params, getHead());
        System.out.println(s);
    }
}

5.執行截圖