-static std::map<std::string, std::string> getEnvVarMap() {
- std::map<std::string, std::string> data;
- for (auto it = environ; *it != nullptr; ++it) {
- std::string key, value;
- split("=", *it, key, value);
- if (key.empty()) {
- continue;
- }
- CHECK(!data.count(key)) << "already contains: " << key;
- data.emplace(move(key), move(value));
- }
- return data;
-}
-
-EnvVarSaver::EnvVarSaver() {
- saved_ = getEnvVarMap();
-}
-
-EnvVarSaver::~EnvVarSaver() {
- for (const auto& kvp : getEnvVarMap()) {
- if (saved_.count(kvp.first)) {
- continue;
- }
- PCHECK(0 == unsetenv(kvp.first.c_str()));
- }
- for (const auto& kvp : saved_) {
- PCHECK(0 == setenv(kvp.first.c_str(), kvp.second.c_str(), (int)true));
- }
-}
-
-} // namespace test
-} // namespace folly