C ++将多个字符串更改为枚举

Question

我正在处理一个遗留代码库，其中一个头文件定义了 5 个字符串，如下所示：

#define NUM_STRINGS 5
extern string string_0;
extern string string_1;
extern string string_2;
extern string string_3;
extern string string_4;

在其他地方的 cpp 文件中：

string string_0 = make_string("my_string_0");
string string_1 = make_string("my_string_1");
string string_2 = make_string("my_string_2");
string string_3 = make_string("my_string_3");
string string_4 = make_string("my_string_4");

现在在代码库中，这些字符串用作函数参数，因为我们有一个函数void func(string str)，其中 str 始终是这 5 个字符串之一。有没有办法将这些字符串放入一个类型中，比如 string_enum，

1. 将以 [near] 零成本从类型中获取正确的字符串

2. 保证别人不能打电话func("hello world")

3. 允许我获取枚举中的元素数量

4. 还有什么你想提的有用的东西

Answer 1

有没有办法将这些字符串放入一个类型中，比如 string_enum，

1. 将以零成本从类型中获取正确的字符串

不是真正的零成本，但相当高效：使用std::map<std::string, my_enum>：

enum my_enum {
    my_string_0 ,
    my_string_1 ,
    my_string_2 ,
    my_string_3 ,
    my_string_4
};

const std::map<std::string,my_enum> enumMap = {
    { "my_string_0" , my_string_0 } ,
    { "my_string_1" , my_string_1 } ,
    { "my_string_2" , my_string_2 } ,
    { "my_string_3" , my_string_3 } ,
    { "my_string_4" , my_string_4 }
};

1. 保证某人不能调用func("hello world")

检查映射键中的字符串是否存在，如果不存在则从func() 抛出异常：

void func(const std::string key) {
    if(enumMap.find(key) == std::end(enumMap)) {
        throw std::invalid_argument("key");
    }
}

1. 允许我获取枚举中的元素数量

std::size_t getNumKeys() {
    return enumMap.size();
}