“使用”功能

问题描述 投票:28回答:9

我已将'使用'功能定义如下:

def using[A, B <: {def close(): Unit}] (closeable: B) (f: B => A): A =
  try { f(closeable) } finally { closeable.close() }

我可以这样使用它:

using(new PrintWriter("sample.txt")){ out =>
  out.println("hellow world!")
}

现在我很好奇如何定义'使用'函数来获取任意数量的参数,并且能够单独访问它们:

using(new BufferedReader(new FileReader("in.txt")), new PrintWriter("out.txt")){ (in, out) =>
  out.println(in.readLIne)
}
scala using tuples
9个回答
11
投票

有人已经这样做了 - 它被称为Scala ARM

从自述文件:

import resource._
for(input <- managed(new FileInputStream("test.txt")) {
  // Code that uses the input as a FileInputStream
}

6
投票

我一直在考虑这个问题,我想可能还有另外一种方法可以解决这个问题。这是我对支持“任意数量”参数的看法(受元组提供的限制):

object UsingTest {

  type Closeable = {def close():Unit }

  final class CloseAfter[A<:Product](val x: A) {
    def closeAfter[B](block: A=>B): B = {
      try {
        block(x);
      } finally {
        for (i <- 0 until x.productArity) {
          x.productElement(i) match { 
            case c:Closeable => println("closing " + c); c.close()
            case _ => 
          }
        }
      }
    }
  }

  implicit def any2CloseAfter[A<:Product](x: A): CloseAfter[A] = 
    new CloseAfter(x)

  def main(args:Array[String]): Unit = {
    import java.io._

    (new BufferedReader(new FileReader("in.txt")), 
     new PrintWriter("out.txt"),
     new PrintWriter("sample.txt")) closeAfter {case (in, out, other) => 
      out.println(in.readLine) 
      other.println("hello world!")
    }
  }
}

我想我正在重用22库元/产品类已经在库中编写的事实......我不认为这种语法比使用嵌套的using(没有双关语)更清晰,但这是一个有趣的难题。

编辑:按照反义词的建议用case (in, out, other)替换val赋值。


2
投票

遗憾的是,标准Scala中不支持任意类型的任意长度参数列表。

您可以通过几种语言更改来执行此类操作(允许将变量参数列表作为HLists传递;请参阅here了解所需内容的大约1/3)。

现在,最好的办法就是做Tuple和Function所做的事情:根据需要使用N实现尽可能多的N.

当然,两个很容易:

def using2[A, B <: {def close(): Unit}, C <: { def close(): Unit}](closeB: B, closeC: C)(f: (B,C) => A): A = {
  try { f(closeB,closeC) } finally { closeB.close(); closeC.close() }
}

如果你需要更多,那么编写一些可以生成源代码的东西可能是值得的。


2
投票

下面是一个示例,它允许您将scala用于理解作为任何java.io.Closeable项的自动资源管理块,但是可以使用close方法轻松扩展它以适用于任何对象。

这种用法似乎非常接近using语句,并允许您轻松地在一个块中定义尽可能多的资源。

object ResourceTest{
  import CloseableResource._
  import java.io._

  def test(){
    for( input <- new BufferedReader(new FileReader("/tmp/input.txt")); output <- new FileWriter("/tmp/output.txt") ){
      output.write(input.readLine)
    }
  }
}

class CloseableResource[T](resource: =>T,onClose: T=>Unit){
  def foreach(f: T=>Unit){
    val r = resource
    try{
      f(r)
    }
    finally{
      try{
        onClose(r)
      }
      catch{
        case e =>
          println("error closing resource")
          e.printStackTrace
      }
    }
  }
}

object CloseableResource{
  implicit def javaCloseableToCloseableResource[T <: java.io.Closeable](resource:T):CloseableResource[T] = new CloseableResource[T](resource,{_.close})
}

1
投票

这个解决方案没有你想要的语法,但我认为它足够接近:)

def using[A <: {def close(): Unit}, B](resources: List[A])(f: List[A] => B): B =
    try f(resources) finally resources.foreach(_.close())

using(List(new BufferedReader(new FileReader("in.txt")), new PrintWriter("out.txt"))) {
  case List(in: BufferedReader, out: PrintWriter) => out.println(in.readLine())
}

当然,缺点是你必须在使用块中输入类型BufferedReaderPrintWrter。您可以添加一些魔法,以便在使用时使用List(in, out)作为A类,只需要multiple ORed type bounds

通过定义一些非常hacky和危险的隐式转换,你可以绕过必须键入List(和另一种方法来解决指定资源的类型),但我没有记录细节,因为它太危险IMO。


1
投票

最好将清理算法从程序路径中分离出来。

此解决方案允许您在范围中累积closeable。 范围清除将在块执行后发生,或者可以分离范围。然后可以在以后完成范围的清洁。

这样我们就可以获得相同的便利,而不仅限于单线程编程。

实用类:

import java.io.Closeable

object ManagedScope {
  val scope=new ThreadLocal[Scope]();
  def managedScope[T](inner: =>T):T={
    val previous=scope.get();
    val thisScope=new Scope();
    scope.set(thisScope);
    try{
      inner
    } finally {
      scope.set(previous);
      if(!thisScope.detatched) thisScope.close();
    }
  }

  def closeLater[T <: Closeable](what:T): T = {
    val theScope=scope.get();
    if(!(theScope eq null)){
      theScope.closeables=theScope.closeables.:+(what);
    }
    what;
  }

  def detatchScope(): Scope={
    val theScope=scope.get();
    if(theScope eq null) null;
    else {
      theScope.detatched=true;
      theScope;
    }
  }
}

class Scope{
  var detatched=false;
  var closeables:List[Closeable]=List();

  def close():Unit={
    for(c<-closeables){
      try{
        if(!(c eq null))c.close();
      } catch{
        case e:Throwable=>{};
      }
    }
  }
}

用法:

  def checkSocketConnect(host:String, portNumber:Int):Unit = managedScope {
    // The close later function tags the closeable to be closed later
    val socket = closeLater( new Socket(host, portNumber) );
    doWork(socket);
  }

  def checkFutureConnect(host:String, portNumber:Int):Unit = managedScope {
    // The close later function tags the closeable to be closed later
    val socket = closeLater( new Socket(host, portNumber) );
    val future:Future[Boolean]=doAsyncWork(socket);

    // Detatch the scope and use it in the future.
    val scope=detatchScope();
    future.onComplete(v=>scope.close());
  }

1
投票

使用结构类型似乎有点矫枉过正,因为java.lang.AutoCloseable是预定的用法:

  def using[A <: AutoCloseable, B](resource: A)(block: A => B): B =
    try block(resource) finally resource.close()

或者,如果您更喜欢扩展方法:

  implicit class UsingExtension[A <: AutoCloseable](val resource: A) extends AnyVal {
    def using[B](block: A => B): B = try block(resource) finally resource.close()
  }

using2是可能的:

  def using2[R1 <: AutoCloseable, R2 <: AutoCloseable, B](resource1: R1, resource2: R2)(block: (R1, R2) => B): B =
    using(resource1) { _ =>
      using(resource2) { _ =>
        block(resource1, resource2)
      }
    }

但imho非常难看 - 我更愿意在客户端代码中简单地嵌套这些使用语句。


1
投票

启动Scala 2.13,标准库提供专用的资源管理实用程序:Using

更具体地说,在处理多种资源时可以使用Using#Manager

在我们的例子中,我们可以管理不同的资源,例如你的PrintWriterBufferedReader,因为它们都实现了AutoCloseable,以便从一个文件读取和写入另一个文件,无论如何,之后关闭输入和输出资源:

import scala.util.Using
import java.io.{PrintWriter, BufferedReader, FileReader}

Using.Manager { use =>

  val in  = use(new BufferedReader(new FileReader("input.txt")))
  val out = use(new PrintWriter("output.txt"))

  out.println(in.readLine)
}
// scala.util.Try[Unit] = Success(())

0
投票

这是我在Scala中的资源管理解决方案:

  def withResources[T <: AutoCloseable, V](r: => T)(f: T => V): V = {
    val resource: T = r
    require(resource != null, "resource is null")
    var exception: Throwable = null
    try {
      f(resource)
    } catch {
      case NonFatal(e) =>
        exception = e
        throw e
    } finally {
      closeAndAddSuppressed(exception, resource)
    }
  }

  private def closeAndAddSuppressed(e: Throwable,
                                    resource: AutoCloseable): Unit = {
    if (e != null) {
      try {
        resource.close()
      } catch {
        case NonFatal(suppressed) =>
          e.addSuppressed(suppressed)
      }
    } else {
      resource.close()
    }
  }

我在多个Scala应用程序中使用它,包括管理Spark执行程序中的资源。并且应该意识到我们是其他更好的管理资源的方法,例如CatsIO:https://typelevel.org/cats-effect/datatypes/resource.html。如果你在Scala中使用纯FP就可以了。

回答你的上一个问题,你绝对可以像这样嵌套资源:

withResource(r: File)(
  r => {
    withResource(a: File)(
      anotherR => {
        withResource(...)(...)
      }
    )
  }
)

这样,不仅可以防止这些资源泄漏,它们也将以正确的顺序(如堆栈)发布。与CatsIO的Resource Monad相同的行为。

© www.soinside.com 2019 - 2024. All rights reserved.