Question

我有一个函数可以接受两个参数，或调用另一个函数来检索这些值（在这种情况下是半默认值）。

假设第一个函数如下所示：

-- the actuall issue function:
function foo( param_1, param_2 )

   local bar_1, bar_2 = param_1, param_2 or getBar()

   -- make funny stuff with parameters
   return funnyStuffMadeWithParameters
end

function getBar()
   -- some magic to get bar.
   return wx , yz
end

在此代码中，如果没有给出参数，bar_2将变为wx，bar_1将保持为零。我知道为什么会发生这种情况，但我不知道如何以一种有效的方式表达这种有条件的指定。

我能做到：

local bar_1 = getBar()
local _,bar_2 = getBar()

但我想避免多个函数调用。还

if not bar_1 or not bar_2 then
   bar_1, bar_2 = getBar()
end

不合法，因为有4种可能性，不仅仅是两种：

 bar_1 == nil and bar_2 == nil
 bar_1 == nil and bar_2 has value
 bar_1 has value and bar_2 is nil
 bar_1 has value and bar_2 has value

在每种情况下，我只想将默认值设为缺失值，而不是两者都已存在。

我的第一个虽然是：

bar_1, bar_2 = (param_1 or getBar() ), (param_2 or _,getBar() )

但这不是合法的语法。

编辑：我能做到：

def_bar_1, def_bar_2 = getBar()
bar_1 = param_1 or def_bar_1
bar_2 = param_2 or def_bar_2

但这可能是一个不必要的函数调用。

Answer 1

function foo(param_1, param_2)

   -- obtain the default values
   local p1, p2 = getBar()

   -- combine the provided values with default values
   local bar_1, bar_2 = (param_1 or p1), (param_2 or p2)

   -- do whatever you need with bar_1 and bar_2

end

如果getBar函数调用很昂贵且应尽可能避免，则必须明确：

function foo(param_1, param_2)

   local bar_1, bar_2
   if param_1 ~= nil and param_2 ~= nil then
      -- both parameters are known, we don't need default values
      bar_1, bar_2 = param_1, param_2
   else
      -- at least one parameter is missing, the getBar call is unavoidable
      local p1, p2 = getBar()
      bar_1, bar_2 = (param_1 or p1), (param_2 or p2)
   end

   -- bar_1 and bar_2 can be used

end

Answer 2

方法＃1

if not (param_1 and param_2) then
  local def_bar_1, def_bar_2 = getBar()
  bar_1 = param_1 or def_bar_1
  bar_2 = param_2 or def_bar_2
end

方法＃2，仅适用于Lua 5.3

function foo(param_1, param_2)
  local def
  local bar_1, bar_2 = table.unpack(setmetatable({param_1, param_2}, {__index = 
    function(t,k) 
      def = def or {getBar()}; 
      return def[k] 
    end
    }), 1, 2)
  -- make funny stuff with parameters
  -- return funnyStuffMadeWithParameters
end

多个值的条件分配

2 个答案: