以前學過MFC,我用過的它裡面的包裝Winsock的兩個類CSocket和CAsyncSocket,我一直覺得它哪個事件通知的功能比較不錯,特別是在連接的兩方收發數據沒有一定的規律的時候比較有用,雖然不用它的事件通知也可以實現功能,但是你需要循環檢測套接字的狀態或者阻塞等待,如果每次都要做這些瑣碎的工作豈不是很麻煩,於是我就想對這些功能作一次封裝。當然在.net裡面采用delegate和event是最好的選擇了。下面就是我實現的一些細節:
先講一下思路:其實這個還是很簡單了,估計高手們會不屑一顧了^_^。當套接字開始工作,比如開始監聽、開始連接的時候,啟動一個線程不停的檢測套接字的狀態,當某一事件的狀態條件滿足時就觸發此事件,要知道具體檢測套接字狀態的方法,請往下看。
平常用套接字編成的時候可能需要用到的功能就是:連接成功或者失敗的通知、在監聽的套接字上有掛起的連接的通知、有數據可以接受的通知、套接字關閉的通知、還有套接字空閒可以發送數據的通知等,我也就使對上面幾個功能作了封裝,不過我覺得最後面一個功能用處不大???
我的Socket類是從System.Net.Sockets.Socket繼承下來的,類名TcpEventSocket,實現如下:
先申明事件的委托類型,具體看代碼
public delegate void AcceptConnectionHandler();
public delegate void ConnectCompletedHandler(bool connected); //connected表明連接成功與否
public delegate void DataCanSendHandler();
public delegate void DataCanReceiveHandler(int buffersize); //當前可以接受的數據量
public delegate void SocketClosedHandler();
這些類型具體什麼意思應該能看出來吧,另外還要申明對應的事件
public event AcceptConnectionHandler AcceptConnection;
public event ConnectCompletedHandler ConnectCompleted;
public event DataCanSendHandler DataCanSend;
public event DataCanReceiveHandler DataCanReceive;
public event SocketClosedHandler SocketClosed;
另外加了幾個虛方法,這幾個方法用來觸發事件
protected virtual void OnAcceptConnection();
protected virtual void OnConnectCompleted(bool connected)
protected virtual void OnDataCanSend()
protected virtual void OnDataCanReceive(int buffersize)
protected virtual void OnSocketClosed()
為了在適當的事機啟動檢測線程,我重寫了幾個基類的方法:
new public void Listen(int backlog)
{
base.Listen(backlog);
sockState = SocketState.Listenning;
if (!checkThread.IsAlive)
checkThread.Start();
}
new public void Connect(EndPoint remoteEP)
{
try
{
base.Connect(remoteEP);
this.Blocking = false; //設置非阻塞狀態,以便事件通知的效率
if (!checkThread.IsAlive)
checkThread.Start();
}
catch(SocketException)
{
OnConnectCompleted(false);
}
}
這兩個方法實際編成中都應該首先被調用的,所以有他們啟動檢測線程比較合適,另外線程啟動了必然也要中止,於是我重寫了Close方法
new public void Close()
{
if(checkThread.IsAlive) //先中止線程再關閉連接
checkThread.Abort();
base.Close();
sockState = SocketState.Disconnected;
OnSocketClosed();
}
於是剩下的工作就是怎麼檢測套接字了,Socket類有個Select的靜態方法,它可以檢測很多套接字的狀態,不過這裡只需要檢測一個,所以直接用Socket的Poll方法,Poll的具體用法可以看MSDN,我這裡用代碼說明我對套接字的檢測方法
while(true) //循環檢查
{
if (sockState == SocketState.Disconnected) //如果當前沒有連接
{
if (Poll(500, SelectMode.SelectWrite))
OnConnectCompleted(true); //如果為可寫狀態,則表示連接成功
}
else if(sockState == SocketState.Listenning)
{
if (Poll(500, SelectMode.SelectRead)) //如果在監聽狀態發現套接字有數據可讀則表示已經有人連接上來可以調用Accept接受連接
OnAcceptConnection();
}
else //這裡sockState = SocketState.Connected
{
if (Poll(500, SelectMode.SelectWrite)) //如果有可寫狀態則表示可以發送數據
OnDataCanSend();
if (Poll(500, SelectMode.SelectRead)) //如果有可讀狀態
{
if (Available > 0) //如果有數據可讀表示可以調用Receive接受數據
OnDataCanReceive(Available);
else
{
OnSocketClosed(); //沒數據可讀表示連接已經關閉
break;
}
}
}
//如果沒有連接並且有錯誤狀態,則連接失敗
if (sockState == SocketState.Disconnected && Poll(500, SelectMode.SelectError))
OnConnectCompleted(false);
}
這裡的OnXXX方法就是執行事件通知的方法,並且派生類中可以重載這些方法直接獲得事件通知而不需要掛界別的事件通知處理函數(類似於MFC的OnAccept等虛函數)。不過派生的函數集的調用基類的相應方法哦。很可惜沒有重寫Socket.Accept方法,讓它返回一個TcpEventSocket,這樣應該就比較完整了,只可惜我不知道怎麼做^_^,如果誰知道的話歡迎指教。
事件通知機制已經基本完成,所缺乏的只是大量的測試(我做了幾下最簡單的測試,慚愧!!^_^)
如果有問題請聯系
[email protected]