lua_ngx_module中http请求和响应的header问题

wujiankaka · 2013-11-13T15:25:45+00:00

Hello! 2013/11/20 杨进 > 在使用srcache和memcache插件时，如果用post&#...

lua_ngx_module中http请求和响应的header问题

wujiankaka

我想通过lua_ngx_module来使用lua来进行一些访问控制，在access_by_lua指令写入lua逻辑。在这当中对于一个http请求，怎么获取请求的头部信息？

怎么区分请求和响应的头部？也就是说ngx.header.HEADER这个接口对应的是请求中的header还是给http响应添加头部？

我想要在lua中给响应添加Set-Cookie，因为需要种多个cookie，所以我写了如下一个函数：

1.function set_cookie(str)

2. local temp = {}

3. if (nil == ngx.header.set_cookie) then

4. table.insert(temp, str)

5. ngx.header["Set-Cookie"] = temp --这个地方写成ngx.header.set_cookie和ngx.header["Set-Cookie"]有何区别呢

6. else

7. table.insert(ngx.header["Set-Cookie"], str)

8. end

9.end

因为设置多个Set-Cookie的头部需要给ngx.header["Set-Cookie"]赋值一个table，所以我需要添加一个Set-Cookie的头部时，就会调用一次上面的set_cookie(str)函数。

但是第一次设置的时候是运行正常的，第二次调用set_cookie时，会执行第7行的代码，这时会报错：bad argument#1 to 'insert' (table expected, got string)

难道是说之前给ngx.header["Set-Cookie"]赋值的table因为只有一个元素被转化成了string？？

agentzh

Hello!

2013/11/12 Jian Wu:
> 我想通过lua_ngx_module来使用lua来进行一些访问控制，在access_by_lua指令写入lua逻辑。在这当中对于一个http请求，怎么获取请求的头部信息？

可以使用 ngx.req.get_headers 函数：

    https://github.com/chaoslawful/lua-nginx-module#ngxreqget_headers

或者可以直接使用 ngx.var.http_HEADER 变量（其中 HEADER 是你要读取的请求头的名字）。见

    http://nginx.org/en/docs/http/ngx_http_core_module.html#var_http_

以及

    https://github.com/chaoslawful/lua-nginx-module#ngxvarvariable

> 怎么区分请求和响应的头部？也就是说ngx.header.HEADER这个接口对应的是请求中的header还是给http响应添加头部？

ngx.header.HEADER 是对指定的响应头进行读写。这在 ngx_lua 模块的官方文档中有明确交待：

    https://github.com/chaoslawful/lua-nginx-module#ngxheaderheader

> 我想要在lua中给响应添加Set-Cookie，因为需要种多个cookie，所以我写了如下一个函数：
> 1.function set_cookie(str)

正确的添加 cookie 的代码应当是下面这个样子的：

    local function add_cookie(cookie)
        local val = ngx.header.set_cookie
        if not val then
            val = cookie
        elseif type(val) == "table" then
            val[#val + 1] = cookie
        else
            val = {val, cookie}
        end
        ngx.header.set_cookie = val
    end

后面等 ngx_lua 添加了 ngx.resp.add_header() API 之后就可以简单多了，即一行代码：

    ngx.resp.add_header("Set-Cookie", cookie)

而且这也会更加高效。

> 2.    local temp = {}
> 3.    if (nil == ngx.header.set_cookie) then
> 4.        table.insert(temp, str)
> 5.        ngx.header["Set-Cookie"] = temp
> --这个地方写成ngx.header.set_cookie和ngx.header["Set-Cookie"]有何区别呢

语义上没有区别。最后生成的响应头的名字会有微小的区别，即前者是全部小写的形式，“set-cookie”，而后者是原始的 Set-Cookie 的形式。

> 6.    else
> 7.         table.insert(ngx.header["Set-Cookie"], str)
> 8.    end
> 9.end
>
> 但是第一次设置的时候是运行正常的，第二次调用set_cookie时，会执行第7行的代码，这时会报错：bad argument#1 to 'insert'
> (table expected, got string)
> 难道是说之前给ngx.header["Set-Cookie"]赋值的table因为只有一个元素被转化成了string？？
>

ngx.header API 是通过元方法 __index 和 __newindex 驱动的，所以你写入的原始的 Lua
值并不会保持。当你读取 ngx.header.HEADER 时，如果只有一个响应头，则总是返回一个 Lua
字符串。只有当实际有多个响应头时，才会返回 Lua table.

我有计划把 ngx.header API 重命名为 ngx.resp.set_header() 和 ngx.resp.get_header
函数，这样就可以避免不必要的困惑了。

Regards,
-agentzh