bloop.logging.TeeOutputStream.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bloop-shared_2.12 Show documentation

bloop-shared

The newest version!

package bloop.logging

import java.io.PrintStream
import java.io.ByteArrayOutputStream
import scala.collection.concurrent.TrieMap

class TeeOutputStream(targetPS: PrintStream) extends PrintStream(targetPS) {

  private val allStreams = TrieMap.empty[Int, ByteArrayOutputStream];

  override def write(b: Int): Unit = {
    allStreams.values.foreach(
      _.write(b)
    )
    targetPS.write(b)
  }

  override def write(buf: Array[Byte], off: Int, len: Int): Unit = {
    allStreams.values.foreach(
      _.write(buf, off, len)
    )
    targetPS.write(buf, off, len)
  }

  override def flush(): Unit = {
    targetPS.flush()
  }

  def addListener(baos: ByteArrayOutputStream): Unit = {
    allStreams += baos.hashCode -> baos
  }

  def removeListener(baos: ByteArrayOutputStream): Unit = {
    allStreams -= baos.hashCode
  }
}