select + set_keepalive 组合操作引起的数据读写错误
在高并发编程中,必须 要使用 连接池技术,通过减少建连、拆连次数来提高通讯速度。
错误示例代码:
local redis = require("resty.redis")
local red = redis:new()
red:set_timeout(1000) -- 1 sec
-- or connect to a unix domain socket file listened
-- by a redis server:
-- local ok, err = red:connect("unix:/path/to/redis.sock")
local ok, err = red:connect("127.0.0.1", 6379)
if not ok then
ngx.say("failed to connect: ", err)
return
end
-- ???
ok, err = red:select(1) -- 这里会是一个坑
if not ok then
ngx.say("failed to select db: ", err)
return
end
-- ???
ngx.say("select result: ", ok)
ok, err = red:set("dog", "an animal")
if not ok then
ngx.say("failed to set dog: ", err)
return
end
ngx.say("set result: ", ok)
-- put it into the connection pool of size 100,
-- with 10 seconds max idle time
local ok, err = red:set_keepalive(10000, 100)
if not ok then
ngx.say("failed to set keepalive: ", err)
return
end
如果单独执行这个用例,没有任何问题,用例是成功的。但是这段 “没问题” 的代码,却导致了诡异的现象。
大部分 Redis 请求的代码应该是类似这样的:
local redis = require("resty.redis")
local red = redis:new()
red:set_timeout(1000) -- 1 sec
-- or connect to a unix domain socket file listened
-- by a redis server:
-- local ok, err = red:connect("unix:/path/to/redis.sock")
local ok, err = red:connect("127.0.0.1", 6379)
if not ok then
ngx.say("failed to connect: ", err)
return
end
ok, err = red:set("cat", "an animal too")
if not ok then
ngx.say("failed to set cat: ", err)
return
end
ngx.say("set result: ", ok)
-- put it into the connection pool of size 100,
-- with 10 seconds max idle time
local ok, err = red:set_keepalive(10000, 100)
if not ok then
ngx.say("failed to set keepalive: ", err)
return
end
这时候第二个示例代码在生产运行中,会出现 cat 偶会被写入到数据库 1 上,且几率大约 1% 左右。
出错的原因在于 错误示例代码使用了 select(1)
操作,并且使用了长连接,并且还潜伏在了连接池中。
当下一个请求刚好从连接池中把它选出来,这里又没有重置数据库的 select(0)
操作,所以后面所有的数据操作就都会默认触发在数据库 1 上了。
怎么解决这个问题?
- 谁制造问题,谁把问题遗留尾巴擦干净;
- 处理业务前,先把 “前辈” 的尾巴擦干净;
这里明显是第一个好,对吧。