go语言tcp协议实现文件上传

一、客户端实现方案：

package main

import (
	"fmt"
	"io"
	"net"
	"os"
)

func sendFile(filePath string, conn net.Conn) {
	defer conn.Close()

	// 获取文件名
	fileInfo, err := os.Stat(filePath)
	if err != nil {
		fmt.Println("Error getting file info:", err)
		return
	}
	fileName := fileInfo.Name()

	// 发送文件名
	conn.Write([]byte(fileName))

	// 打开文件并发送文件内容
	file, err := os.Open(filePath)
	if err != nil {
		fmt.Println("Error opening file:", err)
		return
	}
	defer file.Close()

	_, err = io.Copy(conn, file)
	if err != nil {
		fmt.Println("Error copying file content:", err)
		return
	}

	fmt.Printf("File %s sent.\n", fileName)
}

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

	filePath := "C:\\zhangsan.txt" // 替换为实际文件路径
	sendFile(filePath, conn)
}

二、服务端实现方案：

通过tcp协议的具体应用举例：

package main

import (
	"fmt"
	"io"
	"net"
	"os"
)
/*
	1、
*/
func handleFileTransfer(conn net.Conn) {
	defer conn.Close()

	// 读取文件名
	fileNameBuf := make([]byte, 1024)
	n, err := conn.Read(fileNameBuf)
	if err != nil {
		fmt.Println("Error reading file name:", err)
		return
	}
	fileName := string(fileNameBuf[:n])

	// 打开文件
	file, err := os.Create(fileName)
	if err != nil {
		fmt.Println("Error creating file:", err)
		return
	}
	defer file.Close()

	// 从连接中读取文件内容并写入到文件
	_, err = io.Copy(file, conn)
	if err != nil {
		fmt.Println("Error copying file content:", err)
		return
	}

	fmt.Printf("File %s received and saved.\n", fileName)
}

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

	fmt.Println("File transfer server listening on :8080")

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