在 STL map 中通过引用查找并返回

Question

我有一个包含这两个公共方法的类：

void StateManager::setEntityState(const EntityCTCId id, const EntityCTCState state) {
    auto it = entities.find(id);
    if(it != entities.end())
        it->second.state = state;
    else
        entities.emplace(id, EntityCTC{id, state});
}

EntityCTCState StateManager::getEntityState(const EntityCTCId id) const {
    auto it = entities.find(id);
    if(it != entities.end())
        return it->second.state;

    std::stringstream ss;
    ss << "Entity CTC [" << id << "] NOT found!";
    throw EntityCTCNotFound{ss.str()};
}

其中entities 是std::map<EntityCTCId, EntityCTC>。

我想重构这些方法以隔离常见的find 并通过引用返回找到的EntityCTC，但我不知道如何处理“找不到密钥”的情况：

EntityCTC& findEntity(const EntityCTCId id) {
    auto it = entities.find(id);
    if(it != entities.end())
        return it->second.state;
    else
        // ???
}

我找到的唯一解决方案是在这个新方法中抛出异常，但这意味着在setEntityState 中使用try-catch 来区分更新和新插入（我读到异常管理不应该用作逻辑分支，但仅用于错误管理）。

您能建议我其他方法吗？

Answer 1

另一种方法是充分利用 API。如您所见，emplace 有一个返回值。这是std::pair<iterator, bool>。迭代器可以用来访问key下的item，bool是让你知道是否有插入发生或者key已经存在于map中。

所以，我建议你改写setEntityState：

void StateManager::setEntityState(const EntityCTCId id, const EntityCTCState state) {
    auto status = entities.emplace(id, EntityCTC{id, state});
    if(!status.second)
      status.first->second.state = state;
}

现在无需担心通用功能，您只需在函数中执行一次 O(logN) 操作，而不是两次。

---

附录，因为您可能担心构造值的成本，而不是作为过早优化的一部分。充分利用 API 的另一点是要知道有办法分段构造插入到地图中的对。像这样：

auto status = entities.emplace(std::piecewise_construct,
                               std::forward_as_tuple(id),
                               std::forward_as_tuple(id, state));