Question

我在golang中编写两个服务，需要每秒发送大约200万条消息。每条消息大约50个字节，因此吞吐量应该只有大约100MB / s。我想用tcp来做这件事。但是，结果非常缓慢。我配置了SetNoDelay（false）以确保在发送之前缓冲数据，但这没有任何区别。

我每秒只能发送大约50k个消息，并且消息大小并不重要，所以我假设代码在某处阻塞。这是我的测试代码：

package main

import "net"
import "fmt"
import "bufio"
import (
    //"strings"
    "time"
)

func startserver() {
    fmt.Println("Launching server...")
    ln, _ := net.Listen("tcp", ":8081")
    conn, _ := ln.Accept()

    for {
        bufio.NewReader(conn).ReadString('\n')
        //fmt.Println(message)
        //newmessage := strings.ToUpper(message)
        //conn.Write([]byte(newmessage + "\n"))
    }
}

func startclient() {
    time.Sleep(time.Second) // so that server has time to start
    servAddr := "127.0.0.1:8081"
    tcpAddr, _ := net.ResolveTCPAddr("tcp", servAddr)
    conn, _ := net.DialTCP("tcp", nil, tcpAddr)
    conn.SetNoDelay(false)
    conn.SetWriteBuffer(10000)
    msg := "abc\n"
    start := time.Now()
    for i := 0; i < 1000000; i++ {
        conn.Write([]byte(msg))
        //bufio.NewReader(conn).ReadString('\n')
        //fmt.Print("Message from server: ", response)
    }
    fmt.Println("took:", time.Since(start))
}

func main() {
    go startserver()
    startclient()
}

有什么建议吗？

Answer 1

Syscalls价格昂贵，而且通过环回写入数据的速度非常快，因此这实际上只是计算对write进行100万次调用的速度。

进行大量小读取和写入是使用缓冲io的主要用例。将正常net.TCPConn包裹在bufio.Writer中将提供预期的性能提升。

在golang中通过tcp套接字发送数百万条短消息

1 个答案: