在groovy中重载数组的运算符

时间:2010-12-14 19:55:09

标签: groovy overloading operator-keyword

我是一个时髦的新手。也许这是小菜一碟,但我想重载数组/列表的+运算符来代码像这样

def a= [1,1,1]
def b= [2,2,2]

assert [3,3,3] == a + b 

1 个答案:

答案 0 :(得分:6)

我不建议全球覆盖既定的行为。但是,如果你坚持,这将按照你的要求行事:

ArrayList.metaClass.plus << {Collection b -> 
    [delegate, b].transpose().collect{x, y -> x+y}
}

更加本地化的替代方案是使用类别:

class PlusCategory{
    public static Collection plus(Collection a, Collection b){
        [a, b].transpose().collect{x, y -> x+y}
    }
}
use (PlusCategory){
    assert [3, 3, 3] == [1, 1, 1] + [2, 2, 2]
}

但是,我可能会创建一个通用的zipWith方法(如在函数式编程中),允许人们轻松指定不同的行为......

Collection.metaClass.zipWith = {Collection b, Closure c -> 
    [delegate, b].transpose().collect(c)
}
assert [3, 3, 3] == [1, 1, 1].zipWith([2, 2, 2]){a, b -> a+b}
assert [2, 2, 2] == [1, 1, 1].zipWith([2, 2, 2]){a, b -> a*b}