lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-23 11:33:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Redis ACL (#1381)

Browse Source 
* Support redis ACL
See #1380
This commit is contained in:
云风
2021-04-10 07:02:38 +08:00
committed by GitHub
parent 2b71c2755a
commit 52d65c5dcb
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lualib/skynet/db/redis.lua
View File

@@ -146,10 +146,17 @@ local function compose_message(cmd, msg)
return lines
end
local function redis_login(auth, db)
local function redis_login(conf)
local auth = conf.auth
local db = conf.db
if auth == nil and db == nil then
return
end
if auth then
if conf.username then
auth = { conf.username, auth }
end
end
return function(so)
if auth then
so:request(compose_message("AUTH", auth), read_response)
@@ -164,7 +171,7 @@ function redis.connect(db_conf)
local channel = socketchannel.channel {
host = db_conf.host,
port = db_conf.port or 6379,
auth = redis_login(db_conf.auth, db_conf.db),
auth = redis_login(db_conf),
nodelay = true,
overload = db_conf.overload,
}
@@ -256,10 +263,11 @@ local watchmeta = {
end,
}
local function watch_login(obj, auth)
local function watch_login(conf, obj)
local login_auth = redis_login(conf)
return function(so)
if auth then
so:request(compose_message("AUTH", auth), read_response)
if login_auth then
login_auth(so)
end
for k in pairs(obj.__psubscribe) do
so:request(compose_message ("PSUBSCRIBE", k))
@@ -278,7 +286,7 @@ function redis.watch(db_conf)
local channel = socketchannel.channel {
host = db_conf.host,
port = db_conf.port or 6379,
auth = watch_login(obj, db_conf.auth),
auth = watch_login(db_conf, obj),
nodelay = true,
}
obj.__sock = channel