dev.mongocamp.driver.mongodb.gridfs.GridFSStreamObserver.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongodb-driver_2.12 Show documentation

mongodb-driver

There is a newer version: 2.8.1

package dev.mongocamp.driver.mongodb.gridfs

import java.io.OutputStream
import java.nio.{ Buffer, ByteBuffer }
import java.util.concurrent.atomic.{ AtomicBoolean, AtomicLong }

import com.typesafe.scalalogging.LazyLogging
import org.mongodb.scala.Observer

case class GridFSStreamObserver(outputStream: OutputStream) extends Observer[ByteBuffer] with LazyLogging {
  val completed    = new AtomicBoolean(false)
  val resultLength = new AtomicLong(0)

  override def onNext(buffer: ByteBuffer): Unit = {
    val bytes = new Array[Byte](buffer.remaining())
    resultLength.set(resultLength.get() + bytes.length)
    buffer.get(bytes, 0, bytes.length)
    buffer.asInstanceOf[Buffer].clear()
    outputStream.write(bytes)
  }

  override def onError(e: Throwable): Unit = {
    logger.error(e.getMessage, e)
    outputStream.close()
    resultLength.set(-1)
    completed.set(true)
  }

  override def onComplete(): Unit = {
    outputStream.close()
    completed.set(true)
  }
}