重构组合模式示例，统一接口定义，增强代码可读性和一致性

go/composite/main.go

@@ -7,25 +7,30 @@ import "fmt"
 汤宝宝成为了上市集团老总，他们集团有好多好多家分公司...
 */
 
-// 抽象接口
-type Company interface {
-	// 显示公司名称
+type component interface {
 	Show(prefix string)
-	// 开设分公司
-	Add(company Company)
-	// 分公司关门
-	Remove(company Company)
-	// 获取分公司
-	GetChild(i int) Company
+	getName() string
 }
 
-// 具体实现类
-type ConcreteCompany struct {
+type Department struct {
 	name string
-	children []Company
 }
 
-func (c *ConcreteCompany) Show(prefix string) {
+func (d *Department) Show(prefix string) {
+	fmt.Print(prefix)
+	fmt.Println(d.name)
+}
+
+func (d *Department) getName() string {
+	return d.name
+}
+
+type Company struct {
+	name     string
+	children []component
+}
+
+func (c *Company) Show(prefix string) {
 	fmt.Print(prefix)
 	fmt.Println(c.name)
 	for _, child := range c.children {
@@ -33,22 +38,26 @@ func (c *ConcreteCompany) Show(prefix string) {
 	}
 }
 
-func (c *ConcreteCompany) Add(company Company) {
-	c.children = append(c.children, company)
-	fmt.Printf("在%s下开设了分公司%s\n", c.name, company.(*ConcreteCompany).name)
+func (c *Company) getName() string {
+	return c.name
 }
 
-func (c *ConcreteCompany) Remove(company Company) {
+func (c *Company) Add(company component) {
+	c.children = append(c.children, company)
+	fmt.Printf("在%s下开设了%s\n", c.name, company.getName())
+}
+
+func (c *Company) Remove(company component) {
 	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)
+	fmt.Printf("在%s下裁撤了%s\n", c.name, company.getName())
 }
 
-func (c *ConcreteCompany) GetChild(i int) Company {
+func (c *Company) GetChild(i int) component {
 	if i < 0 || i >= len(c.children) {
 		return nil
 	}
@@ -58,12 +67,12 @@ func (c *ConcreteCompany) GetChild(i int) Company {
 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 := &Company{name: "汤氏总公司"}
+	headOffice.Add(&Company{name: "汤氏A分公司"})
+	headOffice.Add(&Department{name: "人事部"})
+	headOffice.Add(&Department{name: "研发部"})
+	headOffice.GetChild(0).(*Company).Add(&Department{name: "市场部"})
+	headOffice.GetChild(0).(*Company).Add(&Department{name: "财务部"})
 	headOffice.Remove(headOffice.GetChild(2))
 	fmt.Println()
 	headOffice.Show("~ ")

python/composite/main.py

@@ -6,32 +6,31 @@
 from abc import ABC, abstractmethod
 
 
-# 抽象类
-class Company(ABC):
+# 接口
+class component(ABC):
     name: str
-    children: list["Company"]
 
     @abstractmethod
     def show(self, prefix: str) -> None:
         pass
 
-    @abstractmethod
-    def add(self, company: "Company") -> None:
-        pass
 
-    @abstractmethod
-    def remove(self, company: "Company") -> None:
-        pass
-
-    @abstractmethod
-    def get_child(self, index: int) -> "Company":
-        pass
-
-
-# 实现类
-class ConcreteCompany(Company):
+# 具体类
+class Department(component):
     name: str
-    children: list[Company]
+
+    def __init__(self, name: str) -> None:
+        self.name = name
+
+    def show(self, prefix: str) -> None:
+        print(f"{prefix}{self.name}")
+
+    # 实现类
+
+
+class Company(component):
+    name: str
+    children: list[component]
 
     def __init__(self, name: str) -> None:
         self.name = name
@@ -42,15 +41,15 @@ class ConcreteCompany(Company):
         for child in self.children:
             child.show(f"{prefix}-> ")
 
-    def add(self, company: Company) -> None:
+    def add(self, company: component) -> None:
         self.children.append(company)
-        print(f"在{self.name}下开设了分公司{company.name}")
+        print(f"在{self.name}下开设了{company.name}")
 
-    def remove(self, company: Company) -> None:
+    def remove(self, company: component) -> None:
         self.children.remove(company)
-        print(f"在{self.name}下关闭了分公司{company.name}")
+        print(f"在{self.name}下裁撤了{company.name}")
 
-    def get_child(self, index: int) -> Company:
+    def get_child(self, index: int) -> component:
         if index < 0 or index >= len(self.children):
             raise IndexError("子公司索引超出范围")
         return self.children[index]
@@ -59,12 +58,14 @@ class ConcreteCompany(Company):
 if __name__ == "__main__":
     print("============= 组合模式 =============")
 
-    head_office = ConcreteCompany("汤氏总公司")
-    head_office.add(ConcreteCompany("汤氏A分公司"))
-    head_office.add(ConcreteCompany("汤氏B分公司"))
-    head_office.add(ConcreteCompany("汤氏C分公司"))
-    head_office.get_child(0).add(ConcreteCompany("汤氏A分公司-子公司1"))
-    head_office.get_child(0).add(ConcreteCompany("汤氏A分公司-子公司2"))
+    head_office = Company("汤氏总公司")
+    head_office.add(Company("汤氏A分公司"))
+    head_office.add(Department("人事部"))
+    head_office.add(Department("研发部"))
+    child = head_office.get_child(0)
+    if isinstance(child, Company):
+        child.add(Department("市场部"))
+        child.add(Department("财务部"))
     head_office.remove(head_office.get_child(2))
     print("\n")
     head_office.show("~ ")

ts/src/composite/index.ts

@@ -3,16 +3,22 @@
 汤宝宝成为了上市集团老总，他们集团有好多好多家分公司...
 */
 
-interface Company {
+interface component {
   name: string;
-  children: Company[];
   show(prefix: string): void;
-  add(company: Company): void;
-  remove(company: Company): void;
-  getChild(index: number): Company | undefined;
 }
 
-class ConcreteCompany implements Company {
+class Department implements component {
+  name: string;
+  constructor(name: string) {
+    this.name = name;
+  }
+  show(prefix: string): void {
+    console.log(prefix + this.name);
+  }
+}
+
+class Company implements component {
   name: string;
   children: Company[] = [];
   constructor(name: string) {
@@ -24,18 +30,18 @@ class ConcreteCompany implements Company {
       child.show(prefix + "-> ");
     }
   }
-  add(company: Company): void {
-    this.children.push(company);
-    console.log(`在${this.name}下开设了分公司${company.name}`);
+  add(company: component): void {
+    this.children.push(company as Company);
+    console.log(`在${this.name}下开设了${company.name}`);
   }
-  remove(company: Company): void {
-    const index = this.children.indexOf(company);
+  remove(company: component): void {
+    const index = this.children.indexOf(company as Company);
     if (index !== -1) {
       this.children.splice(index, 1);
     }
-    console.log(`在${this.name}下关闭了分公司${company.name}`);
+    console.log(`在${this.name}下裁撤了${company.name}`);
   }
-  getChild(index: number): Company | undefined {
+  getChild(index: number): component {
     return this.children[index];
   }
 }
@@ -43,12 +49,15 @@ class ConcreteCompany implements Company {
 (function () {
   console.log("============= 组合模式 =============");
 
-  const headOffice = new ConcreteCompany("总公司");
-  headOffice.add(new ConcreteCompany("汤氏A分公司"));
-  headOffice.add(new ConcreteCompany("汤氏B分公司"));
-  headOffice.add(new ConcreteCompany("汤氏C分公司"));
-  headOffice.getChild(0)?.add(new ConcreteCompany("汤氏A分公司-子公司1"));
-  headOffice.getChild(0)?.add(new ConcreteCompany("汤氏A分公司-子公司2"));
+  const headOffice = new Company("总公司");
+  headOffice.add(new Company("汤氏A分公司"));
+  headOffice.add(new Department("人事部"));
+  headOffice.add(new Department("研发部"));
+  const child = headOffice.getChild(0);
+  if (child instanceof Company) {
+    child.add(new Department("市场部"));
+    child.add(new Department("财务部"));
+  }
   headOffice.remove(headOffice.getChild(2)!);
 
   console.log();