如何简化C++多态基类实现类的重复方法调用逻辑？

问题背景 #

现有纯虚基类Base，子类Z、X实现该基类；类A持有Z、X的实例，其foo方法需根据enableZ、enableX、runOne三个参数，决定调用Z/X的one()或two()方法，当前代码存在大量重复的if嵌套判断。

核心优化思路 #

将目标对象与待调用成员函数的绑定逻辑抽离，避免在foo中重复写条件分支，通过统一的方式调用目标方法。

具体实现代码 #

1. 基类与子类定义

#include <functional>

class Base {
public:
    virtual void one() = 0;
    virtual void two() = 0;
    virtual ~Base() = default; // 必须声明虚析构，避免内存泄漏
};

class Z : public Base {
public:
    void one() override { /* 此处实现Z::one的逻辑 */ }
    void two() override { /* 此处实现Z::two的逻辑 */ }
};

class X : public Base {
public:
    void one() override { /* 此处实现X::one的逻辑 */ }
    void two() override { /* 此处实现X::two的逻辑 */ }
};

2. 类A的优化实现（成员函数指针方案）

这种方案直接利用基类的多态特性，结合成员函数指针实现：

class A {
private:
    Z z_instance;
    X x_instance;

    // 定义Base类成员函数的指针类型
    using BaseMemFunc = void (Base::*)();

    // 抽离条件判断逻辑，返回目标对象指针+对应成员函数
    std::pair<Base*, BaseMemFunc> getTarget(bool enableZ, bool enableX, bool runOne) {
        Base* target_obj = nullptr;
        if (enableZ) {
            target_obj = &z_instance;
        } else if (enableX) {
            target_obj = &x_instance;
        }

        // 根据runOne选择调用one()还是two()
        BaseMemFunc target_func = runOne ? &Base::one : &Base::two;
        return {target_obj, target_func};
    }

public:
    void foo(bool enableZ, bool enableX, bool runOne) {
        auto [obj, func] = getTarget(enableZ, enableX, runOne);
        if (obj != nullptr) { // 空指针判断，避免非法调用
            (obj->*func)(); // 调用成员函数的语法：对象指针->*成员函数指针
        }
    }
};

3. 类A的优化实现（std::function封装方案）

如果觉得成员函数指针的语法不够直观，可以用std::function和std::bind封装可调用对象：

class A {
private:
    Z z_instance;
    X x_instance;

    std::function<void()> getCallable(bool enableZ, bool enableX, bool runOne) {
        if (enableZ) {
            return runOne ? std::bind(&Z::one, &z_instance) : std::bind(&Z::two, &z_instance);
        } else if (enableX) {
            return runOne ? std::bind(&X::one, &x_instance) : std::bind(&X::two, &x_instance);
        }
        return []() {}; // 返回空操作，避免调用空的function
    }

public:
    void foo(bool enableZ, bool enableX, bool runOne) {
        auto callable = getCallable(enableZ, enableX, runOne);
        callable();
    }
};

常见报错原因说明 #

之前尝试基类指针写法报错，大概率是以下两种情况：

• 未正确使用成员函数指针的调用语法：必须用obj->*func或obj.*func，不能直接像普通函数指针那样调用
• 未做空指针判断：当enableZ和enableX都为false时，对象指针为空，直接调用会导致崩溃

内容的提问来源于stack exchange，提问作者xyf