C++訪問者模式模板函數(shù)無法重載的問題解決

更新時間：2021年12月23日 15:53:27 作者：Erick_Lv

本文主要介紹了C++訪問者模式模板函數(shù)無法重載的問題解決，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下

背景

最近遇到一個比較棘手的場景，我們有一堆模塊，他們有一個通用的基類，我們不防假設(shè)為 BaseClass，該類有一些通用的結(jié)構(gòu)以及需要重載的方法。這些模塊有一個堆同名但是不同類型

參數(shù)的方法，比如：

int DerivedClass1::DoNlpTask(const DerivedReq1& req, DerivedResp* resp);

類似這樣的。每個 DerivedClass 的DoNlpTask都是同名不同參數(shù)的，而且這些要給業(yè)務(wù)去具體實(shí)現(xiàn)。正常來說，很容易想到函數(shù)重載的方式，但是很不幸，模板函數(shù)無法重載
但是，我們想讓框架層和業(yè)務(wù)層相對解耦，不想讓業(yè)務(wù)的具體類型等污染框架調(diào)度模塊的結(jié)構(gòu)。

解決方案

在這里，我們利用訪問者模式的思想，結(jié)合C++的模板來統(tǒng)一處理。先定義一個Visitor，該類定義好具體業(yè)務(wù)模塊的通用操作步驟；實(shí)現(xiàn)的時候利用模板特化來實(shí)例化不同的業(yè)務(wù)模塊及其DoNlpTask 函數(shù)。這樣在框架層只需要調(diào)用Visitor 的統(tǒng)一模板接口即可，具體參看一下代碼。

最終代碼

#include <iostream>
#include <type_traits>

// 以下模擬請求協(xié)議 =====================
class BaseReq {};

class DerivedReq : public BaseReq {};

class BaseResp {};

class DerivedResp : public BaseResp {};

DerivedReq g_derived_req;

// 以下模擬 NLP ============================

class BaseClass {
 public:
  // 單純視為一個需要重載的函數(shù)
  virtual int get_field_id() = 0;
     
  virtual void WormUp () {}
};

#define REGISTER_PROTOTYPE(req_type, resp_type) \
 public:                                        \
  using ReqType = req_type;                     \
  using RespType = resp_type;

class DerivedClass : public BaseClass {
  // 注冊請求類型
  REGISTER_PROTOTYPE(DerivedReq, DerivedResp)
 public:
  int DoNlpTask(const DerivedReq&, DerivedResp* resp) {
    std::cout << "Derived DoNlpTask\n";
  }

  int get_field_id() override { return 1; }
};

// 以下模擬pb反射 =========================
const BaseReq* GetReqType() { return &g_derived_req; }

// 以下是 visitor 的定義 ===========
class Visitor {
 public:
  template <typename ClassType>
  int DoVisit(ClassType* base) {
    static_assert(std::is_base_of<BaseClass, DerivedClass>::value,
                  "type failed");
    int idx = base->get_field_id();
    std::cout << "visitor get field_id " << idx << std::endl;
    const auto* req = GetMessageType<typename ClassType::ReqType>(GetReqType());
    typename ClassType::RespType resp;
    // 返回計(jì)算結(jié)果
    return base->DoNlpTask(*req, &resp);
  }

 private:
  template <typename ReqType>
  const ReqType* GetMessageType(const BaseReq* req) {
    static_assert(std::is_base_of<BaseReq, ReqType>::value,
                  "Message Type Error");
    return static_cast<const ReqType*>(req);
  }
};

int main() {
  DerivedClass dc;
  Visitor vis;
  vis.DoVisit(&dc);
  return 0;
}

到此這篇關(guān)于C++訪問者模式模板函數(shù)無法重載的問題解決的文章就介紹到這了,更多相關(guān)C++模板函數(shù)無法重載內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: