概述
在go语言中构建网络服务器是常见的任务。当我们需要处理基于行的协议，例如某些简单的文本协议时，从tcp连接中逐行读取数据并进行处理就变得尤为重要。本教程将展示如何实现一个tcp服务器，该服务器监听特定端口，接受客户端连接，并将其发送的每一行数据直接输出到服务器的控制台。
核心问题：从net.Conn中逐行读取
net.Conn接口提供了读写字节流的能力，但它本身并没有直接提供按行读取的方法。客户端发送的数据通常是字节流，可能包含多个行，每行以换行符（\n）结束。为了实现逐行读取，我们需要一个能够缓冲并识别行分隔符的工具。
解决方案：使用bufio.Reader
Go标准库中的bufio包提供了一个Reader类型，它能够包装一个io.Reader（net.Conn实现了io.Reader接口），并提供缓冲和高级的读取方法，其中就包括ReadString。
bufio.Reader的ReadString(delim byte)方法会从输入流中读取数据，直到遇到指定的delim字节（例如\n），或者遇到错误（包括io.EOF）。它会返回读取到的字符串（包含分隔符）以及可能发生的错误。
实现步骤
- 监听TCP端口： 使用net.Listen("tcp", ":端口号")来创建一个TCP监听器。
- 接受客户端连接： 在一个无限循环中，调用srv.Accept()来接受新的客户端连接。每个连接都应该在一个独立的 Goroutine 中处理，以避免阻塞主 Goroutine 并支持并发连接。
- 创建bufio.Reader： 在处理每个连接的 Goroutine 中，将net.Conn包装成一个bufio.Reader实例。
- 逐行读取数据： 使用一个循环调用bufio.Reader的ReadString('\n')方法来读取每一行数据。
- 处理错误和EOF： 检查ReadString返回的错误。如果错误是io.EOF，表示客户端关闭了连接，此时应该跳出循环。其他错误则需要进行适当处理。
- 打印到标准输出： 将读取到的每一行数据使用fmt.Print()打印到服务器的标准输出。
- 关闭连接： 在Goroutine结束时，确保调用net.Conn的Close()方法来释放资源。
完整代码示例
下面是实现上述功能的Go语言代码：
立即学习“go语言免费学习笔记（深入）”；
package main
import (
"bufio"
"fmt"
"io"
"log"
"net"
)
func main() {
// 监听TCP端口2000
srv, err := net.Listen("tcp", ":2000")
if err != nil {
log.Fatalf("无法监听端口: %v", err)
}
defer srv.Close() // 确保在main函数退出时关闭监听器
fmt.Println("TCP服务器已启动，监听端口 2000...")
for {
// 接受新的客户端连接
conn, err := srv.Accept()
if err != nil {
log.Printf("接受连接失败: %v", err)
continue // 继续接受下一个连接
}
fmt.Printf("新客户端连接来自: %s\n", conn.RemoteAddr())
// 为每个连接启动一个Goroutine进行处理
go handleConnection(conn)
}
}
// handleConnection 处理单个客户端连接
func handleConnection(c net.Conn) {
defer c.Close() // 确保在函数退出时关闭客户端连接
fmt.Printf("开始处理连接 %s\n", c.RemoteAddr())
// 将net.Conn包装成bufio.Reader，以便逐行读取
reader := bufio.NewReader(c)
for {
// 逐行读取数据，直到遇到换行符 '\n'
line, err := reader.ReadString('\n')
if err == io.EOF {
// 客户端关闭了连接
fmt.Printf("客户端 %s 已断开连接。\n", c.RemoteAddr())
break
} else if err != nil {
// 其他读取错误
log.Printf("从 %s 读取数据时发生错误: %v", c.RemoteAddr(), err)
break
}
// 成功读取到一行数据，打印到服务器的标准输出
// fmt.Print() 会直接输出字符串，包括其中的换行符
fmt.Print(line)
}
fmt.Printf("连接 %s 处理结束。\n", c.RemoteAddr())
}运行与测试
- 保存代码： 将上述代码保存为 server.go 文件。
-
编译： 打开终端，导航到文件所在目录，然后运行：
go build -o server server.go
-
运行服务器：
./server
您将看到输出 TCP服务器已启动，监听端口 2000...
-
使用 telnet 测试： 打开另一个终端，连接到服务器：
telnet localhost 2000
您将看到 Connected to localhost. 等信息。
-
输入数据： 在 telnet 客户端中输入几行文字，每行按回车键结束：
test 123 foobar Go is great!
当您输入并按回车后，这些行将立即显示在运行 ./server 的那个终端窗口中。
- 关闭 telnet： 您可以通过输入 ^] (Ctrl + ]) 然后输入 quit 来退出 telnet。服务器端会打印 客户端 127.0.0.1:xxxxx 已断开连接。
注意事项与最佳实践
- 错误处理： 示例代码中包含了基本的错误处理，如监听失败、接受连接失败以及读取数据时的错误。在生产环境中，应采用更健壮的错误处理机制，例如记录错误日志、返回特定的错误码或优雅地关闭连接，而不是简单地使用 panic。
- 资源管理： 务必使用 defer srv.Close() 和 defer c.Close() 来确保在函数退出时正确关闭监听器和客户端连接，防止资源泄露。
- 标准输出同步： Go语言的 fmt.Print 函数通常是线程安全的，但在高并发场景下，多个 Goroutine 同时向标准输出写入可能会导致输出交错，影响可读性。如果对输出顺序有严格要求，可以考虑使用一个共享的通道（channel）来收集所有 Goroutine 的输出，然后在一个独立的 Goroutine 中统一写入标准输出。
- 缓冲区大小： bufio.NewReader 默认会使用一个内置的缓冲区。对于大量或非常长的行，这个缓冲区大小可能需要根据实际情况进行调整，以优化性能。
- 协议设计： 本示例适用于简单的文本协议。对于更复杂的二进制协议或需要更多状态管理的协议，可能需要实现自定义的读取逻辑或使用特定的协议库。
总结
通过本教程，我们学习了如何利用Go语言的 net 包和 bufio 包构建一个简单的TCP服务器，该服务器能够高效地接收客户端连接，并逐行读取其发送的数据并输出到标准输出。掌握 bufio.Reader 的 ReadString 方法是处理基于行文本协议的关键。在实际应用中，请务必考虑完善的错误处理、资源管理以及在高并发场景下的输出同步问题。
