doobie.postgres.hi.largeobjectmanager.scala Maven / Gradle / Ivy

Go to download
// Copyright (c) 2013-2020 Rob Norris and Contributors
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package doobie.postgres.hi

import cats.syntax.all.*
import doobie.postgres.implicits.*
import java.io.{File, OutputStream, InputStream}
import doobie.postgres.free.{largeobjectmanager as IPFLOM, largeobject as IPFLO}
import doobie.postgres.hi.{largeobject as IPHLO}

object largeobjectmanager {

  val createLO: LargeObjectManagerIO[Long] =
    IPFLOM.createLO

  def createLO(a: Int): LargeObjectManagerIO[Long] =
    IPFLOM.createLO(a)

  def delete(a: Long): LargeObjectManagerIO[Unit] =
    IPFLOM.delete(a)

  def open[A](a: Long, b: Int)(k: LargeObjectIO[A]): LargeObjectManagerIO[A] =
    IPFLOM.open(a, b) >>= (IPFLOM.embed(_, k <* IPFLO.close))

  def open[A](a: Long)(k: LargeObjectIO[A]): LargeObjectManagerIO[A] =
    IPFLOM.open(a) >>= (IPFLOM.embed(_, k <* IPFLO.close))

  def unlink(a: Long): LargeObjectManagerIO[Unit] =
    IPFLOM.unlink(a)

  def createLOFromFile(blockSize: Int, file: File): LargeObjectManagerIO[Long] =
    createLO >>= { oid => open(oid)(IPHLO.copyFromFile(blockSize, file)).as(oid) }

  def createFileFromLO(blockSize: Int, oid: Long, file: File): LargeObjectManagerIO[Unit] =
    open(oid)(IPHLO.copyToFile(blockSize, file))

  def createLOFromStream(blockSize: Int, is: InputStream): LargeObjectManagerIO[Long] =
    createLO >>= { oid =>
      open(oid)(IPHLO.copyFromStream(blockSize, is)).as(oid)
    }

  def createStreamFromLO(blockSize: Int, oid: Long, os: OutputStream): LargeObjectManagerIO[Unit] =
    open(oid)(IPHLO.copyToStream(blockSize, os))
}