博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Golang transfer file with socket
阅读量:5809 次
发布时间:2019-06-18

本文共 3572 字,大约阅读时间需要 11 分钟。

package mainimport (    "bufio"    "code.google.com/p/mahonia"    "fmt"    "io"    "net"    "os")func main() {    fmt.Println("create a server or client?")    reader := bufio.NewReader(os.Stdin)    input, _, _ := reader.ReadLine()    if string(input) == "server" {        Server()    }    if string(input) == "client" {        Client()    } else {        fmt.Println(Show("err arguments,entering again!.\r\n  alternaltive argument is server or client"))        os.Exit(0)    }}func Show(s string) string {    enc := mahonia.NewEncoder("gbk") //中文转码有错误的函数。    return enc.ConvertString(s)}func Server() {    exit := make(chan bool)    ip := net.ParseIP("127.0.0.1")    addr := net.TCPAddr{ip, 8888, ""}    go func() {        listener, err := net.ListenTCP("tcp", &addr) //TCPListener listen        if err != nil {            fmt.Println("Initialize error", err.Error())            exit <- true            return        }        fmt.Println("Server listening...")        tcpcon, err := listener.AcceptTCP() //TCPConn client        if err != nil {            fmt.Println(err.Error())            //continue        }        fmt.Println("Client connect")        data := make([]byte, 1024)        if err != nil {            fmt.Println("tcpcon.Read(data)" + err.Error())        }        //recv file name        wc, err := tcpcon.Read(data)        fo, err := os.Create("F:\\uploads\\" + string(data[0:wc]))        if err != nil {            fmt.Println("os.Create" + err.Error())        }        fmt.Println("the file's name is:", string(data[0:wc]))        //recb file size        wc, err = tcpcon.Read(data)        fmt.Println("the file's size is:", string(data[0:wc]))        defer fo.Close()        for {            c, err := tcpcon.Read(data) //???为何调用conn类的Read            if err != nil {                fmt.Println("tcpcon.Read(data)" + err.Error())            }            if string(data[0:c]) == "filerecvend" {                fmt.Println("string(data[0:c]) == filerecvend is true")                tcpcon.Write([]byte("file recv finished!\r\n"))                tcpcon.Close()                break            }            //write to the file            _, err = fo.Write(data[0:c])            if err != nil {                fmt.Println("write err" + err.Error())            }        }    }()    <-exit    fmt.Println(Show("server close!"))}func Client() {    fmt.Println("send ur file to the destination", "input ur filename:")    reader := bufio.NewReader(os.Stdin)    input, _, _ := reader.ReadLine()    fmt.Println(string(input))    fi, err := os.Open(string(input))    if err != nil {        panic(err)    }    defer fi.Close()    fiinfo, err := fi.Stat()    fmt.Println("the size of file is ", fiinfo.Size(), "bytes") //fiinfo.Size() return int64 type    conn, err := net.Dial("tcp", "127.0.0.1:8888")    if err != nil {        fmt.Println(Show("connect server fail!"), Show(err.Error()))        return    }    defer conn.Close()    //send filename    _, err = conn.Write(input)    if err != nil {        fmt.Println("conn.Write", err.Error())    }    //send file size    _, err = conn.Write([]byte(string(fiinfo.Size())))    if err != nil {        fmt.Println("conn.Write", err.Error())    }    buff := make([]byte, 1024)    for {        n, err := fi.Read(buff)        if err != nil && err != io.EOF {            panic(err)        }        if n == 0 {            conn.Write([]byte("filerecvend"))            fmt.Println("filerecvend")            break        }        _, err = conn.Write(buff)        if err != nil {            fmt.Println(err.Error())        }    }}

转载地址:http://xhjbx.baihongyu.com/

你可能感兴趣的文章
Linux VSFTP服务器
查看>>
DHCP中继数据包互联网周游记
查看>>
Squid 反向代理服务器配置
查看>>
Java I/O操作
查看>>
Tomcat性能调优
查看>>
项目管理心得
查看>>
Android自学--一篇文章基本掌握所有的常用View组件
查看>>
灰度图像和彩色图像
查看>>
通过vb.net 和NPOI实现对excel的读操作
查看>>
TCP segmentation offload
查看>>
java数据类型
查看>>
数据结构——串的朴素模式和KMP匹配算法
查看>>
FreeMarker-Built-ins for strings
查看>>
验证DataGridView控件的数据输入
查看>>
POJ1033
查看>>
argparse - 命令行选项与参数解析(转)
查看>>
一维数组
查看>>
Linux学习笔记之三
查看>>
Floyd最短路算法
查看>>
Class.forName(String name)方法,到底会触发那个类加载器进行类加载行为?
查看>>