openresty开发系列33--openresty执行流程之2重写赋值阶段
一）重写赋值阶段
1）set_by_lua
语法：set_by_lua $res <lua-script-str> [$arg1 $arg2 …]
语境：server、server if、location、location if
阶段：rewrite
设置nginx变量，我们用的set指令即使配合if指令也很难实现负责的赋值逻辑；
传入参数到指定的lua脚本代码中执行，并得到返回值到res中。
<lua-script-str>中的代码可以使从ngx.arg表中取得输入参数(顺序索引从1开始)。
这个指令是为了执行短期、快速运行的代码因为运行过程中nginx的事件处理循环是处于阻塞状态的。
耗费时间的代码应该被避免。
禁止在这个阶段使用下面的API：
1、output api（ngx.say和ngx.send_headers）；
2、control api（ngx.exit）；
3、subrequest api（ngx.location.capture和ngx.location.capture_multi）；
4、cosocket api（ngx.socket.tcp和ngx.req.socket）；
5、sleep api（ngx.sleep）
a、nginx.conf配置文件
location /lua {
set $jump "1";
echo $jump;
}
set 命令对变量进行赋值，但有些场景的赋值业务比较复杂，需要用到lua脚本
所以用到set_by_lua
b、补充知识点：
ngx.var.arg与ngx.req.get_uri_args的区别，都是能够获取请求参数
ngx.var.arg_xx与ngx.req.get_uri_args["xx"]两者都是为了获取请求uri中的参数
例如 http://pureage.info?strider=1
为了获取输入参数strider，以下两种方法都可以：
local strider = ngx.var.arg_strider
local strider = ngx.req.get_uri_args["strider"]
差别在于，当请求uri中有多个同名参数时,ngx.var.arg_xx的做法是取第一个出现的值
ngx.req_get_uri_args["xx"]的做法是返回一个table，该table里存放了该参数的所有值
例如,当请求uri为：http://pureage.info?strider=1&strider=2&strider=3&strider=4时
ngx.var.arg_strider的值为"1",而ngx.req.get_uri_args["strider"]的值为table ["1", "2", "3", "4"]。
因此，ngx.req.get_uri_args属于ngx.var.arg_的增强。
-------------------------------
案例需求：书店网站改造把之前的skuid为8位商品，请求到以前的页面，为9位的请求到新的页面
书的商品详情页进行了改造，美化了一下；上线了时候，不要一下子切换美化页面；；做AB概念
把新录入的书的商品 采用 新的商品详情页，之前维护的书的商品详情页 用老的页面
以前书的id 为8位，新的书id为9位
root /data/www/html;
location /book {
set_by_lua $to_type '
local skuid = ngx.var.arg_skuid
ngx.log(ngx.ERR,"skuid=",skuid)
local r = ngx.re.match(skuid, "^[0-9]{8}$")
local k = ngx.re.match(skuid, "^[0-9]{9}$")
if r then
return "1"
end;
if k then
return "2"
end;
';
if ($to_type = "1") {
echo "skuid为8位";
proxy_pass http://127.0.0.1/old_book/$arg_skuid.html;
}
if ($to_type = "2") {
echo "skuid为9位";
proxy_pass http://127.0.0.1/new_book/$arg_skuid.html;
}
}
# 具体的网页内容
[root@node5 data]# tree /data/www/html/
/data/www/html/
├── new_book
│ └── 123456789.html
└── old_book
└── 12345678.html
2 directories, 2 files
[root@node5 data]# cat /data/www/html/old_book/12345678.html
<h1>old book</h1>
[root@node5 data]# cat /data/www/html/new_book/123456789.html
<h1>new book</h1>
# 测试访问：http://10.11.0.215/book?skuid=123456789
会跳转到new_book/123456789.html
访问：http://10.11.0.215/book?skuid=12345678
会跳转到new_book/12345678.html
=======================================
2）set_by_lua_file
语法set_by_lua_file $var lua_file arg1 arg2...;
在lua代码中可以实现所有复杂的逻辑，但是要执行速度很快，不要阻塞；
location /lua_set_1 {
default_type "text/html";
set_by_lua_file $num /usr/local/luajit/test_set_1.lua;
echo $num;
}
2.1、test_set_1.lua
local uri_args = ngx.req.get_uri_args()
local i = uri_args["i"] or 0
local j = uri_args["j"] or 0
return i + j
得到请求参数进行相加然后返回。
访问如http://192.168.31.138/lua_set_1?i=1&j=10进行测试。 如果我们用纯set指令是无法实现的。
3）注意，这个指令只能一次写出一个nginx变量，但是使用ngx.var接口可以解决这个问题：
location /foo {
set $diff '';
set_by_lua $sum '
local a = 32
local b = 56
ngx.var.diff = a - b; --写入$diff中
return a + b; --返回到$sum中
';
echo "sum = $sum, diff = $diff";
}