Last active
June 6, 2023 22:22
-
-
Save facontidavide/95f20c28df8ec91729f9d8ab01e7d2df to your computer and use it in GitHub Desktop.
Use std::string_view to find() an element in std::unordered_map<string,T>, avoiding potential memory allocations
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include <iostream> | |
#include <cstring> | |
#include <unordered_map> | |
// https://github.com/martinmoene/string-view-lite | |
#include "string_view.hpp" | |
template <typename Value> | |
class StringMap: public std::unordered_map<std::string, Value> | |
{ | |
public: | |
typename std::unordered_map<string,Value>::iterator find(const nonstd::string_view& v ) | |
{ | |
tmp_.reserve( v.size() ); | |
tmp_.assign( v.data(), v.size() ); | |
return std::unordered_map<string, Value>::find(tmp_); | |
} | |
typename std::unordered_map<std::string,Value>::iterator find(const std::string& v ) | |
{ | |
return std::unordered_map<std::string, Value>::find(v); | |
} | |
typename std::unordered_map<std::string,Value>::iterator find(const char* v ) | |
{ | |
tmp_.assign(v); | |
return std::unordered_map<std::string, Value>::find(v); | |
} | |
private: | |
thread_local static std::string tmp_; | |
}; | |
template <typename T> thread_local std::string StringMap<T>::tmp_ = {}; | |
int main() | |
{ | |
std::string key_A("Hello"); | |
nonstd::string_view key_B("Hello"); | |
nonstd::string_view key_C(key_A); | |
StringMap<std::string> smap; | |
smap["Hello"] = "World"; | |
for (auto& it: smap) | |
{ | |
std::cout << it.first << ":" << it.second << std::endl; | |
} | |
std::cout << smap.find(key_A)->second << std::endl; | |
std::cout << smap.find(key_B)->second << std::endl; | |
std::cout << smap.find(key_C)->second << std::endl; | |
std::cout << smap.find("Hello")->second << std::endl; | |
return 0; | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@Vizepi , you are right. It was a silly mistake on my side.
@facontidavide , your solution does not have the problems I mentioned above. It is thread safe for reading.