scala中的caseclass

38 阅读3分钟

case class 是 Scala 中一种特殊的类,它用于创建不可变的数据容器

package caseclass

// 使用Set对自定义对象进行去重
import scala.collection.mutable
object caseclass1 {
  class Book(var id: Int, var name: String) {
    override def equals(obj: Any): Boolean = {
      val other = obj.asInstanceOf[Book]
      other.id == id && other.name == name
    }
    override def hashCode(): Int = id
  }

  def main(args: Array[String]): Unit = {
    // 示例1:使用Set对基本类型进行去重
    val set1: mutable.Set[Int] = mutable.Set(1, 2)

    set1 += 1
    set1 += 1
    set1 += 1
    set1 += 1
    println(set1)

    val book1 = new Book(1, "西游记")
    val book2 = new Book(1, "西游记")

    println(book1 == book2)

   // Set 的类型是 Book,它可以装入的数据是 book 类型。
    val set2: mutable.Set[Book] = mutable.Set()
    set2 += book1
    set2 += book2

    println(set2)
  }
}

case class: 样板类

  1. 可以省略 new
  2. 属性默认是不可变的。val 修饰
  3. 它会自动去实现toString, equals, hasCode等方法
package caseclass
// case class: 样板类

// 1. 可以省略 new
// 2. 属性默认是不可变的。val 修饰
// 3. 它会自动去实现toString, equals, hasCode等方法
object caseclass3 {
  case class Student(name: String, age: Int)

  def main(args: Array[String]): Unit = {
    // val st1 = new Student("小花", 18)
    // 可以省略 new
    val st1 = Student("小花", 18)
    val st2 = Student("小花", 18)

    // 属性默认是不可变的。val 修饰
    // st1.name = "小花花"  // 这行会编译错误

    println(st1)          // 输出: Student(小花,18)
    println(st1 == st2)   // 输出: true
  }
}

实训内容

步骤任务:

1. 使用case class创建图书信息类Book:包含四个属性:ID,书名,作者,价格,数量。

2. 创建一个名为BookList的可变List,它只能用来保存Book的实例。

3. 初始化三本不同的书(注意id不同就是不同的书),加入到BookList中。

4. 添加一本已经存在的书。此时应该去修改BookList中对应书目的数量,而不是去添加一条新数据。

5. 根据图书名称查询是否在列表中(通过遍历列表对比名称)

6. 删除指定书名的书

7. 删除指定ID的书

8. 对于图书按价格从高到低排序。

9. 遍历图书列表,并打印每本书的详细信息

10. 展示全部的总的金额

package caseclass
import scala.collection.mutable.ListBuffer

object caseclass4 {
  // 1. 定义 case class Book
  case class Book(id: Int, name: String, author: String, price: Double, var amount: Int)

  def main(args: Array[String]): Unit = {
    // 2. 创建可变列表
    val BookList: ListBuffer[Book] = ListBuffer()

    // 3. 初始化三本不同的书
    val book1 = Book(1, "凡人修仙记", "梦语", 20.2, 1)
    val book2 = Book(2, "霸道总裁蒙上欺", "梦语", 100, 1)
    val book3 = Book(3, "重生之超级假语", "梦语", 50, 1)

    BookList += book1
    BookList += book2
    BookList += book3

    // 4. 添加图书(如存在则修改数量,否则添加新书)
    // val book4 = Book(3, "重生之超级假语", "梦语", 50, 10) // 同ID的测试用例
    val book4 = Book(4, "18岁的奶奶归来,重整家族", "梦语", 50, 10)

    val rst4 = BookList.find(ele => ele.id == book4.id) // 修正语法错误
    if (rst4.isDefined) { // 修改数量
      rst4.get.amount = rst4.get.amount + book4.amount
    } else {
      BookList += book4
    }

    // 5. 查询图书是否存在
    var bookName = "霸道总裁蒙上欺"
    val rst0 = BookList.find(ele => ele.name == bookName) // 修正语法错误
    if (rst0.isDefined) {
      println(s"《${bookName}》存在") // 修正字符串插值语法
    } else {
      println(s"《${bookName}》不存在") // 修正字符串插值语法
    }
    // 6. 删除书名是"霸道总裁爱上我"的书
    bookName = "霸道总裁爱上我"
    val rstl = BookList.find(ele => ele.name == bookName)
    if (rstl.isDefined) {
      BookList -= rstl.get
      println(s"删除书名是${bookName}的书成功")
    } else {
      println(s"没有找到书名是: ${bookName}的书")
    }

    // 7. 删除id为1的书
    val id = 1
    val rst2 = BookList.find(ele => ele.id == id)
    if (rst2.isDefined) {
      BookList -= rst2.get
      println(s"删除id=${id}的书成功")
    } else {
      println(s"没有找到id=${id}的书")
    }

    // 8. 按价格从高到低排序
    val newList = BookList.sortWith((a, b) => { // 修正:sortWith 不是 sortWidth
      a.price > b.price
    })

    // 9. 遍历打印图书信息
    newList.foreach(ele => { // 修正:foreach 不是 forEach
      println(s"${ele.id} ${ele.name} ${ele.price} ${ele.amount}本")
    })

    // 10. 计算总价格
    var totalPrice = 0.0
    newList.foreach(ele => {
      totalPrice += ele.price * ele.amount
    })
    println(s"总价格: ${totalPrice}")
  }
}