Java 字节数组集

Question

我有一个 byte[]s 的 HashSet，我想测试一个新的 byte[] 是否在该集合中。问题在于 Java 似乎是在测试 byte[] 实例是否相同，而不是测试字节数组中的实际值是否相同。

换句话说，考虑以下代码：

public class Test
{
    public static void main(String[] args)
    {
        java.util.HashSet<byte[]> set=new java.util.HashSet<byte[]>();
        set.add(new String("abc").getBytes());
        System.out.println(set.contains(new String("abc").getBytes()));
    }
}

此代码打印出false，我希望它打印出true。我该怎么做呢？

Answer 1

您可以使用ByteBuffer.wrap 包装每个字节数组，这将为您提供正确的equals 和hashCode 行为。请注意您在 ByteBuffer 上调用的方法（不要修改数组或推进其指针）。

Answer 2

您可以创建一个 ByteArray 类来包装字节数组并按照您想要的方式测试相等性。然后你会有一个Set<ByteArray>。

Answer 3

您可以定义自己的包装类，但可能最简单的做法是将数组“包装”到 ArrayLists 中并使用HashSet<ArrayList>。

Answer 4

现代（截至目前的解决方案）

import com.google.common.collect.ImmutableSet;

import java.nio.ByteBuffer;
import java.util.Set;

import static com.google.common.base.Charsets.UTF_8;
import static java.nio.ByteBuffer.wrap;

public class Scratch
{
    public static void main(String[] args)
    {
        final Set<ByteBuffer> bbs = ImmutableSet.of(wrap("abc".getBytes(UTF_8)).asReadOnlyBuffer());
        System.out.println("bbs.contains(ByteBuffer.wrap(\"abc\".getBytes(Charsets.UTF_8))) = " + bbs.contains(wrap("abc".getBytes(UTF_8)).asReadOnlyBuffer()));
    }
}

注意事项：

您应该从不在不提供Charset 的情况下将String 转换为byte[]，结果将依赖于基于默认Charset 的运行时，这通常不是一个好的结果，并且可以改变。

.asReadOnlyBuffer() 很重要！

创建一个共享此缓冲区的新的只读字节缓冲区 内容。新缓冲区的内容就是这个缓冲区的内容。 此缓冲区内容的更改将在新缓冲区中可见； 但是，新缓冲区本身将是只读的，并且不允许 要修改的共享内容。

两个缓冲区的位置、限制和标记值将是独立的。

新缓冲区的容量、限制、位置和标记值将与此缓冲区的相同。 如果此缓冲区本身是只读的，则此方法的行为与 与复制方法相同。

Answer 5

您可以避免包装器和愚蠢的 hashCode 问题（嘿，像 byte[] 这样的标准东西没有 hashCode 对吗？）：

使用 TreeSet 代替 HashSet 并在实例化时提供一个 byte[] 比较器：

  Set<byte[]> byteATreeSet = new TreeSet<byte[]>(new Comparator<byte[]>() {
    public int compare(byte[] left, byte[] right) {
    for (int i = 0, j = 0; i < left.length && j < right.length; i++, j++) {
        int a = (left[i] & 0xff);
        int b = (right[j] & 0xff);
        if (a != b) {
            return a - b;
        }
    }
    return left.length - right.length;
   }});

如果你从其他地方得到一个 byte[] HashSet b，那么先将你的变量 a 初始化为 TreeSet，然后使用 a.addAll(b);这样，即使 b 包含重复项，a 也不会。