在状态monad中使用flatMap进行映射时,类型推断失败

时间:2019-07-08 04:30:03

标签: scala shapeless scala-cats

概述

我正在尝试推断传递给flatMap方法的cats.data.IndexedStateT[F[_], SA, SB, A]的类型。仅在使用flatMap时,类型推断似乎可以正确推断SA,SB和A的类型参数。但是,当我在flatMap中使用地图时,它将失败。

是否有一种方法可以使这种类型推断工作而无需手动指定传递给flatMap的IndexedStateT的类型参数?

class X
class Y


// Type inference works well when just using flatMap
val res1: IndexedStateT[Eval, Unit, Y, Y] =
    IndexedStateT[Eval, Unit, X, X](_ => Eval.now(new X, new X))
      .flatMap { x =>
        IndexedStateT(_ => Eval.now(new Y, new Y)) // Infers IndexedStateT[Eval, X, Y, Y]
      }


// Type inference fails when mapping inside flatMap
val res2: IndexedStateT[Eval, Unit, Y, (X, Y)] =
    IndexedStateT[Eval, Unit, X, X](_ => Eval.now(new X, new X))
      .flatMap { x =>
        IndexedStateT(_ => Eval.now(new Y, new Y)).map(x -> _) // Fails to infer the types for IndexedStateT[Eval, X, Y, Y] "missing parameter type"
      }

使用环境

我在应用程序代码中使用特殊类型的State monad

type HListState[SA <: HList, A] = IndexedStateT[Eval, SA, A :: SA, A]
  object HListState {
    def apply[SA <: HList, A](fn: SA => A): HListState[SA, A] = IndexedStateT[Eval, SA, A :: SA, A](sa => Eval.now((fn(sa) :: sa, fn(sa))))
  }

// Type inference works here
val res3: IndexedStateT[Eval, HNil, Y :: X :: HNil, Y] =
  HListState[HNil, X](_ => new X).flatMap { x =>
    HListState(_ => new Y)
  }

// Inference not so good :(
val res4: IndexedStateT[Eval, HNil, Y :: X :: HNil, (X, Y)] = 
  HListState[HNil, X](_ => new X).flatMap { x =>
    HListState(_ => new Y).map(x -> _)  // <--- type inference fails here :( "missing parameter type"
  }

在这种情况下,有没有办法使类型推断起作用?

1 个答案:

答案 0 :(得分:0)

在没有map的情况下,编译器可以使用预期的res1类型来确定flatMap以及IndexedStateT的类型参数。但是,当您添加map时,IndexedStateT调用将没有预期的类型。

我不确定是否要进行测试,但是我希望指定参数类型(SA)就足够了,其余的推断应该没有问题:

IndexedStateT { _: X => Eval.now(new Y, new Y) }.map(x -> _)