如何封装可复用的指针类型转换逻辑？基于DType枚举的Array结构体

现有代码定义如下：

typedef enum DType { float32 } DType;

typedef struct Array {
    void *data;
    DType dtype;
    // 其他字段
} Array;

需求是根据dtype字段的值，自动将data指针转换为对应类型。虽然每次使用时可以通过switch判断dtype来转换，但希望把这个逻辑封装成能在任意场景复用的函数或宏，同时未来给DType枚举添加新类型时，能更灵活地扩展，避免重复代码。

我尝试过一种无效方案：

void *cast_data_pointer(Array *arr) {
    switch (arr->dtype) {
        case float32:
            return (float *) arr->data;
        default:
            exit(1);
    }
}

这个方案的类型转换逻辑是对的，但无法满足需求——它要求必须提前知晓返回的指针类型：

Array arr = ...; // dtype为float32
void *data_ptr;

data_ptr = cast_data_pointer(&arr); // 没用，data_ptr还是void*类型

Array arr = ...; // dtype为float32
float *data_ptr;

data_ptr = cast_data_pointer(&arr); // 必须提前知道返回的是float*，失去封装意义

解决方案：封装switch逻辑的宏 #

由于C语言是静态类型语言，函数无法根据运行时的dtype值返回不同类型的指针，因此用宏来封装是最适合的方案。下面的宏将重复的switch逻辑打包，直接给目标指针赋值：

#include <stdlib.h>
#include <stdio.h>

#define ARRAY_CAST_DATA(arr_ptr, target_ptr) \
    do { \
        if (!(arr_ptr) || !(target_ptr)) { \
            fprintf(stderr, "Null pointer passed to ARRAY_CAST_DATA\n"); \
            exit(EXIT_FAILURE); \
        } \
        switch((arr_ptr)->dtype) { \
            case float32: \
                *(target_ptr) = (float *)(arr_ptr)->data; \
                break; \
            /* 未来添加新类型时，只需在这里新增case分支 */ \
            /* case int32: */ \
            /*     *(target_ptr) = (int *)(arr_ptr)->data; */ \
            /*     break; */ \
            /* case uint8: */ \
            /*     *(target_ptr) = (unsigned char *)(arr_ptr)->data; */ \
            /*     break; */ \
            default: \
                fprintf(stderr, "Unsupported DType: %d\n", (arr_ptr)->dtype); \
                exit(EXIT_FAILURE); \
        } \
    } while(0)

使用方式 #

Array arr = ...; // 假设arr的dtype已正确设置
float *data_ptr;

// 调用宏完成类型转换
ARRAY_CAST_DATA(&arr, &data_ptr);

// 现在可以直接使用data_ptr
printf("First element: %f\n", data_ptr[0]);

方案优势 #

• 复用性：所有需要转换指针的地方只需调用宏，避免重复写switch逻辑
• 扩展性：新增DType枚举值时，仅需在宏的switch中添加对应分支，所有调用处自动适配
• 安全性：添加了空指针检查，避免非法访问；默认分支处理未知类型，防止静默错误
• 兼容性：兼容所有C语言编译器，无需依赖C11及以上特性

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