设计模式go语言实现----组合模式Composite

1 定义

组合模式(Composite):将对象组合成树形结构以以表示‘部分-整体’的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。

UML类图：

组合模式包含三个角色：

• Component:是组合中的对象声明接口，在适当的情况下，实现所有类共有接口的默认行为。声明一个接口用于访问和管理Component
• Leaf:在组合中表示叶子节点对象，叶子节点没有子节点
• Composite:定义有枝接节点行为，用来存储子部件，在Composite接口中实现与子部件有关操作

例如，大型公司就是一个典型的组合模式应用的例子，大型公司有总部，还在每个区设置有分部，有可能还在每个城市都设有办事处，而且每个分部或者办事处都具有和总部一样的公司架构。我们下面就以这个为模型实现golang的组合模式。
UML类图：

2 golang实现

2.1 实现ICompany

type ICompany interface {
	Add(ic ICompany)
	Remove(ic ICompany)
	Display(depth int)
	LineOfDuty()
}

2.2 实现ConcreteCompany

type ConcreteCompany struct {
	Name string
	List map[ICompany]ICompany
}

func NewConcreteCompany(name string) *ConcreteCompany {
	list := make(map[ICompany]ICompany)
	return &ConcreteCompany{
		Name: name,
		List: list,
	}
}

func (cc *ConcreteCompany) Add(ic ICompany) {
	cc.List[ic] = ic
}

func (cc *ConcreteCompany) Remove(ic ICompany) {
	delete(cc.List, ic)
}

func (cc *ConcreteCompany) Display(depth int) {
	fmt.Println(strings.Repeat("-", depth), " ", cc.Name)
	for _, ccc := range cc.List {
		ccc.Display(depth + 2)
	}
}

func (cc *ConcreteCompany) LineOfDuty() {
	for _, ccc := range cc.List {
		ccc.LineOfDuty()
	}
}

2.3 实现HRDepartment和FinanceDepartment

type HRDepartment struct {
	Name string
}

func NewHRDepartment(name string) *HRDepartment {
	return &HRDepartment{
		Name: name,
	}
}

func (hrd *HRDepartment) Add(ic ICompany) {
}

func (hrd *HRDepartment) Remove(ic ICompany) {
}

func (hrd *HRDepartment) Display(depth int) {
	fmt.Println(strings.Repeat("-", depth), " ", hrd.Name)
}

func (hrd *HRDepartment) LineOfDuty() {
	fmt.Println(hrd.Name, "员工招聘培训管理")
}

type FinanceDepartment struct {
	Name string
}

func NewFinanceDepartment(name string) *FinanceDepartment {
	return &FinanceDepartment{
		Name: name,
	}
}

func (fd *FinanceDepartment) Add(ic ICompany) {
}

func (fd *FinanceDepartment) Remove(ic ICompany) {
}

func (fd *FinanceDepartment) Display(depth int) {
	fmt.Println(strings.Repeat("-", depth), " ", fd.Name)
}

func (fd *FinanceDepartment) LineOfDuty() {
	fmt.Println(fd.Name, "公司财务收支管理")
}

2.4 客户端调用

func main() {
	root := composite.NewConcreteCompany("北京总公司")
	root.Add(composite.NewHRDepartment("总公司人力资源部"))
	root.Add(composite.NewFinanceDepartment("总公司财务"))

	com := composite.NewConcreteCompany("上海华东分公司")
	com.Add(composite.NewHRDepartment("上海华东分公司人力资源部"))
	com.Add(composite.NewFinanceDepartment("上海华东分公司财务"))
	root.Add(com)

	com1 := composite.NewConcreteCompany("南京办事处")
	com1.Add(composite.NewHRDepartment("南京办事处人力资源部"))
	com1.Add(composite.NewFinanceDepartment("南京办事处财务"))
	com.Add(com1)

	com2 := composite.NewConcreteCompany("杭州办事处")
	com2.Add(composite.NewHRDepartment("杭州办事处人力资源部"))
	com2.Add(composite.NewFinanceDepartment("杭州办事处财务"))
	com.Add(com2)

	fmt.Println("结构图:")
	root.Display(1)
	fmt.Println("职责:")
	root.LineOfDuty()
}