Jak umieścić dane z OutputStream w ByteBuffer?

Question

W Javie muszę umieścić zawartość z OutputStream (I wypełnić dane do tego strumienia samodzielnie) do ByteBuffer. Jak to zrobić w prosty sposób?

Answer 1

Można utworzyć ByteArrayOutputStream i napisać do niego i wyodrębnić zawartość jako byte[] przy użyciu toByteArray(). Następnie ByteBuffer.wrap(byte []) utworzy ByteBuffer z zawartością wyjściowej tablicy bajtów.

Answer 2

Spróbuj użyć PipedOutputStream zamiast OutputStream. Następnie można podłączyć PipedInputStream, aby odczytać dane z powrotem z PipedOutputStream.

Answer 3

Istnieje skuteczniejszy wariant odpowiedzi @ DJClayworth.

Jak @seh prawidłowo zauważył, ByteArrayOutputStream.toByteArray() zwraca kopię obiektu podkład byte[], które mogą być nieskuteczne. Jednak obiekt bazowy oraz obiekt z bajtami są chronionymi elementami klasy ByteArrayOutputStream. Stąd też można stworzyć swój własny wariant ByteArrayOutputStream wystawiając je bezpośrednio:

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

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

    public int getCount() { 
    return count; 
    } 

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

Korzystanie z tej klasy jest proste:

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

W rezultacie, gdy cała moc jest napisany ten sam bufor służy jako podstawa strumienia wejściowego.

Answer 4

Chociaż powyższe odpowiedzi rozwiązują problem, żaden z nich nie jest wydajny, jak można oczekiwać od NIO. ByteArrayOutputStream lub MyByteArrayOutputStream najpierw zapisują dane w pamięci sterty Java, a następnie kopiują ją do ByteBuffer, co znacznie wpływa na wydajność.

Skuteczną implementacją byłoby samodzielne pisanie klasy ByteBufferOutputStream. Właściwie to dość łatwe do zrobienia. Musisz po prostu podać metodę write(). Zobacz ten link dla ByteBufferInputStream.