如何通过检查response header，来决定直接返回结果或者重新发送一个请求

sanmaozhao · 2016-02-18T17:06:42+00:00

hi，春哥：新年好！最近发现resty.mongol库中&#x4f...

如何通过检查response header，来决定直接返回结果或者重新发送一个请求

sanmaozhao

各位大神，新年好！
我在一个集成项目中遇到以下场景，想请教一下应该用什么方式实现：

nginx通过proxy_pass将请求转发到后台服务器之后，会有两种不同的情况（可以通过response header的值来区分）

1、获取到有效的数据，可以直接返回给浏览器
2、身份认证失败，返回登录页
此时我想通过Lua脚本发送一个http请求，通过有效的用户名、密码获取session token
然后从cookie中解析出这个token，重新向后台服务器发送原始请求并带上token，然后再把结果返回给浏览器

多谢！

yuansheng

ngx.location.capture 发起子请求完成 proxy_pass 部分获取应答内容，lua 中判断你的规则，并跳转到你的需求位置。

2016-02-18 17:06 GMT+08:00 Sanmao Zhao <sanm...@gmail.com>:

各位大神，新年好！
我在一个集成项目中遇到以下场景，想请教一下应该用什么方式实现：

nginx通过proxy_pass将请求转发到后台服务器之后，会有两种不同的情况（可以通过response header的值来区分）

1、获取到有效的数据，可以直接返回给浏览器
2、身份认证失败，返回登录页
此时我想通过Lua脚本发送一个http请求，通过有效的用户名、密码获取session token
然后从cookie中解析出这个token，重新向后台服务器发送原始请求并带上token，然后再把结果返回给浏览器

多谢！

--

YuanSheng Wang

---------------------------------------

My Github: https://github.com/membphis

OpenResty lover ^_^

sanmaozhao

感谢您的回复！

按照您的思路我写了一版可行的代码，如下：

nginx.conf

        location ^~/proxy_dom/ {
            proxy_pass  http://mobiledev.smartdot.com/;
            proxy_redirect default; 
            proxy_set_header X-Real-IP $remote_addr;
            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
            proxy_set_header custom-req hello;
            rewrite_by_lua_block {
                ngx.req.set_header("Cookie", ngx.shared.cookie_store:get("token"))
            }
        }

        location ^~/dom/ {
            content_by_lua_file lua_src/add_session_token.lua;
        }

        location = /login {
            proxy_pass  http://mobiledev.smartdot.com/names.nsf?login&username=[user]&password=[password];
            proxy_redirect http://mobiledev.smartdot.com/ /dummy;
        }

        location = /dummy {
            return 200 "dummy";
        }

add_session_token.lua
local function forward_response(res)
    -- 将subrequest的body和header返回
    for k,v in pairs(res.header) do
        ngx.header[k] = v
    end
    ngx.print(res.body)
end

-- 通过当前uri获取后台服务器对应的真实uri
local uri = string.gsub(ngx.var.request_uri,"/dom","/proxy_dom")
local res = ngx.location.capture(uri)
local logintype = res.header["LoginType"]
if logintype == "0" then
    -- 身份验证失败，重新登录
    local res_login = ngx.location.capture("/login")
    -- 从cookie中截取token
    local cookie = res_login.header["Set-Cookie"]
    local m = ngx.re.match(cookie, [[(LtpaToken=.*?);]], "o")
    local token
    if m then
        token = m[1]
        ngx.log(ngx.DEBUG,"get new token:" .. token)
        -- 将token写入共享数据
        local success, err = ngx.shared.cookie_store:set("token", token)
        -- 重新发送原始请求并返回结果
        res = ngx.location.capture(uri)
        forward_response(res)
    else
        -- 未找到token，返回错误信息
        ngx.log(ngx.DEBUG,"login failed")
        ngx.say("login failed")
    end
else
    -- 身份验证通过，直接把结果返回
    ngx.log(ngx.DEBUG,"token is valid")
    forward_response(res)
end

但我觉得有个问题：现有的代码所有请求都需要ngx.location.capture获取再转发出去，是不是性能会比较差？
有没有办法仅当身份验证失败时，再通过

ngx.location.capture发送请求，其他情况任然交给nginx自己处理？

多谢！

在 2016年2月18日星期四 UTC+8下午8:43:40，YuanSheng写道：

ngx.location.capture 发起子请求完成 proxy_pass 部分获取应答内容，lua 中判断你的规则，并跳转到你的需求位置。

2016-02-18 17:06 GMT+08:00 Sanmao Zhao <sanma...@gmail.com>:
各位大神，新年好！
我在一个集成项目中遇到以下场景，想请教一下应该用什么方式实现：

nginx通过proxy_pass将请求转发到后台服务器之后，会有两种不同的情况（可以通过response header的值来区分）

1、获取到有效的数据，可以直接返回给浏览器
2、身份认证失败，返回登录页
此时我想通过Lua脚本发送一个http请求，通过有效的用户名、密码获取session token
然后从cookie中解析出这个token，重新向后台服务器发送原始请求并带上token，然后再把结果返回给浏览器

多谢！

--

--

YuanSheng Wang
---------------------------------------
My Github: https://github.com/membphis
OpenResty lover ^_^

kindy61

试试这个思路:

1. 修改身份验证失败的response code，比如改成 401。

2. 在 nginx conf 的 location 中添加 error_page 的处理，处理 location 中尝试构造新的 request 并 proxy 到后端。

2016-02-19 13:48 GMT+08:00 Sanmao Zhao <sanm...@gmail.com>:

感谢您的回复！

按照您的思路我写了一版可行的代码，如下：

nginx.conf
location ^~/proxy_dom/ { proxy_pass http://mobiledev.smartdot.com/; proxy_redirect default; proxy_set_header X-Real-IP $remote_addr; proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for; proxy_set_header custom-req hello; rewrite_by_lua_block { ngx.req.set_header("Cookie", ngx.shared.cookie_store:get("token")) } } location ^~/dom/ { content_by_lua_file lua_src/add_session_token.lua; } location = /login { proxy_pass http://mobiledev.smartdot.com/names.nsf?login&username=[user]&password=[password]; proxy_redirect http://mobiledev.smartdot.com/ /dummy; } location = /dummy { return 200 "dummy"; }

add_session_token.lua local function forward_response(res) -- 将subrequest的body和header返回 for k,v in pairs(res.header) do ngx.header[k] = v end ngx.print(res.body) end -- 通过当前uri获取后台服务器对应的真实uri local uri = string.gsub(ngx.var.request_uri,"/dom","/proxy_dom") local res = ngx.location.capture(uri) local logintype = res.header["LoginType"] if logintype == "0" then -- 身份验证失败，重新登录 local res_login = ngx.location.capture("/login") -- 从cookie中截取token local cookie = res_login.header["Set-Cookie"] local m = ngx.re.match(cookie, [[(LtpaToken=.*?);]], "o") local token if m then token = m[1] ngx.log(ngx.DEBUG,"get new token:" .. token) -- 将token写入共享数据 local success, err = ngx.shared.cookie_store:set("token", token) -- 重新发送原始请求并返回结果 res = ngx.location.capture(uri) forward_response(res) else -- 未找到token，返回错误信息 ngx.log(ngx.DEBUG,"login failed") ngx.say("login failed") end else -- 身份验证通过，直接把结果返回 ngx.log(ngx.DEBUG,"token is valid") forward_response(res) end 但我觉得有个问题：现有的代码所有请求都需要ngx.location.capture获取再转发出去，是不是性能会比较差？有没有办法仅当身份验证失败时，再通过ngx.location.capture发送请求，其他情况任然交给nginx自己处理？多谢！在 2016年2月18日星期四 UTC+8下午8:43:40，YuanSheng写道：
ngx.location.capture 发起子请求完成 proxy_pass 部分获取应答内容，lua 中判断你的规则，并跳转到你的需求位置。

2016-02-18 17:06 GMT+08:00 Sanmao Zhao <sanma...@gmail.com>:
各位大神，新年好！
我在一个集成项目中遇到以下场景，想请教一下应该用什么方式实现：

nginx通过proxy_pass将请求转发到后台服务器之后，会有两种不同的情况（可以通过response header的值来区分）

1、获取到有效的数据，可以直接返回给浏览器
2、身份认证失败，返回登录页
此时我想通过Lua脚本发送一个http请求，通过有效的用户名、密码获取session token
然后从cookie中解析出这个token，重新向后台服务器发送原始请求并带上token，然后再把结果返回给浏览器

多谢！

--

--

YuanSheng Wang
---------------------------------------
My Github: https://github.com/membphis
OpenResty lover ^_^

--

- - - - - - - - - - - -
林青(Kindy Lin)

pumulu

hi,

怎么实现的，可否分享一下，我也碰到这个问题了

在 2016年2月18日星期四 UTC+8下午5:06:42，Sanmao Zhao写道：

各位大神，新年好！
我在一个集成项目中遇到以下场景，想请教一下应该用什么方式实现：

nginx通过proxy_pass将请求转发到后台服务器之后，会有两种不同的情况（可以通过response header的值来区分）

1、获取到有效的数据，可以直接返回给浏览器
2、身份认证失败，返回登录页
此时我想通过Lua脚本发送一个http请求，通过有效的用户名、密码获取session token
然后从cookie中解析出这个token，重新向后台服务器发送原始请求并带上token，然后再把结果返回给浏览器

多谢！