在处理数据库连接或者输入输出流等场景时,我们经常需要写一些非常繁琐又枯燥乏味的代码来关闭数据库连接或输入输出流。

例如数据库操作:

  def update(sql: String)(conn: Connection): Int = {
var statement: Statement = null
try {
statement = conn.createStatement()
statement.executeUpdate(sql)
} finally {
if (conn != null) conn.close()
if (statement != null) statement.close()
}
}

例如文本操作:

  def save(sql: String, path: String): Unit = {
var pw: PrintWriter = null
try {
pw = new PrintWriter(path)
pw.println(sql)
} finally {
if (pw != null) pw.close()
}
}

从上面的两个例子中(这样的例子还有很多)可以发现两者存在相同模式的代码:

var a: A = null
try {
a = xxx
//对a进行操作
} finally {
if(a != null) a.close()
}

那能不能把这些相同模式的代码抽象出来呢?答案是肯定的,我们可以利用Scala的泛型和高阶函数来完成。先写一个高阶函数:

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

using函数有两个参数a: A、f: A => B,其中a是需要关闭的资源,f是一个输入为A输出为B的函数。现在我们可以利用using函数来重写数据库操作和文本操作了。

数据库操作:

  def update0(sql: String)(conn: Connection): Int = {
using(conn) {
conn => {
using(conn.createStatement()) {
statement => statement.executeUpdate(sql)
}
}
}
}

文本操作:

  def save0(sql: String, path: String): Unit = {
using(new PrintWriter(path)) {
pw => pw.println(sql)
}
}

可以看出重写后的函数相比之前更加精炼,函数只需要关心实现自己的逻辑而不用关心资源关闭操作,这些就交给using函数来处理吧。目前看来using函数似乎可以满足我们的需求了,真的是这样吗?我们再看一个例子:

  def query[A](sql: String, conn: Connection)(implicit f: ResultSet => List[A]): List[A] = {
using(conn) {
conn => {
using(conn.createStatement()) {
statement => {
using(statement.executeQuery(sql)){
resultSet => f(resultSet)
}
}
}
}
}
}

可以看到上面的例子用到了3次using,嵌套了3层函数,代码的可读性变差。而且一旦需要关闭的资源变多,嵌套函数的层数也将相应增加。代码又陷入另一个繁琐枯燥的模式。有什么更好的办法吗?也许可以试一下for表达式这个语法糖,那么代码将如下表示:

  def query0[A](sql: String, conn: Connection)(implicit f: ResultSet => List[A]): List[A] = {
for {
conn <- Closable(conn)
stmt <- Closable(conn.createStatement())
rs <- Closable(stmt.executeQuery(sql))
} yield f(rs)
}

这样没有了复杂的嵌套函数,代码的可读性更好了。可是Closable是什么类型?别急,我们可以从上面的代码分析出Closable类型是什么结构,首先Closable类型有一个可关闭资源的属性;然后Closable类型可以使用for表达式语法糖,那么Closable类型需要实现map和flatMap函数。Closable类型实现如下:

case class Closable[A <: { def close(): Unit }](a: A) {

  def map[B](f: A => B): B =
try f(a)
finally {
if(a != null) a.close()
} def flatMap[B](f: A => B): B = map(f) }

到此代码已经满足我们的需求了,但是还是很想探究其中的魔法。我们知道for表达式其实就是flatMap加map的语法糖,那么让我们剥开糖衣看看这块糖真实的模样:

  def query1[A](sql: String, conn: Connection)(implicit f: ResultSet => List[A]): List[A] = {
Closable(conn).flatMap {
conn => Closable(conn.createStatement()).flatMap {
stmt => Closable(stmt.executeQuery(sql)).map {
rs => f(rs)
}
}
}
}

让我们接着剥开flatMap

  def query2[A](sql: String, conn: Connection)(implicit f: ResultSet => List[A]): List[A] = {
Closable(conn).map {
conn => Closable(conn.createStatement()).map {
stmt => Closable(stmt.executeQuery(sql)).map {
rs => f(rs)
}
}
}
}

最后剥开map

  def query3[A](sql: String, conn: Connection)(implicit f: ResultSet => List[A]): List[A] = {
Closable(conn).map {
conn => try {
Closable(conn.createStatement()).map {
stmt => try {
Closable(stmt.executeQuery(sql)).map {
rs => try {
f(rs)
} finally {
if(rs != null) rs.close()
}
}
} finally {
if (stmt != null) stmt.close()
}
}
} finally {
if (conn != null) conn.close()
}
}
}

到此Closable类型的神秘面纱已经完全揭开,希望Closable类型可以在各位读者工作中在处理一些需要关闭资源的时候提供一种选择,最后再多说两句。有时我们只需要处理两个可关闭资源,而且这两个资源之间没有关联。例如文本操作有一个输入流一个输出流,那么我们使用Closable类型代码将会如下:

  def save1(inPath: String, outPath: String): Unit = {
for {
in <- Closable(new BufferedReader(new FileReader(inPath)))
out <- Closable(new PrintWriter(outPath))
} yield out.println(in.readLine())
}

不是说上面的代码不好,而是我们可以做到更加简练,代码如下:

  def save2(inPath: String, outPath: String): Unit = {
Closable(new BufferedReader(new FileReader(inPath)))
.map2(new PrintWriter(outPath)){ (in, out) => out.println(in.readLine()) }
}

让我们来看下 map2的实现:

  def map2[B <: { def close(): Unit }, C](b: B)(f: (A, B) => C): C =
for {
ac <- Closable(a)
bc <- Closable(b)
} yield f(ac, bc)

聪明的读者可能还会有疑惑,map2用于关闭两个资源,那关闭3个资源我们需要实现一个map3,关闭N个资源岂不是要实现一个mapN。当然我们可以使用for表达式实现,但是还有更好的实现吗?哈哈,这个就交给读者课后思考了,相信聪明的你一定有自己的想法。

Scala实现Try with resources自动关闭IO的更多相关文章

  1. Java使用Try with resources自动关闭资源

    Try-with-resources Try-with-resources是Java7中一个新的异常处理机制,它能够很容易地关闭在try-catch语句块中使用的资源. 利用Try-Catch-Fin ...

  2. 自动关闭IO流-jdk1.7版本

    public static void main(String[] args) throws IOException { try( FileInputStream fis = new FileInput ...

  3. Java注解(二):实战 - 直接使用对象列表生成报表

    通过对Java注解(一):介绍,思想及优点学习了解,相信大家对Java注解有一定程度的了解,本篇文章将实战项目中的应用来加深对Java注解的了解. 本实例实现根据指定字段的JavaBean,生成对应列 ...

  4. scala 随笔

    创建map,并向map添加元素 val idMap = Map( "group_id" -> "GID", "sim_id" -> ...

  5. Scala学习笔记之二--基本数据类型

    前言 本篇主要讲Scala的基本数据类型,更多教程请参考:Scala教程 基本数据类型 Scala一共提供了9中数据类型,Scala的基本数据类型与java中的基本数据类型是一一对应的,这是Scala ...

  6. 转载:scala中:: , +:, :+, :::, ++的区别

    原文链接:https://segmentfault.com/a/1190000005083578 初学Scala的人都会被Seq的各种操作符所confuse.下面简单列举一下各个Seq操作符的区别. ...

  7. Spark集群 + Akka + Kafka + Scala 开发(1) : 配置开发环境

    目标 配置一个spark standalone集群 + akka + kafka + scala的开发环境. 创建一个基于spark的scala工程,并在spark standalone的集群环境中运 ...

  8. A quick tour of JSON libraries in Scala

    A quick tour of JSON libraries in Scala Update (18.11.2015): added spray-json-shapeless libraryUpdat ...

  9. 在Javaweb中使用Scala

    Java 是一门比较优秀的编程语言, 其最大功劳是建立非常繁荣的JVM平台生态.不过 Java 语法比较麻烦,写过 C, Python 的人总是想使用简洁的语法,又希望利用上 Java 平台的强大,因 ...

随机推荐

  1. Linux时间子系统之三:jiffies

    1. jiffies背景介绍 jiffies记录了系统启动以来,经过了多少tick. 一个tick代表多长时间,在内核的CONFIG_HZ中定义.比如CONFIG_HZ=200,则一个jiffies对 ...

  2. jmeter使用csv进行参数化(一)

    先录制一个脚本,具体录制可以参考笔者的随笔:http://www.cnblogs.com/wuyazi/p/8889770.html 1.准备参数化文本内容:mac没有自带的txt文本编辑器,笔者是在 ...

  3. Java 读书笔记 (四) 常量

    常量在程序运行时不能被修改. 在Java中使用final 关键字来修饰常量 ,声明方式和变量类似: final double PI=3.1415927 常量名也可以用小写,但为了便于识别,通常使用大写 ...

  4. VIJOS-P1635 城市连接

    嘿嘿嘿,逆向spfa,貌似不难... #include <cstdio> #include <algorithm> #include <cmath> #includ ...

  5. Postman----Presets(预先设置)的使用

    使用场景: 当我们在进行接口测试post请求时,headers是必填项,我们将一个A接口的headers编写后测试完成,再次进行B接口的测试,需重新编写headers,为了简单操作,我们就用到了Pre ...

  6. python环境下实现OrangePi Zero寄存器访问及GPIO控制

    最近入手OrangePi Zero一块,程序上需要使用板子上自带的LED灯,在网上一查,不得不说OPi的支持跟树莓派无法相比.自己摸索了一下,实现简单的GPIO控制方法,作者的Zero安装的是Armb ...

  7. django 的时区设置

    在Django的配置文件settings.py中,有两个配置参数是跟时间与时区有关的,分别是TIME_ZONE和USE_TZ 如果USE_TZ设置为True时,Django会使用系统默认设置的时区,即 ...

  8. ASP.NET Core 借助 K8S 玩转容器编排

    Production-Grade Container Orchestration - Automated container deployment, scaling, and management. ...

  9. c#批量抓取免费代理并验证有效性

    之前看到某公司的官网的文章的浏览量刷新一次网页就会增加一次,给人的感觉不太好,一个公司的官网给人如此直白的漏洞,我批量发起请求的时候发现页面打开都报错,100多人的公司的官网文章刷新一次你给我看这个, ...

  10. SpringBoot进阶教程(二十九)整合Redis 发布订阅

    SUBSCRIBE, UNSUBSCRIBE 和 PUBLISH 实现了 发布/订阅消息范例,发送者 (publishers) 不用编程就可以向特定的接受者发送消息 (subscribers). Ra ...