com.github.mauricio.async.db.postgresql.encoders.PreparedStatementOpeningEncoder.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2013 Maurício Linhares
 *
 * Maurício Linhares licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

package com.github.mauricio.async.db.postgresql.encoders

import com.github.mauricio.async.db.column.ColumnEncoderRegistry
import com.github.mauricio.async.db.postgresql.messages.backend.ServerMessage
import com.github.mauricio.async.db.postgresql.messages.frontend.{
  ClientMessage,
  PreparedStatementOpeningMessage
}
import com.github.mauricio.async.db.util.{Log, ByteBufferUtils}
import java.nio.charset.Charset
import io.netty.buffer.{Unpooled, ByteBuf}

object PreparedStatementOpeningEncoder {
  val log = Log.get[PreparedStatementOpeningEncoder]
}

class PreparedStatementOpeningEncoder(
  charset: Charset,
  encoder: ColumnEncoderRegistry
) extends Encoder
    with PreparedStatementEncoderHelper {

  import PreparedStatementOpeningEncoder.log

  override def encode(message: ClientMessage): ByteBuf = {

    val m = message.asInstanceOf[PreparedStatementOpeningMessage]

    val statementIdBytes = m.statementId.toString.getBytes(charset)
    val columnCount      = m.valueTypes.size

    val parseBuffer = Unpooled.buffer(1024)

    parseBuffer.writeByte(ServerMessage.Parse)
    parseBuffer.writeInt(0)

    parseBuffer.writeBytes(statementIdBytes)
    parseBuffer.writeByte(0)
    parseBuffer.writeBytes(m.query.getBytes(charset))
    parseBuffer.writeByte(0)

    parseBuffer.writeShort(columnCount)

    if (log.isDebugEnabled) {
      log.debug(s"Opening query (${m.query}) - statement id (${statementIdBytes
          .mkString("-")}) - selected types (${m.valueTypes
          .mkString(", ")}) - values (${m.values.mkString(", ")})")
    }

    for (kind <- m.valueTypes) {
      parseBuffer.writeInt(kind)
    }

    ByteBufferUtils.writeLength(parseBuffer)

    val executeBuffer = writeExecutePortal(
      statementIdBytes,
      m.query,
      m.values,
      encoder,
      charset,
      true
    )

    Unpooled.wrappedBuffer(parseBuffer, executeBuffer)
  }

}