从同一个父调用中调用不同的子类函数

Question

考虑这个简单的 C++11 继承示例：

class A
{
public:
    virtual void func() = 0;
};

class B : public A
{
public:
    void func() { func1(); /* Wish this could be func1() or func2()! */ };

    void func1() { /* Does one thing */ };
    void func2() { /* Does another thing */ };
};

void doSomeStuff(A &a) 
{
    a.func();
}

int main()
{
    B b;

    doSomeStuff(b);

    return 0;
}

我正在努力做到这一点，这样我就不必修改（或复制）类 A 的定义或函数 doSomeStuff，但我希望调用 a.func() 来调用 func1() 或 func2 B 的 ()。理想情况下，我会将行 doSomeStuff(b) 更改为 doSomeStuff(b.butWithFunc1)，但我也可以通过某种方式修改 B 的 func() 版本，以便它可以在内部决定调用 @ 987654325@ 或 func2 基于某些参数。

在调用func 期间，B 类型的同一对象有时可能必须调用func1 或func2，因此我不能使用B 类的持久成员来决定。向 func() 添加一个参数也会使这变得微不足道，但这也不是我能做的。

我有点想知道是否有某种方法可以向 B 类添加一个函数，该函数返回 B 类的突变版本，该版本从 func() 调用 func2()，或者我是否可以使用函数指针或其他东西玩一些技巧.然而，有些事情告诉我我做错了，显而易见的解决方案是盯着我的脸。

如果它对上下文有帮助，A 类类似于 std::lock_guard，它适用于信号量和互斥锁（对于它们只有一种锁定和解锁的定义），但本例中的 B 类是一个 R/W 锁 - 所以有一个“readLock”和“writeLock”，我想能够说“自动锁定这个 RW 锁作为一个读锁”，而不必复制/破坏自动锁代码。

例如：

{
    A_AutoSem(myMutex); // calls lock() on myMutex
    //... do some stuff

    // end of the block, ~A_AutoSem calls unlock on myMutex
}

{
    A_AutoSem(B_RWLock); // how do I say here "call readLock"?
    // ... do some stuff

    // end of the block ~A_AutoSem should call "readUnlock" on B_RWLock
}

Answer 1

只需定义一些额外的类来调用func1() 和func2()，然后将这些类传递给doSomeStuff()，而不是直接传递B。

试试这样的：

class A
{
public:
    virtual void func() = 0;
};

class B
{
public:
    void func1() { /* Does one thing */ };
    void func2() { /* Does another thing */ };
};

class C1 : public A
{
private:
    B &m_b;
public:
    C1(B &b) : m_b(b) {}
    void func() override { m_b.func1(); }
};

class C2 : public A
{
private:
    B &m_b;
public:
    C2(B &b) : m_b(b) {}
    void func() override { m_b.func2(); }
};

void doSomeStuff(A &a) 
{
    a.func();
}

int main()
{
    B b;

    {
    C1 c(b);
    doSomeStuff(c);
    }

    {
    C2 c(b);
    doSomeStuff(c);
    }

    return 0;
}

Live Demo

或者：

class A
{
public:
    virtual void func() = 0;
};

class B
{
private:
    void func1() { /* Does one thing */ };
    void func2() { /* Does another thing */ };

public:
    class C1 : public A
    {
    private:
        B &m_b;
    public:
        C1(B &b) : m_b(b) {}
        void func() override { m_b.func1(); }
    };

    class C2 : public A
    {
    private:
        B &m_b;
    public:
        C2(B &b) : m_b(b) {}
        void func() override { m_b.func2(); }
    };
};

void doSomeStuff(A &a) 
{
    a.func();
}

int main()
{
    B b;

    {
    B::C1 c(b);
    doSomeStuff(c);
    }

    {
    B::C2 c(b);
    doSomeStuff(c);
    }

    return 0;
}

Live Demo