• 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.MonadError
import skunk.~
import skunk.data.Completion
import skunk.net.MessageSocket
import skunk.net.Protocol.{ PreparedCommand, PreparedQuery, CommandPortal, QueryPortal }
import skunk.util.{ Origin, Namer }
import skunk.util.Typer
import natchez.Trace

trait Prepare[F[_]] {
  def apply[A](command: skunk.Command[A], ty: Typer): Resource[F, PreparedCommand[F, A]]
  def apply[A, B](query: skunk.Query[A, B], ty: Typer): Resource[F, PreparedQuery[F, A, B]]
}

object Prepare {

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

      override def apply[A](command: skunk.Command[A], ty: Typer): Resource[F, PreparedCommand[F, A]] =
        for {
          id <- Parse[F].apply(command, ty)
          _  <- Resource.liftF(Describe[F].apply(command, id, ty))
        } yield new PreparedCommand[F, A](id, command) { pc =>
          def bind(args: A, origin: Origin): Resource[F, CommandPortal[F, A]] =
            Bind[F].apply(this, args, origin).map {
              new CommandPortal[F, A](_, pc, args, origin) {
                val execute: F[Completion] =
                  Execute[F].apply(this)
              }
            }
        }

      override def apply[A, B](query: skunk.Query[A, B], ty: Typer): Resource[F, PreparedQuery[F, A, B]] =
        for {
          id <- Parse[F].apply(query, ty)
          rd <- Resource.liftF(Describe[F].apply(query, id, ty))
        } yield new PreparedQuery[F, A, B](id, query, rd) { pq =>
          def bind(args: A, origin: Origin): Resource[F, QueryPortal[F, A, B]] =
            Bind[F].apply(this, args, origin).map {
              new QueryPortal[F, A, B](_, pq, args, origin) {
                def execute(maxRows: Int): F[List[B] ~ Boolean] =
                  Execute[F].apply(this, maxRows, ty)
              }
            }
        }

      }

}