将输入流连接到输出流

Question

java9中的更新： https ：//docs.oracle.com/javase/9​​/docs/api/java/io/InputStream.html#transferTo-java.io.OutputStream-

我看到了一些相似的，但不是我需要的线程。

我有一个服务器，它基本上从客户端，客户端A接收输入，并逐字节转发到另一个客户端，客户端B.

我想将客户端A的输入流与客户端B的输出流连接起来。这可能吗？ 有什么方法可以做到这一点？

此外，这些客户端正在发送彼此有时间敏感的消息，因此缓冲不会。 我不想要500的缓冲区和客户端发送499个字节，然后我的服务器在转发500个字节时保持关闭，因为它没有收到填充缓冲区的最后一个字节。

现在，我正在解析每条消息以找到它的长度，然后读取长度字节，然后转发它们。 我想（并测试）这比读取一个字节并反复转发一个字节更好，因为这将非常慢。 我也不想使用缓冲区或计时器，因为我在上一段中说过 - 我不希望消息等待很长时间才能通过，因为缓冲区未满。

这样做的好方法是什么？

Answer 1

仅仅因为你使用缓冲区并不意味着流必须填充该缓冲区。 换句话说，这应该没问题：

public static void copyStream(InputStream input, OutputStream output)
    throws IOException
{
    byte[] buffer = new byte[1024]; // Adjust if you want
    int bytesRead;
    while ((bytesRead = input.read(buffer)) != -1)
    {
        output.write(buffer, 0, bytesRead);
    }
}

这应该做工精细-基本read调用将阻塞有一些数据可用，直到，但它不会等到这一切都可以填充缓冲区。 （我想它可以，我相信FileInputStream通常会填充缓冲区，但连接到套接字的流更有可能立即为您提供数据。）

我认为至少首先尝试这个简单的解决方案是值得的。

Answer 2

如何使用

void feedInputToOutput(InputStream in, OutputStream out) {
   IOUtils.copy(in, out);
}

并完成它？

来自jakarta apache commons i / o库已被大量项目使用，所以你可能已经在你的类路径中已经有了jar。

Answer 3

为了完整， 番石榴也有一个方便的实用程序

ByteStreams.copy(input, output);

Answer 4

JDK 9为此功能添加了InputStream#transferTo(OutputStream out) 。

Answer 5

您可以使用循环缓冲区：

码

// buffer all data in a circular buffer of infinite size
CircularByteBuffer cbb = new CircularByteBuffer(CircularByteBuffer.INFINITE_SIZE);
class1.putDataOnOutputStream(cbb.getOutputStream());
class2.processDataFromInputStream(cbb.getInputStream());

Maven依赖

<dependency>
    <groupId>org.ostermiller</groupId>
    <artifactId>utils</artifactId>
    <version>1.07.00</version>
</dependency>

模式细节

http://ostermiller.org/utils/CircularBuffer.html

Answer 6

实现它的异步方式。

void inputStreamToOutputStream(final InputStream inputStream, final OutputStream out) {
    Thread t = new Thread(new Runnable() {

        public void run() {
            try {
                int d;
                while ((d = inputStream.read()) != -1) {
                    out.write(d);
                }
            } catch (IOException ex) {
                //TODO make a callback on exception.
            }
        }
    });
    t.setDaemon(true);
    t.start();
}

Answer 7

BUFFER_SIZE是要读入的chucks的大小。应该> 1kb且<10MB。

private static final int BUFFER_SIZE = 2 * 1024 * 1024;
private void copy(InputStream input, OutputStream output) throws IOException {
    try {
        byte[] buffer = new byte[BUFFER_SIZE];
        int bytesRead = input.read(buffer);
        while (bytesRead != -1) {
            output.write(buffer, 0, bytesRead);
            bytesRead = input.read(buffer);
        }
    //If needed, close streams.
    } finally {
        input.close();
        output.close();
    }
}

Answer 8

使用org.apache.commons.io.IOUtils

InputStream inStream = new ...
OutputStream outStream = new ...
IOUtils.copy(inStream, outStream);

或copyLarge的大小> 2GB

Answer 9

这是一个干净而快速的Scala版本（没有stackoverflow）：

  import scala.annotation.tailrec
  import java.io._

  implicit class InputStreamOps(in: InputStream) {
    def >(out: OutputStream): Unit = pipeTo(out)

    def pipeTo(out: OutputStream, bufferSize: Int = 1<<10): Unit = pipeTo(out, Array.ofDim[Byte](bufferSize))

    @tailrec final def pipeTo(out: OutputStream, buffer: Array[Byte]): Unit = in.read(buffer) match {
      case n if n > 0 =>
        out.write(buffer, 0, n)
        pipeTo(out, buffer)
      case _ =>
        in.close()
        out.close()
    }
  }

这使得能够使用>符号，例如输入inputstream > outputstream并且还传递自定义缓冲器/大小。

Answer 10

如果您正在使用它，这是一个用Scala编写的函数，显示如何仅使用val（而不是vars）将输入流复制到输出流。

def copyInputToOutputFunctional(inputStream: InputStream, outputStream: OutputStream,bufferSize: Int) {
  val buffer = new Array[Byte](bufferSize);
  def recurse() {
    val len = inputStream.read(buffer);
    if (len > 0) {
      outputStream.write(buffer.take(len));
      recurse();
    }
  }
  recurse();
}

请注意，建议不要在没有可用内存的Java应用程序中使用，因为使用递归函数可以轻松获得堆栈溢出异常错误