用于“使用/尝试使用资源”的简单 Scala 模式（自动资源管理）

Question

C# 有 using 和 IDisposable 接口。 Java 7+ 具有与try 和AutoCloseable 接口相同的功能。 Scala 允许您选择自己的实现来解决这个问题。

scala-arm 似乎是流行的选择，由 Typesafe 的一名员工维护。但是，对于这样一个简单的行为，它似乎非常复杂。澄清一下，使用说明很简单，但了解所有代码在内部是如何工作的却相当复杂。

我刚刚写了以下超级简单的ARM解决方案：

object SimpleARM {
  def apply[T, Q](c: T {def close(): Unit})(f: (T) => Q): Q = {
    try {
      f(c)
    } finally {
      c.close()
    }
  }
}

• 像单臂这样的东西有什么好处吗？似乎所有额外的复杂性都应该带来额外的好处。
• 通常，与使用自定义代码相比，最好使用其他人支持的公共、开源库来实现通用行为。
• 谁能推荐任何改进？
• 这种简单的方法有什么限制吗？

Answer 1

只要您不需要使用多个资源，所有资源都需要管理，您使用单一简单贷款模式的方法就可以正常工作。这在 scala-arm monadic 方法中是允许的。

import resource.managed

managed(openResA).and(managed(openResB)) acquireFor { (a, b) => ??? }

val res = for {
  a <- managed(openResA)
  b <- managed(openResB)
  c <- managed(openResC)
} yield (a, b, c)

res acquireAndGet {
  case (a, b, c) => ???
}

在 scala-arm 中要了解的主要功能是 resource.managed 和 .acquired{For,AndGet}，顺便说一句，并不是很复杂。

Answer 2

这里是我较新的简单，一目了然，Scala ARM。这完全支持我能想到的每个用例，包括多个资源和收益值。这使用了一个非常简单的理解用法语法：

class AutoCloseableWrapper[A <: AutoCloseable](protected val c: A) {
  def map[B](f: (A) => B): B = {
    try {
      f(c)
    } finally {
      c.close()
    }
  }

  def foreach(f: (A) => Unit): Unit = map(f)

  // Not a proper flatMap.
  def flatMap[B](f: (A) => B): B = map(f)

  // Hack :)    
  def withFilter(f: (A) => Boolean) = this
}

object Arm {
  def apply[A <: AutoCloseable](c: A) = new AutoCloseableWrapper(c)
}

这里是演示使用：

class DemoCloseable(val s: String) extends AutoCloseable {
  var closed = false
  println(s"DemoCloseable create ${s}")

  override def close(): Unit = {
    println(s"DemoCloseable close ${s} previously closed=${closed}")
    closed = true
  }
}

object DemoCloseable {
  def unapply(dc: DemoCloseable): Option[(String)] = Some(dc.s)
}

object Demo {
  def main(args: Array[String]): Unit = {
    for (v <- Arm(new DemoCloseable("abc"))) {
      println(s"Using closeable ${v.s}")
    }

    for (a <- Arm(new DemoCloseable("a123"));
         b <- Arm(new DemoCloseable("b123"));
         c <- Arm(new DemoCloseable("c123"))) {
      println(s"Using multiple resources for comprehension. a.s=${a.s}. b.s=${b.s}. c.s=${c.s}")
    }

    val yieldInt = for (v <- Arm(new DemoCloseable("abc"))) yield 123
    println(s"yieldInt = $yieldInt")

    val yieldString = for (DemoCloseable(s) <- Arm(new DemoCloseable("abc")); c <- s) yield c
    println(s"yieldString = $yieldString")

    println("done")
  }
}

Answer 3

这是我使用的代码：

def use[A <: { def close(): Unit }, B](resource: A)(code: A ⇒ B): B =
    try
        code(resource)
    finally
        resource.close()

与Java try-with-resources 不同，资源不需要实现AutoCloseable。只需要一个close() 方法。 它只支持一种资源。

这是一个使用 InputStream 的示例：

val path = Paths get "/etc/myfile"
use(Files.newInputStream(path)) { inputStream ⇒
    val firstByte = inputStream.read()
    ....
}

Answer 4

http://illegalexception.schlichtherle.de/2012/07/19/try-with-resources-for-scala/

另一个实现，从“遵循 Java 规范”的角度来看可能更干净，但也不支持多种资源

Answer 5

这个非常适合我：

  implicit class ManagedCloseable[C <: AutoCloseable](resource: C) {
    def apply[T](block: (C) => T): T = {
    try {
      block(resource)
    } finally {
      resource.close()
    }
  }

例如在这个 Apache Cassandra 客户端代码中使用它：

val metadata = Cluster.builder().addContactPoint("vader").withPort(1234).build() { cluster =>
  cluster.getMetadata
}

甚至更短：

val metadata = Cluster.builder().addContactPoint("sedev01").withPort(9999).build()(_.getMetadata)

Answer 6

我可以对您建议的方法提出改进建议，即：

  def autoClose[A <: AutoCloseable, B](resource: A)(code: A ⇒ B): B = {
    try
      code(resource)
    finally
      resource.close()
  }

正在使用：

  def autoClose[A <: AutoCloseable, B](resource: A)(code: A ⇒ B): Try[B] = {
    val tryResult = Try {code(resource)}
    resource.close()
    tryResult
  }

恕我直言，拥有 Try[B] 的 tryResult 将允许您稍后更轻松地控制流程。

Answer 7

Choppy 的 Lazy TryClose monad 可能是您正在寻找的（披露：我是作者）。它与Scala的Try很相似，只是自动关闭资源。

val ds = new JdbcDataSource()
val output = for {
  conn  <- TryClose(ds.getConnection())
  ps    <- TryClose(conn.prepareStatement("select * from MyTable"))
  rs    <- TryClose.wrap(ps.executeQuery())
} yield wrap(extractResult(rs))

// Note that Nothing will actually be done until 'resolve' is called
output.resolve match {
    case Success(result) => // Do something
    case Failure(e) =>      // Handle Stuff
}

查看这里了解更多信息：https://github.com/choppythelumberjack/tryclose

Answer 8

这就是我的做法：

  def tryFinally[A, B](acquire: Try[A])(use: A => B)(release: A => Unit): Try[B] =
    for {
      resource <- acquire
      r = Try(use(resource)).fold(
        e => { release(resource); throw e },
        b => { release(resource); b }
      )
    } yield r

你可以在上面调用.get，让它返回B。

Answer 9

这是在 Scala 中执行此操作的一种方法

trait SourceUtils {

  /**
    * Opens a `resource`, passes it to the given function `f`, then
    * closes the resource, returning the value returned from `f`.
    * 
    * Example usage:
    * {{{
    * val myString: String =
    *   using(scala.io.Source.fromFile("file.name")) { source =>
    *     source.getLines.mkString
    *   }
    * }}}
    * 
    * @param resource closeable resource to use, then close
    * @param f function which maps the resource to some return value
    * @tparam T type of the resource to be opened / closed
    * @tparam U type of the return value
    * @return the result of the function `f`
    */
  def using[T <: AutoCloseable, U](resource: => T)(f: T => U): U = {
    scala.util.Try(resource).fold(throw _, source => {
      val result = f(source)
      source.close()
      result
    })
  }
}