KeepAlive

在 OpenResty 中，连接池在使用上如果不加以注意，容易产生数据写错地方，或者得到的应答数据异常等类似的问题。当然使用短连接可以规避这样的问题，但是在一些企业应用环境中，短连接 + 高并发 对企业内部的防火墙是一个巨大的考验。因此，长连接自有其用武之地，使用它的时候要记住，长连接一定要保持其连接池中所有连接的正确性。

  1. -- 错误的代码
  2. local function send()
  3.     for i = 1, count do
  4.         local ssdb_db, err = ssdb:new()
  5.         local ok, err = ssdb_db:connect(SSDB_HOST, SSDB_PORT)
  7.         if not ok then
  8.             ngx.log(ngx.ERR, "create new ssdb failed!")        -- 这里只是记录了 log，并没有 return 语句
  9.         else
  10.             local key,err = ssdb_db:qpop(something)
  11.             if not key then
  12.                 ngx.log(ngx.ERR, "ssdb qpop err:", err)
  13.             else
  14.                 local data, err = ssdb_db:get(key[1])
  15.                 -- other operations
  16.             end
  17.         end
  18.     end
  19.     ssdb_db:set_keepalive(SSDB_KEEP_TIMEOUT, SSDB_KEEP_COUNT)  -- 这里仍然会把失败的连接加入到连接池
  20. end
  22. -- 调用
  23. while true do
  24.     local ths = {}
  25.     for i=1,THREADS do
  26.         ths[i] = ngx.thread.spawn(send)       ----创建线程
  27.     end
  29.     for i = 1, #ths do
  30.         ngx.thread.wait(ths[i])               ----等待线程执行
  31.     end
  32.     ngx.sleep(0.020)
  33. end

以上代码在测试中发现，应该得到 get(key) 的返回值有一定几率为 key。

原因即是在 ssdb 创建连接时可能会失败，但是在得到失败的结果后依然调用 ssdb_db:set_keepalive 将此连接加入连接池中。

正确地做法是，如果连接出现错误，则不要将该连接加入连接池。

  1. local function send()
  2.     for i = 1, count do
  3.         local ssdb_db, err = ssdb:new()
  4.         local ok, err = ssdb_db:connect(SSDB_HOST, SSDB_PORT)
  6.         if not ok then
  7.             ngx.log(ngx.ERR, "create new ssdb failed!")
  8.             return         -- 这里添加了 return 语句
  9.         else
  10.             local key, err = ssdb_db:qpop(something)
  11.             if not key then
  12.                 ngx.log(ngx.ERR, "ssdb qpop err:", err)
  13.             else
  14.                 local data, err = ssdb_db:get(key[1])
  15.                 -- other operations
  16.             end
  17.             -- 把语句移动到了这里，只会把成功的连接加入到连接池
  18.             ssdb_db:set_keepalive(SSDB_KEEP_TIMEOUT, SSDB_KEEP_COUNT)
  19.         end
  20.     end
  21. end

所以，当你使用长连接操作 db 出现结果错乱现象时，首先应该检查下是否存在长连接使用不当的情况。