通过 Socket 在 Go 中检索文件数据-Golang-PHP中文网

通过 Socket 在 Go 中检索文件数据

花韻仙語

发布： 2025-08-12 22:02:20

原创

594人浏览过

通过 socket 在 go 中检索文件数据

在 Go 语言中使用 Socket 进行网络编程时，尤其是在传输文件等大数据时，需要特别注意 TCP 协议的流式特性。TCP 协议是面向连接的、可靠的、基于字节流的传输层通信协议。这意味着数据在传输过程中会被拆分成多个数据包，接收端无法保证一次 Read 操作就能接收到完整的数据。因此，需要一种机制来确定消息的边界，从而正确地解析接收到的数据。

最常见的解决方案是在应用层协议中引入长度字段。发送方在发送数据之前，先发送一个固定长度的字段，该字段表示后续数据 payload 的长度。接收方首先读取长度字段，然后根据长度字段的值读取相应长度的数据，从而得到完整的消息。

以下是一个简单的示例，说明如何在 Go 语言中实现基于长度字段的文件数据传输：

发送端：

package main

import (
    "fmt"
    "net"
    "os"
    "strconv"
)

func main() {
    conn, err := net.Dial("tcp", ":1234")
    if err != nil {
        fmt.Println("Error connecting:", err)
        return
    }
    defer conn.Close()

    // 读取文件
    file, err := os.Open("example.txt")
    if err != nil {
        fmt.Println("Error opening file:", err)
        return
    }
    defer file.Close()

    fileInfo, err := file.Stat()
    if err != nil {
        fmt.Println("Error getting file info:", err)
        return
    }

    fileSize := fileInfo.Size()
    buffer := make([]byte, fileSize)

    _, err = file.Read(buffer)
    if err != nil {
        fmt.Println("Error reading file:", err)
        return
    }

    // 发送文件长度
    length := strconv.Itoa(int(fileSize))
    // 确保长度字符串为固定长度，例如 10 个字符，不足则补 0
    lengthFormatted := fmt.Sprintf("%010s", length)
    _, err = conn.Write([]byte(lengthFormatted))
    if err != nil {
        fmt.Println("Error sending length:", err)
        return
    }

    // 发送文件内容
    _, err = conn.Write(buffer)
    if err != nil {
        fmt.Println("Error sending file data:", err)
        return
    }

    fmt.Println("File sent successfully!")
}

登录后复制

接收端：

百度文心百中

百度大模型语义搜索体验中心

查看详情

package main

import (
    "fmt"
    "io"
    "net"
    "strconv"
)

func main() {
    ln, err := net.Listen("tcp", ":1234")
    if err != nil {
        fmt.Println("Error listening:", err)
        return
    }
    defer ln.Close()

    conn, err := ln.Accept()
    if err != nil {
        fmt.Println("Error accepting:", err)
        return
    }
    defer conn.Close()

    // 读取文件长度
    lengthBuffer := make([]byte, 10) // 假设长度字段为 10 字节
    _, err = io.ReadFull(conn, lengthBuffer) // 确保读取到完整的长度字段
    if err != nil {
        fmt.Println("Error reading length:", err)
        return
    }

    lengthStr := string(lengthBuffer)
    fileSize, err := strconv.Atoi(lengthStr)
    if err != nil {
        fmt.Println("Error converting length:", err)
        return
    }

    // 读取文件内容
    buffer := make([]byte, fileSize)
    _, err = io.ReadFull(conn, buffer) // 确保读取到完整的文件数据
    if err != nil {
        fmt.Println("Error reading file data:", err)
        return
    }

    fmt.Printf("Received file data: %s\n", string(buffer))
}

登录后复制

注意事项：