枚举所有注册表项的最有效方法

Question

我正在使用以下函数来枚举向量中特定键和子键的所有注册表键并将其返回。

但是对于较大的键值，此代码似乎运行得非常慢，例如对于超过200000+个键，则需要5分钟以上的时间才能完成。 只想知道如何使它更有效率。

vector<string> CRegistryAPI::EnumKeys(HKEY RootKey, string SubKey)
{
    DWORD keyLen = 255;
    int index = 0;
    int ret = 0;
    PFILETIME lastWrite = 0;
    HKEY hKey = 0;
    char keyName[255];
    vector<string> keyList;

    if (RegOpenKeyExA(RootKey, SubKey.c_str(), 0, KEY_ENUMERATE_SUB_KEYS,&hKey) != ERROR_SUCCESS)
        return keyList;
    do
    {
        ret = RegEnumKeyExA(hKey, index, keyName, &keyLen, 0, 0, 0, lastWrite);
        if (ret == ERROR_SUCCESS)
        {
            keyList.push_back(keyName);
        }
        index ++;
        keyLen = 255; 
    }
    while (ret == ERROR_SUCCESS);

    if (hKey != NULL)
        RegCloseKey(hKey);

    return keyList;
}

任何帮助表示赞赏。 谢谢。

Answer 1

您可以做的一种优化是使用RegQueryInfoKey()查询有关给定键的信息，然后再枚举其内容。 例如，您可以预分配向量的容量，因此在枚举期间不必多次重新分配：

RegQueryInfoKey(hKey, 0, 0, 0, 0, 0, 0, &dwNumValues, 0, 0, 0, 0);
keyList.reserve(keyList.size()+dwNumValues);

同样，枚举不能确保将缓冲区推入向量时缓冲区为空终止。 即使是这种情况，您当前的推送方式每次都需要std::string来计算缓冲区长度。 您应该改用返回给您的缓冲区长度：

keyList.push_back(string(keyName, keyLen));