一、背景
来看一个项目需求:咖啡订购项目。
咖啡种类有很多:美式、摩卡、意大利浓咖啡;
咖啡加料:牛奶、豆浆、可可。
要求是,扩展新的咖啡种类的时候,能够方便维护,不同种类的咖啡需要快速计算多少钱,客户单点咖啡,也可以咖啡+料。
最差方案
直接想,就是一个咖啡基类,然后所有的单品、所有的组合咖啡都去继承这个基类,每个类都有自己的对应价格。
问题:那么多种咖啡和料的组合,都相当于是售卖的咖啡的一个子类,全都去实现基本就是一个全排列,显然又会类爆炸。并且,扩展起来,多一个调料,都要把所有咖啡种类算上重新组合一次。
改进方案
将调料内置到咖啡基类里,这样不会造成数量过多,当单品咖啡继承咖啡基类的时候,就都拥有了这些调料,同时,点没有点调料,要提供相应的方法,来计算是不是加了这个调料。
问题:这样的方式虽然改进了类爆炸的问题,但是属性内置导致了耦合性很强,如果删了一个调料呢?加了一个调料呢?每一个类都要改,维护量很大。
二、装饰者模式
装饰者模式:动态的将新功能附加到对象上,在对象功能扩展方面,比继承更有弹性。
具体实现起来是这样的,如下类图所示:
可以看到,在装饰者里面拥有一个 Component 对象,这是核心的部分。
也就是不像我们想的,给单品咖啡里加调料,而是反向思维,把单品咖啡拿到调料里来,决定对他的操作。
如果 ConcreteComponent 很多的话,甚至还可以再增加缓冲层。
用装饰者模式来解决上面的咖啡订单问题,类图设计如下,考虑到具体单品咖啡的种类,增加了一个缓冲层,最基本的抽象类叫 Drink:
其中 Drink 就相当于是前面的 Component,Coffee 是缓冲层,下面的不同 Coffee 就是上面的ConcreteConponent。
费用的计算方式一改正常思路的咖啡中,而是在调料中,因为 cost 在 Drink 类里也有,所以到最终的计算,其实是带上之前的 cost 结果,如果多种装饰者进行装饰,比如一个coffee加了很多料,那么其实是 递归的思路计算 最后的 cost 。
这样的话,增加一个单品咖啡,或者增加调料,都不用改变其他地方。
代码如下,类比较多,但是每个都比较简单:
/*
抽象类Drink,相当于Component;
getset方法提供给子类去设置饮品或调料的信息
但是:cost方法留给调料部分实现
*/
public abstract class Drink {
public String description;
private float price = 0.0f;
//价格方法
public abstract float cost();
public float getPrice() {
return price;
}
public void setPrice(float price) {
this.price = price;
}
public String getDescription() {
return description +":"+ price;
}
public void setDescription(String description) {
this.description = description;
}
}
接着就是Coffe缓冲层以及下面的实现类,相当于ConcreteComponent:
public class Coffee extends Drink{
@Override
public float cost() {
return super.getPrice();
}
}
public class MochaCoffee extends Coffee{
public MochaCoffee() {
setDescription(" 摩卡咖啡 ");
setPrice(7.0f);
}
}
public class USCoffee extends Coffee{
public USCoffee() {
setDescription(" 美式咖啡 ");
setPrice(5.0f);
}
}
public class ItalianCoffee extends Coffee {
public ItalianCoffee(){
setDescription(" 意大利咖啡 ");
setPrice(6.0f);
}
}
然后是装饰核心,Decorator,和Drink是继承+组合的关系:
本文转载自:https://www.gylmap.com
/*
Decorator,反客为主去拿已经有price的drink,并加上佐料
加佐料的时候是拿去了Drink对象,但是也是给Drink进行
*/
public class Decorator extends Drink{
private Drink drink;
//提供一个构造器
public Decorator(Drink drink){
this.drink = drink;
}
@Override
public float cost() {
//计算成本,拿到佐料自己的价格+本来一杯Drink的价格
//这里注意调用的是drink.cost不是drink.getPrice,因为cost才是子类实现的,Drink类的getPrice方法默认是返回0
return super.getPrice() + drink.cost();
}
@Override
public String getDescription() {
//自己的信息+被装饰者coffee的信息
return description + " " + getPrice() + " &&" + drink.getDescription();
}
}
以及Decorator的实现类,也就是ConcreteDecorator:
public class Milk extends Decorator{
public Milk(Drink drink) {
super(drink);
setDescription(" 牛奶:");
setPrice(1.0f);
}
}
public class Coco extends Decorator{
public Coco(Drink drink) {
super(drink);
setDescription(" 可可:");
setPrice(2.0f);//调味品价格
}
}
public class Sugar extends Decorator {
public Sugar(Drink drink) {
super(drink);
setDescription(" 糖:");
setPrice(0.5f);
}
}
注意,对于具体的Decorator,这里就体现了逆向思维,拿到的 drink 对象,调用父类构造器得到了一个drink,然后 set 和 get 方法设置调料自己的price和description,父类的方法 cost 就会计算价钱综合。
那里面的 super.getPrice() + drink.cost() 中的 cost(),就是一个递归的过程。
最后我们来写一个客户端测试:
public class Client {
public static void main(String[] args) {
//1.点一个咖啡,用Drink接受,因为还没有完成装饰
Drink usCoffee = new USCoffee();
System.out.println("费用:"+usCoffee.cost()+" 饮品信息:"+usCoffee.getDescription());
//2.加料
usCoffee = new Milk(usCoffee);
System.out.println("加奶后:"+usCoffee.cost()+" 饮品信息:"+usCoffee.getDescription());
//3.再加可可
usCoffee = new Coco(usCoffee);
System.out.println("加奶和巧克力后:"+usCoffee.cost()+" 饮品信息:"+usCoffee.getDescription());
}
}
可以看到,调用的时候,加佐料只要在原来的 drink 对象的基础上,重新构造,将原来的 drink 放进去包装(装饰),最后就达到了效果。
并且,如果要扩展一个类型的 coffee 或者一个调料,只用增加自己一个类就可以。
三、装饰者模式在 JDK 里的应用
java 的 IO 结构,FilterInputStream 就是一个装饰者。
2.1 这里面 InputStream 就相当于 Drink,也就是 Component 部分;
2.2 FileInputStream、StringBufferInputStream、ByteArrayInputStream 就相当于是单品咖啡,也就是ConcreteComponent,是 InputStream 的子类;
2.3 而 FilterInputStream 就相当于 Decorator,继承 InputStream 的同时又组合了InputStream;
2.4 BufferInputStream、DataInputStream、LineNumberInputStream 相当于具体的调料,是FilterInputStream的子类。
我们一般使用的时候:
DataInputStream dataInputStream = new DataInputStream(new FileInputStream("D://test.txt"));
或者:
FileInputStream fi = new FileInputStream("D:test.txt");
DataInputStream dataInputStream = new DataInputStream(fi);
//具体操作
这里面的 fi 就相当于单品咖啡, datainputStream 就是给他加了佐料。
更贴合上面咖啡的写法,声明的时候用 InputStream 接他,就可以:
InputStream fi = new FileInputStream("D:test.txt");
fi = new DataInputStream(fi);
//具体操作
感觉真是完全一样呢。