如何改善向量搜寻功能?

时间:2019-03-23 16:18:36

标签: c++ c++11 search lambda stdvector

因此,我正在用c ++实现一个TCP服务器,并将所有用户存储在一个向量中。现在,我需要一个更通用的函数来搜索多个不同的属性,如何改进此代码?

struct client {
    std::string ip_address = "";
    int socket_id = 0;
    bool blocking = false;
};

enum client_codes {
    ip_address,
    socket_id,
    blocking,
};

template<typename T>
std::vector<client>::iterator search_vector(std::vector<client> &list, int type, T query) {

    std::vector<std::function<bool(client)>> comparators;

    comparators.push_back([&](client ob) {return ob.ip_address == std::to_string(query); });
    comparators.push_back([&](client ob) {return ob.socket_id == query; });
    comparators.push_back([&](client ob) {return ob.blocking == query; });

    return std::find_if(std::begin(list), std::end(list), [&](client obj) {return comparators[type](obj); });

}

// Implementation
std::vector<client> client_list;
auto search1(search_vector(client_list, socket_id, 321));
auto search2(search_vector(client_list, blocking, true));
auto search3(search_vector(client_list, ip_address, "192.168.0.85"));

2 个答案:

答案 0 :(得分:1)

由于您正在搜索相同类型的不同属性,因此我想说这是pointers to members的罕见用例。以下代码应有助于清理此问题:

struct client {
    std::string ip_address = "";
    int socket_id = 0;
    bool blocking = false;
};

// search_vector accepts a pointer to member of type T, and a value of type T
template<typename T>
std::vector<client>::iterator search_vector(std::vector<client>& list, T client::*member, T value){
    return std::find_if(list.begin(), list.end(), [value, member](const client& c){ return c.*member == value; });
}

这就是您现在使用它的方式,而无需任何额外的enums或特殊逻辑。

auto it1 = search_vector(client_list, &client::socket_id, 321);
auto it2 = search_vector(client_list, &client::blocking, true);
auto it3 = search_vector(client_list, &client::ip_address, "192.168.0.85");

这个精确的search_vector函数当然仅限于成员变量和精确相等性测试。但是,使用指向get_ip_address()之类的成员函数的指针来扩展它会非常简单。

出于一般性考虑,如果您不希望完全相等,则使用类似std::function<bool(T)> condition而不是T value的重载也可以帮助您进行更具体的搜索。

答案 1 :(得分:0)

以下是一些建议:

  1. 假设您要多次搜索客户端向量-而不是使用std::find_if,请对向量进行排序并执行二进制搜索(例如,使用std::equal_range)。如果您经常添加新的客户端,请考虑使用一个新近添加的客户端的,较小的,未排序的缓冲区,并偶尔将两者集成到单个已排序的向量中。
  2. 不要在谓词中放置检查搜索类型的代码-这样就可以一次又一次地运行它;而且您还需要比较器的基于堆的向量-啊! ...而是将您的搜索功能模板化为搜索类型,并使其使用单个比较器。使用包装函数选择适当的搜索功能。
相关问题