组合模式（Composite）-CSDN博客

基本概念

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

component为组合中的对象声明接口，在适当情况下，实现所有类共有接口的默认行为。

leaf在组合中表示叶节点对象，叶节点没有子节点。

composite定义有子节点，用来存储字部分，在component接口中实现与字部分有关的操作。

代码实现

公司类抽象

@interface Company : NSObject

@property (nonatomic, copy)NSString *name;

@property (nonatomic, strong)NSMutableArray *childrenArray;

- (instancetype)initWithName:(NSString *)name;

- (void)add:(Company *)company;

- (void)remove:(Company *)company;

- (void)display:(int)depth;//显示

- (void)lineOfDuty;// 履行职责

#import "Company.h"

@interface Company()

@end

@implementation Company

- (instancetype)initWithName:(NSString *)name

{

self = [super init];

if (self) {

self.name = name;

self.childrenArray = [NSMutableArray array];

}

return self;

}

具体公司类

#import "ConcreteCompany.h"

@implementation ConcreteCompany

- (void)add:(Company *)company

{

[self.childrenArray addObject:company];

}

- (void)remove:(Company *)company

{

if ([self.childrenArray containsObject:company]) {

[self.childrenArray removeObject:company];

}

}

- (void)display:(int)depth

{

NSLog(@"%d-%@",depth,self.name);

for (Company *component in self.childrenArray) {

[component display:(depth + 2)];

}

}

- (void)lineOfDuty

{

for (Company *component in self.childrenArray) {

[component lineOfDuty];

}

}

@end

人力资源部和财务部

（1）人力

#import "HRDepartment.h"

@implementation HRDepartment

- (void)add:(Company *)company

{

}

- (void)remove:(Company *)company

{

}

- (void)display:(int)depth

{

NSLog(@"%d-%@",depth,self.name);

}

- (void)lineOfDuty

{

NSLog(@"%@ 员工招聘培训管理",self.name);

}

@end

（财务）

#import "FinanceDepartment.h"

@implementation FinanceDepartment

- (void)add:(Company *)company

{

}

- (void)remove:(Company *)company

{

}

- (void)display:(int)depth

{

NSLog(@"%d-%@",depth,self.name);

}

- (void)lineOfDuty

{

NSLog(@"%@ 公司财务收支管理",self.name);

}

@end

客户端调用

ConcreteCompany *root = [[ConcreteCompany alloc] initWithName:@"北京总公司"];

[root add:[[HRDepartment alloc]initWithName:@"总公司人力资源部"]];

[root add:[[FinanceDepartment alloc]initWithName:@"总公司财务部"]];

ConcreteCompany *comp = [[ConcreteCompany alloc] initWithName:@"上海华东分公司"];

[comp add:[[HRDepartment alloc]initWithName:@"华东分公司人力资源部"]];

[comp add:[[FinanceDepartment alloc]initWithName:@"华东分公司财务部"]];

[root add:comp];

ConcreteCompany *comp1 = [[ConcreteCompany alloc] initWithName:@"南京办事处"];

[comp1 add:[[HRDepartment alloc]initWithName:@"南京办事处人力资源部"]];

[comp1 add:[[FinanceDepartment alloc]initWithName:@"南京办事处财务部"]];

[comp add:comp1];

ConcreteCompany *comp2 = [[ConcreteCompany alloc] initWithName:@"杭州办事处"];

[comp2 add:[[HRDepartment alloc]initWithName:@"杭州办事处人力资源部"]];

[comp2 add:[[FinanceDepartment alloc]initWithName:@"杭州办事处财务部"]];

[comp add:comp2];

[root display:1];

[root lineOfDuty];

输出

2017-04-28 10:56:51.191 PlayTry[1942:63351] 1-北京总公司

2017-04-28 10:56:51.191 PlayTry[1942:63351] 3-总公司人力资源部

2017-04-28 10:56:51.192 PlayTry[1942:63351] 3-总公司财务部

2017-04-28 10:56:51.192 PlayTry[1942:63351] 3-上海华东分公司

2017-04-28 10:56:51.192 PlayTry[1942:63351] 5-华东分公司人力资源部

2017-04-28 10:56:51.193 PlayTry[1942:63351] 5-华东分公司财务部

2017-04-28 10:56:51.193 PlayTry[1942:63351] 5-南京办事处

2017-04-28 10:56:51.193 PlayTry[1942:63351] 7-南京办事处人力资源部

2017-04-28 10:56:51.193 PlayTry[1942:63351] 7-南京办事处财务部

2017-04-28 10:56:51.194 PlayTry[1942:63351] 5-杭州办事处

2017-04-28 10:56:51.194 PlayTry[1942:63351] 7-杭州办事处人力资源部

2017-04-28 10:56:51.194 PlayTry[1942:63351] 7-杭州办事处财务部

2017-04-28 10:56:51.195 PlayTry[1942:63351] 总公司人力资源部员工招聘培训管理

2017-04-28 10:56:51.195 PlayTry[1942:63351] 总公司财务部公司财务收支管理

2017-04-28 10:56:51.196 PlayTry[1942:63351] 华东分公司人力资源部员工招聘培训管理

2017-04-28 10:56:51.222 PlayTry[1942:63351] 华东分公司财务部公司财务收支管理

2017-04-28 10:56:51.224 PlayTry[1942:63351] 南京办事处人力资源部员工招聘培训管理

2017-04-28 10:56:51.224 PlayTry[1942:63351] 南京办事处财务部公司财务收支管理

2017-04-28 10:56:51.226 PlayTry[1942:63351] 杭州办事处人力资源部员工招聘培训管理

2017-04-28 10:56:51.226 PlayTry[1942:63351] 杭州办事处财务部公司财务收支管理