组合模式

组合模式

职务接口

package composite

type jobber interface {
	setJob(string)
	setManager(jobber)
	addSubordinate(jobber)
	print(string)
}

职务节点

package composite

type worker struct {
	job     string
	manager jobber
}

func (w *worker) setJob(job string) {
	w.job = job
}

func (w *worker) setManager(mgr jobber) {
	w.manager = mgr
}

func (w *worker) addSubordinate(subordinate jobber) {}

func (w *worker) Print(str string) {}

管理职务

package composite

import "fmt"

type composite struct {
	worker
	subordinate []jobber
}

func NewComposite(job string) jobber {
	m := new(composite)
	m.job = job
	m.subordinate = make([]jobber, 0)
	return m
}

func (c *composite) addSubordinate(subordinate jobber) {
	subordinate.setManager(c)
	c.subordinate = append(c.subordinate, subordinate)
}

func (c *composite) print(str string) {
	fmt.Printf("%s %s\n", str, c.job)
	str += "--"
	for _, v := range c.subordinate {
		v.print(str)
	}
}

一般职务

package composite

import "fmt"

type leaf struct {
	worker
}

func NewLeaf(job string) jobber {
	l := new(leaf)
	l.job = job
	return l
}

func (l *leaf) print(str string) {
	fmt.Printf("%s %s\n", str, l.job)
}

测试文件

package composite

import "testing"

func TestNewComposite(t *testing.T) {
	m1 := NewComposite("总经理")
	m2 := NewLeaf("秘书")
	m3 := NewComposite("财务经理")
	m4 := NewLeaf("会计")
	m5 := NewLeaf("出纳")
	m6 := NewComposite("研发经理")
	m7 := NewComposite("前端经理")
	m8 := NewLeaf("美工")
	m9 := NewComposite("后端经理")
	m10 := NewLeaf("程序员")

	m1.addSubordinate(m2)
	m1.addSubordinate(m3)
	m1.addSubordinate(m6)
	m3.addSubordinate(m4)
	m3.addSubordinate(m5)
	m6.addSubordinate(m7)
	m6.addSubordinate(m9)
	m7.addSubordinate(m8)
	m9.addSubordinate(m10)

	m1.print("")
}