java socket keep alive 较慢,重新打开 socket 更快

java socket keep alive is slow, reopening a socket is faster

我正在尝试实现一个简单的 HTTP 客户端的 java 实现,该客户端保持套接字打开并重新使用它来查询同一主机上的其他(或相同)URL。

我有一个使用 java.net.Socket 的简单实现,但不知何故,当我保持套接字打开时的性能比我不断创建新套接字时的性能要差。

首先是结果,下面是完整的可执行代码:

使用 KeepAlive:从迭代 #2 开始速度较慢

> java -server -Xms100M -Xmx100M -cp . KeepAlive 10 true
--- Warm up ---
18
61
60
60
78
62
59
60
59
60
Total exec time: 626
--- Run ---
26
59
60
61
60
59
60
60
62
58
Total exec time: 576

每次重新创建套接字都会得到更好的结果:

> java -server -Xms100M -Xmx100M -cp . KeepAlive 10 false
--- Warm up ---
188
34
39
33
33
33
33
33
34
33
Total exec time: 494
--- Run ---
33
35
33
34
44
34
33
34
32
34
Total exec time: 346

KeepAlive.java(独立,无依赖)

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.net.Socket;

public class KeepAlive {

    private static final String NL = "\r\n";
    private static final int READ_SIZE = 1000;
    private Socket socket;
    private DataOutputStream writer;
    private BufferedReader reader;

    public static void main(String[] args) throws Exception {
        if (args.length == 2) {
            KeepAlive ka = new KeepAlive();
            System.out.println("--- Warm up ---");
            ka.query(Integer.parseInt(args[0]), args[1].equals("true"));
            System.out.println("--- Run ---");
            ka.query(Integer.parseInt(args[0]), args[1].equals("true"));
        } else {
            System.out.println("Usage: keepAlive <n queries> <reuse socket>");
        }
    }

    private void query(int n, boolean reuseConnection) throws Exception {
        long t0 = System.currentTimeMillis();
        if (reuseConnection) {
            open();
            for (int i = 0; i < n; i++) {
                long tq0 = System.currentTimeMillis();
                query();
                System.out.println(System.currentTimeMillis() - tq0);
            }
            close();
        } else {
            for (int i = 0; i < n; i++) {
                long tq0 = System.currentTimeMillis();
                open();
                query();
                close();
                System.out.println(System.currentTimeMillis() - tq0);
            }
        }
        System.out.println("Total exec time: " + (System.currentTimeMillis() - t0));
    }

    private void open() throws Exception {
        socket = new Socket();
        socket.setKeepAlive(false);
        socket.connect(new InetSocketAddress("example.org", 80));
        writer = new DataOutputStream(socket.getOutputStream());
        reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
    }

    private void query() throws Exception {
        StringBuilder req = new StringBuilder();
        req.append("GET / HTTP/1.1").append(NL);
        req.append("Host: example.org").append(NL);
        req.append("Connection: Keep-Alive").append(NL);
        req.append(NL);
        String reqStr = req.toString();

        long t0 = System.currentTimeMillis();
        writer.writeBytes(reqStr);
        writer.flush();

        String line;
        int contentLength = 0;
        while ((line = reader.readLine()) != null) {
            if (line.startsWith("Content-Length: ")) {
                contentLength = Integer.parseInt(line.substring(16));
            }
            if (line.equals("")) {
                char[] buf = new char[contentLength];
                int offset = 0;
                while (offset < contentLength) {
                  int len = contentLength - offset;
                  if (len > READ_SIZE) {
                    len = READ_SIZE;
                  }
                  int ret = reader.read(buf, offset, len);
                  if (ret == -1) {
                    System.out.println("End of stream. Exiting");
                    System.exit(1);
                  }
                  offset += ret;
                }

                break;
            }
        }
    }

    private void close() throws Exception {
        writer.close();
        reader.close();
        socket.close();
    }
}

现在,我很确定:

  1. Web 服务器在快速处理新请求方面表现不佳 ()

  2. 我使用缓冲 reader 的方式有问题,因为那是所有时间都在浪费的地方,但看看其他可用的方法(我尝试了一些),我可以'找到我需要做的来解决这个问题...

知道如何使这项工作更快吗?也许要在服务器本身上更改配置?...


解决方案

正如下面 apangin 所解释的,较慢的性能是由默认启用的 Nagle 算法引起的。 使用 setTcpNoDelay(true),我得到更新的以下性能:

没有保活:

java -server -Xms100M -Xmx100M -cp . KeepAlive 10 false
--- Warm up ---
49
22
25
23
23
22
23
23
28
28
Total exec time: 267
--- Run ---
31
23
23
24
25
22
23
25
33
23
Total exec time: 252

带保活:

java -server -Xms100M -Xmx100M -cp . KeepAlive 10 true
--- Warm up ---
13
12
12
14
11
12
13
12
11
12
Total exec time: 168
--- Run ---
14
12
11
12
11
12
13
11
21
28
Total exec time: 158

所以在这里,我们可以看到保持活动版本在每次迭代中的表现都比非保持活动版本好得多,如果比较总执行时间也是如此。 :)

reader.read(buf);

您的测试无效。您不一定要阅读整个回复。您需要将其更改为计算返回数据的循环。如果您不阅读整个回复,您将在 keep-alive 情况下不同步。

这就是Nagle's algorithm的效果。 它延迟发送 TCP 数据包以预期有更多传出数据。

Nagle 的算法在 write-write-read 场景中与 TCP delayed acknowledgment 交互不良。 这正是你的情况,因为 writer.writeBytes(reqStr) 逐字节发送一个字符串。

现在您有两个选项来修复该行为:

  1. 使用socket.setTcpNoDelay(true)禁用Nagle算法;
  2. 一次操作发送完整的请求:writer.write(reqStr.getBytes());

在这两种情况下,重用连接预计会更快地工作。