com.twitter.finagle.postgres.messages.Packet.scala Maven / Gradle / Ivy

Go to download
package com.twitter.finagle.postgres.messages

import com.twitter.finagle.postgres.values.Charsets

import org.jboss.netty.buffer.{ChannelBuffer, ChannelBuffers}

object Packet {
  val INT_SIZE = 4
}

/*
 * Representation of a "packet" sent to Postgres.
 *
 * Converts content into byte format expected by Postgres.
 */
case class Packet(code: Option[Char], length: Int, content: ChannelBuffer, inSslNegotation: Boolean = false) {
  def encode(): ChannelBuffer = {
    val result = ChannelBuffers.dynamicBuffer()
    code.map { c =>
      result.writeByte(c)
    }

    result.writeInt(length + Packet.INT_SIZE)
    result.writeBytes(content)

    result
  }
}

/*
 * Helper class for creating packets from scala types.
 */
class PacketBuilder(val code: Option[Char]) {
  private val underlying = ChannelBuffers.dynamicBuffer()

  def writeByte(byte: Byte) = {
    underlying.writeByte(byte)
    this
  }

  def writeBytes(bytes: Array[Byte]) = {
	  underlying.writeBytes(bytes)
	  this
  }

  def writeBuf(bytes: ChannelBuffer) = {
	  underlying.writeBytes(bytes)
	  this
  }

  def writeChar(char: Char) = {
    underlying.writeByte(char)
    this
  }
  def writeInt(int: Int) = {
    underlying.writeInt(int)
    this
  }

  def writeShort(short: Short) = {
    underlying.writeShort(short)
    this
  }

  def writeCString(str: String) = {
    underlying.writeBytes(str.getBytes(Charsets.Utf8))
    underlying.writeByte(0)
    this
  }

  def toPacket = new Packet(code, underlying.writerIndex(), underlying)
}

object PacketBuilder {
  def apply(): PacketBuilder = new PacketBuilder(None)

  def apply(code: Char): PacketBuilder = new PacketBuilder(Some(code))
}