Generics Example in Scala

GenericSnippet.scala

/** Test One
  *
  * Created by Ajay Gupta on 04/04/18.
  */
object TestOne {
  def main(args: Array[String]): Unit = {
    val employee1 = Employee(1, "Sourab")
    val employee2 = Employee(2, "Ajay")
    val employeeStack = new Stack[Employee]
    employeeStack.push(employee1)
    employeeStack.push(employee2)

    val employee = employeeStack.pop()
    println(employee)

    val manager1 = Manager(1, "Manager1", 11)
    val manager2 = Manager(2, "Manager2", 12)
    val managerStack = new Stack[Manager]
    managerStack.push(manager1)
    managerStack.push(manager2)

    val manager = managerStack.pop()
    println(manager)
  }
}

case class Employee(id: Int, name: String)

case class Manager(id: Int, name: String, managesEmployee: Int)


class Stack[A] {
  private var elements: List[A] = Nil

  def push(x: A) {
    elements = x :: elements
  }

  def peek: A = elements.head

  def pop(): A = {
    val currentTop = peek
    elements = elements.tail
    currentTop
  }
}