深海游弋的鱼 – 默默的点滴

Android 客户端使用 OKHTTP 的时候，需要通过修改底层的 Socket 来解决问题，比如设置 TCP 的心跳、修改TCP 的滑动窗口算法等。

SocketFactory

import java.io.IOException;
import java.net.InetAddress;
import java.net.Socket;

import javax.net.SocketFactory;

public class SocketFactoryTcpNoDelay extends SocketFactory {
    private final SocketFactory socketFactory;

    public SocketFactoryTcpNoDelay() {
        socketFactory = SocketFactory.getDefault();
    }

    @Override
    public Socket createSocket() throws IOException {
        Socket socket = socketFactory.createSocket();
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(String host, int port) throws IOException {
        Socket socket = socketFactory.createSocket(host, port);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(String host, int port, InetAddress localHost, int localPort) throws
            IOException {
        Socket socket = socketFactory.createSocket(host, port, localHost, localPort);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(InetAddress host, int port) throws IOException {
        Socket socket = socketFactory.createSocket(host, port);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(InetAddress address, int port, InetAddress localAddress,
                               int localPort) throws IOException {
        Socket socket = socketFactory.createSocket(address, port, localAddress, localPort);
        socket.setTcpNoDelay(true);
        return socket;
    }
}

import java.io.IOException;

import java.net.InetAddress;

import java.net.Socket;

import javax.net.SocketFactory;

public class SocketFactoryTcpNoDelay extends SocketFactory {

private final SocketFactory socketFactory;

public SocketFactoryTcpNoDelay() {

socketFactory = SocketFactory.getDefault();

}

@Override

public Socket createSocket() throws IOException {

Socket socket = socketFactory.createSocket();

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(String host, int port) throws IOException {

Socket socket = socketFactory.createSocket(host, port);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(String host, int port, InetAddress localHost, int localPort) throws

IOException {

Socket socket = socketFactory.createSocket(host, port, localHost, localPort);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(InetAddress host, int port) throws IOException {

Socket socket = socketFactory.createSocket(host, port);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(InetAddress address, int port, InetAddress localAddress,

int localPort) throws IOException {

Socket socket = socketFactory.createSocket(address, port, localAddress, localPort);

socket.setTcpNoDelay(true);

return socket;

}

SSLSocketFactory

import java.io.IOException;
import java.net.InetAddress;
import java.net.Socket;
import java.security.KeyManagementException;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;
import javax.net.ssl.TrustManagerFactory;
import javax.net.ssl.X509TrustManager;

public class SSLSocketFactoryTcpNoDelay extends SSLSocketFactory {
    private final X509TrustManager trustManager;
    private final SSLSocketFactory sslSocketFactory;

    public SSLSocketFactoryTcpNoDelay() throws NoSuchAlgorithmException, KeyStoreException,
            KeyManagementException {
        // From
        // https://square.github.io/okhttp/4.x/okhttp/okhttp3/-ok-http-client/-builder/ssl-socket-factory
        TrustManagerFactory trustManagerFactory = TrustManagerFactory.getInstance(
                TrustManagerFactory.getDefaultAlgorithm());
        trustManagerFactory.init((KeyStore) null);
        TrustManager[] trustManagers = trustManagerFactory.getTrustManagers();
        if (trustManagers.length != 1 || !(trustManagers[0] instanceof X509TrustManager)) {
            throw new IllegalStateException("Unexpected default trust managers:"
                    + Arrays.toString(trustManagers));
        }
        trustManager = (X509TrustManager) trustManagers[0];

        SSLContext sslContext = SSLContext.getInstance("TLS");
        sslContext.init(null, new TrustManager[] { trustManager }, null);
        sslSocketFactory = sslContext.getSocketFactory();
    }

    public X509TrustManager getTrustManager() {
        return trustManager;
    }

    public SSLSocketFactory getSslSocketFactory() {
        return sslSocketFactory;
    }

    @Override
    public String[] getDefaultCipherSuites() {
        return sslSocketFactory.getDefaultCipherSuites();
    }

    @Override
    public String[] getSupportedCipherSuites() {
        return sslSocketFactory.getSupportedCipherSuites();
    }

    @Override
    public Socket createSocket() throws IOException {
        Socket socket = sslSocketFactory.createSocket();
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(Socket s, String host, int port, boolean autoClose) throws
            IOException {
        Socket socket = sslSocketFactory.createSocket(s, host, port, autoClose);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(String host, int port) throws IOException {
        Socket socket = sslSocketFactory.createSocket(host, port);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(String host, int port, InetAddress localHost, int localPort) throws
            IOException {
        Socket socket = sslSocketFactory.createSocket(host, port, localHost, localPort);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(InetAddress host, int port) throws IOException {
        Socket socket = sslSocketFactory.createSocket(host, port);
        socket.setTcpNoDelay(true);
        return socket;
    }

    @Override
    public Socket createSocket(InetAddress address, int port, InetAddress localAddress,
                               int localPort) throws IOException {
        Socket socket = sslSocketFactory.createSocket(address, port, localAddress, localPort);
        socket.setTcpNoDelay(true);
        return socket;
    }
}

100

101

import java.io.IOException;

import java.net.InetAddress;

import java.net.Socket;

import java.security.KeyManagementException;

import java.security.KeyStore;

import java.security.KeyStoreException;

import java.security.NoSuchAlgorithmException;

import java.util.Arrays;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLSocketFactory;

import javax.net.ssl.TrustManager;

import javax.net.ssl.TrustManagerFactory;

import javax.net.ssl.X509TrustManager;

public class SSLSocketFactoryTcpNoDelay extends SSLSocketFactory {

private final X509TrustManager trustManager;

private final SSLSocketFactory sslSocketFactory;

public SSLSocketFactoryTcpNoDelay() throws NoSuchAlgorithmException, KeyStoreException,

KeyManagementException {

// From

// https://square.github.io/okhttp/4.x/okhttp/okhttp3/-ok-http-client/-builder/ssl-socket-factory

TrustManagerFactory trustManagerFactory = TrustManagerFactory.getInstance(

TrustManagerFactory.getDefaultAlgorithm());

trustManagerFactory.init((KeyStore) null);

TrustManager[] trustManagers = trustManagerFactory.getTrustManagers();

if (trustManagers.length != 1 || !(trustManagers[0] instanceof X509TrustManager)) {

throw new IllegalStateException("Unexpected default trust managers:"

+ Arrays.toString(trustManagers));

}

trustManager = (X509TrustManager) trustManagers[0];

SSLContext sslContext = SSLContext.getInstance("TLS");

sslContext.init(null, new TrustManager[] { trustManager }, null);

sslSocketFactory = sslContext.getSocketFactory();

}

public X509TrustManager getTrustManager() {

return trustManager;

}

public SSLSocketFactory getSslSocketFactory() {

return sslSocketFactory;

}

@Override

public String[] getDefaultCipherSuites() {

return sslSocketFactory.getDefaultCipherSuites();

}

@Override

public String[] getSupportedCipherSuites() {

return sslSocketFactory.getSupportedCipherSuites();

}

@Override

public Socket createSocket() throws IOException {

Socket socket = sslSocketFactory.createSocket();

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(Socket s, String host, int port, boolean autoClose) throws

IOException {

Socket socket = sslSocketFactory.createSocket(s, host, port, autoClose);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(String host, int port) throws IOException {

Socket socket = sslSocketFactory.createSocket(host, port);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(String host, int port, InetAddress localHost, int localPort) throws

IOException {

Socket socket = sslSocketFactory.createSocket(host, port, localHost, localPort);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(InetAddress host, int port) throws IOException {

Socket socket = sslSocketFactory.createSocket(host, port);

socket.setTcpNoDelay(true);

return socket;

}

@Override

public Socket createSocket(InetAddress address, int port, InetAddress localAddress,

int localPort) throws IOException {

Socket socket = sslSocketFactory.createSocket(address, port, localAddress, localPort);

socket.setTcpNoDelay(true);

return socket;

}

Putting It Together

You can create an OkHttpClient that uses these classes as follows:

OkHttpClient.Builder okHttpClientBuilder = new OkHttpClient.Builder();
try {
    SSLSocketFactoryTcpNoDelay sSLSocketFactoryTcpNoDelay =
     new SSLSocketFactoryTcpNoDelay();
    okHttpClientBuilder.sslSocketFactory(sSLSocketFactoryTcpNoDelay.getSslSocketFactory(),
     sSLSocketFactoryTcpNoDelay.getTrustManager());
} catch (NoSuchAlgorithmException | KeyStoreException | KeyManagementException e) {
    // Handle Exception
}
SocketFactoryTcpNoDelay socketFactoryTcpNoDelay = new SocketFactoryTcpNoDelay();
okHttpClientBuilder.socketFactory(socketFactoryTcpNoDelay);
OkHttpClient okHttpClient = okHttpClientBuilder.build();

OkHttpClient.Builder okHttpClientBuilder = new OkHttpClient.Builder();

try {

SSLSocketFactoryTcpNoDelay sSLSocketFactoryTcpNoDelay =

new SSLSocketFactoryTcpNoDelay();

okHttpClientBuilder.sslSocketFactory(sSLSocketFactoryTcpNoDelay.getSslSocketFactory(),

sSLSocketFactoryTcpNoDelay.getTrustManager());

} catch (NoSuchAlgorithmException | KeyStoreException | KeyManagementException e) {

// Handle Exception

}

SocketFactoryTcpNoDelay socketFactoryTcpNoDelay = new SocketFactoryTcpNoDelay();

okHttpClientBuilder.socketFactory(socketFactoryTcpNoDelay);

OkHttpClient okHttpClient = okHttpClientBuilder.build();

WebSocket connections that your OkHttpClient makes will have Nagle’s Algorithm disabled.

Scarlet

Scarlet is a library from Tinder that uses OkHttp for WebSocket connections. If you pass the OkHttpClient that you created above to OkHttpClientUtils#newWebSocketFactory, all WebSocket connections made with Scarlet will have Nagle’s Algorithm disabled.

Note that all Scarlet WebSocket connections use OkHttp. As of writing this post, Scarlet is still using OkHttp version 3.11. OkHttp 4 has been out for over a year. If you are starting a new project, consider using OkHttp directly, rather than using Scarlet.

参考链接

Disable Nagle’s Algorithm with OkHttp or Scarlet

一	二	三	四	五	六	日
						1
2	3	4	5	6	7	8
9	10	11	12	13	14	15
16	17	18	19	20	21	22
23	24	25	26	27	28	29
30

Disable Nagle’s Algorithm with OkHttp or Scarlet

SocketFactory

SSLSocketFactory

Putting It Together

Scarlet

参考链接

发布者

默默

发表回复取消回复

SocketFactory

SSLSocketFactory

Putting It Together

Scarlet

参考链接

发布者

默默

发表回复 取消回复

发表回复取消回复