“重传” 函数参数

Question

我正在使用经典 C 进行此类分配，并且遇到了有关采用可变参数计数和类型的回调函数的问题。

基本上，我正在研究哈希树（每个节点都是哈希树的树），并且我有一定的遍历策略，该策略将出于不同目的多次使用，因此我将其实现为 ht_walk (HashTree 树，(*callback)(Element e))，以便作为回调调用的函数将以任何必要的方式处理 Element。

问题是，在我的问题的大多数情况下，回调函数必须采用不同的参数。 我知道如何使用“可变”函数（使用 stdarg、printf-way）设计带有可变参数列表的函数，但我不知道如何将这些参数“重新传递”给回调函数。

让我提供一个具体的例子：假设我有一个名为 addToList(Element e, List list) 的回调函数，并且我的 ht_walk 声明现在是 ht_walk(HashTree tree, (*callback)(元素 e), ...)。 考虑我想使用 ht_walk，如以下代码片段所示：

HashTree my_tree = ht_create();
/* run some algorithm that populates the tree somehow */
List my_list = list_create();
ht_walk(my_tree, addToList, my_list);

有办法做到这一点吗？ 提前致谢！

Answer 1

您可以使用两种方法之一来解决此问题。

最常见、易于理解且干净的方法是使用“用户”结构：

void ht_walk(HashTree tree, void (*callback)(Element e, void *user), void *user);

void addToList(Element e, void *arg)
{
    STATIC_ASSERT(sizeof(void *) >= sizeof(List));

    List list = arg;

    /* ... */
}

HashTree my_tree = ht_create();
/* run some algorithm that populates the tree somehow */
List my_list = list_create();
ht_walk(my_tree, addToList, my_list);

---

另一种方法是接受 va_list：

#include <stdarg.h>

void ht_walk(HashTree tree, void (*callback)(Element e, va_list args), ...)
{
    for(..)
    {
        va_list args;
        va_start(args, callback);
        callback(element, args);
        va_end(args);
    }
}

void addToList(Element e, va_list args)
{
    List list = va_arg(args, List);

    /* ... */
}

HashTree my_tree = ht_create();
/* run some algorithm that populates the tree somehow */
List my_list = list_create();
ht_walk(my_tree, addToList, my_list);

Answer 2

有一系列函数和宏用于处理可变参数（例如，va_start）
GNU 有一个很好的指南

您似乎在描述某种访客模式。 我不确定我喜欢在这里使用可变参数列表。 如果您通常无法确定所有节点的回调将一致获得哪些参数，那么您一开始就会遇到麻烦。

Answer 3

我不认为你想要做的是传递任意数量的参数。 我认为这里的情况是有多种类型的回调。 在您的示例中，您传递了一个列表，但您也提到有时您会传递一个元素。 所以在我看来，你真正想做的就是声明你的回调函数采用 void* 和一个标志来指示它是什么。 比如：

void callback(void* arg, int type) {
    switch (type) {
    case ARG_TYPE_LIST:
        List* list = (List*)arg;
        ...
        break;
    case ARG_TYPE_ELEMENT:
        Element* ele = (Element*)arg;
        ...
        break;
    ...
}

如果你将东西作为列表传递，列表应该知道计数，所以你不必担心传递它。

或者，您可以定义多种类型的回调，并始终根据需要执行的处理类型回调正确的回调。

如果您不限于直接使用 C（即您可以使用 C++），您可以使用 boost::bind，因为它对于这类事情非常有用。

Answer 4

我会传入一个 void *，因为我认为您不能在回调函数中使用可变数量的参数。 但我可能是错的。

typedef void (*tree_walk_callback)(Element e, void *data);

add_element(Element e, void *data)
{
    List *list = (List *)data;
    add_element_to_list(list, e);
}

...in your function...
List my_list;
ht_walk(my_tree, &add_element, (void *)&my_list);

如果用户想要传递多个参数，那么他们将传递一个包含他们需要的任何内容的结构。

Answer 5

我相信您使用了 va_list 宏。 请参阅stdarg 的手册页。

自从我尝试这个以来已经有一段时间了......