使用 Go 实现 Unix Socket 通信：一个简单的回声服务器教程

碧海醫心

发布时间：2025-07-15 16:06:25

481人浏览过

来源于php中文网

原创

使用 go 实现 unix socket 通信：一个简单的回声服务器教程

本文将引导你使用 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 代码：

BgSub

免费的AI图片背景去除工具

下载

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) // 每秒发送一次数据
    }
}

代码解释: