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 #include "bridge/cj_frontend/cppview/lazy_foreach.h"
17
18 #include "bridge/cj_frontend/cppview/data_change_listener.h"
19 #include "core/components_ng/base/view_stack_model.h"
20
21 using namespace OHOS::Ace;
22 using namespace OHOS::FFI;
23 using namespace OHOS::Ace::Framework;
24
25 namespace OHOS::Ace::Framework {
26
OnExpandChildrenOnInitialInNG()27 void CJLazyForEachBuilder::OnExpandChildrenOnInitialInNG()
28 {
29 auto totalIndex = cjBuilder_->GetTotalCount();
30 for (auto index = 0; index < totalIndex; index++) {
31 auto key = cjBuilder_->GenerateKey(index);
32 ViewStackModel::GetInstance()->PushKey(key);
33 cjBuilder_->GenerateItem(index);
34 ViewStackModel::GetInstance()->PopKey();
35 }
36 }
37
OnGetChildByIndex(int32_t index,std::unordered_map<std::string,std::pair<int32_t,RefPtr<NG::UINode>>> & cachedItems)38 std::pair<std::string, RefPtr<NG::UINode>> CJLazyForEachBuilder::OnGetChildByIndex(
39 int32_t index, std::unordered_map<std::string, std::pair<int32_t, RefPtr<NG::UINode>>>& cachedItems)
40 {
41 std::pair<std::string, RefPtr<NG::UINode>> result;
42 auto key = cjBuilder_->GenerateKey(index);
43 auto cachedIter = cachedItems.find(key);
44 if (cachedIter != cachedItems.end()) {
45 result.first = key;
46 result.second = cachedIter->second.second;
47 return result;
48 }
49 NG::ScopedViewStackProcessor scopedViewStackProcessor;
50 auto* viewStack = ViewStackModel::GetInstance();
51 cjBuilder_->MarkLazyForEachProcess(key);
52 viewStack->PushKey(key);
53 cjBuilder_->GenerateItem(index);
54 viewStack->PopKey();
55 cjBuilder_->ResetLazyForEachProcess();
56 result.first = key;
57 auto view = ViewStackModel::GetInstance()->Finish();
58 result.second = AceType::DynamicCast<NG::UINode>(view);
59
60 return result;
61 }
62
OnGetChildByIndexNew(int32_t index,std::map<int32_t,NG::LazyForEachChild> & cachedItems,std::unordered_map<std::string,NG::LazyForEachCacheChild> & expiringItems)63 std::pair<std::string, RefPtr<NG::UINode>> CJLazyForEachBuilder::OnGetChildByIndexNew(int32_t index,
64 std::map<int32_t, NG::LazyForEachChild>& cachedItems,
65 std::unordered_map<std::string, NG::LazyForEachCacheChild>& expiringItems)
66 {
67 std::pair<std::string, RefPtr<NG::UINode>> result;
68 auto key = cjBuilder_->GenerateKey(index);
69 auto cachedIter = cachedItems.find(index);
70 if (cachedIter != cachedItems.end()) {
71 result.first = key;
72 result.second = cachedIter->second.second;
73 return result;
74 }
75 auto expiringIter = expiringItems.find(key);
76 if (expiringIter != expiringItems.end()) {
77 result.first = key;
78 result.second = expiringIter->second.second;
79 expiringItems.erase(expiringIter);
80 return result;
81 }
82 NG::ScopedViewStackProcessor scopedViewStackProcessor;
83 auto* viewStack = ViewStackModel::GetInstance();
84 cjBuilder_->MarkLazyForEachProcess(key);
85 viewStack->PushKey(key);
86 cjBuilder_->GenerateItem(index);
87 viewStack->PopKey();
88 cjBuilder_->ResetLazyForEachProcess();
89 result.first = key;
90 auto view = ViewStackModel::GetInstance()->Finish();
91 result.second = AceType::DynamicCast<NG::UINode>(view);
92
93 return result;
94 }
95
ReleaseChildGroupById(const std::string & id)96 void CJLazyForEachBuilder::ReleaseChildGroupById(const std::string& id)
97 {
98 cjBuilder_->RemoveChildGroupById(id);
99 }
100
RegisterDataChangeListener(const RefPtr<V2::DataChangeListener> & listener)101 void CJLazyForEachBuilder::RegisterDataChangeListener(const RefPtr<V2::DataChangeListener>& listener)
102 {
103 if (!listener) {
104 return;
105 }
106 auto listenerManager = weakListenerManager_.promote();
107 if (listenerManager) {
108 listenerManager->AddListener(listener);
109 return;
110 }
111 listenerManager = FFIData::Create<CJDataChangeListener>();
112 listenerManager->AddListener(listener);
113 weakListenerManager_ = listenerManager;
114 cjBuilder_->RegisterListenerFunc(listenerManager);
115 }
116
UnregisterDataChangeListener(V2::DataChangeListener * listener)117 void CJLazyForEachBuilder::UnregisterDataChangeListener(V2::DataChangeListener* listener)
118 {
119 if (!listener) {
120 LOGW("CJLazyForEachBuilder listener is null");
121 return;
122 }
123 auto listenerManager = weakListenerManager_.promote();
124 if (!listenerManager) {
125 // this would happen before first RegisterDataChangeListener called
126 // this would happen after cj released listenerManager
127 LOGW("LazyForEach::UnregisterDataChangeListener fail, listenerManager is null.");
128 return;
129 }
130 listenerManager->RemoveListener(WeakClaim(listener));
131 }
132
133 } // namespace OHOS::Ace::Framework
134