设计模式在Kotlin中的实现：组合模式解析-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00230/article/details/148488320

设计模式在Kotlin中的实现：组合模式解析

Design-Patterns-In-Kotlin Design Patterns implemented in Kotlin 项目地址: https://gitcode.com/gh_mirrors/de/Design-Patterns-In-Kotlin

组合模式（Composite Pattern）是一种结构型设计模式，它允许你将对象组合成树形结构来表示"部分-整体"的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。本文将通过一个计算机硬件组装的例子来详细讲解组合模式在Kotlin中的实现方式。

组合模式的核心概念

组合模式主要包含以下几个关键组件：

组件接口（Component）：定义所有对象的通用接口，包括组合对象和叶子对象
叶子对象（Leaf）：表示组合中的叶子节点对象，没有子节点
组合对象（Composite）：包含子组件的容器对象，实现了组件接口的相关操作

Kotlin实现解析

基础组件定义

首先定义了一个基础的Equipment类作为所有设备的基类：

open class Equipment(
    open val price: Int,
    val name: String
)

这个类有两个属性：

price：设备价格，标记为open以便子类可以覆盖
name：设备名称

组合类实现

Composite类继承自Equipment，它能够包含其他设备：

open class Composite(name: String) : Equipment(0, name) {
    private val equipments = ArrayList<Equipment>()
    
    override val price: Int
        get() = equipments.map { it.price }.sum()
    
    fun add(equipment: Equipment) =
        apply { equipments.add(equipment) }
}

关键点：

初始化价格为0，因为组合对象本身没有价格，价格由子组件决定
使用ArrayList来存储子设备
重写price属性，计算所有子设备价格的总和
提供add方法添加子设备，返回this以便链式调用

具体实现类

定义了具体的设备类：

class PersonalComputer : Composite("PC")  // 组合对象
class Processor : Equipment(1070, "Processor")  // 叶子对象
class HardDrive : Equipment(250, "Hard Drive")  // 叶子对象
class Memory : Equipment(280, "Memory")  // 叶子对象

测试用例

测试展示了如何组装一台PC并计算总价：

@Test
fun Composite() {
    val pc = PersonalComputer()
        .add(Processor())
        .add(HardDrive())
        .add(Memory())

    println(pc.price)

    assertThat(pc.name).isEqualTo("PC")
    assertThat(pc.price).isEqualTo(1600)
}

测试验证了：