当前位置：首页 > 文章列表 > Golang > Go教程 > Go语言Socket文件传输全解析

Go语言Socket文件传输全解析

2025-09-04 09:45:47 0浏览收藏

## Go 语言 Socket传输文件方法详解：高效可靠的文件传输方案想知道如何使用 Go 语言的 Socket 实现高效可靠的文件数据传输吗？本文将深入探讨 Go 语言 Socket 文件传输的关键技术。由于 TCP 协议的流式特性，理解消息边界至关重要。本文重点介绍两种常用的解决方案：**使用长度字段**，在消息前添加长度信息，方便接收方准确读取数据；**使用状态机**，解析复杂的消息格式。文章提供详细的代码示例，展示如何通过 TCP 连接发送和接收文件，并处理潜在的错误。无论你是 Go 语言新手还是资深开发者，都能从中学习到实用的 Socket 文件传输技巧，提升你的网络编程能力。

通过 Go 语言 Socket 传输文件数据

本文将深入探讨如何使用 Go 语言的 Socket 进行文件数据传输。理解 TCP 协议的流式特性是解决问题的关键。TCP 是一种面向连接的、可靠的、基于字节流的传输层通信协议。这意味着，与 UDP 等协议不同，TCP 不保证消息的边界。发送方将数据视为字节流发送，接收方也以字节流的形式接收，而不会自动分割成消息。

因此，在处理大数据传输时，简单地使用固定大小的缓冲区读取数据可能会导致问题。例如，如果发送方发送了一个 100KB 的图像数据，接收方使用 1024 字节的缓冲区读取数据，则需要多次读取才能获取完整的数据。更重要的是，每次 Read 调用返回的数据量是不确定的，可能小于缓冲区的大小。

为了解决这个问题，我们需要在应用层协议中定义消息的边界。以下介绍两种常用的方法：

1. 使用长度字段

如果可以控制应用层协议，最简单的方法是在每个消息前添加一个固定长度的字段，用于表示消息体的长度。接收方首先读取长度字段，然后根据长度字段的值读取相应数量的字节，从而获取完整的消息。

package main

import (
    "encoding/binary"
    "fmt"
    "io"
    "net"
    "os"
)

// sendFile 函数通过 TCP 连接发送文件
func sendFile(conn net.Conn, filePath string) error {
    file, err := os.Open(filePath)
    if err != nil {
        return err
    }
    defer file.Close()

    fileInfo, err := file.Stat()
    if err != nil {
        return err
    }

    fileSize := fileInfo.Size()

    // 1. 发送文件大小
    sizeBuf := make([]byte, 8) // 使用 8 字节存储文件大小 (int64)
    binary.BigEndian.PutUint64(sizeBuf, uint64(fileSize))
    _, err = conn.Write(sizeBuf)
    if err != nil {
        return err
    }

    // 2. 发送文件内容
    buf := make([]byte, 4096) // 使用 4KB 的缓冲区
    for {
        bytesRead, err := file.Read(buf)
        if err != nil {
            if err != io.EOF {
                return err
            }
            break // 文件读取完毕
        }

        _, err = conn.Write(buf[:bytesRead])
        if err != nil {
            return err
        }
    }

    return nil
}

// receiveFile 函数通过 TCP 连接接收文件
func receiveFile(conn net.Conn, savePath string) error {
    // 1. 接收文件大小
    sizeBuf := make([]byte, 8)
    _, err := io.ReadFull(conn, sizeBuf)
    if err != nil {
        return err
    }

    fileSize := binary.BigEndian.Uint64(sizeBuf)

    // 2. 接收文件内容
    file, err := os.Create(savePath)
    if err != nil {
        return err
    }
    defer file.Close()

    buf := make([]byte, 4096)
    var totalBytesReceived uint64 = 0

    for totalBytesReceived < fileSize {
        bytesToRead := uint64(len(buf))
        if fileSize-totalBytesReceived < uint64(len(buf)) {
            bytesToRead = fileSize - totalBytesReceived
        }

        bytesRead, err := conn.Read(buf[:bytesToRead])
        if err != nil {
            return err
        }

        _, err = file.Write(buf[:bytesRead])
        if err != nil {
            return err
        }

        totalBytesReceived += uint64(bytesRead)
    }

    return nil
}

func main() {
    // 示例：启动一个简单的 TCP 服务器
    listener, err := net.Listen("tcp", ":8080")
    if err != nil {
        fmt.Println("Error listening:", err.Error())
        os.Exit(1)
    }
    defer listener.Close()

    fmt.Println("Listening on :8080")

    for {
        conn, err := listener.Accept()
        if err != nil {
            fmt.Println("Error accepting: ", err.Error())
            continue
        }

        go func(conn net.Conn) {
            defer conn.Close()

            // 接收文件
            err := receiveFile(conn, "received_file.dat")
            if err != nil {
                fmt.Println("Error receiving file:", err.Error())
                return
            }
            fmt.Println("File received successfully!")

            // (可选) 发送文件
            // err = sendFile(conn, "example.dat")
            // if err != nil {
            //  fmt.Println("Error sending file:", err.Error())
            //  return
            // }
            // fmt.Println("File sent successfully!")

        }(conn)
    }
}

注意事项: