77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
package main
|
|
|
|
import "fmt"
|
|
|
|
/*
|
|
实例
|
|
汤宝宝成为了上市集团老总，他们集团有好多好多家分公司...
|
|
*/
|
|
|
|
// 抽象接口
|
|
type Company interface {
|
|
// 显示公司名称
|
|
Show(prefix string)
|
|
// 开设分公司
|
|
Add(company Company)
|
|
// 分公司关门
|
|
Remove(company Company)
|
|
// 获取分公司
|
|
GetChild(i int) Company
|
|
}
|
|
|
|
// 具体实现类
|
|
type ConcreteCompany struct {
|
|
name string
|
|
children []Company
|
|
}
|
|
|
|
func (c *ConcreteCompany) Show(prefix string) {
|
|
fmt.Print(prefix)
|
|
fmt.Println(c.name)
|
|
for _, child := range c.children {
|
|
child.Show(prefix + "-> ")
|
|
}
|
|
}
|
|
|
|
func (c *ConcreteCompany) Add(company Company) {
|
|
c.children = append(c.children, company)
|
|
fmt.Printf("在%s下开设了分公司%s\n", c.name, company.(*ConcreteCompany).name)
|
|
}
|
|
|
|
func (c *ConcreteCompany) Remove(company Company) {
|
|
for i, child := range c.children {
|
|
if child == company {
|
|
c.children = append(c.children[:i], c.children[i+1:]...)
|
|
break
|
|
}
|
|
}
|
|
fmt.Printf("在%s下关闭了分公司%s\n", c.name, company.(*ConcreteCompany).name)
|
|
}
|
|
|
|
func (c *ConcreteCompany) GetChild(i int) Company {
|
|
if i < 0 || i >= len(c.children) {
|
|
return nil
|
|
}
|
|
return c.children[i]
|
|
}
|
|
|
|
func main() {
|
|
fmt.Println("============= 组合模式 =============")
|
|
|
|
headOffice := &ConcreteCompany{name: "汤氏总公司"}
|
|
headOffice.Add(&ConcreteCompany{name: "汤氏A分公司"})
|
|
headOffice.Add(&ConcreteCompany{name: "汤氏B分公司"})
|
|
headOffice.Add(&ConcreteCompany{name: "汤氏C分公司"})
|
|
headOffice.GetChild(0).Add(&ConcreteCompany{name: "汤氏A分公司-子公司1"})
|
|
headOffice.GetChild(0).Add(&ConcreteCompany{name: "汤氏A分公司-子公司2"})
|
|
headOffice.Remove(headOffice.GetChild(2))
|
|
fmt.Println()
|
|
headOffice.Show("~ ")
|
|
fmt.Println()
|
|
headOffice.GetChild(0).Show("~ ")
|
|
fmt.Println()
|
|
|
|
fmt.Println("=========== 组合模式结束 ===========")
|
|
|
|
}
|