1 /*
2  * Copyright (c) 2023 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 #ifndef UPDATER_HWFAULT_RETRY_H
16 #define UPDATER_HWFAULT_RETRY_H
17 
18 #include "log/log.h"
19 #include <unordered_map>
20 #include <vector>
21 #include <sys/time.h>
22 #include <unistd.h>
23 
24 namespace Updater {
25 class HwFaultRetry {
26 public:
27     HwFaultRetry();
28     ~HwFaultRetry() = default;
29 
30     static HwFaultRetry &GetInstance();
31     using RetryFunc = std::function<void ()>;
32     void DoRetryAction();
33     void RegisterFunc(const std::string &faultInfo, RetryFunc func);
34     void SetFaultInfo(const std::string &faultInfo);
35     void SetRetryCount(const uint32_t count);
36 
37 private:
38     void RebootRetry();
39 
40     std::unordered_map<std::string, RetryFunc> retryMap_;
41     std::string faultInfo_;
42     uint32_t retryCount_ {};
43 };
44 } // Updater
45 #endif // UPDATER_HWFAULT_RETRY_H
46