com.avsystem.commons.mongo.BsonWriterOutput.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons-mongo_2.13 Show documentation

AVSystem commons library for Scala

There is a newer version: 2.22.0

package com.avsystem.commons
package mongo

import com.avsystem.commons.serialization.{ListOutput, ObjectOutput}
import org.bson.types.{Decimal128, ObjectId}
import org.bson.{BsonBinary, BsonValue, BsonWriter}

final class BsonWriterOutput(bw: BsonWriter, override val legacyOptionEncoding: Boolean = false)
  extends BsonOutput {

  override def writeNull(): Unit =
    bw.writeNull()

  override def writeString(str: String): Unit =
    bw.writeString(str)

  override def writeBoolean(boolean: Boolean): Unit =
    bw.writeBoolean(boolean)

  override def writeInt(int: Int): Unit =
    bw.writeInt32(int)

  override def writeLong(long: Long): Unit =
    if (long.isValidInt) writeInt(long.toInt)
    else bw.writeInt64(long)

  override def writeTimestamp(millis: Long): Unit =
    bw.writeDateTime(millis)

  override def writeDouble(double: Double): Unit =
    bw.writeDouble(double)

  override def writeBinary(binary: Array[Byte]): Unit =
    bw.writeBinaryData(new BsonBinary(binary))

  override def writeList(): BsonWriterListOutput = {
    bw.writeStartArray()
    new BsonWriterListOutput(bw, legacyOptionEncoding)
  }

  override def writeObject(): BsonWriterObjectOutput = {
    bw.writeStartDocument()
    new BsonWriterObjectOutput(bw, legacyOptionEncoding)
  }

  override def writeObjectId(objectId: ObjectId): Unit =
    bw.writeObjectId(objectId)

  override def writeDecimal128(decimal128: Decimal128): Unit =
    bw.writeDecimal128(decimal128)

  override def writeBsonValue(bsonValue: BsonValue): Unit =
    BsonValueUtils.encode(bw, bsonValue)
}

final class BsonWriterListOutput(bw: BsonWriter, legacyOptionEncoding: Boolean) extends ListOutput {
  override def writeElement(): BsonWriterOutput =
    new BsonWriterOutput(bw, legacyOptionEncoding)

  override def finish(): Unit =
    bw.writeEndArray()
}

final class BsonWriterObjectOutput(bw: BsonWriter, legacyOptionEncoding: Boolean) extends ObjectOutput {
  override def writeField(key: String): BsonWriterOutput = {
    bw.writeName(KeyEscaper.escape(key))
    new BsonWriterOutput(bw, legacyOptionEncoding)
  }

  override def finish(): Unit =
    bw.writeEndDocument()
}