2016-11-06 18 views
5

我隔壁班:無形的轉換情況下類HList並跳過所有選項字段

case class Foo(a: Option[Int], b: Option[String], c: Option[Double]) 

,你可以看到,所有字段是可選的,我想轉換這個類爲HList或元組,像

val f1 = Foo(Some(1) , None, Some(3D)) 
val f2 = Foo(None, "foo") 

val result1 = f1.to[Int::Double::HNil] // => 1::3D 
val result2 = f2.to[String::HNil] // "foo" 

有沒有可能,沒有反思?

+1

你想在例如'f1.to [String :: HNil]'的情況?這當然是可能的,但是你需要決定如何處理只在運行時才知道缺失的情況。 –

+0

我認爲 - 異常,但可能返回Option [T <:HList]就像無形的'cast',並且在'f1.to [String :: HNil]'的情況下返回None – mike

回答

6

對於Shapeless中的現有類型類(類似於NatTRelRemoveAll),可能會這樣做,但我並不是100%確定的,並且這種情況下我只需編寫自己的類型類:

import shapeless._ 

trait OptionalPieces[L <: HList, S <: HList] { 
    def apply(l: L): Option[S] 
} 

object OptionalPieces extends LowPriorityOptionalPieces { 
    implicit val hnilOptionalPieces: OptionalPieces[HNil, HNil] = 
    new OptionalPieces[HNil, HNil] { 
     def apply(l: HNil): Option[HNil] = Some(HNil) 
    } 

    implicit def hconsOptionalPiecesMatch[H, T <: HList, S <: HList](implicit 
    opt: OptionalPieces[T, S] 
): OptionalPieces[Option[H] :: T, H :: S] = 
    new OptionalPieces[Option[H] :: T, H :: S] { 
     def apply(l: Option[H] :: T): Option[H :: S] = for { 
     h <- l.head 
     t <- opt(l.tail) 
     } yield h :: t 
    } 
} 

sealed class LowPriorityOptionalPieces { 
    implicit def hconsOptionalPiecesNoMatch[H, T <: HList, S <: HList](implicit 
    opt: OptionalPieces[T, S] 
): OptionalPieces[Option[H] :: T, S] = 
    new OptionalPieces[Option[H] :: T, S] { 
     def apply(l: Option[H] :: T): Option[S] = opt(l.tail) 
    } 
} 

此作見證L包含至少所有的S包裹在Option,爲了元素,併爲您提供了一種方式在運行時(安全)解開他們。

然後,我們可以定義一個語法輔助類是這樣的:

implicit class OptionalPiecesSyntax[A, R <: HList](a: A)(implicit 
    gen: Generic.Aux[A, R] 
) { 
    def to[S <: HList](implicit op: OptionalPieces[gen.Repr, S]): Option[S] = 
    op(gen.to(a)) 
} 

然後:

scala> val f1 = Foo(Some(1) , None, Some(3D)) 
f1: Foo = Foo(Some(1),None,Some(3.0)) 

scala> val f2 = Foo(None, Some("foo"), None) 
f2: Foo = Foo(None,Some(foo),None) 

scala> val result1 = f1.to[Int :: Double :: HNil] 
result1: Option[shapeless.::[Int,shapeless.::[Double,shapeless.HNil]]] = Some(1 :: 3.0 :: HNil) 

scala> val result2 = f2.to[String :: HNil] 
result2: Option[shapeless.::[String,shapeless.HNil]] = Some(foo :: HNil) 

如果你真的想例外,你可以只調用.get在語法類,但似乎是一個壞主意。

相關問題