web-dev-qa-db-ja.com

ASP NET Coreはリクエストボディを変更/置換します

HttpContext.Request.Bodyの置換を行う必要があります。

私はミドルウェアの中でそれをやろうとしました

public async Task Invoke(HttpContext context)
    {
        if (context.Request.Path.Value.Contains("DataSourceResult"))
        {
            var originalBody = new StreamReader(context.Request.Body).ReadToEnd();
            DataSourceRequest dataSource = null;

            try
            {
                dataSource = JsonConvert.DeserializeObject<DataSourceRequest>(originalBody);
            } catch
            {
                await _next.Invoke(context);
            }

            if (dataSource != null && dataSource.Take > 2000)
            {
                dataSource.Take = 2000;

                var bytesToWrite = dataSource.AsByteArray();
                await context.Request.Body.WriteAsync(bytesToWrite, 0, bytesToWrite.Length);
            }
            else
            {
                var bytesToWrite = originalBody.AsByteArray();
                await context.Request.Body.WriteAsync(bytesToWrite, 0, bytesToWrite.Length);
            }
        }

        await _next.Invoke(context);
    }

1つ目の問題は、本文を1回だけ読み取ることができることです。2つ目の問題は、ストリームが読み取り専用であり、書き込むことができないことです。 request.bodyを変更/置換するにはどうすればよいですか(リクエスト本文のプロパティ値を変更する必要があります)?

11
ovasylenko

リクエストの本文を取得し、その内容を読み取り、変更が必要な場合は変更を加えてから、新しいストリームを作成してパイプラインを渡します。アクセスしたら、要求ストリームを置き換える必要があります。

public async Task Invoke(HttpContext context) {
    var request = context.Request;
    if (request.Path.Value.Contains("DataSourceResult")) {
        //get the request body and put it back for the downstream items to read
        var stream = request.Body;// currently holds the original stream                    
        var originalContent = new StreamReader(stream).ReadToEnd();
        var notModified = true;
        try {
            var dataSource = JsonConvert.DeserializeObject<DataSourceRequest>(originalContent);
            if (dataSource != null && dataSource.Take > 2000) {
                dataSource.Take = 2000;
                var json = JsonConvert.SerializeObject(dataSource);
                //replace request stream to downstream handlers
                var requestContent = new StringContent(json, Encoding.UTF8, "application/json");
                stream = await requestContent.ReadAsStreamAsync();//modified stream
                notModified = false;
            }
        } catch {
            //No-op or log error
        }
        if (notModified) {
            //put original data back for the downstream to read
            var requestData = Encoding.UTF8.GetBytes(originalContent);
            stream = new MemoryStream(requestData);
        }

        request.Body = stream;
    }
    await _next.Invoke(context);
}
13
Nkosi