go语言传输文件

发送文件

package main

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

func SendFile(path string, conn net.Conn) {
    //以只读方式打开文件
    f, err := os.Open(path)
    if err != nil {
        fmt.Println("conn.Read err = ", err)
        return
    }
    defer f.Close()
    buf := make([]byte, 1024*4)

    // 读文件内容，读多少发多少，一点不差
    for {
        n, err := f.Read(buf)
        if err != nil {
            if err == io.EOF {
                fmt.Println("文件发送完毕")
            } else {
                fmt.Println("f.Read err = ", err)
            }

            return
        }
        //发送内容
        conn.Write(buf[:n])
    }
}
func main() {
    // 提示输入文件
    fmt.Println("请输入需要传输的文件")
    var path string
    fmt.Scan(&path)

    //获取文件名
    info, err := os.Stat(path)
    if err != nil {
        fmt.Println("err = ", err)
        return
    }

    //主动连接服务器
    conn, err1 := net.Dial("tcp", "127.0.0.1:8000")
    if err1 != nil {
        fmt.Println("net.Dial err1 = ", err1)
        return
    }
    defer conn.Close()

    // 给接受方，先发送文件名
    _, err = conn.Write([]byte(info.Name()))
    if err != nil {
        fmt.Println("conn.Write err = ", err)
        return
    }
    // 接受对方的会服，如果回复“OK”说明对方准备好可以发送
    var n int
    buf := make([]byte, 1024)
    n, err = conn.Read(buf)
    if err != nil {
        fmt.Println("conn.Read err = ", err)
        return
    }

    if "ok" == string(buf[:n]) {
        //发送文件内容
        SendFile(path, conn)
    }

}

接受文件：

package main

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

//接受文件
func ReciveFile(fileName string, conn net.Conn) {
    f, err := os.Create(fileName)
    if err != nil {
        fmt.Println("os.Create err = ", err)
        return
    }
    buf := make([]byte, 1024*4)
    for {
        n, err := conn.Read(buf) //接受对方发送的文件内容
        if err != nil {
            if err == io.EOF {
                fmt.Println("文件接受 完毕")
            } else {
                fmt.Println("f.Read err = ", err)
            }

            return
        }
        f.Write(buf[:n])
    }

}
func main() {
    //监听
    listenner, err := net.Listen("tcp", "127.0.0.1:8000")
    if err != nil {
        fmt.Println("net.Listen err = ", err)
        return
    }
    defer listenner.Close()

    // 阻塞等待用户连接
    conn, err1 := listenner.Accept()
    if err1 != nil {
        fmt.Println("listenner.Accept err1 = ", err1)
        return
    }
    buf := make([]byte, 1024)
    var n int
    n, err = conn.Read(buf) //读取对方发送的文件名
    if err != nil {
        fmt.Println("conn.Read err = ", err)
        return
    }
    fileName := string(buf[:n])
    defer conn.Close()
    // 回复ok
    conn.Write([]byte("ok"))
    //接受文件
    ReciveFile(fileName, conn)
}