如何改进我的向量搜索功能？

Question

因此，我在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"));

Answer 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的重载也可以帮助您执行更具体的搜索。

Answer 2

以下是一些建议：

1. 假设您多次搜索客户端向量--而不是使用std::find_if，对向量进行排序并执行二进制搜索(例如使用std::equal_range)。如果您经常添加新客户端，请考虑使用最近添加的客户端的附加、小的、未排序的缓冲区，并偶尔将两者集成到一个排序向量中。
2. 不要将检查搜索类型的代码放在谓词中--这样它就会一次又一次地运行；而且您还需要基于堆的比较器向量！相反，模板您的搜索函数的搜索类型，并让它使用一个比较器。使用包装器函数来选择适当的搜索函数。