-
Notifications
You must be signed in to change notification settings - Fork 597
/
CollectorPlatform.scala
78 lines (65 loc) · 3.13 KB
/
CollectorPlatform.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
* Copyright (c) 2013 Functional Streams for Scala
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package fs2
import scala.collection.immutable.ArraySeq
import scala.collection.{Factory, IterableFactory, MapFactory}
import scala.reflect.ClassTag
private[fs2] trait CollectorPlatform { self: Collector.type =>
implicit def supportsCollectionFactory[A, C](f: Factory[A, C]): Collector.Aux[A, C] =
make(Builder.fromCollectionFactory(f))
def supportsFactory[A, C[_], B](f: Factory[A, C[B]]): Collector.Aux[A, C[B]] =
make(Builder.fromFactory(f))
implicit def supportsIterableFactory[A, C[_]](f: IterableFactory[C]): Collector.Aux[A, C[A]] =
make(Builder.fromIterableFactory(f))
implicit def supportsMapFactory[K, V, C[_, _]](f: MapFactory[C]): Collector.Aux[(K, V), C[K, V]] =
make(Builder.fromMapFactory(f))
/** Use `ArraySeq.untagged` to build a `Collector` where a `ClassTag` is not available.
*/
implicit def supportsTaggedArraySeq[A: ClassTag](
a: ArraySeq.type
): Collector.Aux[A, ArraySeq[A]] = {
val _ = a
make(implicitly[ClassTag[A]] match {
case ClassTag.Byte =>
Builder.byteArray.mapResult(ArraySeq.unsafeWrapArray).asInstanceOf[Builder[A, ArraySeq[A]]]
case _ => Builder.taggedArraySeq[A]
})
}
implicit def supportsIArray[A: ClassTag](a: IArray.type): Collector.Aux[A, IArray[A]] = {
val _ = a
new Collector[A] {
type Out = IArray[A]
def newBuilder = supportsArray[A](Array).newBuilder.mapResult(IArray.unsafeFromArray)
}
}
private[fs2] trait BuilderPlatform { self: Collector.Builder.type =>
def fromCollectionFactory[A, C](f: Factory[A, C]): Builder[A, C] =
fromBuilder(f.newBuilder)
def fromFactory[A, C[_], B](f: Factory[A, C[B]]): Builder[A, C[B]] =
fromBuilder(f.newBuilder)
def fromIterableFactory[A, C[_]](f: IterableFactory[C]): Builder[A, C[A]] =
fromBuilder(f.newBuilder)
def fromMapFactory[K, V, C[_, _]](f: MapFactory[C]): Builder[(K, V), C[K, V]] =
fromBuilder(f.newBuilder)
def taggedArraySeq[A: ClassTag]: Builder[A, ArraySeq[A]] =
array[A].mapResult(ArraySeq.unsafeWrapArray)
}
}