本文将引导你使用 Go 语言创建一个简单的基于 Unix Socket 的回声客户端和服务器。我们将重点讲解服务器和客户端的实现，以及如何处理数据读写，并提供完整的代码示例，帮助你理解 Unix Socket 在 Go 中的应用。通过本文，你将掌握使用 Unix Socket 进行进程间通信的基础知识。
Unix Socket 简介
Unix Socket 是一种进程间通信（IPC）机制，它允许在同一台机器上的不同进程之间进行数据交换。与 TCP Socket 不同，Unix Socket 使用文件系统路径名作为地址，而不是 IP 地址和端口号。这使得 Unix Socket 在本地进程通信时更加高效和安全。
创建 Unix Socket 回声服务器
以下是一个简单的 Unix Socket 回声服务器的 Go 代码：
package main
import (
"log"
"net"
)
func echoServer(c net.Conn) {
defer c.Close() // 确保连接关闭
for {
buf := make([]byte, 512)
nr, err := c.Read(buf)
if err != nil {
log.Println("Read error:", err)
return // 客户端断开连接或发生错误时退出循环
}
data := buf[:nr]
log.Printf("Server received: %s", string(data))
_, err = c.Write(data) // 将收到的数据写回客户端
if err != nil {
log.Println("Write error:", err)
return // 客户端断开连接或发生错误时退出循环
}
}
}
func main() {
socketPath := "/tmp/echo.sock" // 定义 socket 文件路径
l, err := net.Listen("unix", socketPath) // 监听 Unix Socket
if err != nil {
log.Fatalf("Listen error: %v", err)
}
defer l.Close() // 确保 listener 关闭
log.Println("Server listening on", socketPath)
for {
fd, err := l.Accept() // 接受客户端连接
if err != nil {
log.Println("Accept error:", err)
continue // 继续监听新的连接
}
go echoServer(fd) // 为每个连接启动一个 goroutine
}
}代码解释:
- net.Listen("unix", socketPath): 创建一个 Unix Socket 监听器，socketPath 指定了 socket 文件的路径。
- l.Accept(): 接受客户端连接。
- echoServer(fd): 为每个客户端连接启动一个 goroutine，处理客户端的请求。
- c.Read(buf): 从客户端读取数据。
- c.Write(data): 将数据写回客户端。
- defer c.Close() and defer l.Close(): 使用 defer 确保在函数退出时关闭连接和监听器，释放资源。
注意事项:
- 确保 /tmp/echo.sock 文件不存在，否则 net.Listen 会失败。你可以手动删除它，或者在程序启动前使用 os.Remove("/tmp/echo.sock") 删除。
- 错误处理是关键。在实际应用中，需要更完善的错误处理机制。
创建 Unix Socket 客户端
以下是一个简单的 Unix Socket 客户端的 Go 代码：
package main
import (
"io"
"log"
"net"
"time"
)
func reader(r io.Reader) {
buf := make([]byte, 1024)
for {
n, err := r.Read(buf[:])
if err != nil {
if err != io.EOF {
log.Println("Read error:", err)
}
return
}
log.Printf("Client received: %s", string(buf[0:n]))
}
}
func main() {
socketPath := "/tmp/echo.sock" // 定义 socket 文件路径
c, err := net.Dial("unix", socketPath) // 连接到 Unix Socket
if err != nil {
log.Fatalf("Dial error: %v", err)
}
defer c.Close() // 确保连接关闭
go reader(c) // 启动 goroutine 读取服务器的响应
for {
_, err := c.Write([]byte("hi\n")) // 向服务器发送数据
if err != nil {
log.Println("Write error:", err)
break // 发生错误时退出循环
}
time.Sleep(1 * time.Second) // 每秒发送一次数据
}
}代码解释:
- net.Dial("unix", socketPath): 连接到指定的 Unix Socket。
- reader(c): 启动一个 goroutine 用于读取服务器的响应。这避免了客户端在发送数据后阻塞等待响应。
- c.Write([]byte("hi\n")): 向服务器发送数据。
- *`time.Sleep(1 time.Second)`:** 每秒发送一次数据，避免客户端过快地发送数据。
- defer c.Close(): 使用 defer 确保在函数退出时关闭连接，释放资源。
注意事项:
- 客户端也需要处理可能的错误，例如连接错误、读写错误等。
- reader goroutine 需要处理 io.EOF 错误，这是服务器关闭连接时返回的错误。
运行示例
- 保存服务器代码为 server.go，客户端代码为 client.go。
- 编译并运行服务器：go run server.go
- 编译并运行客户端：go run client.go
你应该能在服务器和客户端的控制台中看到相应的输出，表明客户端成功连接到服务器，并成功发送和接收数据。
总结
本文介绍了如何使用 Go 语言创建基于 Unix Socket 的回声客户端和服务器。通过这个简单的示例，你了解了 Unix Socket 的基本概念和使用方法。在实际应用中，你可以根据需要扩展这个示例，实现更复杂的功能。Unix Sockets 在高性能、本地进程间通信的场景中非常有用。
