1 /*
2  * Copyright (c) 2021-2022 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 
16 #ifndef FOUNDATION_ACE_FRAMEWORKS_CORE_COMMON_WATCH_DOG_H
17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMMON_WATCH_DOG_H
18 
19 #include <unordered_map>
20 
21 #include "base/thread/task_executor.h"
22 
23 namespace OHOS::Ace {
24 class ThreadWatcher;
25 
26 struct Watchers {
27     RefPtr<ThreadWatcher> jsWatcher;
28     RefPtr<ThreadWatcher> uiWatcher;
29 };
30 
31 class WatchDog final : public Referenced {
32 public:
33     WatchDog();
34     ~WatchDog() override;
35 
36     void Register(int32_t instanceId, const RefPtr<TaskExecutor>& taskExecutor, bool useUIAsJSThread);
37     void Unregister(int32_t instanceId);
38     void BuriedBomb(int32_t instanceId, uint64_t bombId);
39     void DefusingBomb(int32_t instanceId);
40 
41 private:
42     std::unordered_map<int32_t, Watchers> watchMap_;
43 
44     ACE_DISALLOW_COPY_AND_MOVE(WatchDog);
45 };
46 
47 } // namespace OHOS::Ace
48 
49 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMMON_WATCH_DOG_H
50