Log-parser/LogFilter.cpp at main · CodesbyNeo/Log-parser · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#include "LogFilter.h"
#include "Utils.h"
#include <algorithm>

std::vector<LogEntry> LogFilter::apply(
    const std::vector<LogEntry>& entries,
    const FilterOptions& opts)
{
    std::vector<LogEntry> result = entries;

    if (!opts.level.empty()) {
        LogLevel lvl = LogEntry::levelFromString(opts.level);
        result = byLevel(result, lvl);
    }

    if (!opts.keyword.empty()) {
        result = byKeyword(result, opts.keyword);
    }

    if (!opts.source.empty()) {
        result = bySource(result, opts.source);
    }

    if (!opts.afterTime.empty()) {
        result.erase(std::remove_if(result.begin(), result.end(),
            [&](const LogEntry& e) {
                return e.timestamp < opts.afterTime;
            }), result.end());
    }

    if (!opts.beforeTime.empty()) {
        result.erase(std::remove_if(result.begin(), result.end(),
            [&](const LogEntry& e) {
                return e.timestamp > opts.beforeTime;
            }), result.end());
    }

    return result;
}

std::vector<LogEntry> LogFilter::byLevel(
    const std::vector<LogEntry>& entries,
    LogLevel level)
{
    std::vector<LogEntry> result;
    std::copy_if(entries.begin(), entries.end(), std::back_inserter(result),
        [level](const LogEntry& e) { return e.level == level; });
    return result;
}

std::vector<LogEntry> LogFilter::byKeyword(
    const std::vector<LogEntry>& entries,
    const std::string& keyword)
{
    std::vector<LogEntry> result;
    std::copy_if(entries.begin(), entries.end(), std::back_inserter(result),
        [&](const LogEntry& e) {
            return Utils::containsIgnoreCase(e.message, keyword) ||
                   Utils::containsIgnoreCase(e.source, keyword);
        });
    return result;
}

std::vector<LogEntry> LogFilter::bySource(
    const std::vector<LogEntry>& entries,
    const std::string& source)
{
    std::vector<LogEntry> result;
    std::copy_if(entries.begin(), entries.end(), std::back_inserter(result),
        [&](const LogEntry& e) {
            return Utils::containsIgnoreCase(e.source, source);
        });
    return result;
}