golang文件传输服务
本文内容纲要:
续上篇,本篇介绍一个完整的golang文件传输服务器。
完整的代码可以看服务器,客户端
网络使用的框架如上篇介绍,这里就不再复述.
首先定义3个命令码:
const(
request_file=1
file_size=2
transfering=3
)
request_file用于请求文件传输,附带的命令参数是文件key.
file_size用于通告客户端文件的大小.
transfering用于传输文件内容,附带参数是文件内容的二进制数据.
服务器的文件配置示例
../learnyouhaskell.pdf=haskell
../golang.1.1.2.chm=golang
../NodeJS.pdf=NodeJS
上面的文件配置了3个文件可供传输=左边是文件路径,右边是请求文件时使用的key.
服务器启动时首先调用loadfile将文件导入到内存中,然后根据定义的key,将文件内容插入到字典filemap中:
funcloadfile(){
//从配置导入文件
F,err:=os.Open("./config.txt")
iferr!=nil{
fmt.Printf("config.txtopenfailed\n")
return
}
filemap=make(map[string][]byte)
bufferReader:=bufio.NewReader(F)
eof:=false
for!eof{
line,err:=bufferReader.ReadString('\n')
iferr==io.EOF{
err=nil
eof=true
}elseiferr!=nil{
fmt.Printf("parsefileerror\n")
return
}
iflen(line)>1{
line=line[0:len(line)-1]//drop'\n'
fileconfig:=strings.Split(line,"=")
iflen(fileconfig)==2{
buf,err:=ioutil.ReadFile(fileconfig[0])
iferr!=nil{
fmt.Printf("%sloaderror\n",fileconfig[0])
}else{
filemap[fileconfig[1]]=buf
fmt.Printf("%sloadsuccess,key%s\n",fileconfig[0],fileconfig[1])
}
}
}
}
iffilemap["golang"]==nil{
fmt.Printf("golangnotfound\n")
}
fmt.Printf("loadfilefinish\n")
}
接着是服务其的packet_handler:
funcprocess_client(session*tcpsession.Tcpsession,rpk*packet.Rpacket){
cmd,_:=rpk.Uint16()
ifcmd==request_file{
ifsession.Ud()!=nil{
fmt.Printf("alreadyintransfersession\n")
}else
{
filename,_:=rpk.String()
filecontent:=filemap[filename]
iffilecontent==nil{
fmt.Printf("%snotfound\n",filename)
session.Close()
}else{
fmt.Printf("requestfile%s\n",filename)
tsession:=&transfer_session{filecontent:filecontent,ridx:0}
session.SetUd(tsession)
wpk:=packet.NewWpacket(packet.NewByteBuffer(64),false)
wpk.PutUint16(file_size)
wpk.PutUint32(uint32(len(filecontent)))
session.Send(wpk,nil)
tsession.send_file(session)
}
}
}else{
fmt.Printf("cmderror,%d\n",cmd)
session.Close()
}
}
如果收到的消息是requestfile,首先查看请求的文件是否存在,如果存在则创建一个文件传输过程transfersession,
并将它与tcpsession绑定,然后发出一个文件大小通告包,紧接着立即调用send_file开始发送文件内容.
func(this*transfer_session)send_file(session*tcpsession.Tcpsession){
remain:=len(this.filecontent)-this.ridx
sendsize:=0
ifremain>=16000{
sendsize=16000
}else{
sendsize=remain
}
wpk:=packet.NewWpacket(packet.NewByteBuffer(uint32(sendsize)),false)
wpk.PutUint16(transfering)
wpk.PutBinary(this.filecontent[this.ridx:this.ridx+sendsize])
session.Send(wpk,send_finish)
this.ridx+=sendsize
}
sendfile中根据当前发送位置判断还有多少内容需要发送,如果剩余内容小于16000字节就将所剩数据一次性
发出,否则发送16000字节的数据,并调整发送位置。注意到Send函数带了一个sendfinish函数作为参数,其作用
是当数据包发送完成后回调send_finish函数.
funcsend_finish(sinterface{},wpk*packet.Wpacket){
session:=s.(*tcpsession.Tcpsession)
tsession:=session.Ud().(*transfer_session)
iftsession.check_finish(){
session.Close()
return
}
tsession.send_file(session)
}
send_finish的作用是判断文件是否已经发送完,如果发完断开连接,否则接着发送剩余部分.
总结一下,golang用来编写服务器应用还是相当方便的,很多细节问题在语言层面或系统库里已经帮你解决掉了
,可以将主要的精力放在逻辑的处理上.
本文内容总结:
原文链接:https://www.cnblogs.com/sniperHW/p/3587313.html