reverted wfoMap_ to unordered_map and added error checking to getWFOName

This commit is contained in:
AdenKoperczak 2024-09-09 08:56:53 -04:00
parent ea23986ead
commit ed00cec4cc
2 changed files with 10 additions and 5 deletions

View file

@ -31,7 +31,7 @@ typedef std::unordered_map<char, StateMap> FormatMap;
static bool initialized_ {false};
static FormatMap countyDatabase_;
static std::unordered_map<std::string, std::string> stateMap_;
static std::map<std::string, std::string> wfoMap_;
static std::unordered_map<std::string, std::string> wfoMap_;
void Initialize()
{
@ -264,14 +264,20 @@ const std::unordered_map<std::string, std::string>& GetStates()
return stateMap_;
}
const std::map<std::string, std::string>& GetWFOs()
const std::unordered_map<std::string, std::string>& GetWFOs()
{
return wfoMap_;
}
const std::string& GetWFOName(const std::string& wfoId)
{
return wfoMap_.at(wfoId);
auto wfo = wfoMap_.find(wfoId);
if (wfo == wfoMap_.end())
{
return wfoId;
}
return wfo->second;
}
} // namespace CountyDatabase

View file

@ -4,7 +4,6 @@
#include <string>
#include <unordered_map>
#include <vector>
#include <map>
namespace scwx
{
@ -20,7 +19,7 @@ std::string GetCountyName(const std::string& id);
std::unordered_map<std::string, std::string>
GetCounties(const std::string& state);
const std::unordered_map<std::string, std::string>& GetStates();
const std::map<std::string, std::string>& GetWFOs();
const std::unordered_map<std::string, std::string>& GetWFOs();
const std::string& GetWFOName(const std::string& wfoId);
} // namespace CountyDatabase