线程安全的SendAsync /等待的FIFO

时间:2019-02-03 11:22:06

标签: c# multithreading websocket async-await

所以我正在为System.Net.WebSockets.ClientWebSocket做一个线程安全包装器。

您如何看待我提出的以下方法,线程安全吗? 重要的是一次只能执行1个线程,_ws.SendAsync并且等待的线程将按FIFO顺序执行。

另一种解决方案是改为使用BlockingCollection和生产者-消费者模式。但是BlockingCollection没有异步等待功能,因此如果它是线程安全的,我希望使用当前的解决方案。我也非常感谢其他建议。

private ClientWebSocket _ws;
private int _sendTicketCount = 0;
private int _sendTicketCurrent = 1;

public async Task SendAsync(string message)
{
    if (_ws.State != WebSocketState.Open)
        throw new Exception("Connection is not open");

    try
    {
        int mySendTicket = Interlocked.Increment(ref _sendTicketCount);

        if (mySendTicket != _sendTicketCurrent)
        {
            await Task.Run(() => SpinWait.SpinUntil(() => mySendTicket == _sendTicketCurrent)).ConfigureAwait(false);
        }

        var cancel = new CancellationTokenSource(5000);
        var bytes = new ArraySegment<byte>(Encoding.UTF8.GetBytes(message));
        await _ws.SendAsync(bytes, WebSocketMessageType.Text, true, cancel.Token).ConfigureAwait(false);
    }
    catch(Exception ex)
    {
        throw ex;
    }
    finally
    {
        Interlocked.Increment(ref _sendTicketCurrent);
    }
}

2 个答案:

答案 0 :(得分:4)

您有多个生产者和一个消费者的情况。

您可以通过设置适当选项的ActionBlock来做到这一点。

答案 1 :(得分:0)

read_numbers 001.csv [1, 2, 3] read_numbers 003.csv [7, 8, 9] read_numbers 002.csv [4, 5, 6] process_numbers 003.csv [7, 8, 9] process_numbers 002.csv [4, 5, 6] process_numbers 001.csv [1, 2, 3] write_numbers 003.csv [70, 80, 90] write_numbers 002.csv [40, 50, 60] write_numbers 001.csv [10, 20, 30] 必须是易失性的,否则允许JIT将值缓存在寄存器中,这可能导致无限循环。除此之外,它看起来是线程安全的,除了这会浪费大量CPU(SpinWait完全使用内核)。

_sendTicketCurrent确实没有意义。无论如何,您都会持有一个线程,因此您可以在当前线程中进行等待。

要以更优化的方式完成您想做的事情,已经提供了一个兼容异步的同步原语:SemaphoreSlim。

await Task.Run(() => SpinWait.SpinUntil(() => mySendTicket == _sendTicketCurrent))