groovy - groovy 采取如何重新实现

Question

take 和 drop 方法被添加到 groovy v 1.8.1 中的列表对象中，工作方式如下：

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) == []