在 ASP.NET 5 Core 中保存请求和响应标头

List < string > AllResponseHeaders = new List < string > ();
var uniqueResponseHeaders = httpContext.Response.Headers.Where(x => AllResponseHeaders.All(h => h != x.Key)).Select(x => x.Key);

构建一个中间件来读取请求和响应头

中间件是处理请求和响应的软件组件。ASP.Net Core 通常有一系列中间件组件——这些中间件组件要么处理请求，要么将它们传递给请求处理管道中的下一个组件。

在项目的根目录中创建一个新类并将其命名为 RequestResponseHeaderMiddleware。接下来，将默认代码替换为以下内容：

public class RequestResponseHeadersMiddleware {
    private readonly RequestDelegate _next;
    public RequestResponseHeadersMiddleware(RequestDelegate next) {
        _next = next;
    }
    public async Task Invoke(HttpContext httpContext) {
        await _next.Invoke(httpContext);
    }
}

这是 ASP.NET Core 中中间件的极简实现。但是，除了将控制权传递给管道中的下一个组件之外，它不会做任何事情。

我们需要添加读取请求和响应头的逻辑。我们已经在上一节中编写了代码。这是我们中间件的更新版本：

public class RequestResponseHeadersMiddleware {
    private readonly RequestDelegate _next;
    public readonly List < string > AllRequestHeaders = new List < string > ();
    public readonly List < string > AllResponseHeaders = new List < string > ();
    public RequestResponseHeadersMiddleware(RequestDelegate next) {
        _next = next;
    }
    public async Task Invoke(HttpContext httpContext) {
        var requestHeaders = httpContext.Request.Headers.Where(x => AllRequestHeaders.All(h => h != x.Key)).Select(x => x.Key);
        AllRequestHeaders.AddRange(requestHeaders);
        await this._next.Invoke(httpContext);
        var uniqueResponseHeaders = httpContext.Response.Headers.Where(x => AllResponseHeaders.All(h => h != x.Key)).Select(x => x.Key);
        AllResponseHeaders.AddRange(uniqueResponseHeaders);
        await _next.Invoke(httpContext);
    }
}

为中间件创建扩展方法

我们现在将为我们的中间件创建一个扩展方法，以便能够在 Startup 类的 Configure 方法中方便地配置中间件。

为此，请在名为 RequestResponseHeadersMiddlewareExtensions.cs 的文件中创建一个名为 RequestResponseHeadersMiddlewareExtensions 的新类，其中包含以下代码：

public static class RequestResponseHeadersMiddlewareExtensions {
    public static IApplicationBuilder UseRequestResponseHeadersMiddleware(this IApplicationBuilder builder) {
        return builder.UseMiddleware < RequestResponseHeadersMiddleware > ();
    }
}

在启动类中配置中间件

您可以在 Startup 类的 Configure 方法中编写以下代码来启用我们迄今为止构建的自定义中间件。

app.UseRequestResponseHeadersMiddleware();

创建模型

现在在扩展名为 .cs 的同名文件中创建以下类。我们将使用它来存储请求和响应标头值。

public class RequestResponseHeader {
    public string RequestHeader {
        get;
        set;
    }
    public string ResponseHeader {
        get;
        set;
    }
}

创建 DbManager 类

DbManager 类使用 dotConnect for Postgresql 包装对数据库的所有调用。它只有一种名为 Save 的方法来将请求和响应头值保存在 Postgresql 数据库中。

public class DbManager {
    private readonly string connectionString = "Some connection string...";
    public int Save(RequestResponseHeader requestResponseHeader) {
        try {
            using(PgSqlConnection pgSqlConnection = new PgSqlConnection(connectionString)) {
                using(PgSqlCommand cmd = new PgSqlCommand()) {
                    cmd.CommandText = "INSERT INTO
                    public.requestresponseheaders " +
                    "(requestheader, responseheader) " + "VALUES (@requestheader, @responseheader)";
                    cmd.Connection = pgSqlConnection;
                    cmd.Parameters.AddWithValue("requestkey", requestResponseHeader.RequestHeader);
                    cmd.Parameters.AddWithValue("responsevalue", requestResponseHeader.ResponseHeader);
                    if (pgSqlConnection.State != System.Data.ConnectionState.Open) pgSqlConnection.Open();
                    return cmd.ExecuteNonQuery();
                }
            }
        } catch {
            throw;
        }
    }
}

从中间件调用 Save 方法

我们现在将修改我们实现的中间件，以使用 dotConnect for PostgreSQL 在 PostgreSQL 数据库中存储请求和响应标头。

您可以使用以下代码片段从我们之前实现的中间件中调用 DbManager 类的 Save 方法。

for (int index = 0; index < AllRequestHeaders.Count; index++) {
    RequestResponseHeader requestResponseHeader = new
    RequestResponseHeader();
    requestResponseHeader.RequestHeader = AllRequestHeaders[index];
    requestResponseHeader.ResponseHeader = AllResponseHeaders[index];
    dbManager.Save(requestResponseHeader);
}

正如您在此处看到的，有两个字符串集合，即 AllRequestHeaders 和 AllResponseHeaders，分别保存请求和响应标头值。我们在文章前面看到了它是如何完成的。

在此代码片段中，创建了 RequestResponseHeader 的一个实例，初始化其属性，然后将此实例作为参数传递给 Save 方法。

下面是中间件的完整代码，供大家参考：

public class RequestResponseHeadersMiddleware {
    private readonly RequestDelegate _next;
    public readonly List < string > AllRequestHeaders = new List < string > ();
    public readonly List < string > AllResponseHeaders = new List < string > ();
    private readonly DbManager dbManager = new DbManager();
    public RequestResponseHeadersMiddleware(RequestDelegate next) {
        _next = next;
    }
    public async Task Invoke(HttpContext httpContext) {
        var requestHeaders = httpContext.Request.Headers.Where(x => AllRequestHeaders.All(h => h != x.Key)).Select(x => x.Key);
        AllRequestHeaders.AddRange(requestHeaders);
        await this._next.Invoke(httpContext);
        var uniqueResponseHeaders = httpContext.Response.Headers.Where(x => AllResponseHeaders.All(h => h != x.Key)).Select(x => x.Key);
        AllResponseHeaders.AddRange(uniqueResponseHeaders);
        await _next.Invoke(httpContext);
        for (int index = 0; index < AllRequestHeaders.Count; index++) {
            RequestResponseHeader requestResponseHeader = new
            RequestResponseHeader();
            requestResponseHeader.RequestHeader = AllRequestHeaders[index];
            requestResponseHeader.ResponseHeader = AllResponseHeaders[index];
            dbManager.Save(requestResponseHeader);
        }
    }
}