因此,我正在用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"));
答案 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)
以下是一些建议:
std::find_if
,请对向量进行排序并执行二进制搜索(例如,使用std::equal_range
)。如果您经常添加新的客户端,请考虑使用一个新近添加的客户端的,较小的,未排序的缓冲区,并偶尔将两者集成到单个已排序的向量中。