• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• skunk-master
  • .github
    • workflows
      • build.yml
      • release.yml
  • LICENSE
  • project
    • build.properties
    • project
      • plugins.sbt
    • plugins.sbt
  • .gitattributes
  • CHANGELOG.md
  • README.md
  • world
    • server.key
    • README.md
    • server.crt
    • Dockerfile
  • build.sbt
  • .gitignore
  • modules
    • circe
      • src
        • main
          • scala
            • JsonCodecs.scala
    • example
      • src
        • main
          • scala
            • Minimal1.scala
            • NatchezHttp4sModule.scala
            • Channel.scala
            • Error.scala
            • Math1.scala
            • Values.scala
            • Main.scala
            • Join.scala
            • Http4s.scala
            • Math2.scala
            • Minimal3.scala
            • Minimal2.scala
            • Transaction.scala
    • refined
      • src
        • main
          • scala
            • codec
              • NumericCodecs.scala
    • tests
      • src
        • main
          • scala
            • ffstest
              • FFramework.scala
        • test
          • scala
            • CommandTest.scala
            • StartupTest.scala
            • codec
              • BinaryCodecTest.scala
              • EnumCodecTest.scala
              • NumericCodecTest.scala
              • JsonCodecTest.scala
              • TextCodecTest.scala
              • BooleanCodecTest.scala
              • UuidCodecTest.scala
              • TemporalCodecTest.scala
              • CodecTest.scala
            • ExtendedQueryErrorTest.scala
            • SkunkTest.scala
            • SslTest.scala
            • TransactionStatusTest.scala
            • ListTest.scala
            • ErrorResponseTest.scala
            • PoolTest.scala
            • QuickQueryErrorTest.scala
    • core
      • src
        • main
          • scala-2.12
            • utll
              • package.scala
          • scala
            • Decoder.scala
            • PreparedCommand.scala
            • Statement.scala
            • Channel.scala
            • codec
              • BooleanCodec.scala
              • UuidCodec.scala
              • BinaryCodecs.scala
              • AllCodecs.scala
              • EnumCodec.scala
              • TemporalCodecs.scala
              • NumericCodecs.scala
              • TextCodecs.scala
            • Cursor.scala
            • Command.scala
            • Session.scala
            • package.scala
            • net
              • SSLNegotiation.scala
              • package.scala
              • Protocol.scala
              • AbstractMessageSocket.scala
              • message
                • AuthenticationRequest.scala
                • ParseComplete.scala
                • PortalSuspended.scala
                • AuthenticationKerberosV5.scala
                • Terminate.scala
                • ErrorResponse.scala
                • AuthenticationSSPI.scala
                • AuthenticationSASLContinue.scala
                • package.scala
                • NoticeResponse.scala
                • FrontendMessage.scala
                • AuthenticationSASL.scala
                • Flush.scala
                • Execute.scala
                • AuthenticationCleartextPassword.scala
                • RowDescription.scala
                • Sync.scala
                • StartupMessage.scala
                • AuthenticationMD5Password.scala
                • AuthenticationSASLFinal.scala
                • BackendMessage.scala
                • CommandComplete.scala
                • ParameterDescription.scala
                • Describe.scala
                • Close.scala
                • ParameterStatus.scala
                • NotificationResponse.scala
                • Parse.scala
                • RowData.scala
                • Bind.scala
                • BindComplete.scala
                • UnknownMessage.scala
                • CloseComplete.scala
                • AuthenticationSCMCredential.scala
                • Query.scala
                • AuthenticationOk.scala
                • NoData.scala
                • PasswordMessage.scala
                • ReadyForQuery.scala
                • AuthenticationGSS.scala
                • BackendKeyData.scala
                • AuthenticationGSSContinue.scala
              • MessageSocket.scala
              • BufferedMessageSocket.scala
              • protocol
                • Unroll.scala
                • Exchange.scala
                • package.scala
                • Execute.scala
                • Startup.scala
                • Describe.scala
                • Close.scala
                • Parse.scala
                • Bind.scala
                • Query.scala
                • Prepare.scala
              • BitVectorSocket.scala
            • SSL.scala
            • PreparedQuery.scala
            • Encoder.scala
            • util
              • Text.scala
              • Twiddler.scala
              • Pretty.scala
              • Recycler.scala
              • Namer.scala
              • Pool.scala
              • Typer.scala
            • Void.scala
            • syntax
              • ListOps.scala
              • IdOps.scala
              • AllOps.scala
              • StringContextOps.scala
            • exception
              • NoDataException.scala
              • UnknownTypeException.scala
              • StartupException.scala
              • UnknownOidException.scala
              • package.scala
              • DecodeException.scala
              • SkunkException.scala
              • ProtocolError.scala
              • UnexpectedColumnsException.scala
              • ColumnAlignmentException.scala
              • PostgresErrorException.scala
            • data
              • TypedRowDescription.scala
              • Completion.scala
              • Identifier.scala
              • Notification.scala
              • Type.scala
              • TransactionStatus.scala
            • SqlState.scala
            • Fragment.scala
            • Query.scala
            • Codec.scala
            • Transaction.scala
          • scala-2.13
            • utll
              • package.scala
    • docs
      • src
        • main
          • paradox
            • tutorial
              • Setup.md
              • Channels.md
              • Transactions.md
              • Command.md
              • Tracing.md
              • Index.md
              • Query.md
            • index.md
            • reference
              • TwiddleLists.md
              • SchemaTypes.md
              • Fragments.md
              • Sessions.md
              • Concurrency.md
              • Index.md
              • Identifiers.md
              • Encoders.md
    • macros
      • src
        • main
          • scala
            • util
              • Origin.scala
  • docker-compose.yml

// Copyright (c) 2018-2020 by Rob Norris
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package skunk.net.protocol

import cats.effect.Resource
import cats.implicits._
import cats.MonadError
import skunk.exception.PostgresErrorException
import skunk.net.message.{ Parse => ParseMessage, Close => _, _ }
import skunk.net.MessageSocket
import skunk.net.Protocol.StatementId
import skunk.Statement
import skunk.util.Namer
import skunk.util.Typer
import skunk.exception.UnknownTypeException
import natchez.Trace

trait Parse[F[_]] {
  def apply[A](statement: Statement[A], ty: Typer): Resource[F, StatementId]
}

object Parse {

  def apply[F[_]: MonadError[?[_], Throwable]: Exchange: MessageSocket: Namer: Trace]: Parse[F] =
    new Parse[F] {

      override def apply[A](statement: Statement[A], ty: Typer): Resource[F, StatementId] =
        statement.encoder.oids(ty) match {

          case Right(os) =>
            Resource.make {
              exchange("parse") {
                for {
                  id <- nextName("statement").map(StatementId)
                  _  <- Trace[F].put(
                          "statement-name"            -> id.value,
                          "statement-sql"             -> statement.sql,
                          "statement-parameter-types" -> os.map(n => ty.typeForOid(n, -1).getOrElse(n)).mkString("[", ", ", "]")
                        )
                  _  <- send(ParseMessage(id.value, statement.sql, os))
                  _  <- send(Flush)
                  _  <- flatExpect {
                          case ParseComplete       => ().pure[F]
                          case ErrorResponse(info) => syncAndFail(statement, info)
                        }
                } yield id
              }
            } { Close[F].apply }

          case Left(err) =>
            Resource.liftF(UnknownTypeException(statement, err).raiseError[F, StatementId])

        }

      def syncAndFail(statement: Statement[_], info: Map[Char, String]): F[Unit] =
        for {
          hi <- history(Int.MaxValue)
          _  <- send(Sync)
          _  <- expect { case ReadyForQuery(_) => }
          a  <- new PostgresErrorException(
                  sql       = statement.sql,
                  sqlOrigin = Some(statement.origin),
                  info      = info,
                  history   = hi,
                ).raiseError[F, Unit]
        } yield a

    }

}