Добавление регистрации, если возвращаемое значение Нет - PullRequest
Новая
       7

Добавление регистрации, если возвращаемое значение Нет

2 голосов
/ 25 июля 2011

Предположим, есть две функции findUser(id:String):Option[User] и findAddress(user:User):Option[Address], вызываемые следующим образом: 

for(user <- findUser(id); address <- findAddress(user)) println(address)

Теперь я хотел бы добавить регистрацию ошибок к этому for-comprehension.Я хотел бы вызвать функцию log(msg:String), если либо user, либо address не найдено.

for(user <- findUser(id) ifNone log("user not found"); 
    address <- findAddress(user) ifNone log("address not found")) 
       println(address)

Можно ли это сделать без изменения сигнатур функций?

Ответы [ 3 ]

3 голосов
/ 25 июля 2011

Lift's Box - более подходящий класс для вашего случая использования.A Box похож на Option, но с двумя пустыми состояниями: ok и error.Вы можете использовать его следующим образом: 

val addr = for {
  user <- findUser(id) ?~ "user not found"
  address <- findAddress(user) ?~ "address not found"
} yield address

address match {
  case Full(addr) => println(addr)
  case oops: Failure => println(oops.msg) // see Failure for more details
}

См. этот блог для различных предложений, связанных с вашей проблемой.

1 голос
/ 25 июля 2011

Это может быть излишним, но это похоже на то, что вы хотели;) 

object Extensions {
  // You need a wrapper since Option is sealed
  class OptionWrapper[E](option: Option[E]) {
    def foreach[U](f: E => U) {
      option foreach f
    }
    def isEmpty = option.isEmpty
  }

  // Modification trait for OptionWrapper
  trait ErrorLogging[E] extends OptionWrapper[E] {
    abstract override def foreach[U](f: E => U) {
      if (isEmpty)
        println("error")
      else
        super.foreach(f)
    }
  }

  // Accessor for the new mixin
  def log[E](option: Option[E]) = new OptionWrapper(option) with ErrorLogging[E]
}

object TestingLogger extends App {
  case class User(address: String)
  def findUser(id: Int): Option[User] = if (id == 1) Some(User("address")) else None
  def findAddress(user: User): Option[String] = Some(user.address)

  import Extensions._

  for {
    user <- log(findUser(1)) // prints out address
    address <- log(findAddress(user))
  } println(address)

  for {
    user <- log(findUser(2)) // prints out error
    address <- log(findAddress(user))
  } println(address)
}

Если вы не знаете, что только что произошло, прочитайте это .

1 голос
/ 25 июля 2011

Может быть 

implicit def withIfNone[A](o: Option[A]) = new {
  def ifNone(action: => Unit) = { if (o == None) action; o }
}

Вы также можете рассмотреть возможность использования либо варианта, либо параметра (или преобразования ваших параметров в любой вариант).Это не сработает с foreach (a для без доходности), но вы можете сделать 

for(
  a <- option1.toRight("option1 missing").right; 
  b <- option2.toRight("option2 missing").right)
yield f(a,b)

Тогда вы можете сопоставить шаблон с результатом 

case Left(error) => log (error)
case Right(result) => // use result
...