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()
}