歡迎光臨
每天分享高質量文章

corefx 原始碼學習:SqlClient 是如何同步建立 Socket 連接的

在昨天的技術周會上發現 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 有什麼用 。

已同步到看一看
赞(0)

分享創造快樂