Java:如何为`Collection`实现`toArray`

时间:2010-10-24 23:01:35

标签: java collections types casting toarray

现在,我有:

    public <T> T[] toArray(T[] old) {
        T[] arr = Arrays.copyOf(old, old.length + size());
        int i = old.length;
        for(E obj : this) {
            arr[i] = old.getClass().getComponentType().cast(obj);
            ++i;
        }
        return arr;
    }

(请注意,这不符合axtavt指出的合同。)

我收到此警告:

Type safety: Unchecked cast from capture#2-of ? to T

这仍然是实施它的最好/最直接的方法吗?我可以以某种方式编码它而没有那个警告吗?我将如何实施呢?


编辑:我目前的解决方案。首先,我真的希望在toArray本身没有这样的警告。因此,我编写了这些小辅助函数(read here)以进一步讨论这些函数:

@SuppressWarnings("unchecked") static <T> Class<? extends T> classOf(T obj) {
    return (Class<? extends T>) obj.getClass();
}

@SuppressWarnings("unchecked") static <T> Class<? extends T> classOf(T[] array) {
    return (Class<? extends T>) array.getClass().getComponentType();
}

@SuppressWarnings("unchecked") static <T> T[] newArray(Class<T> clazz, int size) {
    return (T[]) Array.newInstance(clazz, size);
}   

现在,我的toArray实现如下:

    public <T> T[] toArray(T[] array) { 
        int size = size();
        if (array.length < size) { 
            array = newArray(classOf(array), size);
        } else if (array.length > size) {
            array[size] = null;
        }

        int i = 0;
        for (E e : this) {
            array[i] = classOf(array).cast(e);
            i++;
        }
        return array;
    } 

2 个答案:

答案 0 :(得分:8)

  

这仍然是实施它的最好/最直接的方法吗?我将如何实施呢?

这不是Josh Bloch的表现。看看AbstractCollection#toArray()的来源。以下是JDK 1.6.0_22的相关摘录。

public <T> T[] toArray(T[] a) {
    // Estimate size of array; be prepared to see more or fewer elements
    int size = size();
    T[] r = a.length >= size 
        ? a 
        : (T[]) Array.newInstance(a.getClass().getComponentType(), size);
    Iterator<E> it = iterator();

    for (int i = 0; i < r.length; i++) {
        if (!it.hasNext()) { // fewer elements than expected
            if (a != r)
                return Arrays.copyOf(r, i);
            r[i] = null; // null-terminate
            return r;
        }
        r[i] = (T) it.next();
    }
    return it.hasNext() ? finishToArray(r, it) : r;
}

源代码可在JDK的src.zip文件中找到。您可以将它集成到Eclipse / IDEA / Netbeans等任何体面的IDE中,以便在打开AbstractCollection类时可以看到它。

  

我可以以某种方式编码而没有警告吗?

没有。如果它困扰你,请使用@SuppressWarnings("unchecked")

也就是说,如果可能的话,我建议延长AbstractCollection而不是实施Collection,这样您至少已经为您实现了基本功能。

答案 1 :(得分:5)

首先,如果它应该是Collection.toArray()的实现,它不遵循合同 - 你不应该在数组中保留旧元素(参见javadoc)。

正确的实现如下所示:

public <T> T[] toArray(T[] array) { 
    int size = size();
    if (array.length < size) { 
        // If array is too small, allocate the new one with the same component type
        array = Array.newInstance(array.getClass().getComponentType(), size);
    } else if (array.length > size) {
        // If array is to large, set the first unassigned element to null
        array[size] = null;
    }

    int i = 0;
    for (E e: this) {
        // No need for checked cast - ArrayStoreException will be thrown 
        // if types are incompatible, just as required
        array[i] = (T) e;
        i++;
    }
    return array;
}