如何在编译时为自定义String结构体实现短字符串优化的条件初始化？

问题背景 #

定义了如下String结构体，通过匿名联合实现小字符串优化：

typedef struct String {
    size_t size;
    union {
        char *ptr, buf[sizeof(char *)];
    };
} String;

提取字符串指针的逻辑为：当size >= sizeof(char*)时使用堆分配的ptr，否则使用栈内的buf：

extern String string;
char *str = string.size >= sizeof(char *) ? string.ptr : string.buf;

需要编译时初始化const static或全局的String结构体，但无法用预处理指令#if判断字符串长度（sizeof是编译期操作，无法在预处理阶段使用），之前的尝试写法无效：

static const String string = {
    sizeof(CONSTANT_STRING)-1,
#if sizeof(CONSTANT_STRING) > sizeof(char *)
    .ptr
#else
    .buf
#endif
        = CONSTANT_STRING
};

解决方案 #

以下两种方法均基于编译期常量表达式实现，无需预处理指令：

方法1：复合字面量+编译期条件判断 #

通过C99引入的复合字面量，结合编译期条件运算符，直接初始化匿名联合的对应成员：

#define STRING_INIT(s) \
    { \
        .size = sizeof(s) - 1, \
        (sizeof(s) - 1 >= sizeof(char *)) ? \
            (union { char *ptr; char buf[sizeof(char *)]; }){.ptr = s} : \
            (union { char *ptr; char buf[sizeof(char *)]; }){.buf = s} \
    }

// 使用示例
#define CONSTANT_STRING "Hello"
static const String string = STRING_INIT(CONSTANT_STRING);

编译期会计算sizeof(s)-1与sizeof(char*)的大小关系，选择对应的联合成员初始化方式，最终生成的初始化表达式是编译时常量。

方法2：C11 _Generic类型匹配 #

利用C11的_Generic编译期类型匹配特性，根据字符串数组的长度选择初始化分支：

#define STRING_INIT(s) \
    { \
        .size = sizeof(s) - 1, \
        ._ = _Generic( (s), \
            char[sizeof(char *) + 1]: (union { char *ptr; char buf[sizeof(char *)]; }){.buf = s}, \
            default: (union { char *ptr; char buf[sizeof(char *)]; }){.ptr = s} \
        ) \
    }

// 使用示例
#define CONSTANT_STRING "A longer string that exceeds buf size"
static const String string = STRING_INIT(CONSTANT_STRING);

当字符串长度（含终止符）为sizeof(char*)+1时，对应size = sizeof(char*)，此时使用buf；更长的字符串则使用ptr。._是GCC对匿名联合的初始化语法，部分编译器可直接省略成员名，仅保留复合字面量。

补充：静态断言保障正确性 #

可在宏中加入_Static_assert确保字符串非空，避免无效初始化：

#define STRING_INIT(s) \
    { \
        .size = sizeof(s) - 1, \
        _Static_assert(sizeof(s) > 0, "String cannot be empty"), \
        (sizeof(s) - 1 >= sizeof(char *)) ? \
            (union { char *ptr; char buf[sizeof(char *)]; }){.ptr = s} : \
            (union { char *ptr; char buf[sizeof(char *)]; }){.buf = s} \
    }

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