设计模式18 - 访问者模式【Visitor Pattern】

原创

阿甘兄_ 2022-03-25 10:48:01 ©著作权

文章标签 访问者 ide 数据结构 文章分类 代码人生

©著作权归作者所有：来自51CTO博客作者阿甘兄_的原创作品，请联系作者获取转载授权，否则将追究法律责任

访问者模式

定义：

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

举例：（记账例子）：

账本只有收入和消费类型，而访问者有多种，如老板和会计。

/**
 * 账单接口，有接受访问者的功能
 */
interface Bill {
    void accept(AccountBookView viewer);
}

/**
*消费账单
*/
class ConsumerBill implements Bill {
    private String item;
    private double amount;

    public ConsumerBill(String item, double amount) {
        this.item = item;
        this.amount = amount;
    }

    @Override
    public void accept(AccountBookView viewer) {
        viewer.view(this);
    }

    public String getItem() {
        return item;
    }

    public double getAmount() {
        return amount;
    }
}

/*
*支出账单
*/
class IncomingBill implements Bill {
    private String item;
    private double amount;

    public IncomingBill(String item, double amount) {
        this.item = item;
        this.amount = amount;
    }

    @Override
    public void accept(AccountBookView viewer) {
        viewer.view(this);
    }

    public String getItem() {
        return item;
    }

    public double getAmount() {
        return amount;
    }
}


/**
*访问者接口
*/
interface AccountBookView {
    //查看消费的单子
    void view(ConsumerBill consumerBill);

    //查看收入单子
    void view(IncomingBill incomingBill);
}


/**
*访问者老板
*/
class Boss implements AccountBookView {

    private double totalConsumer;
    private double totalIncome;


    @Override
    public void view(ConsumerBill consumerBill) {
        totalConsumer = totalConsumer + consumerBill.getAmount();
    }

    @Override
    public void view(IncomingBill incomingBill) {
        totalIncome = totalIncome + incomingBill.getAmount();
    }

    public void getTotalConsumer() {
        System.out.println("老板一共消费了" + totalConsumer);
    }

    public void getTotalIncome() {
        System.out.println("老板一共收入了" + totalIncome);
    }

}

/**
*访问者会计
*/
class CPA implements AccountBookView {

    int count = 0;

    // 查看消费的单子
    public void view(ConsumerBill consumerBill) {
        count++;
        if (consumerBill.getItem().equals("消费")) {
                    System.out.println("第" + count + "个单子消费了：" + consumerBill.getAmount());
        }
    }

    // 查看收入单子

    public void view(IncomingBill incomeBill) {

        if (incomeBill.getItem().equals("收入")) {
            System.out.println("第" + count + "个单子收入了：" + incomeBill.getAmount());
        }
    }

}

/**
*账单类：用于添加账单，和为每一个账单添加访问者
*/
class AccountBook {

    private List<Bill> listBill = new ArrayList<Bill>();

    // 添加单子
    public void add(Bill bill) {
        listBill.add(bill);
    }

    // 为每个账单添加访问者
    public void show(AccountBookView viewer) {
        for (Bill b : listBill) {
            b.accept(viewer);
        }
    }
}
/**
* 客户端
*/
public class Client{

    public void main(String[] args) {

        // 创建消费和收入单子
        Bill consumerBill = new ConsumerBill("消费", 3000);
        Bill incomeBill = new IncomingBill("收入", 5000);
        Bill consumerBill2 = new ConsumerBill("消费", 4000);
        Bill incomeBill2 = new IncomingBill("收入", 8000);

        // 添加单子
        AccountBook accountBook = new AccountBook();
        accountBook.add(consumerBill);
        accountBook.add(incomeBill);
        accountBook.add(consumerBill2);
        accountBook.add(incomeBill2);

        // 创建访问者
        AccountBookView boss = new Boss();
        AccountBookView cpa = new CPA();

        // 接受访问者
        accountBook.show(boss);
        accountBook.show(cpa);

        // boss查看总收入和总消费
        ((Boss) boss).getTotalConsumer();
        ((Boss) boss).getTotalIncome();
    }
}