原文:C#高性能大容量SOCKET并發(fā)（十一）：編寫上傳客戶端

客戶端封裝整體框架

客戶端編程基于阻塞同步模式，只有數(shù)據(jù)正常發(fā)送或接收才返回，如果發(fā)生錯(cuò)誤則拋出異常，基于TcpClient進(jìn)行封裝，主要類結(jié)構(gòu)如下圖：

TcpClient：NET系統(tǒng)封裝，實(shí)現(xiàn)了底層Socket操作，提供了阻塞和非阻塞調(diào)用；

OutgoingDataAssembler m_outgoingDataAssembler：協(xié)議組裝器，用來(lái)組裝往外發(fā)送的命令，主要用于組裝協(xié)議格式；

DynamicBufferManager m_sendBuffer：用于把命令和數(shù)據(jù)同時(shí)寫入到緩存中，調(diào)用一次發(fā)送，這樣服務(wù)器就只會(huì)產(chǎn)生一次IOCP回調(diào)，可以提高性能；

IncomingDataParser m_incomingDataParser：收到數(shù)據(jù)的解析器，用于解析返回的內(nèi)容，主要是解析文本格式；

protected DynamicBufferManager m_recvBuffer：接收數(shù)據(jù)的緩存，數(shù)據(jù)存到緩存中后，可以解析命令和數(shù)據(jù)；

TcpClient說(shuō)明，阻塞和非阻塞

TcpClient封裝了NET的底層Socket操作，基于TCP協(xié)議，提供了阻塞和非阻塞模式調(diào)用，具體是設(shè)置m_tcpClient.Client.Blocking = true表示使用阻塞模式，反之則使用非阻塞模式。阻塞模式表示接收完指定長(zhǎng)度的數(shù)據(jù)才返回，非阻塞模式表示收到一點(diǎn)數(shù)據(jù)就返回。

如我們調(diào)用m_tcpClient.Client.Receive(m_recvBuffer.Buffer, sizeof(int), packetLength, SocketFlags.None)，假設(shè)傳入的長(zhǎng)度為1024，阻塞模式一點(diǎn)要等到數(shù)據(jù)達(dá)到1024長(zhǎng)度才返回，否則一直等待Socket超時(shí)或者鏈路斷了，非阻塞模式則不同，加入收到8字節(jié)了，則返回調(diào)用者，調(diào)用者使用循環(huán)繼續(xù)接受1024-8=1016的數(shù)據(jù)。

發(fā)送命令

發(fā)送數(shù)據(jù)和服務(wù)端相同，主要是對(duì)數(shù)據(jù)進(jìn)行組包，然后調(diào)用發(fā)送函數(shù)發(fā)送，具體代碼如下：

public void SendCommand(byte[] buffer, int offset, int count){string commandText = m_outgoingDataAssembler.GetProtocolText();byte[] bufferUTF8 = Encoding.UTF8.GetBytes(commandText);int totalLength = sizeof(int) + bufferUTF8.Length + count; //獲取總大小m_sendBuffer.Clear();m_sendBuffer.WriteInt(totalLength, false); //寫入總大小m_sendBuffer.WriteInt(bufferUTF8.Length, false); //寫入命令大小m_sendBuffer.WriteBuffer(bufferUTF8); //寫入命令內(nèi)容m_sendBuffer.WriteBuffer(buffer, offset, count); //寫入二進(jìn)制數(shù)據(jù)m_tcpClient.Client.Send(m_sendBuffer.Buffer, 0, m_sendBuffer.DataCount, SocketFlags.None);}

接收命令

接收命令和發(fā)送相反，先接收長(zhǎng)度，然后接收內(nèi)容，然后對(duì)數(shù)據(jù)進(jìn)行解包，具體代碼如下：

public bool RecvCommand(out byte[] buffer, out int offset, out int size){m_recvBuffer.Clear();m_tcpClient.Client.Receive(m_recvBuffer.Buffer, sizeof(int), SocketFlags.None);int packetLength = BitConverter.ToInt32(m_recvBuffer.Buffer, 0); //獲取包長(zhǎng)度if (NetByteOrder)packetLength = System.Net.IPAddress.NetworkToHostOrder(packetLength); //把網(wǎng)絡(luò)字節(jié)順序轉(zhuǎn)為本地字節(jié)順序m_recvBuffer.SetBufferSize(sizeof(int) + packetLength); //保證接收有足夠的空間m_tcpClient.Client.Receive(m_recvBuffer.Buffer, sizeof(int), packetLength, SocketFlags.None);int commandLen = BitConverter.ToInt32(m_recvBuffer.Buffer, sizeof(int)); //取出命令長(zhǎng)度string tmpStr = Encoding.UTF8.GetString(m_recvBuffer.Buffer, sizeof(int) + sizeof(int), commandLen);if (!m_incomingDataParser.DecodeProtocolText(tmpStr)) //解析命令{buffer = null;offset = 0;size = 0;return false;}else{buffer = m_recvBuffer.Buffer;offset = commandLen + sizeof(int) + sizeof(int);size = packetLength - offset;return true;}}

命令交互

封裝了底層Socket操作和協(xié)議解析后，實(shí)現(xiàn)一個(gè)命令交互如登錄代碼如下：

public bool DoLogin(string userName, string password){try{m_outgoingDataAssembler.Clear();m_outgoingDataAssembler.AddRequest();m_outgoingDataAssembler.AddCommand(AsyncSocketServer.ProtocolKey.Login);m_outgoingDataAssembler.AddValue(AsyncSocketServer.ProtocolKey.UserName, userName);m_outgoingDataAssembler.AddValue(AsyncSocketServer.ProtocolKey.Password, AsyncSocketServer.BasicFunc.MD5String(password));SendCommand();bool bSuccess = RecvCommand();if (bSuccess){bSuccess = CheckErrorCode();if (bSuccess){m_userName = userName;m_password = password;}return bSuccess;}elsereturn false;}catch (Exception E){//記錄日志m_errorString = E.Message;return false;}}

上傳協(xié)議

上傳協(xié)議主要分為三個(gè)命令，第一個(gè)是Upload，向服務(wù)器請(qǐng)求上傳的文件，如果服務(wù)器有相同的文件，則返回是否傳完，如果未傳完，返回需要續(xù)傳的文件位置，然后客戶端則從上一個(gè)位置開(kāi)始傳輸，傳輸數(shù)據(jù)服務(wù)器只接收，不應(yīng)答，客戶端傳輸完后，發(fā)完成（EOF）命令。因此三個(gè)命令封裝代碼如下：

public bool DoUpload(string dirName, string fileName, ref long fileSize){bool bConnect = ReConnectAndLogin(); //檢測(cè)連接是否還在，如果斷開(kāi)則重連并登錄if (!bConnect)return bConnect;try{m_outgoingDataAssembler.Clear();m_outgoingDataAssembler.AddRequest();m_outgoingDataAssembler.AddCommand(AsyncSocketServer.ProtocolKey.Upload);m_outgoingDataAssembler.AddValue(AsyncSocketServer.ProtocolKey.DirName, dirName);m_outgoingDataAssembler.AddValue(AsyncSocketServer.ProtocolKey.FileName, fileName);SendCommand();bool bSuccess = RecvCommand();if (bSuccess){bSuccess = CheckErrorCode();if (bSuccess){bSuccess = m_incomingDataParser.GetValue(AsyncSocketServer.ProtocolKey.FileSize, ref fileSize);}return bSuccess;}elsereturn false;}catch (Exception E){//記錄日志m_errorString = E.Message;return false;}}public bool DoData(byte[] buffer, int offset, int count){try{m_outgoingDataAssembler.Clear();m_outgoingDataAssembler.AddRequest();m_outgoingDataAssembler.AddCommand(AsyncSocketServer.ProtocolKey.Data);SendCommand(buffer, offset, count);return true;}catch (Exception E){//記錄日志m_errorString = E.Message;return false;}}public bool DoEof(Int64 fileSize){try{m_outgoingDataAssembler.Clear();m_outgoingDataAssembler.AddRequest();m_outgoingDataAssembler.AddCommand(AsyncSocketServer.ProtocolKey.Eof);SendCommand();bool bSuccess = RecvCommand();if (bSuccess)return CheckErrorCode();elsereturn false;}catch (Exception E){//記錄日志m_errorString = E.Message;return false;}}調(diào)用過(guò)程：

protected static bool SendFile(string fileName, ClientUploadSocket uploadSocket){FileStream fileStream = new FileStream(fileName, FileMode.Open, FileAccess.ReadWrite);try{try{long fileSize = 0;if (!uploadSocket.DoUpload("", Path.GetFileName(fileName), ref fileSize))throw new Exception(uploadSocket.ErrorString);fileStream.Position = fileSize;byte[] readBuffer = new byte[PacketSize];while (fileStream.Position < fileStream.Length){int count = fileStream.Read(readBuffer, 0, PacketSize);if (!uploadSocket.DoData(readBuffer, 0, count))throw new Exception(uploadSocket.ErrorString);}if (!uploadSocket.DoEof(fileStream.Length))throw new Exception(uploadSocket.ErrorString);return true;}catch (Exception E){Console.WriteLine("Upload File Error: " + E.Message);return false;}}finally{fileStream.Close();}}
DEMO下載地址：http://download.csdn.net/detail/sqldebug_fan/7467745
免責(zé)聲明：此代碼只是為了演示C#完成端口編程，僅用于學(xué)習(xí)和研究，切勿用于商業(yè)用途。水平有限，C#也屬于初學(xué)，錯(cuò)誤在所難免，歡迎指正和指導(dǎo)。郵箱地址：fansheng_hx@163.com。

總結(jié)

以上是生活随笔為你收集整理的C#高性能大容量SOCKET并发（十一）：编写上传客户端的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。