Lua - Come posso ottenere un ritorno?

Question

Sono interessato ad acquisire qualsiasi formato di ritorno di una funzione. Per esempio

function foo() 
    return 1 
end 

local result = foo() -- foo is numeric 1 

function foo() 
    return {1,2,3} 
end 
local result1, result2, result3 = foo() 
local result = foo() -- this is bad as result is `1` but `2` and `3` are lost 

function foo() 
    return 1, 2, 3 
end 
local result = foo() -- foo is a table with all the numbers, that's ok 

Sto costruendo un profiler che sovrascriverà funzioni con le funzioni di proxy, ma ho bisogno di sapere i dati restituiti, quindi controllare type() di esso e l'accesso accordingly` ma dal codice si può vedere che io sono impossibile accedere a tutte e 3 le situazioni con un solo metodo. C'è qualche ?

Answer 1

Se si conosce il numero massimo dei rendimenti, usare qualcosa come

v1,v2,v3 = foo() 

, ma non sarà in grado di dire se foo restituito due valori o tre, con l'ultima delle quali a zero.

La soluzione robusta è quello di raccogliere tutti i ritorni in una tabella:

v = table.pack(foo()) 

Poi v.n contiene il numero di valori restituiti, tra cui tutti i nils.

Answer 2

Ecco una versione che funziona su qualsiasi Lua versione 5.1+.

local function WrapAndInspect(Inspector, FunctionToCall) 
    local function Inspect(...) 
     Inspector(...) 
     return ... 
    end 
    local function Wrapper(...) 
     return Inspect(FunctionToCall(...)) 
    end 
    return Wrapper 
end 

Cosa WrapAndInspect fa è generare una funzione che chiamerà la funzione data, quindi passare i suoi valori di ritorno per una seconda funzione che fornisci. Quella funzione può fare qualunque elaborazione tu ritenga necessaria su di loro. Ma il framework garantirà che i valori di ritorno dalla funzione originale siano passati come erano.

Ecco una variante che fa qualcosa di simile, ma invece di avvolgere la FunctionToCall, restituisce una funzione che prende una funzione per essere chiamato (insieme con i suoi parametri):

local function CallAndInspect(Inspector) 
    local function Inspect(...) 
     Inspector(...) 
     return ... 
    end 
    local function Caller(FunctionToCall, ...) 
     return Inspect(FunctionToCall(...)) 
    end 
    return Caller 
end 

si potrebbe usare questo uno su qualsiasi funzione particolare che si desidera esaminare.

Answer 3

Ecco una soluzione alternativa per coloro che non hanno accesso a table.pack. Per me sembra semplice e dovrebbe funzionare su lua 5.1 e versioni successive - e forse anche prima in lua.

table_pack dovrebbe funzionare come table.pack

function table_pack(...) 
    return {n=select("#", ...), ...} 
end 
function foo() 
    return 1, 2, 3 
end 

local v = table_pack(foo()) 
print(v.n)