31 lines
791 B
C++
31 lines
791 B
C++
|
|
#include <iostream>
|
|
|
|
#include "CompanyEntry.h"
|
|
#include "FindVisitor.h"
|
|
#include "GroupEntry.h"
|
|
#include "PersonEntry.h"
|
|
|
|
FindVisitor::FindVisitor(std::string query) : _query(query) {}
|
|
|
|
const std::vector<const Entry *> &FindVisitor::getFindings() { return findings; }
|
|
|
|
void FindVisitor::visit(const PersonEntry &s) {
|
|
if (s.getName().find(_query) != std::string::npos) {
|
|
findings.push_back(&s);
|
|
}
|
|
}
|
|
|
|
void FindVisitor::visit(const GroupEntry &g) {
|
|
if (g.getName().find(_query) != std::string::npos) {
|
|
findings.push_back(&g);
|
|
}
|
|
}
|
|
|
|
void FindVisitor::visit(const CompanyEntry &c) {
|
|
|
|
if (c.getName().find(_query) != std::string::npos || c.getCategory().find(_query) != std::string::npos) {
|
|
findings.push_back(&c);
|
|
}
|
|
}
|