Question

在groovy v 1.8.1中将take和drop方法添加到列表对象中，并按以下方式工作：

def list = ['Simple', 'list', 'with', 5, 'items']   
assert list.take(2) == ['Simple', 'list']

我只有v 1.8.0可用。如何使用Groovy v1.8.0重新实现我自己的list.take（2）版本？

是否可以执行以下操作：

def list = ['Simple', 'list', 'with', 5, 'items']  
def limit = 2 
assert list['0..'+limit] == ['Simple', 'list']

当我尝试这个时，我得到一个例外。

Answer 1

你可以按照groovy的方式做到这一点

https://github.com/groovy/groovy-core/blob/master/src/main/org/codehaus/groovy/runtime/DefaultGroovyMethods.java#L6498

public static <T> List<T> take(List<T> self, int num) {
    if (self.isEmpty() || num <= 0) {
        return createSimilarList(self, 0);
    }
    if (self.size() <= num) {
        List<T> ret = createSimilarList(self, self.size());
        ret.addAll(self);
        return ret;
    }
    List<T> ret = createSimilarList(self, num);
    ret.addAll(self.subList(0, num));
    return ret;
}

方法createSimilarList可以在这里找到

https://github.com/groovy/groovy-core/blob/master/src/main/org/codehaus/groovy/runtime/DefaultGroovyMethodsSupport.java#L162

Answer 2

切片可能会实现相同的目标：

def list = ['Simple', 'list', 'with', 5, 'items']   
assert list[0..1] == ['Simple', 'list']

Answer 3

与GargantuChet的答案一样，你可以将它切片，但使用专属范围：

List.metaClass.take { n ->
    def len = [n,delegate.size()].min()
    delegate[0..<len]
}

assert [1,2,3,4].take(2) == [1,2]
assert [1,2].take(2) == [1,2]
assert [1].take(2) == [1]
assert [].take(2) == []

groovy采取如何重新实现

3 个答案: