client.go
package main
import (
"fmt"
"io"
"net"
"os"
)
func sendFile(conn net.Conn,filePath string) {
//只读打开文件
f, err := os.Open(filePath)
if err != nil {
fmt.Println("os.Open err:",err)
return
}
defer f.Close()
//从本文件中,读数据,写给网络接收端。读多少,写多少。原封不动
buf := make([]byte,4096)
for {
n,err := f.Read(buf)
if err != nil {
if err == io.EOF {
fmt.Println("发送文件完成。")
return
}else {
fmt.Println("os.Open err:",err)
return
}
}
//写道网络socket中
_,err = conn.Write(buf[:n])
if err != nil {
fmt.Println("conn.Write err:",err)
return
}
}
}
func main() {
list := os.Args
if len(list) != 2 {
fmt.Println("格式为:go run xxx.go 文件绝对路径")
return
}
//提取文件绝对路径
filePath := list[1]
//提取文件名
fileInfo, err := os.Stat(filePath)
if err != nil {
fmt.Println("os.Stat err:",err)
return
}
fileName := fileInfo.Name()
//主动发起连接请求
conn,err := net.Dial("tcp","127.0.0.1:8008")
if err != nil {
fmt.Println("net.Dial err:",err)
return
}
defer conn.Close()
//发送文件名给接收端
n,err := conn.Write([]byte(fileName))
if err != nil{
fmt.Println("conn.Write err:",err)
return
}
//读取服务器回发的 OK
buf := make([]byte,16)
n, err = conn.Read(buf)
if err != nil{
fmt.Println("conn.Read err:",err)
return
}
if "ok" == string(buf[:n]) {
//写文件内容给服务器-- 借助conn
sendFile(conn,filePath)
}
}
server.go
package main
import (
"fmt"
"net"
"os"
)
func recvFile(conn net.Conn, fileName string) {
//按照文件名创建新文件
f, err := os.Create(fileName)
if err != nil {
fmt.Println("os.Create err:",err)
return
}
defer f.Close()
//从网络中读数据,写入本地文件
buf := make([]byte,4096)
for {
n,_ := conn.Read(buf)
if n == 0 {
fmt.Println("接收文件完成。")
return
}
//写入本地文件,都多少,写多少
f.Write(buf[:n])
}
}
func main(){
//创建用于监听的socket
listenner,err := net.Listen("tcp","127.0.0.1:8008")
if err != nil{
fmt.Println("net.Listen err: ",err)
return
}
defer listenner.Close()
//阻塞监听
conn,err := listenner.Accept()
if err != nil{
fmt.Println("listenner.Accept err:",err)
return
}
defer conn.Close()
//获取文件名,保存
buf := make([]byte,4096)
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(conn,fileName)
}