При программировании в java я всегда регистрирую входной параметр и возвращаемое значение метода, но в scala последней строкой метода является возвращаемое значение. поэтому я должен сделать что-то вроде:
def myFunc() = {
val rs = calcSomeResult()
logger.info("result is:" + rs)
rs
}
чтобы сделать его легким, я пишу утилиту:
class LogUtil(val f: (String) => Unit) {
def logWithValue[T](msg: String, value: T): T = { f(msg); value }
}
object LogUtil {
def withValue[T](f: String => Unit): ((String, T) => T) = new LogUtil(f).logWithValue _
}
Затем я использовал его как:
val rs = calcSomeResult()
withValue(logger.info)("result is:" + rs, rs)
он зарегистрирует значение и вернет его. он работает для меня, но кажется странным. поскольку я старый программист Java, но новичок в scala, я не знаю, есть ли более идиоматический способ сделать это в scala.
спасибо за вашу помощь, теперь я создаю лучшего использования, используя комбинатор Kestrel, измеренный romusz
object LogUtil {
def kestrel[A](x: A)(f: A => Unit): A = { f(x); x }
def logV[A](f: String => Unit)(s: String, x: A) = kestrel(x) { y => f(s + ": " + y)}
}
Я добавляю параметр f, чтобы передать его логгером из slf4j, а тестовый пример:
class LogUtilSpec extends FlatSpec with ShouldMatchers {
val logger = LoggerFactory.getLogger(this.getClass())
import LogUtil._
"LogUtil" should "print log info and keep the value, and the calc for value should only be called once" in {
def calcValue = { println("calcValue"); 100 } // to confirm it called only once
val v = logV(logger.info)("result is", calcValue)
v should be === 100
}
}