安卓開發中，http 網絡請求的框架有很多，有 HttpUrlConnection、 Volley 、 Okhttp很多種，現在由于 retrofit 和 rxjava 的出現， Okhttp 的使用越來越多，因為 Retrofit 也僅僅是在 Okhttp 做的一層封裝，那么需求上需要我們在每一個 URL 請求中添加一個固定的參數時候，或者將請求的消息頭統一改為我們需要的形式，我們要怎么實現呢？
我們用過 OkHttp 的都知道它獨特而且強大的攔截器功能，比如 提供的 HttpLoggingInterceptor()，我們也可以通過自定義 攔截器來實現上述的公共參數需求。首先我們實現 Interceptor 接口
public interface Interceptor {
Response intercept(Chain chain) throws IOException;
interface Chain {
Request request();
Response proceed(Request request) throws IOException;
/**
* Returns the connection the request will be executed on. This is only available in the chains
* of network interceptors; for application interceptors this is always null.
*/
@Nullable Connection connection();
}
}
在 intercept (Chain chain) 方法中 Chain 對象可以拿到當前請求的 Request 對象，然后我們可以對Request做二次處理，最后生成我們需要的請求，然后再通過網絡發送請求到服務端，這樣就完成了一次攔截。由于代碼很簡單，我們先介紹使用方法，源碼貼在了后面：
//添加post公共請求參數 Constants.PACKAGE 和 Constants.CFROM
BasicParamsInterceptor basicParamsInterceptor = new BasicParamsInterceptor.Builder()
.addParam("from", "android") //添加公共參數到 post 請求體
.addQueryParam("version","1") // 添加公共版本號，加在 URL 后面
.addHeaderLine("X-Ping: Pong") // 示例： 添加公共消息頭
.addParamsMap(map) // 可以添加 Map 格式的參數
.build();
然后在 OkHttpClient中添加攔截器就完成了
okHttpClient = new OkHttpClient.Builder()
.readTimeout(READ_TIME_OUT, TimeUnit.MILLISECONDS)
.connectTimeout(CONNECT_TIME_OUT, TimeUnit.MILLISECONDS)
.writeTimeout(CONNECT_TIME_OUT, TimeUnit.SECONDS)
.addInterceptor(basicParamsInterceptor) // 添加公共參數攔截器
.addInterceptor(logInterceptor)
.addInterceptor(noNetInterceptor)
.addNetworkInterceptor(cacheInterceptor)
.retryOnConnectionFailure(true)
.cache(cache)
.build();
整個攔截器源碼如下：
/**
* Created by : Liu
* date: 2017/2/16 14:04 <br/>
*/
public class BasicParamsInterceptor implements Interceptor {
Map<String, String> queryParamsMap = new HashMap<>(); // 添加到 URL 末尾，Get Post 方法都使用
Map<String, String> paramsMap = new HashMap<>(); // 添加到公共參數到消息體，適用 Post 請求
Map<String, String> headerParamsMap = new HashMap<>(); // 公共 Headers 添加
List<String> headerLinesList = new ArrayList<>(); // 消息頭 集合形式，一次添加一行
// 私有構造器
private BasicParamsInterceptor() {}
@Override
public Response intercept(Chain chain) throws IOException {
Request request = chain.request();
Request.Builder requestBuilder = request.newBuilder();
// process header params inject
Headers.Builder headerBuilder = request.headers().newBuilder();
// 以 Entry 添加消息頭
if (headerParamsMap.size() > 0) {
Iterator iterator = headerParamsMap.entrySet().iterator();
while (iterator.hasNext()) {
Map.Entry entry = (Map.Entry) iterator.next();
headerBuilder.add((String) entry.getKey(), (String) entry.getValue());
}
}
// 以 String 形式添加消息頭
if (headerLinesList.size() > 0) {
for (String line: headerLinesList) {
headerBuilder.add(line);
}
requestBuilder.headers(headerBuilder.build());
}
// process header params end
// process queryParams inject whatever it's GET or POST
if (queryParamsMap.size() > 0) {
request = injectParamsIntoUrl(request.url().newBuilder(), requestBuilder, queryParamsMap);
}
// process post body inject
if (paramsMap.size() > 0) {
if (canInjectIntoBody(request)) {
FormBody.Builder formBodyBuilder = new FormBody.Builder();
for(Map.Entry<String, String> entry : paramsMap.entrySet()) {
formBodyBuilder.add((String) entry.getKey(), (String) entry.getValue());
}
RequestBody formBody = formBodyBuilder.build();
String postBodyString = bodyToString(request.body());
postBodyString += ((postBodyString.length() > 0) ? "&" : "") + bodyToString(formBody);
requestBuilder.post(RequestBody.create(MediaType.parse("application/x-www-form-urlencoded;charset=UTF-8"), postBodyString));
}
}
request = requestBuilder.build();
return chain.proceed(request);
}
/**
* 確認是否是 post 請求
* @param request 發出的請求
* @return true 需要注入公共參數
*/
private boolean canInjectIntoBody(Request request) {
if (request == null) {
return false;
}
if (!TextUtils.equals(request.method(), "POST")) {
return false;
}
RequestBody body = request.body();
if (body == null) {
return false;
}
MediaType mediaType = body.contentType();
if (mediaType == null) {
return false;
}
if (!TextUtils.equals(mediaType.subtype(), "x-www-form-urlencoded")) {
return false;
}
return true;
}
// func to inject params into url
private Request injectParamsIntoUrl(HttpUrl.Builder httpUrlBuilder, Request.Builder requestBuilder, Map<String, String> paramsMap) {
if (paramsMap.size() > 0) {
Iterator iterator = paramsMap.entrySet().iterator();
while (iterator.hasNext()) {
Map.Entry entry = (Map.Entry) iterator.next();
httpUrlBuilder.addQueryParameter((String) entry.getKey(), (String) entry.getValue());
}
requestBuilder.url(httpUrlBuilder.build());
return requestBuilder.build();
}
return null;
}
private static String bodyToString(final RequestBody request){
try {
final RequestBody copy = request;
final Buffer buffer = new Buffer();
if(copy != null)
copy.writeTo(buffer);
else
return "";
return buffer.readUtf8();
}
catch (final IOException e) {
return "did not work";
}
}
public static class Builder {
BasicParamsInterceptor interceptor;
public Builder() {
interceptor = new BasicParamsInterceptor();
}
// 添加公共參數到 post 消息體
public Builder addParam(String key, String value) {
interceptor.paramsMap.put(key, value);
return this;
}
// 添加公共參數到 post 消息體
public Builder addParamsMap(Map<String, String> paramsMap) {
interceptor.paramsMap.putAll(paramsMap);
return this;
}
// 添加公共參數到消息頭
public Builder addHeaderParam(String key, String value) {
interceptor.headerParamsMap.put(key, value);
return this;
}
// 添加公共參數到消息頭
public Builder addHeaderParamsMap(Map<String, String> headerParamsMap) {
interceptor.headerParamsMap.putAll(headerParamsMap);
return this;
}
// 添加公共參數到消息頭
public Builder addHeaderLine(String headerLine) {
int index = headerLine.indexOf(":");
if (index == -1) {
throw new IllegalArgumentException("Unexpected header: " + headerLine);
}
interceptor.headerLinesList.add(headerLine);
return this;
}
// 添加公共參數到消息頭
public Builder addHeaderLinesList(List<String> headerLinesList) {
for (String headerLine: headerLinesList) {
int index = headerLine.indexOf(":");
if (index == -1) {
throw new IllegalArgumentException("Unexpected header: " + headerLine);
}
interceptor.headerLinesList.add(headerLine);
}
return this;
}
// 添加公共參數到 URL
public Builder addQueryParam(String key, String value) {
interceptor.queryParamsMap.put(key, value);
return this;
}
// 添加公共參數到 URL
public Builder addQueryParamsMap(Map<String, String> queryParamsMap) {
interceptor.queryParamsMap.putAll(queryParamsMap);
return this;
}
public BasicParamsInterceptor build() {
return interceptor;
}
}
}
