OpenResty 操作cookies

在or中简单的使用cookies 复杂的操作请使用 [lua_resty_cookies](https://github.com/cloudflare/lua-resty-cookie)

基本操作

  • 获取一个cookies, cookies的name叫做session
local cookie_name = "cookie_session"
ngx.say(ngx.var[cookie_name])
  • 设置 cookies
ngx.header["Set-Cookie"] = "session=af54; Path=/; Expires=" ..
ngx.cookie_time(ngx.time() + 86400)

测试代码

        location ~ /header_add {
            content_by_lua_block {
                ngx.header["Set-Cookie"] = "session=lzz; Path=/; Expires=" ..
                                           ngx.cookie_time(ngx.time() + 86400)
                ngx.say("orangleliu")
            }
        }

看看效果

# curl -i 127.0.0.1:8001/header_add
HTTP/1.1 200 OK
Server: openresty/1.9.7.1
Date: Fri, 16 Dec 2016 07:00:30 GMT
Content-Type: text/html
Transfer-Encoding: chunked
Connection: keep-alive
Set-Cookie: session=lzz; Path=/; Expires=Sat, 17-Dec-16 07:00:30 GMT
  • set cookie 使用的是直接操作 header 的方式,这个时候如果单独添加一个cookie就会覆盖掉其他的值,所以需要多做一些处理。
function get_cookies()
  local cookies = ngx.header["Set-Cookie"] or {}
  if type(cookies) == "string" then
    cookies = {cookies}
  end
  return cookies
end

function add_cookie(cookie)
  local cookies = get_cookies()
  table.insert(cookies, cookie)
  ngx.header['Set-Cookie'] = cookies
end

add_cookie("session=aff12d; Path=/")

⚠️:上面的代码需要在 header_filter 阶段运行,在这个阶段可以获取 应用或者是后端 proxy 设置的cookies, 如果你想要 添加,删除,替换某一个cookies,注意不要覆盖所有的 cookies了。

  • 删除cookies跟上面类似

一个例子,在content阶段设置了多个cookies,然后在 header_filter 阶段删除cookies

        location ~ /header_delete {
            content_by_lua_block {
                ngx.header["Set-Cookie"] = {'Foo=abc; path=/', 'age=18; path=/'}
                ngx.say("openresty")
            }

            header_filter_by_lua_block {
                local match = string.match

                local function get_cookies()
                    -- local cookies = ngx.header["Set-Cookie"] or {}
                    local cookies = ngx.header.set_cookie or {}
                    if type(cookies) == "string" then
                        cookies = {cookies}
                    end
                    return cookies
                end

                local function remove_cookie(cookie_name)
                    local cookies = get_cookies()

                    ngx.log(ngx.ERR, "source cookies ", table.concat(cookies, " "))
                    for key, value in ipairs(cookies) do
                        local name = match(value, "(.-)=")
                        ngx.log(ngx.ERR, key.."<=>", value)
                        if name == cookie_name then
                            table.remove(cookies, key)
                        end
                    end

                    ngx.header['Set-Cookie'] = cookies or {}
                    ngx.log(ngx.ERR, "new cookies ", table.concat(cookies, " "))
                end

                remove_cookie("Foo")
            }
        }
}

测试下

# curl '127.0.0.1:8001/header_delete' -i
HTTP/1.1 200 OK
Server: openresty/1.9.7.1
Date: Fri, 16 Dec 2016 08:07:34 GMT
Content-Type: text/html
Transfer-Encoding: chunked
Connection: keep-alive
Set-Cookie: age=18; path=/

其他需求

  • 在做 proxy_pass 之后想给请求统一修改cookies值这么做呢?反正都很类似了。。

使用 header_filter_by_lua 指令

header_filter_by_lua '
    local cookies = ngx.header.set_cookie
    if cookies then
        if type(cookies) ~= "table" then
            cookies = {cookies}
        end
        local gsub = string.gsub
        local changed
        for i, cookie in ipairs(cookies) do
            local new_cookie = gsub(cookie, "^target%-cookie=[^;]*", "target-cookie=myvalue", 1)
            if new_cookie ~= cookie then
                cookies[i] = new_cookie
                changed = true
            end
        end
        if changed then
            ngx.header.set_cookie = cookies
        end
    end
';

https://github.com/openresty/headers-more-nginx-module/issues/18

ref

上一篇:java Spring集合


下一篇:JavaScript遍历对象-总结一