summaryrefslogtreecommitdiff
path: root/src/library/scala/runtime/AbstractPartialFunction.scala
blob: cbe778f09b110edf25d6b947b7040f63c6bbf30d (plain) (blame)
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
/*                     __                                               *\
**     ________ ___   / /  ___     Scala API                            **
**    / __/ __// _ | / /  / _ |    (c) 2011, LAMP/EPFL                  **
**  __\ \/ /__/ __ |/ /__/ __ |    http://scala-lang.org/               **
** /____/\___/_/ |_/____/_/ | |                                         **
**                          |/                                          **
\*                                                                      */

package scala.runtime

import scala.annotation.unchecked.uncheckedVariance

/** This class provides a default implementation of partial functions
 *  that is used for all partial function literals.
 *  It contains an optimized `orElse` method which supports
 *  chained `orElse` in linear time, and with no slow-down
 *  if the `orElse` part is not needed.
 *  The implementation of `orElse` works by cloning the abstract function object
 *  and modifying a private `fallBack` variable that encodes the `getorElse` part.
 */
abstract class AbstractPartialFunction[-T1, +R]
    extends AbstractFunction1[T1, R]
    with PartialFunction[T1, R]
    with Cloneable {

  private var fallBackField: PartialFunction[T1 @uncheckedVariance, R @uncheckedVariance] = _

  def fallBack: PartialFunction[T1, R] = synchronized {
    if (fallBackField eq null) fallBackField = PartialFunction.empty
    fallBackField
  }

  override protected def missingCase(x: T1): R = fallBack(x)

  // Question: Need to ensure that fallBack is overwritten before any access
  // Is the `synchronized` here the right thing to achieve this?
  // Is there a cheaper way?
  override def orElse[A1 <: T1, B1 >: R](that: PartialFunction[A1, B1]) : PartialFunction[A1, B1] = {
    val result = this.clone.asInstanceOf[AbstractPartialFunction[A1, B1]]
    result.synchronized {
      result.fallBackField = if (this.fallBackField eq null) that else this.fallBackField orElse that
      result
    }
  }

  def isDefinedAt(x: T1): scala.Boolean = _isDefinedAt(x) || fallBack.isDefinedAt(x)
  def _isDefinedAt(x: T1): scala.Boolean

}