是否可以创建具有多个代码块的自定义控制结构,以before { block1 } then { block2 } finally { block3 }
? 问题仅与糖部分有关-我知道通过将三个块传递给方法可以轻松实现该功能,例如doInSequence(block1, block2, block3)
.
一个真实的例子。对于我的测试实用程序,我想创建一个这样的结构:
getTime(1000) {
// Stuff I want to repeat 1000 times.
} after { (n, t) =>
println("Average time: " + t / n)
}
编辑:
最后我想出了这个解决方案:
object MyTimer {
def getTime(count: Int)(action : => Unit): MyTimer = {
val start = System.currentTimeMillis()
for(i <- 1 to count) { action }
val time = System.currentTimeMillis() - start
new MyTimer(count, time)
}
}
class MyTimer(val count: Int, val time: Long) {
def after(action: (Int, Long) => Unit) = {
action(count, time)
}
}
// Test
import MyTimer._
var i = 1
getTime(100) {
println(i)
i += 1
Thread.sleep(10)
} after { (n, t) =>
println("Average time: " + t.toDouble / n)
}
输出是:
1
2
3
...
99
100
Average time: 10.23
它主要基于Thomas Lockney的回答,我只是添加了伴随对象以便能够import MyTimer._
谢谢大家,伙计们。