使用Nginx+Lua搭建AB测试平台

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/lzyzsd/article/details/48580495

首先推荐一本Nginx+Lua开发的电子书

Nginx Lua是由国人开发的一个Nginx模块,核心开发者微博agentzh,支持Lua 5.1或者LuaJIT(2.0/2.1)。需要注意的是这个模块中的Lua代码都是已非阻塞模式运行的,这样就可以保证在Lua里使用Redis,Mysql,Memcached是没有问题的。

为了方便开发者开发web,微博agentzh开发了openresty这个nginx lua web框架,我们就要使用这个框架来完成我们的工作,因为它对nginx的很多东西提供了更高级的封装,也提供了更多高级的web开发api,避免了我们直接在lua拼接http cookie,header等原始的写法。具体的环境配置我就不啰嗦了,agentzh提供了一个bundle,建议直接从GitHub上clone下来编译,这个bundle包含了nginx,lua以及很多常用的nginx module。

AB测试这里我使用的方法是根据cookie中的值来决定分配到哪个服务器,如果第一次访问,会随机生成一个值,并将这个值写入cookie。代码其实很简单

local ck = require("resty.cookie")
local cookie = ck:new()
local abtest = cookie:get("abtest")
if abtest == nil then
  math.randomseed(tostring(os.time()):reverse():sub(1, 6))
  r = math.floor(math.random()*10)
  if r % 2 == 0 then
    if ngx.req.get_headers()["Cookie"] then
      ngx.header["Set-Cookie"]="abtest=1; path=/;"..ngx.req.get_headers()["Cookie"]
    else
      ngx.header["Set-Cookie"]="abtest=1; path=/;"
    end
    ngx.log(ngx.ERR, "set abtest=1 go to webv2")
    ngx.redirect("http://www.xxx.com")
  else
    if ngx.req.get_headers()["Cookie"] then
      ngx.header["Set-Cookie"]="abtest=0; path=/;"..ngx.req.get_headers()["Cookie"]
    else
      ngx.header["Set-Cookie"]="abtest=0; path=/;"
    end
    ngx.log(ngx.ERR, "set abtest=0 go to webv1")
    ngx.redirect("http://www.xxx.com")
  end
else
  if abtest == "1" then
    ngx.log(ngx.ERR, "abtest=1 go to webv2")
    ngx.exec("@sitev2")
  else
    ngx.log(ngx.ERR, "abtest=0 go to webv1")
    ngx.exec("@sitev1")
  end
end

另外新浪技术保障部开源了一套自己的Nginx Lua AB测试框架,包含了基于ip,user id等分流策略,还有一套web管理界面,方便随时修改策略。GitHub地址。我后面会写几篇博客来介绍新浪的代码,也推荐大家尽量使用新浪的这套解决方案,非常的赞。

阅读更多

没有更多推荐了,返回首页