1c29fa5a6Sopenharmony_ci/*
2c29fa5a6Sopenharmony_ci * Copyright (c) 2024 Huawei Device Co., Ltd.
3c29fa5a6Sopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License");
4c29fa5a6Sopenharmony_ci * you may not use this file except in compliance with the License.
5c29fa5a6Sopenharmony_ci * You may obtain a copy of the License at
6c29fa5a6Sopenharmony_ci *
7c29fa5a6Sopenharmony_ci * http://www.apache.org/licenses/LICENSE-2.0
8c29fa5a6Sopenharmony_ci *
9c29fa5a6Sopenharmony_ci * Unless required by applicable law or agreed to in writing, software
10c29fa5a6Sopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS,
11c29fa5a6Sopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12c29fa5a6Sopenharmony_ci * See the License for the specific language governing permissions and
13c29fa5a6Sopenharmony_ci * limitations under the License.
14c29fa5a6Sopenharmony_ci */
15c29fa5a6Sopenharmony_ci
16c29fa5a6Sopenharmony_ci#ifndef INPUT_AGGREGATOR_H
17c29fa5a6Sopenharmony_ci#define INPUT_AGGREGATOR_H
18c29fa5a6Sopenharmony_ci
19c29fa5a6Sopenharmony_ci#include <iostream>
20c29fa5a6Sopenharmony_ci#include <map>
21c29fa5a6Sopenharmony_ci#include <set>
22c29fa5a6Sopenharmony_ci#include <string>
23c29fa5a6Sopenharmony_ci#include <utility>
24c29fa5a6Sopenharmony_ci
25c29fa5a6Sopenharmony_ci#include "mmi_log.h"
26c29fa5a6Sopenharmony_ci
27c29fa5a6Sopenharmony_cinamespace OHOS {
28c29fa5a6Sopenharmony_cinamespace MMI {
29c29fa5a6Sopenharmony_ciclass Aggregator {
30c29fa5a6Sopenharmony_cipublic:
31c29fa5a6Sopenharmony_ci    Aggregator(std::function<int32_t(int32_t, int32_t, std::function<void()>)> addTimer,
32c29fa5a6Sopenharmony_ci               std::function<int32_t(int32_t)> resetTimer, uint32_t maxRecordCount = 100)
33c29fa5a6Sopenharmony_ci        : addTimer_(std::move(addTimer)), resetTimer_(std::move(resetTimer)), maxRecordCount_(maxRecordCount)
34c29fa5a6Sopenharmony_ci    {}
35c29fa5a6Sopenharmony_ci
36c29fa5a6Sopenharmony_ci    bool Record(const LogHeader &lh, const std::string &key, const std::string &record);
37c29fa5a6Sopenharmony_ci
38c29fa5a6Sopenharmony_ci    ~Aggregator();
39c29fa5a6Sopenharmony_ci
40c29fa5a6Sopenharmony_ciprivate:
41c29fa5a6Sopenharmony_ci    struct RecordInfo {
42c29fa5a6Sopenharmony_ci        std::string record;
43c29fa5a6Sopenharmony_ci        std::chrono::system_clock::time_point timestamp;
44c29fa5a6Sopenharmony_ci    };
45c29fa5a6Sopenharmony_ci    std::string key_;
46c29fa5a6Sopenharmony_ci    std::vector<RecordInfo> records_;
47c29fa5a6Sopenharmony_ci    std::function<int32_t(int32_t, int32_t, std::function<void()>)> addTimer_;
48c29fa5a6Sopenharmony_ci    std::function<int32_t(int32_t)> resetTimer_;
49c29fa5a6Sopenharmony_ci    int32_t timerId_ { -1 };
50c29fa5a6Sopenharmony_ci    uint32_t maxRecordCount_ { 0 };
51c29fa5a6Sopenharmony_ci
52c29fa5a6Sopenharmony_ci    void FlushRecords(const LogHeader &lh, const std::string &key = "", const std::string &extraRecord = "");
53c29fa5a6Sopenharmony_ci};
54c29fa5a6Sopenharmony_ci} // namespace MMI
55c29fa5a6Sopenharmony_ci} // namespace OHOS
56c29fa5a6Sopenharmony_ci#endif // INPUT_AGGREGATOR_H
57