scala实现组合模式

package com.linewell.modeldesgin.Composite

import scala.collection.mutable.ArrayBuffer

/**
* 抽象构件，透明组合模式
* Created by ctao on 2015/8/16.
*/
trait AbstractFile {
def add(abstractFile: AbstractFile) = println("对不起，不支持该方法")

def remove(abstractFile: AbstractFile) = println("对不起，不支持该方法")

def child(i: Int): AbstractFile = {
println("对不起，不支持该方法")
null
}

def killVirus(): Unit

}

/**
* 图片文件样例类
* @param name 文件名
*/
case class ImageFile(name: String) extends AbstractFile {
override def killVirus(): Unit = println(s"------对图像$name 进行杀毒")
}

/**
* 文本文件样例类
* @param name 文件名
*/
case class TextFile(name: String) extends AbstractFile {
override def killVirus(): Unit = println(s"------对文本文件$name 进行杀毒")
}

/**
* 视频文件样例类
* @param name 文件名
*/
case class VideoFile(name: String) extends AbstractFile {
override def killVirus(): Unit = println(s"------对视频文件$name 进行杀毒")
}


/**
* 文件夹样例类
* @param name 文件夹名
*/
case class Folder(name: String) extends AbstractFile {
/**
* 队列，存放文件或文件夹
*/
var fileList = new ArrayBuffer[AbstractFile]()


/**
* 插入
* @param abstractFile 抽象构件
*/
override def add(abstractFile: AbstractFile) = fileList += abstractFile


/**
* 删除
* @param abstractFile 抽象构件
*/
override def remove(abstractFile: AbstractFile) = fileList -= abstractFile

/**
* 查找孩子
* @param i 索引
* @return 子对象
*/
override def child(i: Int) = fileList(i)

/**
* 递归调用
*/
override def killVirus(): Unit = {
println(s"******对文件夹$name 进行杀毒")
for (file <- fileList) file.killVirus()
}
}
package com.linewell.modeldesgin.Composite

/**
* 测试客户端
* Created by ctao on 2015/8/16.
*/
object Client extends App {
/**
* 文件夹
*/
val folder1 = Folder("ctao的资料")
val folder2 = Folder("图像资料")
val folder3 = Folder("文本文件")
val folder4 = Folder("视频资料")


/**
* 文件
*/
val file1 = ImageFile("qq截图.jpg")
val file2 = ImageFile("美图.png")
val file3 = TextFile("solr.log")
val file4 = TextFile("迭代.doc")
val file5 = VideoFile("测试.avi")

folder2.add(file1)
folder2.add(file2)
folder3.add(file3)
folder3.add(file4)
folder4.add(file5)
folder1.add(folder2)
folder1.add(folder3)
folder1.add(folder4)

folder1.remove(folder3)
// folder1.child(2).killVirus()
/**
* 对根文件夹杀毒，递归调用
*/
folder1.killVirus()

}