Go语言设计模式实战：中介者模式详解

package main

import (
	"fmt"
	"sync"
)

// 火车类
type Train struct {
	number   string
	mediator *Mediator
}

// 请求进站
func (t *Train) RequestGetIn() {
	fmt.Println(t.number + "请求进站")
	if t.mediator.canGetIn(t) {
		fmt.Println("mediator:" + t.number + "可以进站停靠")
	} else {
		fmt.Println("mediator:" + t.number + "不能进站，请等待通知")
	}
}

// 出站
func (t *Train) Departure() {
	fmt.Println(t.number + "出站")
	// 中介者通知其他火车可以进站了
	t.mediator.notifyFree()
}

// 进站
func (t *Train) GetIn() {
	fmt.Println(t.number + "已进站")
}

// 中介者类
type Mediator struct {
	isPlatformFree bool
	lock           *sync.Mutex
	trainQueue     []Train
}

// 创建中介者
func newMediator() *Mediator {
	return &Mediator{
		isPlatformFree: true,
		lock:           &sync.Mutex{},
	}
}

// 判断火车能否进站
func (m *Mediator) canGetIn(t *Train) bool {
	m.lock.Lock()
	defer m.lock.Unlock()
	if m.isPlatformFree {
		m.isPlatformFree = false
		return true
	}
	m.trainQueue = append(m.trainQueue, *t)
	return false
}

// 通知火车进站
func (m *Mediator) notifyFree() {
	m.lock.Lock()
	defer m.lock.Unlock()
	if !m.isPlatformFree {
		m.isPlatformFree = true
		fmt.Println("mediator:" + "站台已空闲")
	}
	if len(m.trainQueue) > 0 {
		firstTrainInQueue := m.trainQueue[0]
		m.trainQueue = m.trainQueue[1:]

		fmt.Println("mediator:" + "允许排队中的火车进站")
		firstTrainInQueue.GetIn()
	}
}

func main() {
	m := newMediator()

	// 火车 K6801
	K6801 := Train{
		number:   "K6801",
		mediator: m,
	}

	// 火车 K6308
	K6308 := Train{
		number:   "K6308",
		mediator: m,
	}

	K6308.RequestGetIn()
	K6801.RequestGetIn()
	K6308.Departure()
}

K6308请求进站
mediator:K6308可以进站停靠
K6801请求进站
mediator:K6801不能进站，请等待通知
K6308出站
mediator:站台已空闲
mediator:允许排队中的火车进站
K6801已进站