如何在MVC 3中正确进行长轮询

时间:2012-06-20 10:17:21

标签: asp.net-mvc asynchronous long-polling

我正在尝试连接AsyncController,以便当用户点击订单页面上的订单上的保存时,查看相同订单的所有用户都应该收到订单已更改的通知。我实现这一点的方法是在订单页面上进行长轮询ajax请求,但是如何制作可扩展的AsyncController来处理这个问题对我来说并不明显。

所以这就是我到目前为止,ID是指示为更改或轮询更改的订单的ID。

public class MessageController : AsyncController
{
    static readonly ConcurrentDictionary<int, AutoResetEvent> Events = new ConcurrentDictionary<int, AutoResetEvent>();

    public ActionResult Signal(int id)
    {
        AutoResetEvent @event;
        if (Events.TryGetValue(id, out @event))
            @event.Set();

        return Content("Signal");
    }

    public void WaitAsync(int id)
    {
        Events.TryAdd(id, new AutoResetEvent(false));

        // TODO: This "works", but I should probably not block this thread.
        Events[id].WaitOne();
    }

    public ActionResult WaitCompleted()
    {
        return Content("WaitCompleted");
    }
}

我看过How to do long-polling AJAX requests in ASP.NET MVC?。我试图了解有关此代码的所有细节,但据我了解此代码,它阻止了线程池中的每个工作线程,据我所知,最终会导致线程饥饿。

那么,我该如何以一种漂亮,可扩展的方式实现它呢?请记住,我不想再使用任何第三方组件,我希望能够很好地理解如何正确实现这种情况。

2 个答案:

答案 0 :(得分:3)

实际上我能够在不阻塞工作线程的情况下实现它,我缺少的东西是ThreadPool.RegisterWaitForSingleObject。

public class ConcurrentLookup<TKey, TValue>
{
    private readonly Dictionary<TKey, List<TValue>> _lookup = new Dictionary<TKey, List<TValue>>();

    public void Add(TKey key, TValue value)
    {
        lock (_lookup)
        {
            if (!_lookup.ContainsKey(key))
                _lookup.Add(key, new List<TValue>());

            _lookup[key].Add(value);
        }
    }

    public List<TValue> Remove(TKey key)
    {
        lock (_lookup)
        {
            if (!_lookup.ContainsKey(key))
                return new List<TValue>();

            var values = _lookup[key];
            _lookup.Remove(key);

            return values;
        }
    }
}

[SessionState(SessionStateBehavior.Disabled)]
public class MessageController : AsyncController
{
    static readonly ConcurrentLookup<int, ManualResetEvent> Events = new ConcurrentLookup<int, ManualResetEvent>();

    public ActionResult Signal(int id)
    {
        foreach (var @event in Events.Remove(id))
            @event.Set();

        return Content("Signal " + id);
    }

    public void WaitAsync(int id)
    {
        AsyncManager.OutstandingOperations.Increment();

        var @event = new ManualResetEvent(false);

        Events.Add(id, @event);

        RegisteredWaitHandle handle = null;
        handle = ThreadPool.RegisterWaitForSingleObject(@event, (state, timeout) => 
        {
            handle.Unregister(@event);
            @event.Dispose();

            AsyncManager.Parameters["id"] = id;
            AsyncManager.Parameters["timeout"] = timeout;
            AsyncManager.OutstandingOperations.Decrement();
        }, null, new TimeSpan(0, 2, 0), false);
    }


    public ActionResult WaitCompleted(int id, bool timeout)
    {
        return Content("WaitCompleted " + id + " " + (timeout? "Timeout" : "Signaled"));
    }
}

答案 1 :(得分:0)

现在,使用async / await实现长时间轮询要容易得多。

public class MessageController : ApiController
{
    private static readonly ConcurrentDictionary<int, ManualResetEventAsync> ManualResetEvents = new ConcurrentDictionary<int, ManualResetEventAsync>();

    [HttpGet]
    public IHttpActionResult Signal(int id)
    {
        if (ManualResetEvents.TryGetValue(id, out var manualResetEvent) == false)
        {
            return Content(HttpStatusCode.OK, "Signal: No one waiting for signal");
        }

        manualResetEvent.Set();

        return Content(HttpStatusCode.OK, "Signaled: " + id);
    }

    [HttpGet]
    public async Task<IHttpActionResult> Wait(int id)
    {
        var manualResetEvent = ManualResetEvents.GetOrAdd(id, _ => new ManualResetEventAsync());

        var signaled = await manualResetEvent.WaitAsync(TimeSpan.FromSeconds(100));

        var disposed = manualResetEvent.DisposeIfNoWaiters();
        if (disposed)
        {
            ManualResetEvents.TryRemove(id, out var _);
        }

        return Content(HttpStatusCode.OK, "Wait: " + (signaled ? "Signaled" : "Timeout") + " " + id);
    }
}

internal class ManualResetEventAsync
{
    private readonly SemaphoreSlim semaphore = new SemaphoreSlim(0, int.MaxValue);
    private int waiters;

    public void Set()
    {
        semaphore.Release(int.MaxValue);
    }

    public async Task<bool> WaitAsync(TimeSpan timeSpan)
    {
        lock (semaphore)
        {
            waiters++;
        }

        var task = await semaphore.WaitAsync(timeSpan);

        lock (semaphore)
        {
            waiters--;
        }

        return task;
    }

    public bool DisposeIfNoWaiters()
    {
        lock (semaphore)
        {
            if (waiters != 0)
            {
                return false;
            }

            semaphore.Dispose();
            return true;
        }
    }
}