1 /*
2  * Copyright (c) 2024 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_NAPI_NATIVE_ENGINE_ARK_FINALIZERS_PACK_H
17 #define FOUNDATION_ACE_NAPI_NATIVE_ENGINE_ARK_FINALIZERS_PACK_H
18 
19 #include "ark_crash_holder.h"
20 #include "interfaces/inner_api/napi/native_node_api.h"
21 
22 class NativeEngine;
23 
24 using RefFinalizer = std::pair<NapiNativeFinalize, std::tuple<NativeEngine*, void*, void*>>;
25 using ArkFinalizersPackFinishNotify = std::function<void(size_t totalNativeBindingSize)>;
26 
27 class ArkFinalizersPack {
28 public:
29     ArkFinalizersPack() = default;
30     ~ArkFinalizersPack() = default;
31     DEFAULT_MOVE_SEMANTIC(ArkFinalizersPack);
32     DEFAULT_COPY_SEMANTIC(ArkFinalizersPack);
33 
Clear()34     void Clear()
35     {
36         finalizers_.clear();
37         totalNativeBindingSize_ = 0;
38         notify_ = nullptr;
39     }
Empty()40     bool Empty() const
41     {
42         return finalizers_.empty();
43     }
RegisterFinishNotify(ArkFinalizersPackFinishNotify notify)44     void RegisterFinishNotify(ArkFinalizersPackFinishNotify notify)
45     {
46         notify_ = notify;
47     }
GetNumFinalizers()48     size_t GetNumFinalizers() const
49     {
50         return finalizers_.size();
51     }
ProcessAll()52     void ProcessAll() const
53     {
54         INIT_CRASH_HOLDER(holder);
55         for (auto &iter : finalizers_) {
56             NapiNativeFinalize callback = iter.first;
57             auto &[p0, p1, p2] = iter.second;
58             holder.UpdateCallbackPtr(reinterpret_cast<uintptr_t>(callback));
59             callback(reinterpret_cast<napi_env>(p0), p1, p2);
60         }
61         NotifyFinish();
62     }
GetTotalNativeBindingSize()63     size_t GetTotalNativeBindingSize() const
64     {
65         return totalNativeBindingSize_;
66     }
AddFinalizer(RefFinalizer & finalizer,size_t nativeBindingSize)67     void AddFinalizer(RefFinalizer &finalizer, size_t nativeBindingSize)
68     {
69         finalizers_.emplace_back(finalizer);
70         totalNativeBindingSize_ += nativeBindingSize;
71     }
72 private:
NotifyFinish()73     void NotifyFinish() const
74     {
75         if (notify_ != nullptr) {
76             notify_(totalNativeBindingSize_);
77         }
78     }
79     std::vector<RefFinalizer> finalizers_ {};
80     size_t totalNativeBindingSize_ {0};
81     ArkFinalizersPackFinishNotify notify_ {nullptr};
82 };
83 
84 #endif /* FOUNDATION_ACE_NAPI_NATIVE_ENGINE_ARK_FINALIZERS_PACK_H */
85