scala实现中介者模式

package com.linewell.modeldesgin.mediator

import scala.collection.mutable.ArrayBuffer

/**
* 中介者模式，抽象中介者，通信软件
* Created by ctao on 2015/9/1.
*/
abstract class AbstractSoftware(val name: String) {
    private var persons = new ArrayBuffer[Person]()

    /**
     * 增加用户
     * @param person 用户
     */
    def addPerson(person: Person): Unit = {
        if (persons.forall(!person.eq(_))) {
            println(s"群${name}增加${person.name}")
            persons += person
        } else {
            println(s"用户${person.name}已在群${name}内")
        }
    }

    /**
     * 删除用户
     * @param person 用户
     */
    def removePerson(person: Person): Unit = {
        if (persons.exists(person.eq(_))) {
            persons -= person
        } else {
            println("该用户已被删除")
        }
    }

    /**
     * 发放通知
     * @param person 发起者
     * @param message 信息
     */
    def notify(person: Person, message: String): Unit = {
        if (persons.exists(person.eq(_))) {
            persons.filter(!person.eq(_)).foreach(p => println(s"${p.name}从${person.name}接收到信息:$message"))
        } else {
            println(s"${person.name}您已经不在群组:$name")
        }

    }

    /**
     * 私聊
     * @param send 发送者
     * @param receive 接收者
     * @param message 信息
     */
    def pm(send: Person, receive: Person, message: String): Unit = send match {
        case p if persons.exists(p.eq(_)) => receive match {
            case r if persons.exists(r.eq(_)) => println(s"${send.name}发送信息:$message 给${receive.name}")
            case _ => println(s"接收者${receive.name}没有获得该软件的许可")
        }
        case _ => println(s"发送者${send.name}没有获得该软件的许可")
    }
}

/**
* qq，具体中介者
* @param name 群名
*/
class QQSoftware(name: String) extends AbstractSoftware(name) {
    override def notify(person: Person, message: String): Unit = {
        println(s"这里是qq:$name")
        super.notify(person, message)
    }

    override def pm(send: Person, receive: Person, message: String): Unit = {
        println(s"使用qq软件进行私聊")
        super.pm(send, receive, message)
    }


}

/**
* msn，具体中介者
* @param name 群名
*/
class MSNSoftware(name: String) extends AbstractSoftware(name) {
    override def notify(person: Person, message: String): Unit = {
        println(s"这里是msn群:$name")
        super.notify(person, message)
    }

    override def pm(send: Person, receive: Person, message: String): Unit = {
        println(s"使用msn软件进行私聊")
        super.pm(send, receive, message)
    }


}


/**
* 抽象同事类
* @param name 名称
*/
abstract class Person(val name: String) {
    /**
     * 设置群组
     * @param software 群组
     */
    def setAbstractSoftware(software: AbstractSoftware)

    /**
     * 发言
     * @param message 信息
     */
    def speak(message: String): Unit

    /**
     * 删除用户
     * @param person 用户
     */
    def remove(person: Person): Unit

    /**
     * 增加用户
     * @param person 用户
     */
    def add(person: Person): Unit

    /**
     * 私聊
     * @param person 接收者
     * @param message 信息
     */
    def privateChat(person: Person, message: String):Unit
}

/**
* 管理员角色，属于同事
* @param name 名称
*/
class Admin(name: String) extends Person(name) {
    private var abstractSoftware: AbstractSoftware = null

    def setAbstractSoftware(software: AbstractSoftware) = abstractSoftware = software

    override def speak(message: String) = abstractSoftware.notify(this, message)

    /**
     * 删除
     * @param person 用户
     */
    def remove(person: Person) = abstractSoftware.removePerson(person)

    /**
     * 增加
     * @param person 用户
     */
    def add(person: Person) = {
        println(s"${name}进行添加用户${person.name}的操作")
        abstractSoftware.addPerson(person)
    }

    /**
     * 私聊
     * @param person 接收者
     * @param message 信息
     */
    def privateChat(person: Person, message: String) = abstractSoftware.pm(this, person, message)
}

/**
* 普通用户角色，属于同事
* @param name 名称
*/
class Member(name: String) extends Person(name) {
    private var abstractSoftware: AbstractSoftware = null

    def setAbstractSoftware(software: AbstractSoftware) = abstractSoftware = software

    override def speak(message: String) = abstractSoftware.notify(this, message)

    /**
     * 增加用户
     * @param person 用户
     */
    override def add(person: Person): Unit = {
        println(s"${name}您不是管理员，不具备增加用户权限")

    }

    /**
     * 判断是否为删除自己，如果是删除自己则为退群
     * @param person 用户
     */
    override def remove(person: Person): Unit = {
        if (person.eq(this)) {
            println(s"$name，您将退出${abstractSoftware.name}")
            abstractSoftware.removePerson(person)
        } else {
            println(s"${name}您不是管理员，不具备删除用户权限")
        }
    }

    /**
     * 私聊
     * @param person 接收者
     * @param message 信息
     */
    def privateChat(person: Person, message: String) = abstractSoftware.pm(this, person, message)
}



package com.linewell.modeldesgin.mediator

/**
* 测试客户端
* Created by ctao on 2015/9/1.
*/
object Client extends App {
    val admin: Person = new Admin("admin")
    val member1: Person = new Member("member1")
    val member2: Person = new Member("member2")
    val member3: Person = new Member("member3")
    val member4: Person = new Member("member4")

    val qqSoftware: AbstractSoftware = new QQSoftware("研发中心")
    admin.setAbstractSoftware(qqSoftware)
    member1.setAbstractSoftware(qqSoftware)
    member2.setAbstractSoftware(qqSoftware)
    member3.setAbstractSoftware(qqSoftware)
    member4.setAbstractSoftware(qqSoftware)

    admin.add(admin)
    admin.add(member1)
    admin.add(member2)
    admin.add(member3)
    admin.add(member4)
    admin.add(member1)

    admin.speak("hello")
    admin.remove(member1)

    member1.speak("hi")

    member2.add(member1)
    member2.remove(member2)

    member2.speak("admin")

    member3.privateChat(admin, "你好")
    member3.privateChat(member2, "你好")



    member2.privateChat(admin, "加我")

    println("------------------------------------------")

    val msnSoftware: AbstractSoftware = new MSNSoftware("通研事业部")
    admin.setAbstractSoftware(msnSoftware)
    member1.setAbstractSoftware(msnSoftware)
    member2.setAbstractSoftware(msnSoftware)
    member3.setAbstractSoftware(msnSoftware)
    member4.setAbstractSoftware(msnSoftware)


    admin.add(admin)
    admin.add(member1)
    admin.add(member2)
    admin.add(member3)
    admin.add(member4)
    admin.add(member1)

    admin.speak("hello")
    admin.remove(member1)

    member1.speak("hi")

    member2.add(member1)

    member2.speak("admin")

    member2.privateChat(member3,"test")

}