声明后添加好友类

Question

我正在尝试用 C++ 编写一个命名管道服务器。我有一个名为 client_pool 的类，其中包含管道实例的容器，以及一个公共成员函数 write，该函数将数据异步发送到所有连接的客户端。

问题是，客户端往往会意外断开连接。发生这种情况时，对 WriteFileEx 的调用将失败并显示 ERROR_NO_DATA。当发生这种情况时，我想转到 client_pool 类并告诉它关闭客户端句柄并将其从容器中删除。然而，由于 WriteFileEx 太难用了，我在匿名命名空间中创建了一个名为 write_context 的帮助器类。

所以最终的结果是，我想从类 write_context 中调用 client_pool 中的私有方法，该方法在 clients.h 中声明，它在 clients.cpp 中声明。类似的东西（省略细节/错误处理）：

clients.h

class client_pool {
    struct implementation;
    std::unique_ptr<implementation> pimpl;
public:
    void write(uint8_t *data, size_t size);
};

clients.cpp

struct client_pool::implementation {
    set<HANDLE> connected;
    // ...
    void disconnect(HANDLE victim)
    {
        CloseHandle(victim);
        connected.erase(victim);
    }
};

namespace { struct write_context {
    OVERLAPPED overlapped;
    client_pool *owner;
    HANDLE target;
    const uint8_t *buffer;
    size_t total_size;
    size_t written;
    // ...
    void next_chunk()
    {
        if(!WriteFileEx(/* ... */, write_context::completion_routine)) {
            if(GetLastError() == ERROR_NO_DATA) {
                // I want to do something like
                owner->pimpl->disconnect(target);
            }
        }
    }
    static void CALLBACK completion_routine(DWORD errcode, DWORD transferred, LPOVERLAPPED overlapped)
    {
        auto self = reinterpret_cast<write_context*>(overlapped);
        self->written += transferred;
        if(errcode == ERROR_MORE_DATA) {
            self->next_chunk();
        } else {
            delete self;
        }
    }
}; }

void client_pool::write(uint8_t *data, size_t size)
{
    for each handle in pimpl->connected {
        auto context = new write_context(this, handle, data, size);
        context->next_chunk();
    }
}

显然，行 owner->pimpl->disconnect(target) ; 无法编译，因为 pimpl 是私有的。我能做什么/我有什么选择？

Answer 1

直接在 client_pool::write 方法中直接访问 pimpl->connected 和 write_context 与 pimpl 习惯用法的要点有点相反。你可能会提出其他的理由，直到你遇到这样的问题。

我只想创建一个implementation::write 方法，您可以为其传递参数和指向client_pool 的指针。

Answer 2

我认为如果您使用命名名称空间而不是匿名名称空间，您可以将此行放在类定义中：

friend void namespace_name::next_chunk()

或者将所有额外的内容作为类中的静态函数放置在匿名名称空间中。由于静态方法和结构不会更改 ABI，因此您可以通过预处理器技巧对所有其他实例隐藏它。

或者是残酷而恐怖的：

#define class struct
#define private public
#define protected public

Answer 3

让write_context成为implementation的朋友。将 pimpl 作为 write_context 的 owner 传递。

Answer 4

抱歉，但这不是使用 PIMPL 惯用语的最佳方式。

PIMPL 隐藏其所有者的实现细节，并且只能通过其所有者接口进行访问。因此，如果您想调用“client_pool::implementation”方法，则应将其移至“client_pool”接口，并且其实现应将工作委托给“client_pool::implementation”类。在其他情况下，这看起来像是设计错误。