fix problem with recursive tables when using the 'process' option

This commit is contained in:
Andreas Hofer 2016-03-29 22:14:10 +02:00
parent a998635207
commit d372d2ab08
2 changed files with 30 additions and 15 deletions

View File

@ -165,27 +165,36 @@ local function makePath(path, ...)
end end
local function processRecursive(process, item, path) local function processRecursive(process, item, path)
local visited = {}
local function processRecursive2(item, path)
if item == nil then return nil end if item == nil then return nil end
if visited[item] then return visited[item] end
local processed = process(item, path) local processed = process(item, path)
if type(processed) == 'table' then if type(processed) == 'table' then
local processedCopy = {} local processedCopy = {}
visited[item] = processedCopy
local processedKey local processedKey
for k,v in pairs(processed) do for k,v in pairs(processed) do
processedKey = processRecursive(process, k, makePath(path, k, inspect.KEY)) processedKey = processRecursive2(k, makePath(path, k, inspect.KEY))
if processedKey ~= nil then if processedKey ~= nil then
processedCopy[processedKey] = processRecursive(process, v, makePath(path, processedKey)) processedCopy[processedKey] = processRecursive2(v, makePath(path, processedKey))
end end
end end
local mt = processRecursive(process, getmetatable(processed), makePath(path, inspect.METATABLE)) local mt = processRecursive2(getmetatable(processed), makePath(path, inspect.METATABLE))
setmetatable(processedCopy, mt) setmetatable(processedCopy, mt)
processed = processedCopy processed = processedCopy
end end
return processed return processed
end end
return processRecursive2(item, path)
end
------------------------------------------------------------------- -------------------------------------------------------------------

View File

@ -315,6 +315,12 @@ describe( 'inspect', function()
}, items) }, items)
end) end)
it('handles recursive tables correctly', function()
local tbl = { 1,2,3}
tbl.loop = tbl
inspect(tbl, { process=function(x) return x end})
end)
end) end)
describe('metatables', function() describe('metatables', function()