在ByteBuffer上缺少一些绝对方法

时间:2013-03-14 12:47:59

标签: java

也许我会以错误的方式解决这个问题,但我在ByteBuffer上错过了一些绝对 put方法。

如果你看ByteBuffer,你会发现大多数put方法都有绝对变量和相对变量。

除了:

  • byte数组的部分内容写入ByteBuffer
  • ByteBuffer写入ByteBuffer

..我确实需要那些。

要明确ByteBuffer有方法:

 put(byte[] src, int offset, int length)
 put(ByteBuffer src)

但缺乏:

 put(int index, byte[] src, int offset, int length)
 put(int index, ByteBuffer src)

我有理由不想移动缓冲区的位置指针,因此我只想使用绝对put方法。

知道为什么这些方法被遗漏了吗?

我当然可以在不移动缓冲区位置指针的情况下模仿丢失的方法,但这将涉及在源字节上循环。 Javadoc明确指出,这些方法(可能)比移动更有效,而不是循环并逐个移动字节。我相信Javadoc,因为我的测试表明相同。我需要从实施中尽可能多地挤出速度,因此我当然倾向于利用我可以获得的任何批量方法......如果它们存在的话。

Incedentially ByteBuffer也缺少部分字节数组移动的绝对get方法。但我目前实际上并不需要这样的方法。但又奇怪的是它不存在。

4 个答案:

答案 0 :(得分:4)

获得所需方法的一种方法是让第二个ByteBuffer共享相同的内存,这样您就可以在不改变原始位置的情况下更改其位置。

不幸的是,slice方法也没有采用位置参数;相反,它使用原始缓冲区的当前位置。所以你做不到:

dstBuffer.slice(100).put(srcBuffer);

以下是一些想法,没有特别的顺序,除了它是我想到的顺序:

  • 如果它符合您使用缓冲区的方式,您可以使用slice()准备缓冲区的副本,并在需要将数据放置在独立于原来的位置。

  • 如果要绝对放置的位置始终大于或等于原始缓冲区的位置指针,则可以执行以下操作:

    dstBuffer.slice().position(desiredPosition - dstBuffer.position()).put(srcBuffer);
    

不幸的是,由于切片上的位置不允许为负数,因此无法放在较早的位置。 编辑:没关系,我忘记了duplicate方法。请参阅@ BorisBrodski的好答案。

  • 如果您不使用直接字节缓冲区,System.arraycopy既简单又快捷:

    System.arraycopy(
        srcBuffer.array(), srcBuffer.arrayOffset() + srcBuffer.position(),
        dstBuffer.array(), dstBuffer.arrayOffset() + desiredPosition,
        srcBuffer.remaining()
    );
    
  • 如果不需要并发访问,则可以在需要执行绝对放置时临时更改缓冲区的位置,然后将其放回。如果您需要并发访问但线程争用率较低,则可以同步对缓冲区的所有访问(可能很明显,但为了完整性而包括在内):

    synchronize (lock) {
        int originalPosition = dstBuffer.position();
        dstBuffer.position(desiredPosition);
        dstBuffer.put(srcBuffer);
        dstBuffer.position(originalPosition);
    }
    
  • 如果其他想法都不适合您,您可以破解缓冲区。这很麻烦,但这是一个例子:

    private static final sun.misc.Unsafe UNSAFE;
    static {
        Object result = null;
        try {
            Class<?> klass = Class.forName("sun.misc.Unsafe");
            for (Field field : klass.getDeclaredFields()) {
                if (field.getType() == klass &&
                    (field.getModifiers() & (Modifier.FINAL | Modifier.STATIC)) ==
                        (Modifier.FINAL | Modifier.STATIC)) {
                    field.setAccessible(true);
                    result = field.get(null);
                    break;
                }
            }
        } catch (Throwable t) {}
        UNSAFE = result == null ? null : (sun.misc.Unsafe)result;
    }
    
    private static final Field ADDRESS_FIELD;
    static {
        Field f;
        try {
            f = Buffer.class.getDeclaredField("address");
            f.setAccessible(true);
        } catch (NoSuchFieldException | SecurityException e) {
            f = null;
        }
        ADDRESS_FIELD = f;
    }
    
    
    public static void absolutePut(ByteBuffer dstBuffer, int dstPosition, ByteBuffer srcBuffer) {
        if (!srcBuffer.isDirect()) {
            absolutePut(dstBuffer, dstPosition,
                srcBuffer.array(), srcBuffer.arrayOffset() + srcBuffer.position(),
                srcBuffer.remaining());
            return;
        }
    
        if (UNSAFE != null && ADDRESS_FIELD != null && dstBuffer.isDirect()) {
            try {
                long dstAddress = (long)ADDRESS_FIELD.get(dstBuffer) + dstPosition;
                long srcAddress = (long)ADDRESS_FIELD.get(srcBuffer) + srcBuffer.position();
                UNSAFE.copyMemory(srcAddress, dstAddress, srcBuffer.remaining());
            } catch (IllegalAccessException e) {
                throw new RuntimeException(e);
            }
        } else {
            // fallback to basic loop
            for (int i = srcBuffer.position(); i < srcBuffer.limit(); i++) {
                dstBuffer.put(dstPosition + i, srcBuffer.get(i));
            }
        }
    }
    
    public static void absolutePut(ByteBuffer dstBuffer, int dstPosition, byte[] src, int srcOffset, int length) {
        if (UNSAFE != null && ADDRESS_FIELD != null && dstBuffer.isDirect()) {
            try {
                long dstAddress = (long)ADDRESS_FIELD.get(dstBuffer) + dstPosition;
                UNSAFE.copyMemory(
                    src, UNSAFE.arrayBaseOffset(byte[].class) + srcOffset,
                    null, dstAddress,
                    length);
            } catch (IllegalAccessException e) {
                throw new RuntimeException(e);
            }
        } else {
            // fallback to System.arraycopy
            System.arraycopy(
                src, srcOffset,
                dstBuffer.array(), dstBuffer.arrayOffset() + dstPosition,
                length);
        }
    }
    

我已经使用直接和非直接缓冲区的混合物对此代码进行了一些最小的测试,并且 似乎 没问题。如果反射技术失败(例如,因为您在applet安全沙箱中或Java实现不兼容),它可以回退到更简洁的方法。

答案 1 :(得分:3)

在我看来,那些遗漏的方法,因为没有其他这样的方法存在。有许多绝对方法采用包含“byte”的原语,但没有绝对方法采用原语的 array

解决方案可能是:

((ByteBuffer)buffer.duplicate().position(index)).put(src, offset, length);
((ByteBuffer)buffer.duplicate().position(index)).put(otherByteBuffer);

答案 2 :(得分:2)

static void put(ByteBuffer buffer, int index, ByteBuffer src)
{
    int p0 = buffer.position();
    buffer.position(index);
    buffer.put(src);
    buffer.position(p0);
}

答案 3 :(得分:0)

以下是我用于绝对批量get()的内容:

// Perform a bulk get() that doesn't modify the buffer
public void get(ByteBuffer buf, int position, byte[] dest, int off, int len) {
    if (buf.hasArray())
        System.arraycopy(buf.array(), buf.arrayOffset() + position, dest, off, len);
    else if (len < 256) {        // 256 is a wild guess TODO: performance testing
        while (len-- > 0)
            dest[off++] = buf.get(position++);
    } else
        ((ByteBuffer)buf.duplicate().position(position)).get(dest, off, len);
}

类似的方法适用于put()