ホームページ >バックエンド開発 >Golang >golang でのファイル転送の小さなデモを共有する

golang でのファイル転送の小さなデモを共有する

藏色散人
藏色散人転載
2020-11-04 14:45:252587ブラウズ

以下は、golang チュートリアル コラムからの golang でのファイル転送の小さなデモです。必要とする友人の役に立てば幸いです。

ファイル情報を取得するには、os.Stat インターフェイスを使用する必要があります。ファイルを送信する前に、受信機 (サーバー) を開きます。クライアントを起動し、最初にファイル名を受信機に送信します。受信機は、ファイルを受信します。ファイル名を返し、ローカル ファイルを読み取って受信者に送信する前に、確認メッセージ "ok" " を返します。

送信者:

package main

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

func main() {
    fmt.Println("请输入接收者地址(IP:PORT):")    var addr string
    fmt.Scan(&addr)
    fmt.Println("请输入需要传输的文件:")    var path string
    fmt.Scan(&path)
    info, errf := os.Stat(path)    if errf != nil{
        fmt.Println("os.Stat errf =", errf)        return
    }

    conn, err := net.Dial("tcp", addr)    if err != nil{
        fmt.Println("net.Dial err =",err)        return
    }
    defer conn.Close()

    _, err = conn.Write([]byte(info.Name()))    if err != nil{
        fmt.Println("conn.Write info.Name err =",err)        return
    }    var n int
    buf := make([]byte, 1024)
    n, err = conn.Read(buf)    if err != nil{
        fmt.Println("conn.Read ok err =", err)        return
    }    if "ok" == string(buf[:n]){
        fmt.Println("ok")
        SendFile(path, conn)
    }
}

func SendFile(path string, conn net.Conn){
     file , err := os.Open(path)     if err != nil{
         fmt.Println("os.Open err =", err)         return
     }
     defer file.Close()
     buf := make([]byte, 1024 * 4)     for {
         n, err := file.Read(buf)         if err != nil{             if err == io.EOF{
                 fmt.Println("文件发送完毕")
            } else{
                fmt.Println("file.Read err =",err)
            }             return
         }         if n == 0{
             fmt.Println("文件发送完毕")             break
         }
         conn.Write(buf[:n])
     }
}

受信者:

package main

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

func main() {
    fmt.Println("请你的地址(IP:PORT):")    var addr string
    fmt.Scan(&addr)
    listenner, err := net.Listen("tcp", addr)    if err != nil{
        fmt.Println("net.Listen err =", err)        return
    }
    defer listenner.Close()

    conn, errl := listenner.Accept()    if errl != nil{
        fmt.Println("listenner.Accept err =", errl)        return
    }    var n int
    buf := make([]byte, 1024)
    n, err = conn.Read(buf)    if err != nil{
        fmt.Println("conn.Read fileName err =", err)        return
    }
    fileName := string(buf[:n])
    n, err = conn.Write([]byte("ok"))    if err != nil{
        fmt.Println("conn.Write ok err =", err)        return
    }

    RecvFile(fileName, conn)
}

func RecvFile(fileName string, conn net.Conn){
     file , err := os.Create(fileName)     if err != nil{
         fmt.Println("os.Create err =", err)         return
     }

     defer file.Close()

     buf := make([]byte, 1024 * 4)     for{
         n, err := conn.Read(buf)         if err != nil{             if err == io.EOF{
                 fmt.Println("文件接收完成")
            } else {
                fmt.Println("conn.Read err =", err)
            }            return
        }

         n, err = file.Write(buf[:n])         if err != nil{
             fmt.Println("file.Write err =", err)             break
        }
     }
}

以上がgolang でのファイル転送の小さなデモを共有するの詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。

声明:
この記事はcnblogs.comで複製されています。侵害がある場合は、admin@php.cn までご連絡ください。