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 有什么用 。

posted @ 2019-04-27 14:44 dudu 阅读(...) 评论(...) 编辑 收藏