在Golang的网络编程场景中，TCP客户端发送文件是常见需求，实现过程需要结合网络I/O操作和数据分块逻辑，避免大文件传输时出现内存占用过高或者传输中断的问题。
TCP文件发送的基本流程
TCP客户端发送文件的核心流程可以分为四个步骤：建立TCP连接、打开本地待发送文件、分块读取文件内容并发送、传输完成后关闭连接和文件句柄。其中数据分块是核心环节，需要根据网络状况和内存使用情况设置合理的分块大小，通常建议分块大小在1KB到64KB之间。
核心实现代码
客户端发送文件实现
下面是完整的Golang TCP客户端发送文件的代码示例，包含连接建立、文件读取、分块发送和异常处理逻辑：
package main
import (
"fmt"
"io"
"net"
"os"
)
// 定义每次读取的文件分块大小，设置为32KB
const chunkSize = 32 * 1024
func main() {
// 建立TCP连接，目标地址为本地8080端口的服务端
conn, err := net.Dial("tcp", "127.0.0.1:8080")
if err != nil {
fmt.Println("建立TCP连接失败:", err)
return
}
defer conn.Close()
// 打开待发送的本地文件
filePath := "./test_file.txt"
file, err := os.Open(filePath)
if err != nil {
fmt.Println("打开文件失败:", err)
return
}
defer file.Close()
// 循环读取文件内容，分块发送
buffer := make([]byte, chunkSize)
for {
// 从文件中读取最多chunkSize大小的内容
n, err := file.Read(buffer)
if err != nil {
// 读取到文件末尾，正常退出循环
if err == io.EOF {
fmt.Println("文件发送完成")
break
}
// 其他读取错误，打印错误信息后退出
fmt.Println("读取文件失败:", err)
break
}
// 将读取到的内容通过TCP连接发送出去
_, writeErr := conn.Write(buffer[:n])
if writeErr != nil {
fmt.Println("发送数据失败:", writeErr)
break
}
}
}
配套服务端接收代码
为了验证文件发送效果，这里提供简单的TCP服务端接收代码，用于接收客户端发送的文件内容并写入本地：
package main
import (
"fmt"
"io"
"net"
"os"
)
func main() {
// 监听本地8080端口
listener, err := net.Listen("tcp", "127.0.0.1:8080")
if err != nil {
fmt.Println("监听端口失败:", err)
return
}
defer listener.Close()
fmt.Println("服务端已启动，等待客户端连接")
// 接受客户端连接
conn, err := listener.Accept()
if err != nil {
fmt.Println("接受连接失败:", err)
return
}
defer conn.Close()
// 创建接收文件的句柄
receiveFile, err := os.Create("./received_file.txt")
if err != nil {
fmt.Println("创建接收文件失败:", err)
return
}
defer receiveFile.Close()
// 从连接中读取数据并写入文件
_, err = io.Copy(receiveFile, conn)
if err != nil {
fmt.Println("接收文件失败:", err)
return
}
fmt.Println("文件接收完成，保存路径为./received_file.txt")
}
关键注意事项
- 分块大小选择：如果分块过小，会增加系统调用次数，降低传输效率；如果分块过大，可能导致内存占用过高，建议根据实际场景调整
chunkSize的值。 - 错误处理：网络传输过程中可能出现连接中断、写入失败等问题，需要在代码中添加完善的错误处理逻辑，避免程序异常崩溃。
- 文件完整性校验：实际生产环境中，建议在发送文件前计算文件的哈希值，接收完成后再次计算哈希值进行比对，确保文件传输完整。
- 连接关闭顺序：先关闭文件句柄，再关闭TCP连接，避免资源泄漏。
常见问题解答
为什么发送大文件时需要分块？
如果不分块直接读取整个大文件到内存再发送，会导致内存占用急剧上升，甚至触发内存溢出。分块读取可以每次只占用少量内存，适合大文件传输场景。
发送过程中连接中断怎么办？
可以在发送逻辑中添加重试机制，当检测到连接中断时，重新建立TCP连接，从断点位置继续发送剩余的文件内容，减少重复传输的开销。