nginx+lua获取POST参数问题

Question

各位大神，我有个需求就是需要判断url中的参数来获取service参数的值，url有GET和POST请求。我是用nginx+lua做的。代码如下：

        location / {
                set_by_lua $service '
                        local request_method = ngx.var.request_method
                        if request_method == "GET" then
                                local arg = ngx.req.get_uri_args()["service"] or 0
                                return arg
                        elseif request_method == "POST" then
                                ngx.req.read_body()
                                local arg = ngx.req.get_post_args()["service"] or 0
                                return arg
                        end;';
                
                if ($service = 'register')
                        {       
                                proxy_pass http://userinfo;
                        }
                                
                proxy_redirect off;
                proxy_set_header HOST $host;
                proxy_set_header X-Real-IP $remote_addr;
                proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
        }

现在的问题是，当我用GET方式请求的时候，一切正常，但是用POST方式请求的时候，Nginx就报500错误。

我用如下的代码调试：

local request_method = ngx.var.request_method
if request_method == "GET" then
        local arg = ngx.req.get_uri_args()["service"] or 0
        ngx.say(arg)
elseif request_method == "POST" then
        ngx.req.read_body()
        local arg = ngx.req.get_post_args()["service"] or 0
        ngx.say(arg)
end

GET和POST请求方式打印出来的service参数的值都是正确的。

现在不知道问题出在哪儿了？求大神告知。感激。。。

Answer 1

set $service '';

       rewrite_by_lua  '
                local request_method = ngx.var.request_method
                if request_method == "GET" then
                        local arg = ngx.req.get_uri_args()["service"] or 0
                        ngx.var.service = arg
                elseif request_method == "POST" then
                        ngx.req.read_body()
                        local arg = ngx.req.get_post_args()["service"] or 0
                        ngx.var.service = arg
                end;';
        
        if ($service = 'register')
                {       
                        proxy_pass http://userinfo;
                }
                

if ($service = 'register') 要比rewrite_by_lua先执行，所以$service根本获取不到内容，不知道有什么解决获取post参数，然后通过参数值判断跳转到不同的upstream，试过用set_form_input $service;也不行，获取不到变量的内容！！！有大神有什么解决方案吗

Answer 2

解决了，只能说是一个替代方案吧：（proxy_pass的值设置成变量，在rewrite_by_lua模块内条件判断进行赋值）

set $userinfo http://userinfo; 
set $p_pass http://default;  # proxy_pass的值设置成变量, 默认值
  rewrite_by_lua  '
            local request_method = ngx.var.request_method
            if request_method == "GET" then
                    local arg = ngx.req.get_uri_args()["service"] or 0
                    ngx.var.service = arg
            elseif request_method == "POST" then
                    ngx.req.read_body()
                    local arg = ngx.req.get_post_args()["service"] or 0
                    ngx.var.service = arg
            end;
            if ngx.var.service == "register" then
                ngx.var.p_pass = ngx.var.userinfo
            end;
  ';
   proxy_pass $p_pass;

（项目遇到问题刚看两天lua，写的不好，或者大牛有更好的办法记得留言，谢谢）

Answer 3

ngx.req.get_post_args()只能在rewrite_by_lua, access_by_lua, content_by_lua*阶段使用，且在使用前需要先调用ngx.req.read_body()，或打开
lua_need_request_body 选项强制本模块读取请求体（此方法不推荐）

Answer 4

楼主请教一下。不知道为什么我的ngx.req.get_post_args()一直获取不到值，但ngx.req.get_body_data()可以。可能是什么原因呢？

Answer 5

当你使用set_by_lua时，如果是POST提交，在调用ngx.req.read_body()时，set_by_lua模块中read_body是被禁止调用的（ failed to run set_by_lua: set_by_lua:6: API disabled in the context of set_by_lua
stack traceback: [C]: in function 'read_body' set_by_lua:6: in function <set_by_lua:1>）；
但是你可以使用rewrite_by_lua，如：
location / {

           set $service '';
           rewrite_by_lua  '
                    local request_method = ngx.var.request_method
                    if request_method == "GET" then
                            local arg = ngx.req.get_uri_args()["service"] or 0
                            ngx.var.service = arg
                    elseif request_method == "POST" then
                            ngx.req.read_body()
                            local arg = ngx.req.get_post_args()["service"] or 0
                            ngx.var.service = arg
                    end;';
            
            if ($service = 'register')
                    {       
                            proxy_pass http://userinfo;
                    }
                            
            proxy_redirect off;
            proxy_set_header HOST $host;
            proxy_set_header X-Real-IP $remote_addr;
            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
    }

希望能够帮到你，另外建议变量名字不要取$service，可能是关键字与系统变量有冲突。