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)
  6. if not ok then
  7. ngx.log(ngx.ERR, "create new ssdb failed!")
  8. else
  9. local key,err = ssdb_db:qpop(something)
  10. if not key then
  11. ngx.log(ngx.ERR, "ssdb qpop err:", err)
  12. else
  13. local data, err = ssdb_db:get(key[1])
  14. -- other operations
  15. end
  16. end
  17. end
  18. ssdb_db:set_keepalive(SSDB_KEEP_TIMEOUT, SSDB_KEEP_COUNT)
  19. end
  20. -- 调用
  21. while true do
  22. local ths = {}
  23. for i=1,THREADS do
  24. ths[i] = ngx.thread.spawn(send) ----创建线程
  25. end
  26. for i = 1, #ths do
  27. ngx.thread.wait(ths[i]) ----等待线程执行
  28. end
  29. ngx.sleep(0.020)
  30. 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)
  5. if not ok then
  6. ngx.log(ngx.ERR, "create new ssdb failed!")
  7. return
  8. else
  9. local key,err = ssdb_db:qpop(something)
  10. if not key then
  11. ngx.log(ngx.ERR, "ssdb qpop err:", err)
  12. else
  13. local data, err = ssdb_db:get(key[1])
  14. -- other operations
  15. end
  16. ssdb_db:set_keepalive(SSDB_KEEP_TIMEOUT, SSDB_KEEP_COUNT)
  17. end
  18. end
  19. end

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