lua-resty-http库中body_reader流式读取的正确使用方式

在使用OpenResty进行HTTP请求处理时，lua-resty-http是一个非常常用的库。很多开发者会遇到需要流式处理响应体的场景，比如处理大文件下载或分块传输编码的响应。本文将深入探讨如何正确使用body_reader功能。

常见误区分析

很多开发者会直接使用request_uri方法，然后尝试通过返回的res.body_reader进行流式读取，这会导致"can't resume a dead coroutine"的错误。这是因为request_uri方法设计上是用于简单请求场景，它会自动读取整个响应体到内存中，不适合流式处理。

正确的流式处理方法

要实现真正的流式处理，应该使用request方法而不是request_uri。以下是正确的实现方式：

location / {
    content_by_lua_block {
        local http = require "resty.http"
        local http_client = http.new()
        http_client:set_timeouts(60000, 60000, 60000) -- 设置超时时间为60秒
        
        -- 建立连接并发送请求
        local ok, err = http_client:connect("example.com", 80)
        if not ok then
            ngx.log(ngx.ERR, "连接失败: ", err)
            return ngx.exit(500)
        end
        
        -- 发送请求
        local res, err = http_client:request{
            path = "/",
            method = "GET"
        }
        
        if not res then
            ngx.log(ngx.ERR, "请求失败: ", err)
            return ngx.exit(502)
        end
        
        -- 获取流式读取器
        local reader = res.body_reader
        local buffer_size = 8192 -- 建议使用8KB的缓冲区
        
        -- 流式读取响应体
        repeat
            local chunk, err = reader(buffer_size)
            if err then
                ngx.log(ngx.ERR, "读取错误: ", err)
                break
            end
            
            if chunk then
                ngx.print(chunk)
                ngx.flush(true) -- 立即刷新输出
            end
        until not chunk
        
        -- 关闭连接
        http_client:close()
    }
}

关键点解析

1. 连接管理：必须显式建立连接(connect)和关闭连接(close)，这是流式处理的基础

2. 请求方法：使用request方法而不是request_uri，前者支持流式处理而后者不支持

3. 缓冲区大小：建议使用8KB(8192字节)的缓冲区，这是网络传输的常用块大小

4. 错误处理：需要妥善处理连接错误、请求错误和读取错误

5. 资源释放：务必在结束时关闭连接，避免资源泄漏

性能优化建议

1. 对于大文件传输，可以适当增大缓冲区大小(如32KB)以提高吞吐量

2. 考虑使用ngx.ctx来保持长连接，避免频繁建立新连接的开销

3. 在代理场景下，可以设置适当的接收和发送缓冲区大小

4. 对于高并发场景，建议实现连接池管理

适用场景

这种流式处理方法特别适合以下场景：

• 大文件下载或上传
• 视频流处理
• 实时数据流处理
• 内存受限环境下处理大响应体
• 需要边接收边处理的场景

通过正确使用lua-resty-http的流式处理功能，可以显著降低内存使用，提高处理效率，特别是在处理大响应体时效果更为明显。