Как сопоставление с образцом в Scala реализовано на уровне байт-кода?

Как сопоставление с образцом в Scala реализовано на уровне байт-кода?

Это как серия if (x instanceof Foo) конструкции или что-то еще? Каковы его показатели производительности?

Например, учитывая следующий код (из страниц Scala By Example 46-48), как будет выглядеть эквивалентный Java-код для eval метод похож?

abstract class Expr
case class Number(n: Int) extends Expr
case class Sum(e1: Expr, e2: Expr) extends Expr

def eval(e: Expr): Int = e match {
  case Number(x) => x
  case Sum(l, r) => eval(l) + eval(r)
}

PS Я могу читать байт-код Java, поэтому представление байт-кода было бы достаточно для меня, но, возможно, для других читателей было бы лучше знать, как это будет выглядеть как код Java.

PPS Дает ли книга " Программирование в Scala" ответ на этот и другие подобные вопросы о том, как реализована Scala? Я заказал книгу, но она еще не пришла.

3 ответа

Решение

Низкий уровень можно исследовать с помощью дизассемблера, но краткий ответ заключается в том, что это набор элементов if/elses, где предикат зависит от шаблона.

case Sum(l,r) // instance of check followed by fetching the two arguments and assigning to two variables l and r but see below about custom extractors 
case "hello" // equality check
case _ : Foo // instance of check
case x => // assignment to a fresh variable
case _ => // do nothing, this is the tail else on the if/else

Есть гораздо больше, что вы можете сделать с шаблонами типа или шаблонами и комбинациями, такими как "case Foo(45, x)", но в целом это всего лишь логические расширения того, что я только что описал. У шаблонов также могут быть элементы защиты, которые являются дополнительными ограничениями для предикатов. Есть также случаи, когда компилятор может оптимизировать сопоставление с образцом, например, когда есть некоторое совпадение между случаями, он может немного объединить вещи. Расширенные шаблоны и оптимизация являются активной областью работы компилятора, поэтому не удивляйтесь, если байт-код существенно улучшится по сравнению с этими основными правилами в текущих и будущих версиях Scala.

В дополнение ко всему этому вы можете написать свои собственные пользовательские экстракторы в дополнение или вместо стандартных, которые Scala использует для классов дел. Если вы это сделаете, то стоимость сопоставления с образцом - это стоимость всего, что делает экстрактор. Хороший обзор находится в http://lamp.epfl.ch/~emir/written/MatchingObjectsWithPatterns-TR.pdf

Джеймс (выше) сказал это лучше всего. Однако, если вам интересно, всегда полезно посмотреть на разобранный байт-код. Вы также можете вызвать scalac с -print опция, которая будет печатать вашу программу со всеми удаленными специфичными для Scala функциями. Это в основном Ява в одежде Скалы. Вот соответствующий scalac -print вывод для фрагмента кода, который вы дали:

def eval(e: Expr): Int = {
  <synthetic> val temp10: Expr = e;
  if (temp10.$isInstanceOf[Number]())
    temp10.$asInstanceOf[Number]().n()
  else
    if (temp10.$isInstanceOf[Sum]())
      {
        <synthetic> val temp13: Sum = temp10.$asInstanceOf[Sum]();
        Main.this.eval(temp13.e1()).+(Main.this.eval(temp13.e2()))
      }
    else
      throw new MatchError(temp10)
};

Начиная с версии 2.8, Scala имеет аннотацию @switch. Цель состоит в том, чтобы гарантировать, что сопоставление с образцом будет скомпилировано в табличный или поисковый переключатель вместо серии условных if заявления.

Чтобы расширить комментарий @Zifre: если вы читаете это в будущем, и компилятор scala принял новые стратегии компиляции, и вы хотите знать, что они из себя представляют, вот как вы узнаете, что он делает.

Скопируйте и вставьте свой matchкод в автономный файл примера. Бежать scalacв этом файле. Тогда беги javap -v -c theClassName$.class.

Например, я помещаю в /tmp/question.scala:

object question {
  abstract class Expr
  case class Number(n: Int) extends Expr
  case class Sum(e1: Expr, e2: Expr) extends Expr

  def eval(e: Expr): Int = e match {
    case Number(x) => x
    case Sum(l, r) => eval(l) + eval(r)
  }
}

Затем я побежал scalac question.scala, который произвел кучу *.classфайлы. Поковырявшись немного, я нашел выражение соответствия внутри question$.class. В javap -c -v question$.class вывод доступен ниже.

Поскольку мы ищем конструкцию потока управления условием, знание набора инструкций байт-кода Java предполагает, что поиск "если" должен быть хорошим местом для начала.

В двух местах мы находим пару последовательных строк на форме isinstanceof <something>; ifeq <somewhere>, что означает: если последнее вычисленное значение не является экземпляром something затем перейти somewhere. (ifeq является jump if zero, и isinstanceof дает вам ноль, чтобы представить ложь.)

Если вы проследите за потоком управления, вы увидите, что он соответствует ответу @Jorge Ortiz: мы делаем if (blah isinstanceof something) { ... } else if (blah isinstanceof somethingelse) { ... }.

Здесь javap -c -v question$.class вывод:

Classfile /tmp/question$.class
  Last modified Nov 20, 2020; size 956 bytes
  MD5 checksum cfc788d4c847dad0863a797d980ad2f3
  Compiled from "question.scala"
public final class question$
  minor version: 0
  major version: 50
  flags: (0x0031) ACC_PUBLIC, ACC_FINAL, ACC_SUPER
  this_class: #2                          // question$
  super_class: #4                         // java/lang/Object
  interfaces: 0, fields: 1, methods: 3, attributes: 4
Constant pool:
   #1 = Utf8               question$
   #2 = Class              #1             // question$
   #3 = Utf8               java/lang/Object
   #4 = Class              #3             // java/lang/Object
   #5 = Utf8               question.scala
   #6 = Utf8               MODULE$
   #7 = Utf8               Lquestion$;
   #8 = Utf8               <clinit>
   #9 = Utf8               ()V
  #10 = Utf8               <init>
  #11 = NameAndType        #10:#9         // "<init>":()V
  #12 = Methodref          #2.#11         // question$."<init>":()V
  #13 = Utf8               eval
  #14 = Utf8               (Lquestion$Expr;)I
  #15 = Utf8               question$Number
  #16 = Class              #15            // question$Number
  #17 = Utf8               n
  #18 = Utf8               ()I
  #19 = NameAndType        #17:#18        // n:()I
  #20 = Methodref          #16.#19        // question$Number.n:()I
  #21 = Utf8               question$Sum
  #22 = Class              #21            // question$Sum
  #23 = Utf8               e1
  #24 = Utf8               ()Lquestion$Expr;
  #25 = NameAndType        #23:#24        // e1:()Lquestion$Expr;
  #26 = Methodref          #22.#25        // question$Sum.e1:()Lquestion$Expr;
  #27 = Utf8               e2
  #28 = NameAndType        #27:#24        // e2:()Lquestion$Expr;
  #29 = Methodref          #22.#28        // question$Sum.e2:()Lquestion$Expr;
  #30 = NameAndType        #13:#14        // eval:(Lquestion$Expr;)I
  #31 = Methodref          #2.#30         // question$.eval:(Lquestion$Expr;)I
  #32 = Utf8               scala/MatchError
  #33 = Class              #32            // scala/MatchError
  #34 = Utf8               (Ljava/lang/Object;)V
  #35 = NameAndType        #10:#34        // "<init>":(Ljava/lang/Object;)V
  #36 = Methodref          #33.#35        // scala/MatchError."<init>":(Ljava/lang/Object;)V
  #37 = Utf8               this
  #38 = Utf8               e
  #39 = Utf8               Lquestion$Expr;
  #40 = Utf8               x
  #41 = Utf8               I
  #42 = Utf8               l
  #43 = Utf8               r
  #44 = Utf8               question$Expr
  #45 = Class              #44            // question$Expr
  #46 = Methodref          #4.#11         // java/lang/Object."<init>":()V
  #47 = NameAndType        #6:#7          // MODULE$:Lquestion$;
  #48 = Fieldref           #2.#47         // question$.MODULE$:Lquestion$;
  #49 = Utf8               question
  #50 = Class              #49            // question
  #51 = Utf8               Sum
  #52 = Utf8               Expr
  #53 = Utf8               Number
  #54 = Utf8               Code
  #55 = Utf8               LocalVariableTable
  #56 = Utf8               LineNumberTable
  #57 = Utf8               StackMapTable
  #58 = Utf8               SourceFile
  #59 = Utf8               InnerClasses
  #60 = Utf8               ScalaInlineInfo
  #61 = Utf8               Scala
{
  public static final question$ MODULE$;
    descriptor: Lquestion$;
    flags: (0x0019) ACC_PUBLIC, ACC_STATIC, ACC_FINAL

  public static {};
    descriptor: ()V
    flags: (0x0009) ACC_PUBLIC, ACC_STATIC
    Code:
      stack=1, locals=0, args_size=0
         0: new           #2                  // class question$
         3: invokespecial #12                 // Method "<init>":()V
         6: return

  public int eval(question$Expr);
    descriptor: (Lquestion$Expr;)I
    flags: (0x0001) ACC_PUBLIC
    Code:
      stack=3, locals=9, args_size=2
         0: aload_1
         1: astore_2
         2: aload_2
         3: instanceof    #16                 // class question$Number
         6: ifeq          27
         9: aload_2
        10: checkcast     #16                 // class question$Number
        13: astore_3
        14: aload_3
        15: invokevirtual #20                 // Method question$Number.n:()I
        18: istore        4
        20: iload         4
        22: istore        5
        24: goto          69
        27: aload_2
        28: instanceof    #22                 // class question$Sum
        31: ifeq          72
        34: aload_2
        35: checkcast     #22                 // class question$Sum
        38: astore        6
        40: aload         6
        42: invokevirtual #26                 // Method question$Sum.e1:()Lquestion$Expr;
        45: astore        7
        47: aload         6
        49: invokevirtual #29                 // Method question$Sum.e2:()Lquestion$Expr;
        52: astore        8
        54: aload_0
        55: aload         7
        57: invokevirtual #31                 // Method eval:(Lquestion$Expr;)I
        60: aload_0
        61: aload         8
        63: invokevirtual #31                 // Method eval:(Lquestion$Expr;)I
        66: iadd
        67: istore        5
        69: iload         5
        71: ireturn
        72: new           #33                 // class scala/MatchError
        75: dup
        76: aload_2
        77: invokespecial #36                 // Method scala/MatchError."<init>":(Ljava/lang/Object;)V
        80: athrow
      LocalVariableTable:
        Start  Length  Slot  Name   Signature
            0      81     0  this   Lquestion$;
            0      81     1     e   Lquestion$Expr;
           20      61     4     x   I
           47      34     7     l   Lquestion$Expr;
           54      27     8     r   Lquestion$Expr;
      LineNumberTable:
        line 6: 0
        line 7: 2
        line 8: 27
        line 6: 69
      StackMapTable: number_of_entries = 3
        frame_type = 252 /* append */
          offset_delta = 27
          locals = [ class question$Expr ]
        frame_type = 254 /* append */
          offset_delta = 41
          locals = [ top, top, int ]
        frame_type = 248 /* chop */
          offset_delta = 2
}
SourceFile: "question.scala"
InnerClasses:
  public static #51= #22 of #50;          // Sum=class question$Sum of class question
  public static abstract #52= #45 of #50; // Expr=class question$Expr of class question
  public static #53= #16 of #50;          // Number=class question$Number of class question
  ScalaInlineInfo: length = 0xE (unknown attribute)
   01 01 00 02 00 0A 00 09 01 00 0D 00 0E 01
  Scala: length = 0x0 (unknown attribute)
Другие вопросы по тегам