我最近需要為一個.net項目准備一個內部線程通信機制. 項目有多個使用ASP.NET,Windows 表單和控制台應用程序的服務器和客戶端構成. 考慮到實現的可能性,我下定決心要使用原生的socket,而不是許多.NET中已經提前為我們構建好的組件, 像是所謂的管道, NetTcpClient 還有 Azure 服務總線.
這篇文章中的服務器基於System.Net.Sockets類異步方法. 這些允許你支持大量的socket客戶端, 而一個客戶端的連接是唯一的阻塞機制. 阻塞的時間是可以忽略不記得,所以服務器基本上是在當做一個多線程socket服務器在運作的.
原生的socket在為你提供通信層面的完全控制權上具有優勢, 而在處理不同的數據類型是具有很大的靈活性. 你甚至可以通過socket發送序列化了的CLR對象,盡管我在這裡不會那樣做. 這個項目將會想你展示如何在socket之間發送文本.
使用下面的代碼,你初始化了一個Server類,並運行了Start()方法:
1
Server myServer = new Server();
2
myServer.Start();
如果你計劃在一個Windows表單中管理服務器的話,我建議使用一個BackgroundWorker, 因為socket方法(一般會是ManualResentEvent) 將會阻塞GUI線程的運行.
Server 類:
01
using System.Net.Sockets;
02
03
public class Server
04
{
05
private static Socket listener;
06
public static ManualResetEvent allDone = new ManualResetEvent(false);
07
public const int _bufferSize = 1024;
08
public const int _port = 50000;
09
public static bool _isRunning = true;
10
11
class StateObject
12
{
13
public Socket workSocket = null;
14
public byte[] buffer = new byte[bufferSize];
15
public StringBuilder sb = new StringBuilder();
16
}
17
18
// Returns the string between str1 and str2
19
static string Between(string str, string str1, string str2)
20
{
21
int i1 = 0, i2 = 0;
22
string rtn = "";
23
24
i1 = str.IndexOf(str1, StringComparison.InvariantCultureIgnoreCase);
25
if (i1 > -1)
26
{
27
i2 = str.IndexOf(str2, i1 + 1, StringComparison.InvariantCultureIgnoreCase);
28
if (i2 > -1)
29
{
30
rtn = str.Substring(i1 + str1.Length, i2 - i1 - str1.Length);
31
}
32
}
33
return rtn;
34
}
35
36
// Checks if the socket is connected
37
static bool IsSocketConnected(Socket s)
38
{
39
return !((s.Poll(1000, SelectMode.SelectRead) && (s.Available == 0)) || !s.Connected);
40
}
41
42
// Insert all the other methods here.
43
}
ManualResetEvent 是一個實現了你的socket服務器中事件的.NET類. 我們需要這個項目在我們想要發布阻塞操作的時候向代碼發送信號. 你可以試驗一下用bufferSize來適配你的需求. 如果能預期到消息的大小, 使用byte單位來設置消息的大小參數bufferSize. port是偵聽TCP的端口參數. 要意識到為其它應用程序伺服所使用的接口. 如果你想要能夠方便地停止服務器,你需要實現一些機制來將_isRunning設置成false. 這一般可以借助於使用一個 BackgroundWorker做到, 其中你可以使用myWorker.CancellationPending替換_isRunning. 我提到_isRunning的原因是給你在處理取消操作的問題上提供一個方向, 並向你展示偵聽器可以方便的停止的.
Between() 和IsSocketConnected() 是輔助方法.
現在轉過來看看方法. 首先是Start()方法:
01
public void Start()
02
{
03
IPHostEntry ipHostInfo = Dns.GetHostEntry(Dns.GetHostName());
04
IPEndPoint localEP = new IPEndPoint(IPAddress.Any, _port);
05
listener = new Socket(localEP.Address.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
06
listener.Bind(localEP);
07
08
while (_IsRunning)
09
{
10
allDone.Reset();
11
listener.Listen(10);
12
listener.BeginAccept(new AsyncCallback(acceptCallback), listener);
13
bool isRequest = allDone.WaitOne(new TimeSpan(12, 0, 0)); // Blocks for 12 hours
14
15
if (!isRequest)
16
{
17
allDone.Set();
18
// Do some work here every 12 hours
19
}
20
}
21
listener.Close();
22
}
這個方法初始化了偵聽器socket, 並開始等待用戶連接的到來. 項目中主要的模式是使用異步委派. 異步委派是在調用者中的狀態改變時被異步調用的方法. isRequest 告訴你WaitOne 是否已經因為有客戶端連接或者超時而退出.
如果你有大量的客戶端連接同時發生, 考慮提高Listen()方法的隊列參數.
現在來看看下一個方法, acceptCallback . 這個方法由listener.BeginAccept異步調用. 當方法完成執行時,偵聽器會立即偵聽新的客戶端.
01
static void acceptCallback(IAsyncResult ar)
02
{
03
// Get the listener that handles the client request.
04
Socket listener = (Socket)ar.AsyncState;
05
06
if (listener != null)
07
{
08
Socket handler = listener.EndAccept(ar);
09
10
// Signal main thread to continue
11
allDone.Set();
12
13
// Create state
14
StateObject state = new StateObject();
15
state.workSocket = handler;
16
handler.BeginReceive(state.buffer, 0, _bufferSize, 0, new AsyncCallback(readCallback), state);
17
}
18
}
acceptCallback 會派生出另外一個異步指派: readCallback. 這個方法會讀取來自socket的實際數據. 我已經為收發數據作了我自己的控制, 對於_bufferSize來說是不變的. 所有發送到服務器的字符串都必須用<!--SOCKET--> 和 <!--ENDSOCKET-->包起來. 同樣,客戶端在收到服務器的響應式,必須解除響應信息的包裹, 後者被<!--RESPONSE--> 和 <!--ENDRESPONSE-->包了起來。
01
static void readCallback(IAsyncResult ar)
02
{
03
StateObject state = (StateObject)ar.AsyncState;
04
Socket handler = state.workSocket;
05
06
if (!IsSocketConnected(handler))
07
{
08
handler.Close();
09
return;
10
}
11
12
int read = handler.EndReceive(ar);
13
14
// Data was read from the client socket.
15
if (read > 0)
16
{
17
state.sb.Append(Encoding.UTF8.GetString(state.buffer, 0, read));
18
19
if (state.sb.ToString().Contains("<!--ENDSOCKET-->"))
20
{
21
string toSend = "";
22
string cmd = ts.Strings.Between(state.sb.ToString(), "<!--SOCKET-->", "<!--ENDSOCKET-->");
23
24
switch (cmd)
25
{
26
case "Hi!":
27
toSend = "How are you?";
28
break;
29
case "Milky Way?":
30
toSend = "No I am not.";
31
break;
32
}
33
34
toSend = "<!--RESPONSE-->" + toSend + "<!--ENDRESPONSE-->";
35
36
byte[] bytesToSend = Encoding.UTF8.GetBytes(toSend);
37
handler.BeginSend(bytesToSend, 0, bytesToSend.Length, SocketFlags.None
38
, new AsyncCallback(sendCallback), state);
39
}
40
else
41
{
42
handler.BeginReceive(state.buffer, 0, _bufferSize, 0
43
, new AsyncCallback(readCallback), state);
44
}
45
}
46
else
47
{
48
handler.Close();
49
}
50
}
readCallback 會派生另外一個方法, sendCallback, 它將會向客戶端發送請求. 如果客戶端沒有關閉連接, sendCallback 將會向socket發送信號以獲得更多的數據.
01
static void sendCallback(IAsyncResult ar)
02
{
03
StateObject state = (StateObject)ar.AsyncState;
04
Socket handler = state.workSocket;
05
handler.EndSend(ar);
06
07
StateObject newstate = new StateObject();
08
newstate.workSocket = handler;
09
handler.BeginReceive(newstate.buffer, 0, StateObject.BufferSize, 0, new AsyncCallback(readCallback), newstate);
10
}
我會將寫一個socket客戶端作為聯系留給讀者. socket客戶端應該使用同異步調用同樣的編程模式. 我希望你能從這篇文章中收獲樂趣,並且會像一個socket程序員那樣付諸實踐!
我在生產環境下使用了此代碼,其中的socket服務器是一個自由文本搜索引擎。 SQL Server缺乏對自由文本搜索支持(你可以使用自由文本索引,但它們是緩慢和昂貴的)。socket服務器負載了大量導向IEnumerables的文本數據,並使用Linq來搜索文本。來自socket服務器的響應從數百萬行的Unicode文本數據中搜索時間在幾毫秒內。我們還使用了三個分布式的Sphinx服務器(www.sphinxsearch.com)。socket服務器充當了Sphinx服務器的高速緩存。如果你需要一個快速的自由文本搜索引擎,我強烈建議使用Sphinx。
原文地址:http://www.codeproject.com/Articles/745134/csharp-async-socket-server