C++20 访问者模式:std::variant, std::visit, if conexpr,奇异递归模板

 访问者模式原始版

#include <iostream>
#include "common/log.h"
#include <limits>
#include <vector>
#include <span>
#include <array>
#include <type_traits>
#include <cmath>
#include <memory>
using namespace AdsonLib;

//访问者模式:Element的种类固定,但是操作Element的方法经常变
//如果在Element中增加操作方法,需要经常修改Element,为此把操作
//方法封装到Visitor中,扩展Visitor方便。
struct Element;
struct ElementA;
struct ElementB;

struct Visitor {
    virtual void VisitElementA(ElementA &a) = 0;
    virtual void VisitElementB(ElementB &b) = 0;
    virtual ~Visitor(){}
};

struct Element {
    virtual void accept(Visitor &vis) = 0;
    virtual ~Element(){}
    int val = 1;
};
struct ElementA : public Element{
    virtual void accept(Visitor &vis) override {
        vis.VisitElementA(*this);
    }
};
struct ElementB  : public Element{
    virtual void accept(Visitor &vis) override {
        vis.VisitElementB(*this);
    }
};

struct VisitorA : public Visitor {
    int sum = 1;
    virtual void VisitElementA(ElementA &a) override {
        LOG(INFO) << "VisitorA vis ElementA: " << a.val;
        sum += a.val;
    }
    virtual void VisitElementB(ElementB &b) override {
        LOG(INFO) << "VisitorA vis ElementB: " << b.val;
        sum *=b.val;
    }
    int total() {
        return sum;
    }
};
struct VisitorB : public Visitor {
    int mul = 1;
    virtual void VisitElementA(ElementA &a) override {
        LOG(INFO) << "VisitorA vis ElementA: " << a.val;
        mul *= a.val;
    }
    virtual void VisitElementB(ElementB &b) override {
        LOG(INFO) << "VisitorA vis ElementB: " << b.val;
        mul += b.val;
    }
    int multi() {
        return mul;
    }
};

struct ObjectStructure{
    ObjectStructure(){
        ElementA a1;
        a1.val = 1;
        objs.push_back(std::make_shared<ElementA>(a1));
        a1.val = 2;
        objs.push_back(std::make_shared<ElementA>(a1));
        ElementB b1;
        b1.val = 3;
        objs.push_back(std::make_shared<ElementB>(b1));
        b1.val = 4;
        objs.push_back(std::make_shared<ElementB>(b1));
    }
    void calc(Visitor &vis){
        for(auto && e : objs) {
            e->accept(vis);
        }
    }
    std::vector<std::shared_ptr<Element>> objs;
};

void common_test(){
    ObjectStructure os;
    VisitorA VA;
    VisitorB VB;
    os.calc(VA);
    LOG(INFO) << "visit by VA: " << VA.total();
    os.calc(VB);
    LOG(INFO) << "visit by VB: " << VB.multi();
}
int main(int argc, char *argv[]) {
   common_test();
}

奇异模板递归,把dispatch放到父类

template<typename T>
struct Base{};

//子类传递到父类中了,父类就能用子类类型了,这就是奇异模板递归
struct Der:public Base<Der>{}; 
#include <iostream>
#include "common/log.h"
#include <limits>
#include <vector>
#include <span>
#include <array>
#include <type_traits>
#include <cmath>
#include <memory>
using namespace AdsonLib;

//访问者模式:Element的种类固定,但是操作Element的方法经常变
//如果在Element中增加操作方法,需要经常修改Element,为此把操作
//方法封装到Visitor中,扩展Visitor方便。
struct Element;
struct ElementA;
struct ElementB;

struct Visitor {
    virtual void Visit(ElementA &a) = 0;
    virtual void Visit(ElementB &b) = 0;
    virtual ~Visitor(){}
};

struct Element {
    virtual void accept(Visitor &vis)  = 0;
    virtual ~Element(){}
    int val = 1;
};
template<typename T>
struct AutoDispatchElement : public Element {
    virtual void accept(Visitor &vis) override {
        vis.Visit(static_cast<T&>(*this));
    }
};

struct ElementA : public AutoDispatchElement<ElementA>{

};
struct ElementB  : public AutoDispatchElement<ElementB>{

};

struct VisitorA : public Visitor {
    int sum = 1;
    virtual void Visit(ElementA &a) override {
        LOG(INFO) << "VisitorA vis ElementA: " << a.val;
        sum += a.val;
    }
    virtual void Visit(ElementB &b) override {
        LOG(INFO) << "VisitorA vis ElementB: " << b.val;
        sum *=b.val;
    }
    int total() {
        return sum;
    }
};
struct VisitorB : public Visitor {
    int mul = 1;
    virtual void Visit(ElementA &a) override {
        LOG(INFO) << "VisitorA vis ElementA: " << a.val;
        mul *= a.val;
    }
    virtual void Visit(ElementB &b) override {
        LOG(INFO) << "VisitorA vis ElementB: " << b.val;
        mul += b.val;
    }
    int multi() {
        return mul;
    }
};

struct ObjectStructure{
    ObjectStructure(){
        ElementA a1;
        a1.val = 1;
        objs.push_back(std::make_shared<ElementA>(a1));
        a1.val = 2;
        objs.push_back(std::make_shared<ElementA>(a1));
        ElementB b1;
        b1.val = 3;
        objs.push_back(std::make_shared<ElementB>(b1));
        b1.val = 4;
        objs.push_back(std::make_shared<ElementB>(b1));
    }
    void calc(Visitor &vis){
        for(auto && e : objs) {
            e->accept(vis);
        }
    }
    std::vector<std::shared_ptr<Element>> objs;
};

void common_test(){
    ObjectStructure os;
    VisitorA VA;
    VisitorB VB;
    os.calc(VA);
    LOG(INFO) << "visit by VA: " << VA.total();
    os.calc(VB);
    LOG(INFO) << "visit by VB: " << VB.multi();
}
int main(int argc, char *argv[]) {
   common_test();
}

std::variant, std::visit, if constexpr版访问者

#include <iostream>
#include "common/log.h"
#include <limits>
#include <vector>
#include <span>
#include <array>
#include <type_traits>
#include <cmath>
#include <memory>
#include <variant>
using namespace AdsonLib;

struct ElementA{
};
struct ElementB{};

using Element = std::variant<ElementA, ElementB>;

struct Visitor{
    void operator()(ElementA &e) {
        LOG(INFO) << "visit a";
    }
    void operator()(ElementB &e) {
        LOG(INFO) << "visit b";
    }
};

void common_test(){
    std::vector<Element> vecs;
    vecs.push_back(ElementA());
    vecs.push_back(ElementB());
    vecs.push_back(ElementA());
    vecs.push_back(ElementB());
    for(auto && e: vecs){
        std::visit([](auto &&arg){
            LOG(INFO) << "arg";
        }, e);
    }

    LOG(INFO) << "------------";
    for(auto && e: vecs){
        std::visit([](auto &&arg){
            using T = std::decay_t<decltype(arg)>;
            if constexpr(std::is_same_v<T, ElementA>){
                LOG(INFO) << "ElementA";
            } else if constexpr(std::is_same_v<T, ElementB>){
                LOG(INFO) << "ElementB";
            }
            
        }, e);
    }
    LOG(INFO) << "------------";
    for(auto && e: vecs){
        std::visit(Visitor(), e);
    }
}
int main(int argc, char *argv[]) {
   common_test();
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值