kong--kong
wehub-resource-sync
45 行
1.2 KiB
Lua
45 行
1.2 KiB
Lua
local Workspaces = {}
|
|
|
|
|
|
local constants = require("kong.constants")
|
|
local lmdb = require("resty.lmdb")
|
|
|
|
|
|
local DECLARATIVE_DEFAULT_WORKSPACE_KEY = constants.DECLARATIVE_DEFAULT_WORKSPACE_KEY
|
|
local DECLARATIVE_DEFAULT_WORKSPACE_ID = constants.DECLARATIVE_DEFAULT_WORKSPACE_ID
|
|
|
|
|
|
function Workspaces:truncate()
|
|
self.super.truncate(self)
|
|
if kong.configuration.database == "off" then
|
|
return true
|
|
end
|
|
|
|
local default_ws, err = self:insert({ name = "default" })
|
|
if err then
|
|
kong.log.err(err)
|
|
return
|
|
end
|
|
|
|
ngx.ctx.workspace = default_ws.id
|
|
kong.default_workspace = default_ws.id
|
|
end
|
|
|
|
|
|
function Workspaces:select_by_name(key, options)
|
|
if kong.configuration.database == "off" and key == "default" then
|
|
-- TODO: Currently, only Kong workers load the declarative config into lmdb.
|
|
-- The Kong master doesn't get the default workspace from lmdb, so we
|
|
-- return the default constant value. It would be better to have the
|
|
-- Kong master load the declarative config into lmdb in the future.
|
|
--
|
|
-- it should be a table, not a single string
|
|
return { id = lmdb.get(DECLARATIVE_DEFAULT_WORKSPACE_KEY) or DECLARATIVE_DEFAULT_WORKSPACE_ID, }
|
|
end
|
|
|
|
return self.super.select_by_name(self, key, options)
|
|
end
|
|
|
|
|
|
return Workspaces
|