wehub-resource-sync
57 lines
1.2 KiB
Lua
57 lines
1.2 KiB
Lua
local get_request = require("resty.core.base").get_request
|
|
|
|
|
|
local workspaces = {}
|
|
|
|
|
|
function workspaces.upsert_default()
|
|
local old_default_ws_id = kong.default_workspace
|
|
local default_ws, err = kong.db.workspaces:select_by_name("default")
|
|
if err then
|
|
return nil, err
|
|
end
|
|
if not default_ws then
|
|
default_ws, err = kong.db.workspaces:insert({ name = "default" })
|
|
if not default_ws then
|
|
return nil, err
|
|
end
|
|
end
|
|
ngx.ctx.workspace = default_ws and default_ws.id
|
|
kong.default_workspace = default_ws and default_ws.id
|
|
|
|
if old_default_ws_id ~= default_ws.id then
|
|
kong.log.debug("default workspace id changed from ", old_default_ws_id,
|
|
" to ", default_ws.id)
|
|
end
|
|
|
|
return default_ws
|
|
end
|
|
|
|
|
|
function workspaces.get_workspace()
|
|
local ws_id = ngx.ctx.workspace or kong.default_workspace
|
|
return kong.db.workspaces:select({ id = ws_id })
|
|
end
|
|
|
|
|
|
function workspaces.get_workspace_name()
|
|
return "default"
|
|
end
|
|
|
|
|
|
function workspaces.set_workspace(ws)
|
|
ngx.ctx.workspace = ws and ws.id
|
|
end
|
|
|
|
|
|
function workspaces.get_workspace_id(ctx)
|
|
if not get_request() then
|
|
return nil
|
|
end
|
|
|
|
return (ctx or ngx.ctx).workspace or kong.default_workspace
|
|
end
|
|
|
|
|
|
return workspaces
|