web-dev-qa-db-ja.com

OutputStreamからByteBufferにデータを配置する方法は?

In Java私はOutputStreamからのコンテンツをByteBufferに入れる必要があります(私は自分でそのストリームにデータを入力します)。簡単な方法でそれを行う方法は?

20
Rasto

ByteArrayOutputStreamを作成して書き込み、toByteArray()を使用して内容を_byte[]_として抽出できます。次に、ByteBuffer.wrap(byte [])は出力バイト配列の内容を使用してByteBufferを作成します。

32
DJClayworth

@DJClayworthの答えのより効率的な変形があります。

@sehが正しく認識しているように、ByteArrayOutputStream.toByteArray()はバッキングbyte[]オブジェクトのcopyを返しますが、これは非効率的である可能性があります。ただし、バッキング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());

その結果、すべての出力が書き込まれると、同じバッファーが入力ストリームのベースとして使用されます。

22
mark

上記の回答は問題を解決しますが、NIOに期待するほど効率的なものはありません。 ByteArrayOutputStreamまたはMyByteArrayOutputStreamは、最初にデータをJavaヒープメモリに書き込み、次にそれをByteBufferにコピーします。これはパフォーマンスに大きく影響します。

効率的な実装は、ByteBufferOutputStreamクラスを自分で作成することです。実際、それは非常に簡単です。 write()メソッドを提供するだけです。 ByteBufferInputStream については、このリンクを参照してください。

5
RoboAlex

OutputStreamの代わりにPipedOutputStreamを使用してみてください。次に、PipedInputStreamを接続して、PipedOutputStreamからデータを読み戻すことができます。

1
Marcus Adams

//拡張クラスから保護されたメンバーにアクセスbufcount

class ByteArrayOutputStream2ByteBuffer extends ByteArrayOutputStream {
    public ByteBuffer toByteBuffer() {
        return ByteBuffer.wrap(buf, 0, count);
    }
}
0
Yessy

あなたは自分でこのストリームに書いていると言いますか?もしそうなら、多分あなたはあなた自身のByteBufferOutputStreamを実装して、プラグアンドプレイをすることができます。

基本クラスは次のようになります。

public class ByteBufferOutputStream extends OutputStream {
    //protected WritableByteChannel wbc; //if you need to write directly to a channel
    protected static int bs = 2 * 1024 * 1024; //2MB buffer size, change as needed
    protected ByteBuffer bb = ByteBuffer.allocate(bs);

    public ByteBufferOutputStream(...) {
        //wbc = ... //again for writing to a channel
    }

    @Override
    public void write(int i) throws IOException {
        if (!bb.hasRemaining()) flush();
        byte b = (byte) i;
        bb.put(b);
    }

    @Override
    public void write(byte[] b, int off, int len) throws IOException {
        if (bb.remaining() < len) flush();
        bb.put(b, off, len);
    }

    /* do something with the buffer when it's full (perhaps write to channel?)
    @Override
    public void flush() throws IOException {
        bb.flip();
        wbc.write(bb);
        bb.clear();
    }

    @Override
    public void close() throws IOException {
        flush();
        wbc.close();
    }
    /*
}
0
MeetTitan