Как поместить данные из OutputStream в ByteBuffer?

Question

В Java мне нужно поместить содержимое из OutputStream (я сам заполняю данные в этот поток) в ByteBuffer. Как это сделать простым способом?

Answer 1

Вы можете создать ByteArrayOutputStream и написать ему и извлечь содержимое в виде byte[], используя toByteArray(). Затем ByteBuffer.wrap(byte []) создаст ByteBuffer с содержимым выходного массива байтов.

Answer 2

Попробуйте использовать PipedOutputStream вместо OutputStream. Затем вы можете подключить PipedInputStream для чтения данных из PipedOutputStream.

Answer 3

Существует более эффективный вариант ответа @ DJClayworth.

Как @seh правильно заметили, ByteArrayOutputStream.toByteArray() возвращает копию объекта подложки byte[], который может быть неэффективным. Однако объект поддержки byte[], а также количество байтов являются защищенными членами класса ByteArrayOutputStream. Таким образом, вы можете создать свой собственный вариант ByteArrayOutputStream подвергая их непосредственно:

public class MyByteArrayOutputStream extends ByteArrayOutputStream { 
    public MyByteArrayOutputStream() { 
    } 

    public MyByteArrayOutputStream(int size) { 
    super(size); 
    } 

    public int getCount() { 
    return count; 
    } 

    public byte[] getBuf() { 
    return buf; 
    } 
} 

Используя этот класс легко:

MyByteArrayOutputStream out = new MyByteArrayOutputStream(); 
fillTheOutputStream(out); 
return new ByteArrayInputStream(out.getBuf(), 0, out.getCount()); 

В результате, когда все выходные данные записываются тот же буфер используется как основу входного потока.

Answer 4

Несмотря на то, что вышеупомянутые ответы решают вашу проблему, ни одна из них не эффективна, как вы ожидаете от NIO. ByteArrayOutputStream или MyByteArrayOutputStream сначала записывают данные в память кучи Java, а затем копируют их в ByteBuffer, что существенно влияет на производительность.

Эффективная реализация будет писать класс ByteBufferOutputStream самостоятельно. На самом деле это довольно легко сделать. Вы должны просто предоставить метод write(). См. Эту ссылку для ByteBufferInputStream.