Tour of Scala

Трейты

Language

Трейты (Traits) используются чтоб обмениваться между классами информацией о структуре и полях. Они похожи на интерфейсы из Java 8. Классы и объекты могут расширять трейты, но трейты не могут быть созданы и поэтому не имеют параметров.

Объявление трейта

Минимальное объявление трейта - это просто ключевое слово trait и его имя:

trait HairColor

Трейты наиболее полезны в качестве обобщенного типа с абстрактными методами.

trait Iterator[A] {
  def hasNext: Boolean
  def next(): A
}

При наследовании от трейта Iterator[A] требует указание типа A а также реализация методов hasNext и next.

Использование трейтов

Чтоб использовать трейты, необходимо наследовать класс от него используя ключевое слово extends. Затем необходимо реализовать все абстрактные члены трейта, используя ключевое слово override:

trait Iterator[A] {
  def hasNext: Boolean
  def next(): A
}

class IntIterator(to: Int) extends Iterator[Int] {
  private var current = 0
  override def hasNext: Boolean = current < to
  override def next(): Int =  {
    if (hasNext) {
      val t = current
      current += 1
      t
    } else 0
  }
}


val iterator = new IntIterator(10)
iterator.next()  // вернет 0
iterator.next()  // вернет 1

Этот класс IntIterator использует параметр to в качестве верхней границы. Он наследуется от Iterator[Int], что означает, что метод next должен возвращать Int.

Подтипы

Туда где требуется определенный тип трейта, мы можем передавать любой наследованный от требуемого трейта класс

import scala.collection.mutable.ArrayBuffer

trait Pet {
  val name: String
}

class Cat(val name: String) extends Pet
class Dog(val name: String) extends Pet

val dog = new Dog("Harry")
val cat = new Cat("Sally")

val animals = ArrayBuffer.empty[Pet]
animals.append(dog)
animals.append(cat)
animals.foreach(pet => println(pet.name))  // выведет "Harry" и "Sally"

У трейта Pet есть абстрактное поле name, которое реализовано в классах Cat and Dog. В последней строке мы вызываем pet.name, который должен быть реализован в любом подтипе унаследованным от трейта Pet.

Contributors to this page: