Compare commits

...

3 Commits

5 changed files with 142 additions and 64 deletions

View File

@ -31,6 +31,11 @@ local function isExistRole(id)
end end
-- 查询数据表中的所有角色信息 -- 查询数据表中的所有角色信息
function _M.getAllSystemRoles()
return roleModel:all()
end
-- 根据页码和数量查询数据表中的角色信息
function _M.getSystemRoles(pageNum, pageSize) function _M.getSystemRoles(pageNum, pageSize)
return roleModel:paginate(pageNum, pageSize) return roleModel:paginate(pageNum, pageSize)
end end
@ -97,7 +102,7 @@ function _M:getPermission2roleId(role_id, status)
if status ~= nil then if status ~= nil then
sql = sql.." AND \"A\".status='"..status.."'" sql = sql.." AND \"A\".status='"..status.."'"
end end
return roleModel.exec(sql) return roleModel:exec(sql)
end end
return _M return _M

View File

@ -7,9 +7,11 @@
"ngx_lua""init_by_lua_file"; "ngx_lua""init_by_lua_file";
nginx时初始化一次 nginx时初始化一次
--]] --]]
require("config") --只在第一个worker进程中执行一次
if ngx.worker.id() ~= 0 then
return
end
print("init application...")
--判断程序是否加载权限数据 --判断程序是否加载权限数据
--local dict = ngx.shared.dict --local dict = ngx.shared.dict
--local load = dict:get("RBAC") --local load = dict:get("RBAC")
@ -17,28 +19,12 @@ print("init application...")
-- return -- return
--end --end
--只在第一个worker进程中执行一次 require("config")
if ngx.worker.id() ~= 0 then print("init application woker id:", ngx.worker.id())
return
end
--初始化获取系统默认的用户权限为实现RBAC框架做权限数据准备 --初始化获取系统默认的用户权限为实现RBAC框架做权限数据准备
local function handler() local function handler()
--读取用户表、角色表和权限表中配置的权限数据 --与redis进行连接
local roleDao = require("dao.role")
--获取数据表中的记录数
local code, res = roleDao:all()
if res == nil then return end
ngx.log(ngx.INFO, "user count:"..res)
--读取角色id和角色名称
for _, row in pairs(res) do
for key, value in pairs(row) do
ngx.say(key .. ":" .. tostring(value))
end
end
--“admin-system::users::edit“ ”1“
--将取到的数据存储到redis中后续进行验证使用
local redis = require("resty.redis") local redis = require("resty.redis")
local red = redis:new() local red = redis:new()
-- 设置超时时间 -- 设置超时时间
@ -62,42 +48,33 @@ local function handler()
-- 从连接池中获取连接 -- 从连接池中获取连接
--red:set_keepalive(SYSTEM_CONFIG.REDIS.POOL_MAX_IDLE_TIME, SYSTEM_CONFIG.REDIS.POOL_SIZE) --red:set_keepalive(SYSTEM_CONFIG.REDIS.POOL_MAX_IDLE_TIME, SYSTEM_CONFIG.REDIS.POOL_SIZE)
-- 设置 key-value --读取用户表、角色表和权限表中配置的权限数据
local ok, err = red:set("admin-system:user:add", "1") local roleDao = require("dao.role")
--获取数据表中的记录数
local code, res = roleDao:getAllSystemRoles()
if res == nil then return end
--读取角色id和角色名称
for _, row in pairs(res) do
local id = row.id --:1
local name = row.role_name --:admin
--row.status:0,
local code, rest = roleDao:getPermission2roleId(id)
for _, ret in pairs(rest) do
--获取数据表中的数据
local permid = ret.permission_id
local perm = ret.permission_code
local key = name.."-"..perm
--role_name-permission_code 组成key进行验证 存储到redis中
local ok, err = red:set(key, "1")
if not ok then if not ok then
ngx.log(ngx.ERR, "redis failed to set key: "..err) ngx.log(ngx.ERR, "redis failed to set key: "..err)
return
end end
local ok, err = red:set("admin-system:user:edit", "1")
if not ok then
ngx.log(ngx.ERR, "failed to set key: "..err)
return
end end
local ok, err = red:set("admin-system:user:delete", "1")
if not ok then
ngx.log(ngx.ERR, "failed to set key: "..err)
return
end end
local ok, err = red:set("admin-system:user:view", "1")
if not ok then
ngx.log(ngx.ERR, "failed to set key: "..err)
return
end
local ok, err = red:set("admin-system:user:list", "1")
if not ok then
ngx.log(ngx.ERR, "failed to set key: "..err)
return
end
ngx.log(ngx.INFO, "set key successfully")
--关闭redis连接 --关闭redis连接
red:close() red:close()
--共享数据字典进行数据存储
--dict:set("RBAC", "1") --dict:set("RBAC", "1")
end end

View File

@ -59,4 +59,8 @@ function _M:ttl(key)
return self.red:ttl(key) return self.red:ttl(key)
end end
function _M:call(key, ...)
return self.red:call(key, ...)
end
return _M return _M

View File

@ -333,17 +333,30 @@ else
end end
--]] --]]
local perm = require("util.permissionfilter")
local userModel = require("dao.user") local rest = perm.getRolePermissions("admin")
local code, res = userModel:userRole("1") for _, row in pairs(rest) do
--显示查询到的数据记录 print(row.key)
if res ~= nil then end
print("id value: -- ", res[1].id)
perm.clearRolePermissions("admin")
--读取用户表、角色表和权限表中配置的权限数据
local roleDao = require("dao.role")
--获取数据表中的记录数
local code, res = roleDao:getAllSystemRoles()
if res == nil then return end
--读取角色id和角色名称
for _, row in pairs(res) do for _, row in pairs(res) do
for key, value in pairs(row) do --row.id:1
ngx.say(key .. ":" .. tostring(value)) --row.create_by:admin
end --row.create_time:
end --row.role_name:admin
--row.status:0,
--row.remark:超级管理员
--row.role_key:超级管理员
print(row.id..row.create_by..row.role_name)
end end
--[[ --[[

View File

@ -0,0 +1,79 @@
---
--- Generated by EmmyLua(https://github.com/EmmyLua)
--- Created by admin.
--- DateTime: 2025/11/8 11:15
--- 权限过滤器用于验证用户是否有权限进行接口访问通过redis进行权限获取
local red = require("share.redis")
local _M = {}
-- 为角色增加权限
function _M:addRolePerms(role_name, permissions, description)
local key = role_name.."-"..permissions
local res, err = red:get(key)
if res ~= nil then return end
--键值不存在则增加到redis中
res, err = red:set(key, "1")
end
-- 检查角色是否拥有指定权限
function _M:hasPermission(role_name, permission)
if role_name == nil or permission == nil then
return false
end
-- 检查直接权限
local key = role_name.."-"..permission
local res, err = red:get(key)
if res ~= nil then
return true
end
return false
end
-- 检查多角色是否拥有所有指定权限
function _M:hasMultiRoleNamePermission(rolenameTable, permission)
if rolenameTable == nil or permission == nil then
return false
end
-- 检查直接权限
for i = 1, #rolenameTable do
local role_name = rolenameTable[i]
local key = role_name.."-"..permission
local res, err = red:get(key)
if res ~= nil then
return true
end
end
return false
end
-- 获取角色的所有权限
function _M:getRolePermissions(role_name)
local allPermissions = {}
if role_name == nil then
return allPermissions
end
--获取redis中所有匹配的数据内容
local prefix = role_name.."-"
local cursor = "0"
local result = red:call('SCAN', cursor, 'MATCH', prefix .. '*', 'COUNT')
cursor = result[1]
for _, key in ipairs(result[2]) do
table.insert(allPermissions, key)
end
return allPermissions
end
-- 清除角色的所有权限
function _M:clearRolePermissions(role_name)
--将redis中角色相关的键值去掉
local keys = red:call("KEYS", role_name.."-*")
if #keys > 0 then
red:call("DEL", unpack(keys))
end
end
-- 导出模块
return _M