corefx 源碼學習:SqlClient 是如何同步建立 Socket 連接的
阿新 • • 發佈:2019-04-27
als 中學 blank clas .data net 發現 code .com
在昨天的技術周會上發現 EnyimMemcached 中建立 Socket 連接的代碼有問題,今天坐車的時候在手機上閱讀 .net core 2.2 的 SqlClient 中同步建立 Socket 連接的代碼 SNITcpHandle.cs#L180 學習了一下。
建立 Socket 連接需要處理2個問題:1)處理連接失敗的問題;2)處理連接超時的問題。
SNITcpHandle 中這樣處理連接失敗的,如果 Connected 為 false ,就 dispose 當前 socket ,代碼如下:
sockets[i].Connect(ipAddresses[i], port);if (sockets[i] != null) // sockets[i] can be null if cancel callback is executed during connect() { if (sockets[i].Connected) { availableSocket = sockets[i]; break; } else { sockets[i].Dispose(); sockets[i] = null; } }
連接超時的處理,SNITcpHandle 是借助 CancellationTokenSource 實現的
CancellationTokenSource cts = new CancellationTokenSource(); cts.CancelAfter(timeout); void Cancel() { for (int i = 0; i < sockets.Length; ++i) { try { if (sockets[i] != null && !sockets[i].Connected) { sockets[i].Dispose(); sockets[i]= null; } } catch { } } } cts.Token.Register(Cancel);
第一次見到在同步方式中使用 CancellationTokenSource ,而且神奇地解決了檢測連接超時以及在超時發生時 dispose socket 的問題。
從中學到了 CancellationTokenSource 可以當作更簡單的定時器來使用,在博問中寫了段簡單的代碼驗證了一下,詳見 C# 中的 CancellationTokenSource 有什麽用 。
corefx 源碼學習:SqlClient 是如何同步建立 Socket 連接的