// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Microsoft.AspNetCore.Connections.Features;
 
namespace Microsoft.AspNetCore.Http.Features;
 
/// <summary>
/// Default implementation of <see cref="IConnectionLifetimeNotificationFeature"/>.
/// </summary>
internal sealed class DefaultConnectionLifetimeNotificationFeature : IConnectionLifetimeNotificationFeature
{
    private readonly IHttpResponseFeature? _httpResponseFeature;
 
    /// <summary>
    ///
    /// </summary>
    /// <param name="httpResponseFeature"></param>
    public DefaultConnectionLifetimeNotificationFeature(IHttpResponseFeature? httpResponseFeature)
    {
        _httpResponseFeature = httpResponseFeature;
    }
 
    ///<inheritdoc/>
    public CancellationToken ConnectionClosedRequested { get; set; }
 
    ///<inheritdoc/>
    public void RequestClose()
    {
        if (_httpResponseFeature != null)
        {
            if (!_httpResponseFeature.HasStarted)
            {
                _httpResponseFeature.Headers.Connection = "close";
            }
        }
    }
}