工作填坑 - OkHttp 支持 TCP 请求
我们都知道， OkHttp 是我们常用的网络请求框架，可是都是用在了 HTTP 请求上去了，但是我们的项目中既有 HTTP ，又有 TCP 请求，我已经使用了Retrofit+OkHttp+Rxjava的架构，难道再写一套网络请求框架吗，能再这上面改造一下，也同时支持 TCP 请求吗？然后就有了这篇研究。 在这之前，对 OkHttp 了解少的，可以先查阅 源码分析 - OkHttp 对 OkHttp 源码有一个基本认识。
OkHttp 真正请求的地方是 CallServerInterceptor 中的。 所有的 Interceptor 存放到了一个集合中，一个一个遍历，但是中间都会执行到 Chain.proceed(），这样才能进去到下一个 Interceptor 中，直到 CallServerInterceptor 这个拦截器中，然后请求网络， 如果我自定义的 Interceptor 不执行 Chain.proceed(），那么就不会执行到了 CallServerInterceptor ， 并且这个 自定义的 IInterceptor 直接通过 TCP 方式请求数据，然后直接把数据返回，这样不就可以了吗？
思路有了，可是具体步骤呢？我想应该分三步骤吧！
- 从 RequestBody 中得到请求的 action 和参数
- 发送 TCP 请求得到数据
- 把请求结果封装成一个 Response 并返回
从 Request 中得到请求的 action 和参数
得到action
这个比较容易拿到的，通过 Request 可以得到 url ，然后直接调用 url 的 encodedPath() ，就行了，不过这里面有一个坑
我这边得到的参数 是 /busi/login,而真正的 action 是busi/login,所以需要截取一下。
即 String action = url.encodedPath().substring(1);
得到参数
这个就有点麻烦了，因为参数是 封装到了 RequestBody 中，所以在我们自定义的 Interceptor 中，就需要解析 RequestBody ， 下图是 RequestBody 的结构
红框里面就是我们想要的参数数据！！！
可是怎么拿到红框里面的参数呢我们知道，这 RequestBody 是一个 抽象类，这个实现类是 RequestBuilder.ContentTypeOverridingRequestBody, 可是这个类也仅仅是一个代理类而已，
private static class ContentTypeOverridingRequestBody extends RequestBody {
private final RequestBody delegate;
private final MediaType contentType;
ContentTypeOverridingRequestBody(RequestBody delegate, MediaType contentType) {
this.delegate = delegate;
this.contentType = contentType;
}
@Override public MediaType contentType() {
return contentType;
}
@Override public long contentLength() throws IOException {
return delegate.contentLength();
}
@Override public void writeTo(BufferedSink sink) throws IOException {
delegate.writeTo(sink);
}
}
我之前是想着通过添加 addFactory() ,自定义一个 RequestBody ，然后来处理的，可是发现就算我自定义一个类，也白搭，最后得到的还是这个代理类，根本不行。
忽然我看到了里面有一个 writeTo(BufferedSink sink)方法，这是干嘛的啊，之前没见过 BufferedSink 这个类。
查资料才知道，这是 Okio 中的一个类，是不是可以在这上面做文章呢？
在了解一下 BufferedSink 这个类的用法的时候，知道了 Buffer 这个类，它实现了 BufferedSink ，Buffer 这个类可以说是神一般的存在，既能读数据，又能写数据。
Buffer buffer = new Buffer();
body.writeTo(buffer);
buffer.close();//数据的真正写入是在 close() 中，之前的只是在缓存中。
创建一个 Buffer 对象，然后 writeTo() ,最后 close() ,
刚开始我使用的是 flush() ,想着和以前的 IO 操作一样，可是发现 flush() 之后并没有作用，数据还是没过来，然后我又研究 CallServerInterceptor ，看他们的数据是怎么过来的，最后发现竟然是调用 close() 之后，才会写入操作，有点意思。数据真正写入是在 close() 中，详情参考 OKio - 重新定义了“短小精悍”的 IO 框架 ,这里面有详细介绍 close() 写入数据的流程。
Buffer 转 String
写入 Buffer 后，可是怎么转 String 呢，发现了 readByteString() 转成 ByteString ,关于 ByteString ，我刚好看到过，可以直接转 String 。
ByteString byteString = buffer.readByteString();
String utf8 = byteString.utf8();
其实 Buffer 也提供了转 String 的方法，buffer.readString(Charset.forName(“utf-8”))，这样也转成了utf-8类型的字符串 然后打印结果，发现就是我想要的数据
intercept utf8: {"action":"busi/login","data":{"DevID":"M15","DevType":0,"Password":"4QrcOUm6Wau+VuBX8g+IPg==","Username":"test3"}}
那这样不就齐活了嘛。这一段代码就如下
Buffer buffer = new Buffer();
body.writeTo(buffer);
buffer.close();//数据的真正写入是在 close() 中，之前的只是在缓存中。
String utf8 = buffer.readString(Charset.forName("utf-8"));
Log.d("hoyouly", "intercept utf8: "+utf8);
Response response = null;
if (request.url() != null) {
// 得到 action 和参数 json
HttpUrl url = request.url();
//因为得到的 action 多带有一个“/”,所以需要单独处理一下，
String action = url.encodedPath().substring(1);
...
}
第一步已经完成，第二步发送数据也必将简单
发送 TCP 请求得到数据
这是之前已经现成的，直接通过 action 和参数发送即可
//调用 NettyClient.getInstance().send()
String result = NettyClient.getInstance().send(action, utf8);
关于 NettyClient 的代码，其实就是使用 Netty 框架，封装的 TCP 请求。这里就不再多说了。
还剩下最后一步，把得到的结果封装成一个 Response ，然后返回，这里面同样用到了 Okio 操作
把请求结果封装成一个Response
其实主要是还是 ResponseBody ， 在创建 Response 的时候，有一个参数是 ResponseBody ，这里面就是网络请求返回的数据，所以需要想办法封装一个 ResponseBody 。 可是查看 ResponseBody ,发现有三个参数必须要设置的， contentLength ， contentType 和 BufferedSource ，里面并没有
// Response.java
public abstract @Nullable MediaType contentType();
public abstract long contentLength();
public abstract BufferedSource source();
Type 还行，直接通过 request 就能拿到， Length 是内容长度，前面得到请求结果，也能设置长度 可是 BufferedSource 是啥啊，没见过啊，肯定还是 Okio 咯，那么怎么把 String 转换 BufferedSource 呢
前面我们能把 BufferedSink 转成一个 String ，那么也应该可以的，上面说了 Buffer 是一个神一般的存在，能读取数据，也能写入数据，那么这次就让他写吧。
String 转换 BufferedSource
buffer = new Buffer();
buffer.writeString(result, Charset.forName("utf-8"));
Source source = Okio.source(buffer.inputStream());
BufferedSource bufferedSource = Okio.buffer(source))
刚开始我还这样写，后来我发现， Buffer 就是 BufferedSource 的实现类，直接调用 Buffer 的 writeUtf8() 就行，然后 close() ,结束战斗
buffer = new Buffer();
buffer.writeUtf8(result);
buffer.close();
response = new Response.Builder()
.body(new RealResponseBody(body.contentType().type(), result.length(), buffer))
.request(request)
.protocol(Protocol.HTTP_1_0)
.code(200)
.message(action + " is success")
.build();
build()
因为 build() 中有规定，所以需要单独设置 request , protocol , code ,message
public Response build() {
if (request == null) throw new IllegalStateException("request == null");
if (protocol == null) throw new IllegalStateException("protocol == null");
if (code < 0) throw new IllegalStateException("code < 0: " + code);
if (message == null) throw new IllegalStateException("message == null");
return new Response(this);
}
- Request 容易 ，直接设置 ，
.request(request) - code也容易，请求成功，当然是 200 ，
.code(200) - message 也简单，请求成功，直接返回 success ，为了能清楚是哪个 action 的，所以带上了 action 。即
message(action + " is success")
定义protocol()
protocol 比较头疼，因为 Protocol 是一个 Enum ，一共支持的协议一共就那么几种。
public enum Protocol {
HTTP_1_0("http/1.0"),
HTTP_1_1("http/1.1"),
SPDY_3("spdy/3.1"),
HTTP_2("h2"),
H2_PRIOR_KNOWLEDGE("h2_prior_knowledge"),
QUIC("quic");
}
唯一看着像的就是 SPDY_3 ，因为 Socket 使用的是 Netty 框架，而 Netty 中是可以实现 SPDY 的，虽然我不知道这到底是啥，但是既然可以实现 SPDY ，那就 SPDY 了，可是发现这个竟然 deprecated ，不管了，先这样设置吧，好像没多大关系的。
完整代码
所以 这个 Interceptor 的代码就是
public class SocketInterceptor implements Interceptor {
@Override
public Response intercept(Chain chain) throws IOException {
Request request = chain.request();
if (request == null) {
Log.d("SocketInterceptor", "intercept request is null");
return null;
}
RequestBody body = request.body();
if (body == null) {
Log.d("SocketInterceptor", "intercept body is null");
return null;
}
Buffer buffer = new Buffer();
body.writeTo(buffer);
buffer.close();//数据的真正写入是在 close() 中，之前的只是在缓存中。
String utf8 = buffer.readString(Charset.forName("utf-8"));
Response response = null;
if (request.url() != null) {
// 得到 action 和参数 json
HttpUrl url = request.url();
String action = url.encodedPath().substring(1);
try {
//调用 NettyClient.getInstance().send()
String result = NettyClient.getInstance().send(action, utf8);
if (result == null) {
Log.d("SocketInterceptor", "intercept result is null");
return null;
}
//把得到的结果 封装成一个 response 返回即可
buffer = new Buffer();
buffer.writeUtf8(result);
buffer.close();
response = new Response.Builder()
.body(new RealResponseBody(body.contentType().type(), result.length(), buffer))
.request(request)
.protocol(Protocol.SPDY_3)
.code(200)
.message(action + " is success")
.build();
} catch (Exception e) {
e.printStackTrace();
}
}
return response;
}
}
在创建 Okhttp 的时候， addInterceptor() 上去就行了。
httpClientBuilder.addInterceptor(new SocketInterceptor());
注意，不能使用 addNetworkInterceptor() ,因为会报错，原因还在查找中。
搬运地址：
既已览卷至此，何不品评一二：