在Asp.net core返回PushStream

最近用asp.net core webapi实现了一个实时视频流的推送功能，在Asp.net中，这个是通过PushStreamContent来实现的。

基于对asp.net core的知识，随手写了一个（要求控制器继承自Controller基类）

[HttpGet]
public async Task Get()
{
    var response = HttpContext.Response;
    response.ContentType = "text/html";
    response.StatusCode = ;
    var stream = HttpContext.Response.Body;

    while (true)
    {
        await Task.Delay();
        var content = DateTime.Now + @"<br>";
        var data = Encoding.Default.GetBytes(content);
        await stream.WriteAsync(data, , data.Length);
        await stream.FlushAsync();
    }
}

使用chrome调试这个接口时，发现它确实行之有效的将当前的时间推送到了浏览器的页面上。

然而，当我进一步的调试它的异常情况时，发现就算将chrome关掉，这个程序却依然在继续运行。从调试器中看到stream的状态为Aborted，已经识别到位终止的流了。

　　

并且从VS的调试窗口也能看到异常信息：

　　

但下面这两行就是不抛异常：

await stream.WriteAsync(data, , data.Length);
await stream.FlushAsync();

单单从接口的实现角度上来看，这个已经不合理了。这是一个很大的坑，功能看上去还是正确的，没有详细调试还看不出来。一个不留神就踩上了。不知道微软为什么要这么设计。

埋怨归埋怨，问题还是要解决的。我查看了下FileStreamResult的源码，发现它是靠HttpContext.RequestAborted来判断客户端是否终止了的。这是一个CancellationToken类型的对象，当客户端连接断开后，它就处于被取消的状态。

知道原因后，就可以知道如何修改我的程序了。

[HttpGet]
public async Task Get()
{
    var cancel = HttpContext.RequestAborted;

    var response = HttpContext.Response;
    response.ContentType = "text/html";
    response.StatusCode = ;
    var stream = HttpContext.Response.Body;

    while (true)
    {
        cancel.ThrowIfCancellationRequested();
        await Task.Delay(, cancel);
        var content = DateTime.Now + @"<br>";
        var data = Encoding.Default.GetBytes(content);
        await stream.WriteAsync(data, , data.Length, cancel);
        await stream.FlushAsync(cancel);
    }
}

再然后就是封装了，我这里将其封装为了一个PushStreamResult，这样就可以在PocoController中使用了。

class MyPushStreamResult :IActionResult
{
    Func<Stream, CancellationToken, Task> _pushAction;
    string _contentType;

    public MyPushStreamResult(Func<Stream, CancellationToken, Task> pushAction, string contentType)
    {
        _pushAction = pushAction;
        _contentType = contentType;
    }

    public Task ExecuteResultAsync(ActionContext context)
    {
        var response = context.HttpContext.Response;
        response.ContentType = _contentType;
        response.StatusCode = ;

        return _pushAction(response.Body, context.HttpContext.RequestAborted);
    }
}

使用方法如下：

[HttpGet]
public IActionResult Get()
{
    return new MyPushStreamResult(pushData, "text/html");
}

async Task pushData(Stream stream, CancellationToken cancel)
{
    while (true)
    {
        if (cancel.IsCancellationRequested)
            return;
        await Task.Delay(, cancel);
        var content = DateTime.Now + @"<br>";
        var data = Encoding.Default.GetBytes(content);
        await stream.WriteAsync(data, , data.Length, cancel);
        await stream.FlushAsync(cancel);
    }
}