如何在自定义中间件中提取Azure Functions HTTP响应并修改？

在Isolated Worker模型中，直接调用context.GetHttpResponseData()会返回null，因为函数的响应结果并非直接存储在这个位置，而是保存在函数调用结果的绑定数据里。要修改HTTP响应，需要从InvocationResult中获取并修改返回值。

修正后的中间件代码 #

internal sealed class MyCustomMiddleware : IFunctionsWorkerMiddleware
{
    public async Task Invoke(FunctionContext context, FunctionExecutionDelegate next)
    {
        await next(context);

        // 获取函数调用结果
        var invocationResult = context.GetInvocationResult();
        if (invocationResult?.Value == null) return;

        // 处理函数返回字符串的情况
        if (invocationResult.Value is string originalResponse)
        {
            // 获取请求对象，用于创建响应
            var request = await context.GetHttpRequestDataAsync();
            if (request == null) return;

            // 创建新的响应并修改内容
            var newResponse = request.CreateResponse(System.Net.HttpStatusCode.OK);
            await newResponse.WriteStringAsync($"{originalResponse} World");
            // 替换原有返回值
            invocationResult.Value = newResponse;
        }
        // 处理函数直接返回HttpResponseData的情况
        else if (invocationResult.Value is HttpResponseData existingResponse)
        {
            // 重置流位置并读取原有响应内容
            existingResponse.Body.Position = 0;
            var originalContent = await new StreamReader(existingResponse.Body).ReadToEndAsync();
            
            // 清空流并写入修改后的内容
            existingResponse.Body.SetLength(0);
            await existingResponse.WriteStringAsync($"{originalContent} World");
        }
    }
}

关键说明 #

• context.GetInvocationResult()：用于获取函数执行后的返回结果，无论是字符串还是HttpResponseData都存储在这里。
• 双场景兼容：覆盖了函数返回字符串和直接返回HttpResponseData的常见情况，确保不同返回类型的HTTP触发器都能修改响应。
• 流操作注意：修改已有HttpResponseData时，必须重置流位置并清空原有内容，避免新旧内容叠加导致响应异常。

内容的提问来源于stack exchange，提问作者Fylix