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

聖光之護

发布时间：2025-08-12 21:44:12

834人浏览过

来源于php中文网

原创

通过 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)
    }
}

注意事项:

Psky企业网站系统1.1.2

Psky企业网站是基于PHP+MYSQL开发的一款企业官网宣传网站，支持PC、wap浏览。主要功能包括导航管理、轮播新闻、单位动态、通知通报、单位简介、后台管理等。运行环境下载安装PHPStudy（小皮）即可。系统采用PHP语言（版本：7.3.4nts）和MYSQL（版本：5.7.26）数据库进行开发，安装PHPStudy（小皮）集成环境进行测试通过。安装步骤：1、导入SQL文件到PHPmyad

下载