设计模式 - 访问者模式(Visitor)

Visitor模式表示一个作用于某对象结构中的各元素的操作。它使你可以在不改变各元素的类的前提下定义作用于这些元素的新操作。

类图：

• 一个对象结构包含很多类对象，它们有不同的接口，而你想对这些对象实施一些依赖于其具体类的操作。
• 需要对一个对象结构中的对象进行很多不同的并且不相关的操作，而你想避免让这些操作“污染”这些对象的类。Visitor使得你可以将相关的操作集中起来定义在一个类中。
• 定义对象结构的类很少改变，但经常需要在此结构上定义新的操作。

场景：一个方法从不同的类中聚集信息，把聚集工作搬移到一个能够访问每个类以便聚集信息的Visitor中。

示例： 购物车里有不同的产品（书，水果）， 购物车类提供一个功能就是分别统计不同产品的费用，并打印出信息，但书和水果类没有统一的接口统计费用和打印信息。

public interface Item {
    public void accept(Visitor visitor);
}

public class Book implements Item {
    private int price;
    private String isbnNum;
    private String name;

    public Book(String isbn, String name, int price) {
        this.price   = price;
        this.isbnNum = isbn;
        this.name    = name;
    }
    public int getPrice() {
        return price;
    }
    public String getIsbnNum() {
        return isbnNum;
    }
    public String getName() {
        return name;
    }
    public void accept(Visitor visitor) {
        visitor.visit(this);
    }
}

public class Fruit implements Item {
    private int pricePerKg;
    private int weight;
    private String name;

    public Fruit(String name, int priceKg, int wt){
        this.pricePerKg = priceKg;
        this.weight     = wt;
        this.name       = name;
    }
    public int getPricePerKg() {
        return pricePerKg;
    }
    public int getWeight() {
        return weight;
    }
    public String getName(){
        return this.name;
    }
    public void accept(Visitor visitor) {
        visitor.visit(this);
    }
}

public interface Visitor {
    void visit(Book book);
    void visit(Fruit fruit);
}

public class ShoppingCartVisitor implements Visitor {
    private StringBuilder info = new StringBuilder();//聚集信息
    public void visit(Book book) {
        int cost = 0;
        // apply 5$ discount if book price is greater than 50
        if (book.getPrice() > 50) {
            cost = book.getPrice() - 5;
        } else {
            cost = book.getPrice();
    	}
        info.append("Book ISBN::" + book.getIsbnNum() + " cost = " + cost+"\n");//聚集信息
    }

    public void visit(Fruit fruit) {
        int cost = fruit.getPricePerKg() * fruit.getWeight();
        info.append(fruit.getName() + " cost = " + cost + "\n");//聚集信息
    }
    
    public String getInfo() {
    	return info.toString();
    }
}

public class ShoppingCart {
    private Item[] items = new Item[] { new Book("1234", "Design Patterns", 38),
            new Book("5678", "Refactoring", 100), new Fruit("Banana", 10, 2), new Fruit("Apple", 5, 5) };

    public void printCartInfo() {
    	ShoppingCartVisitor visitor = new ShoppingCartVisitor();
        for(Item item : items){
            item.accept(visitor); //通过每个对象的访问者来聚集信息
        }
        System.out.println(visitor.getInfo()); //从访问者获取聚集信息
    }
}

测试代码

public class ShoppingCartTest {
	public static void main(String[] args) {
	    ShoppingCart cart =  new ShoppingCart();
	    cart.printCartInfo();
	}
}
//输出
Book ISBN::1234 cost = 38
Book ISBN::5678 cost = 95
Banana cost = 20
Apple cost = 25