0203bd3305
* refactor: use base::Contains() in KeyWeakMap::Has() * refactor: use base::Contains() in WebRequest::RequestFilter::MatchesType() * refactor: use base::Contains() in BaseWindow::AddBrowserView() * refactor: use base::Contains() in DeepFreeze() * refactor: use base::Contains() in Clipboard::Read() * Revert "refactor: use base::Contains() in BaseWindow::AddBrowserView()" This reverts commit 60152359d3978451ebdd7c8eed602c2fb8a9cafa. * refactor: use base::Contains() in BaseWindow::AddBrowserView() * refactor: use base::Contains() in IsDevToolsFileSystemAdded() * refactor: use base::Contains() in MessagePort::DisentanglePorts() * refactor: use base::Contains() in PowerSaveBlocker::IsStarted() * refactor: use base::Contains() in SpellCheckClient::OnSpellCheckDone() * refactor: use base::Contains() in ShowTaskDialogWstr() * refactor: use base::Contains() in PrintViewManagerElectron::ScriptedPrint() * refactor: use base::Contains() in PrintViewManagerElectron::DidGetPrintedPagesCount() * refactor: use base::Contains() in NativeWindow::AddDraggableRegionProvider() * refactor: use base::Contains() in ElectronBindings::ActivateUVLoop() * refactor: use base::Contains() in NativeWindowViews::IsVisibleOnAllWorkspaces() * refactor: use base::Contains() in HidChooserController::OnDeviceAdded() * refactor: use base::Contains() in ElectronSandboxedRendererClient::WillReleaseScriptContext() * refactor: use base::Contains() in ElectronRendererClient::WillDestroyWorkerContextOnWorkerThread() * refactor: use base::Contains() in GlobalShortcut::OnKeyPressed()
90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
// Copyright (c) 2016 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef ELECTRON_SHELL_COMMON_KEY_WEAK_MAP_H_
|
|
#define ELECTRON_SHELL_COMMON_KEY_WEAK_MAP_H_
|
|
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "base/containers/contains.h"
|
|
#include "base/memory/raw_ptr.h"
|
|
#include "v8/include/v8.h"
|
|
|
|
namespace electron {
|
|
|
|
// Like ES6's WeakMap, but the key is Integer and the value is Weak Pointer.
|
|
template <typename K>
|
|
class KeyWeakMap {
|
|
public:
|
|
// Records the key and self, used by SetWeak.
|
|
struct KeyObject {
|
|
K key;
|
|
raw_ptr<KeyWeakMap> self;
|
|
};
|
|
|
|
KeyWeakMap() {}
|
|
virtual ~KeyWeakMap() {
|
|
for (auto& p : map_)
|
|
p.second.second.ClearWeak();
|
|
}
|
|
|
|
// disable copy
|
|
KeyWeakMap(const KeyWeakMap&) = delete;
|
|
KeyWeakMap& operator=(const KeyWeakMap&) = delete;
|
|
|
|
// Sets the object to WeakMap with the given |key|.
|
|
void Set(v8::Isolate* isolate, const K& key, v8::Local<v8::Object> object) {
|
|
KeyObject key_object = {key, this};
|
|
auto& p = map_[key] =
|
|
std::make_pair(key_object, v8::Global<v8::Object>(isolate, object));
|
|
p.second.SetWeak(&(p.first), OnObjectGC, v8::WeakCallbackType::kParameter);
|
|
}
|
|
|
|
// Gets the object from WeakMap by its |key|.
|
|
v8::MaybeLocal<v8::Object> Get(v8::Isolate* isolate, const K& key) {
|
|
auto iter = map_.find(key);
|
|
if (iter == map_.end())
|
|
return v8::MaybeLocal<v8::Object>();
|
|
else
|
|
return v8::Local<v8::Object>::New(isolate, iter->second.second);
|
|
}
|
|
|
|
// Whether there is an object with |key| in this WeakMap.
|
|
constexpr bool Has(const K& key) const { return base::Contains(map_, key); }
|
|
|
|
// Returns all objects.
|
|
std::vector<v8::Local<v8::Object>> Values(v8::Isolate* isolate) const {
|
|
std::vector<v8::Local<v8::Object>> keys;
|
|
keys.reserve(map_.size());
|
|
for (const auto& it : map_)
|
|
keys.emplace_back(v8::Local<v8::Object>::New(isolate, it.second.second));
|
|
return keys;
|
|
}
|
|
|
|
// Remove object with |key| in the WeakMap.
|
|
void Remove(const K& key) {
|
|
auto iter = map_.find(key);
|
|
if (iter == map_.end())
|
|
return;
|
|
|
|
iter->second.second.ClearWeak();
|
|
map_.erase(iter);
|
|
}
|
|
|
|
private:
|
|
static void OnObjectGC(
|
|
const v8::WeakCallbackInfo<typename KeyWeakMap<K>::KeyObject>& data) {
|
|
KeyWeakMap<K>::KeyObject* key_object = data.GetParameter();
|
|
key_object->self->Remove(key_object->key);
|
|
}
|
|
|
|
// Map of stored objects.
|
|
std::unordered_map<K, std::pair<KeyObject, v8::Global<v8::Object>>> map_;
|
|
};
|
|
|
|
} // namespace electron
|
|
|
|
#endif // ELECTRON_SHELL_COMMON_KEY_WEAK_MAP_H_
|