2015-03-25 7 views
3

В контексте another stackoverflow question, у меня есть этот фрагмент:Infer параметр типа от аргумента функции

def orderedGroupBy[T, P](seq: Traversable[T], f: T => P): Traversable[Tuple2[P, Traversable[T]]] = { 
    @tailrec 
    def accumulator(seq: Traversable[T], f: T => P, res: List[Tuple2[P, Traversable[T]]]): Traversable[Tuple2[P, Traversable[T]]] = seq.headOption match { 
    case None => res.reverse 
    case Some(h) => { 
     val key = f(h) 
     val subseq = seq.takeWhile(f(_) == key) 
     accumulator(seq.drop(subseq.size), f, (key -> subseq) :: res) 
    } 
    } 
    accumulator(seq, f, Nil) 
} 

я хотел бы использовать его так же, как можно использовать .groupBy, например:

orderedGroupBy(1 to 100, (_/10)) 

Но компилятор дает ошибку о не имея достаточно информации типа

<console>:10: error: missing parameter type for expanded function ((x$1) => x$1.$div(10)) 
       orderedGroupBy(1 to 100, (_/10)) 

Что идиоматических способ сделать это?

ответ

4

Вы можете изменить параметры, так что T определяется только от seq: Traversable[T].

def orderedGroupBy[T, P](seq: Traversable[T])(f: T => P): Traversable[Tuple2[P, Traversable[T]]] = ??? 

scala> orderedGroupBy(1 to 100)(_/10) 
res110: Traversable[(Int, Traversable[Int])] = List((0,Range(1, 2, 3, 4, 5, 6, 7, 8, 9)), (1,Range(10, 11, 12, 13, 14, 15, 16, 17, 18, 19)), (2,Range(20, 21, 22, 23, 24, 25, 26, 27, 28, 29)), (3,Range(30, 31, 32, 33, 34, 35, 36, 37, 38, 39)), (4,Range(40, 41, 42, 43, 44, 45, 46, 47, 48, 49)), (5,Range(50, 51, 52, 53, 54, 55, 56, 57, 58, 59)), (6,Range(60, 61, 62, 63, 64, 65, 66, 67, 68, 69)), (7,Range(70, 71, 72, 73, 74, 75, 76, 77, 78, 79)), (8,Range(80, 81, 82, 83, 84, 85, 86, 87, 88, 89)), (9,Range(90, 91, 92, 93, 94, 95, 96, 97, 98, 99)), (10,Range(100))) 
+0

Является ли это ожидаемой вещью? Как это задержит в обзоре кода? Спасибо за ответ кстати. – hraban

+1

На самом деле я предпочел бы карри, потому что это позволит вам использовать его с кодовым блоком, не выглядя ужасно. 'orderedGroupBy (mySeq) {element => .....}' .. Представьте, что у вас больше логики. 'foldLeft' в библиотеке коллекций scala делает то же самое. –

+1

@hraban Это совершенно стандартно в Scala. –

Смежные вопросы