94 lines
2.1 KiB
C++
94 lines
2.1 KiB
C++
// Copyright (c) 2015 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "atom/common/id_weak_map.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "native_mate/converter.h"
|
|
|
|
namespace atom {
|
|
|
|
namespace {
|
|
|
|
struct ObjectKey {
|
|
ObjectKey(int id, IDWeakMap* map) : id(id), map(map) {}
|
|
int id;
|
|
IDWeakMap* map;
|
|
};
|
|
|
|
void OnObjectGC(const v8::WeakCallbackInfo<ObjectKey>& data) {
|
|
ObjectKey* key = data.GetParameter();
|
|
key->map->Remove(key->id);
|
|
delete key;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
IDWeakMap::IDWeakMap() : next_id_(0) {
|
|
}
|
|
|
|
IDWeakMap::~IDWeakMap() {
|
|
}
|
|
|
|
void IDWeakMap::Set(v8::Isolate* isolate,
|
|
int32_t id,
|
|
v8::Local<v8::Object> object) {
|
|
auto global = make_linked_ptr(new v8::Global<v8::Object>(isolate, object));
|
|
ObjectKey* key = new ObjectKey(id, this);
|
|
global->SetWeak(key, OnObjectGC, v8::WeakCallbackType::kParameter);
|
|
map_[id] = global;
|
|
}
|
|
|
|
int32_t IDWeakMap::Add(v8::Isolate* isolate, v8::Local<v8::Object> object) {
|
|
int32_t id = GetNextID();
|
|
Set(isolate, id, object);
|
|
return id;
|
|
}
|
|
|
|
v8::MaybeLocal<v8::Object> IDWeakMap::Get(v8::Isolate* isolate, int32_t id) {
|
|
auto iter = map_.find(id);
|
|
if (iter == map_.end())
|
|
return v8::MaybeLocal<v8::Object>();
|
|
else
|
|
return v8::Local<v8::Object>::New(isolate, *iter->second);
|
|
}
|
|
|
|
bool IDWeakMap::Has(int32_t id) const {
|
|
return map_.find(id) != map_.end();
|
|
}
|
|
|
|
std::vector<int32_t> IDWeakMap::Keys() const {
|
|
std::vector<int32_t> keys;
|
|
keys.reserve(map_.size());
|
|
for (const auto& iter : map_)
|
|
keys.emplace_back(iter.first);
|
|
return keys;
|
|
}
|
|
|
|
std::vector<v8::Local<v8::Object>> IDWeakMap::Values(v8::Isolate* isolate) {
|
|
std::vector<v8::Local<v8::Object>> keys;
|
|
keys.reserve(map_.size());
|
|
for (const auto& iter : map_)
|
|
keys.emplace_back(v8::Local<v8::Object>::New(isolate, *iter.second));
|
|
return keys;
|
|
}
|
|
|
|
void IDWeakMap::Remove(int32_t id) {
|
|
auto iter = map_.find(id);
|
|
if (iter == map_.end())
|
|
LOG(WARNING) << "Removing unexist object with ID " << id;
|
|
else
|
|
map_.erase(iter);
|
|
}
|
|
|
|
void IDWeakMap::Clear() {
|
|
map_.clear();
|
|
}
|
|
|
|
int32_t IDWeakMap::GetNextID() {
|
|
return ++next_id_;
|
|
}
|
|
|
|
} // namespace atom
|