C语言中函数如何根据运行逻辑返回module或level两种不同结构体

C是强静态类型语言，函数返回值类型必须在编译期确定，你当前的写法返回值类型固定为struct module，返回level类型会直接触发编译错误，无法运行。以下是三种可行的实现方案：

方案1：带标签联合体（最通用推荐） #

通过枚举标记返回类型，用联合体承载不同结构体，无需手动管理内存，类型安全。

类型定义 #

// 定义返回类型标签枚举
typedef enum {
    RET_TYPE_MODULE,
    RET_TYPE_LEVEL
} SelectRetType;

// 定义带标签的联合体作为返回值类型
typedef struct {
    SelectRetType type;
    union {
        module mod;
        level lvl;
    } data;
} SelectResult;

函数实现 #

SelectResult SELECT(FILE *f, int id, int db) {
    SelectResult res = {0};
    if (id == 1) {
        res.type = RET_TYPE_MODULE;
        // 此处补充module结构体的业务赋值逻辑
        return res;
    } else if (id == 2) {
        res.type = RET_TYPE_LEVEL;
        // 此处补充level结构体的业务赋值逻辑
        return res;
    }
    return res;
}

调用示例 #

FILE *tmp = fopen(str, "r");
SelectResult raw_res = SELECT(tmp, 2, 1);
if (raw_res.type == RET_TYPE_LEVEL) {
    level res = raw_res.data.lvl;
    // 后续业务逻辑
}

方案2：void*返回指针（灵活但需手动管理内存） #

函数返回堆上分配的结构体指针，调用方按预期类型强转使用，用完需手动释放内存。

函数实现 #

void* SELECT(FILE *f, int id, int db) {
    if (id == 1) {
        module *mod = malloc(sizeof(module));
        // 此处补充module结构体的业务赋值逻辑
        return mod;
    } else if (id == 2) {
        level *lvl = malloc(sizeof(level));
        // 此处补充level结构体的业务赋值逻辑
        return lvl;
    }
    return NULL;
}

调用示例 #

FILE *tmp = fopen(str, "r");
level *res = (level*)SELECT(tmp, 2, 1);
// 后续业务逻辑
free(res); // 必须手动释放，避免内存泄漏

方案3：输出参数（最适合调用方已知返回类型的场景） #

调用方提前传入对应类型的结构体指针接收结果，函数返回状态码判断是否成功，完全栈上操作无内存风险，贴合你给出的调用示例场景。

函数实现 #

// 返回值：0 成功，-1 类型不匹配/参数错误
int SELECT(FILE *f, int id, int db, void *out_buf, SelectRetType expect_type) {
    if (id == 1 && expect_type == RET_TYPE_MODULE) {
        module *mod = (module*)out_buf;
        // 此处补充module结构体的业务赋值逻辑
        return 0;
    } else if (id == 2 && expect_type == RET_TYPE_LEVEL) {
        level *lvl = (level*)out_buf;
        // 此处补充level结构体的业务赋值逻辑
        return 0;
    }
    return -1;
}

调用示例 #

FILE *tmp = fopen(str, "r");
level res;
int ret = SELECT(tmp, 2, 1, &res, RET_TYPE_LEVEL);
if (ret == 0) {
    // 后续业务逻辑
}

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