在线时间:8:00-16:00
迪恩网络APP
随时随地掌握行业动态
扫描二维码
关注迪恩网络微信公众号
package main
import (
"fmt"
"os"
"net"
"io"
)
//发送文件内容
func SendFile(path string,conn net.Conn){
//以只读的方式打开文件
f,err := os.Open(path)
if err != nil{
fmt.Println("os.Open 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.Name()
info,err := os.Stat(path)
if err != nil{
fmt.Println("os.Stat err= ",err)
return
}
//主动连接我们的服务器
conn,err := net.Dial("tcp","127.0.0.1:8000")
if err !=nil{
fmt.Println("net.Dial err=",err)
return
}
//给接收方发送文件名
_,err = conn.Write([]byte(info.Name()))
if err != nil{
fmt.Println("conn.Write err =",err)
return
}
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 (
"net"
"fmt"
"os"
"io"
)
//文件的接收操作
func RecvFile(filename string,conn net.Conn) {
//新建文件
f,err := os.Create(filename)
if err != nil{
fmt.Println("os.Create err=",err)
return
}
buf := make([]byte,1024)
for{
n,err := conn.Read(buf) //接收对方发送过来的文件内容
if err != nil{
if err == io.EOF{
fmt.Println("文件接收完毕")
}else{
fmt.Println("conn.Read err=",err)
}
return
}
if n == 0{
fmt.Println("n==0 文件接收完毕")
break
}
f.Write(buf[:n])
}
}
func main() {
//监听
listernner,err := net.Listen("tcp","127.0.0.1:8000")
if err != nil{
fmt.Println("net.Listen err =",err)
return
}
defer listernner.Close()
//阻塞等待用户连接
conn,err := listernner.Accept()
if err != nil{
fmt.Println("listenner.Accept err=",err)
return
}
defer conn.Close()
//缓冲
buf := make([]byte,1024)
n,err := conn.Read(buf)
if err != nil{
fmt.Println("conn.Read err =",err)
return
}
filename := string(buf[:n])
//回复ok
conn.Write([]byte("ok"))
//接收文件内容
RecvFile(filename,conn)
}
|
请发表评论