设计模式之装饰器模式

装饰器模式： 动态的将责任附加到对象上。若要扩展功能，装饰器提供了比继承更有弹性的替代方案。

装饰器中有两个比较重要的角色，装饰者和被装饰者。被装饰者实现我们的核心逻辑，装饰者只是对这些逻辑进行增强。比如点一杯饮料，冰，加豆浆等都是装饰者，是对我们饮料的增强，而饮料 是被装饰者。

装饰器模式UML

装饰器UML

从UML中可以看到装饰者和被装饰者实现同一接口。装饰者中有一个被装饰者的对象引用。被装饰者的行为增强是通过旧逻辑前后添加新逻辑来实现的。

下面以一个饮料费用计算为例，来简单介绍装饰者模式的使用。购买一杯饮料的时候，可以选择添加不同的材料，价格也随之变化。这种情况我们就可以使用装饰者模式实现对扩展开放，对修改关闭。

饮料抽象类

public abstract class Beverage {

    String description = "Unknown Beverage";

    public String getDescription(){
        return description;
    }

    abstract double cost();
}

被装饰者

public class Espresso extends Beverage{

    public Espresso() {
        description = "Espresso";
    }

    @Override
    double cost() {
        return 10;
    }
}

public class HouseBlend  extends Beverage{

    public HouseBlend() {
        description = "House Blend Coffee";
    }

    @Override
    double cost() {
        return 20;
    }
}

装饰则抽象类

public abstract class CondimentDecorator extends Beverage{
    public abstract String getDescription();
}

装饰者

public class Mocha extends CondimentDecorator{

    private Beverage beverage;

    public Mocha(Beverage beverage) {
        this.beverage = beverage;
    }

    @Override
    public String getDescription() {
        return beverage.getDescription() + ", Mocha";
    }

    @Override
    double cost() {
        return 2+beverage.cost();
    }
}

public class Soy extends CondimentDecorator {
   private Beverage beverage;

    public Soy(Beverage beverage) {
        this.beverage = beverage;
    }

    @Override
    public String getDescription() {
        return beverage.getDescription() + ", Soy";
    }

    @Override
    double cost() {
        return 0.5 +beverage.cost();
    }
}

UML如下所示

image.png

执行

public class Main {

    public static void main(String[] args) {
      Espresso espresso = new Espresso();
      Mocha mocha = new Mocha(espresso);
      Soy soy = new Soy(mocha);
      System.out.println(soy.getDescription());
      System.out.println(soy.cost());

      HouseBlend houseBlend = new HouseBlend();
      Mocha mocha1 = new Mocha(houseBlend);
      Soy soy1 = new Soy(mocha1);
      System.out.println(soy1.getDescription());
      System.out.println(soy1.cost());

    }
}

结果

Espresso, Mocha, Soy
12.5
House Blend Coffee, Mocha, Soy
22.5

在没有修改任何代码的情况下就可以实现对原有功能的增强。且代码有很高的复用性。